仁智企后端
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
zhangjinyang
2025-05-22 065d7aee856305cd7f6392f25d0e0ac578726be3 
 aiflowy-modules/aiflowy-module-ai/src/main/java/tech/aiflowy/ai/controller/AiWorkflowController.java


@@ -1,35 +1,30 @@


package tech.aiflowy.ai.controller;




import com.agentsflex.core.document.Document;


import com.agentsflex.core.llm.Llm;


import com.agentsflex.core.store.DocumentStore;


import com.agentsflex.core.store.SearchWrapper;


import com.agentsflex.core.store.StoreOptions;


import cn.dev33.satoken.annotation.SaIgnore;


import cn.hutool.core.collection.CollectionUtil;


import cn.hutool.core.io.IoUtil;


import cn.hutool.core.util.StrUtil;


import com.agentsflex.core.chain.*;


import com.alibaba.fastjson.JSONObject;


import dev.tinyflow.core.Tinyflow;


import dev.tinyflow.core.knowledge.Knowledge;


import dev.tinyflow.core.node.KnowledgeNode;


import dev.tinyflow.core.parser.ChainParser;


import dev.tinyflow.core.provider.KnowledgeProvider;


import dev.tinyflow.core.provider.LlmProvider;


import org.springframework.beans.factory.annotation.Value;


import tech.aiflowy.ai.entity.AiKnowledge;


import tech.aiflowy.ai.entity.AiLlm;


import dev.tinyflow.core.parser.NodeParser;


import org.springframework.http.MediaType;


import org.springframework.web.bind.annotation.*;


import org.springframework.web.multipart.MultipartFile;


import tech.aiflowy.ai.entity.AiWorkflow;


import tech.aiflowy.ai.node.*;


import tech.aiflowy.ai.service.AiKnowledgeService;


import tech.aiflowy.ai.service.AiLlmService;


import tech.aiflowy.ai.service.AiWorkflowService;


import tech.aiflowy.ai.utils.TinyFlowConfigService;


import tech.aiflowy.common.domain.Result;


import tech.aiflowy.common.filestorage.FileStorageService;


import tech.aiflowy.common.util.SpringContextUtil;


import tech.aiflowy.common.web.controller.BaseCurdController;


import tech.aiflowy.common.web.jsonbody.JsonBody;


import com.agentsflex.core.chain.*;


import org.springframework.web.bind.annotation.*;


import tech.aiflowy.system.service.SysApiKeyService;




import javax.annotation.Resource;


import javax.servlet.http.HttpServletRequest;


import java.io.InputStream;


import java.math.BigInteger;


import java.nio.charset.StandardCharsets;


import java.util.ArrayList;


import java.util.List;


import java.util.Map;




@@ -45,15 +40,27 @@


    private final AiLlmService aiLlmService;




    @Resource


    private AiKnowledgeService aiKnowledgeService;


    @Resource


    private TinyFlowConfigService tinyFlowConfigService;


    private SysApiKeyService apiKeyService;




    public AiWorkflowController(AiWorkflowService service, AiLlmService aiLlmService) {


        super(service);


        this.aiLlmService = aiLlmService;


    }




    @PostMapping("/importWorkFlow")


    public Result importWorkFlow(AiWorkflow workflow, MultipartFile jsonFile) throws Exception {


        InputStream is = jsonFile.getInputStream();


        String content = IoUtil.read(is, StandardCharsets.UTF_8);


        workflow.setContent(content);


        save(workflow);


        return Result.success();


    }




    @GetMapping("/exportWorkFlow")


    public Result exportWorkFlow(BigInteger id) {


        AiWorkflow workflow = service.getById(id);


        return Result.success("content", workflow.getContent());


    }




    @GetMapping("getRunningParameters")


    public Result getRunningParameters(@RequestParam BigInteger id) {


@@ -73,7 +80,10 @@


            return Result.fail(2, "节点配置错误，请检查! ");


        }


        List<Parameter> chainParameters = chain.getParameters();


        return Result.success("parameters", chainParameters);


        return Result.success("parameters", chainParameters)


                .set("title", workflow.getTitle())


                .set("description", workflow.getDescription())


                .set("icon", workflow.getIcon());


    }




    @PostMapping("tryRunning")


