feat: 新增工作流试运行时显示每一步的执行结果。close #IC7KLQ

zhangjinyang

2025-05-23 aff1785df0f25c60f0a248154f30886e2b081478

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
package tech.aiflowy.ai.node;
 
import com.agentsflex.core.chain.Chain;
import com.agentsflex.core.chain.node.BaseNode;
import com.agentsflex.core.llm.functions.Function;
import com.alibaba.fastjson.JSON;
import tech.aiflowy.ai.entity.AiPluginTool;
import tech.aiflowy.ai.service.AiPluginToolService;
import tech.aiflowy.common.util.SpringContextUtil;
 
import java.math.BigInteger;
import java.util.Collections;
import java.util.Map;
 
public class PluginToolNode extends BaseNode {
 
    private final BigInteger pluginId;
 
    public PluginToolNode(BigInteger pluginId) {
        this.pluginId = pluginId;
    }
 
    @SuppressWarnings("unchecked")
    @Override
    protected Map<String, Object> execute(Chain chain) {
        Map<String, Object> map = chain.getParameterValues(this);
        AiPluginToolService bean = SpringContextUtil.getBean(AiPluginToolService.class);
        AiPluginTool tool = bean.getById(pluginId);
        if (tool == null) {
            return Collections.emptyMap();
        }
 
        Function function = tool.toFunction();
        if (function == null) {
            return Collections.emptyMap();
        }
 
        Object result = function.invoke(map);
        if (result == null) {
            return Collections.emptyMap();
        }
 
        if (result instanceof Map) {
            return (Map<String, Object>) result;
        }
 
        return JSON.parseObject(JSON.toJSONString(result), Map.class);
    }
}