@@ -85,21 +95,19 @@


        }




        Tinyflow tinyflow = workflow.toTinyflow();




        tinyFlowConfigService.setAll(tinyflow);




        Chain chain = tinyflow.toChain();


        chain.addEventListener(new ChainEventListener() {


            @Override


            public void onEvent(ChainEvent event, Chain chain) {


                //System.out.println("onEvent : " + event);


                System.out.println("onEvent : " + event);


            }


        });






        chain.addOutputListener(new ChainOutputListener() {


            @Override


            public void onOutput(Chain chain, ChainNode node, Object outputMessage) {


                //System.out.println("output : " + node.getId() + " : " + outputMessage);


                System.out.println("output : " + node.getId() + " : " + outputMessage);


            }


        });




@@ -107,4 +115,89 @@




        return Result.success("result", result).set("message", chain.getMessage());


    }




    @SaIgnore


    @GetMapping(value = "/external/getRunningParams", produces = MediaType.APPLICATION_JSON_VALUE)


    @ResponseBody


    public Result externalGetRunningParameters(HttpServletRequest request,


                                               @RequestParam BigInteger id) {


        String apiKey = request.getHeader("Authorization");


        apiKeyService.checkApiKey(apiKey);


        return getRunningParameters(id);


    }




    @SaIgnore


    @PostMapping(value = "/external/run", produces = MediaType.APPLICATION_JSON_VALUE)


    @ResponseBody


    public Result externalRun(HttpServletRequest request,


                              @JsonBody(value = "id", required = true) BigInteger id,


                              @JsonBody("variables") Map<String, Object> variables) {


        String apiKey = request.getHeader("Authorization");


        apiKeyService.checkApiKey(apiKey);


        return tryRunning(id, variables);


    }




    @PostMapping("/singleRun")


    public Result singleRun(


            @JsonBody(value = "id", required = true) BigInteger id,


            @JsonBody(value = "node", required = true) Map<String, Object> node,


            @JsonBody("variables") Map<String, Object> variables) {




        AiWorkflow workflow = service.getById(id);


        if (workflow == null) {


            return Result.fail(1, "工作流不存在");


        }


        List<ChainNode> nodes = new ArrayList<>();


        Tinyflow tinyflow = workflow.toTinyflow();


        Chain fullChain = tinyflow.toChain();


        if (fullChain != null) {


            nodes = fullChain.getNodes();


        }


        Map<String, NodeParser> map = tinyflow.getChainParser().getNodeParserMap();


        NodeParser parser = map.get(node.get("type").toString());


        if (parser == null) {


            return Result.fail(1, "节点类型不存在");


        }


        ChainNode currentNode = parser.parse(new JSONObject(node), tinyflow);


        if (currentNode == null) {


            return Result.fail(1, "节点不存在");


        }


        currentNode.setInwardEdges(null);


        currentNode.setOutwardEdges(null);


        fixParamType(nodes, currentNode);


        Chain chain = new Chain();


        chain.addNode(currentNode);


        Map<String, Object> res = chain.executeForResult(variables);


        return Result.success(res);


    }




    /**


     * 修正引用类的值类型


     */


    private void fixParamType(List<ChainNode> allNodes, ChainNode currentNode) {


        List<Parameter> currentParams = currentNode.getParameters();


        if (CollectionUtil.isEmpty(currentParams)) {


            return;


        }


        for (Parameter parameter : currentParams) {


            RefType refType = parameter.getRefType();


            if (refType.equals(RefType.REF)) {


                parameter.setRefType(RefType.INPUT);


                String ref = parameter.getRef();


                if (StrUtil.isNotEmpty(ref)) {


                    for (ChainNode node : allNodes) {


                        List<Parameter> parameters = node.getParameters();


                        if (parameters != null) {


                            for (Parameter nodeParameter : parameters) {


                                String nodeAttr = node.getId() + "." + nodeParameter.getName();


                                if (ref.equals(nodeAttr)) {


                                    parameter.setDataType(nodeParameter.getDataType());


                                }


                            }


                        }


                    }


                }


            }


        }


    }


}