feat(ai): 新增公司工作流配置查询功能

Administrator

14 hours ago 438b5375d320af6d18f3f34cb26b803200e1ca81

 src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliApplicationLlmStrategyServiceImpl.java

@@ -39,6 +39,7 @@
    private final String bizParam_1 = "prompt_ai_system";
    private final String bizParam_2 = "question";
    private final String bizParam_3 = "query";
    private final String bizParam_4 = "messages";

    private HashMap getPrompt(List<LlmStrategyDto> dto) {
        HashMap<String, String> bizParamsMap = new HashMap<>();
@@ -52,25 +53,11 @@
            if (StrUtil.equals(dtoItem.getRole(),Role.USER.getValue())){
                bizParamsMap.put(bizParam_3, dtoItem.getContent());
            }
        }
        return bizParamsMap;
    }

    private List<Message> getMessages(List<LlmStrategyDto> dto) {
        List<Message> messages = new ArrayList<>();
        for (LlmStrategyDto dtoItem : dto){
            if (StrUtil.equals(AiTypeEnum.MESSAGES.getName(), dtoItem.getRole())){
                List<LlmStrategyDto> messages1 = dtoItem.getMessages();
                for (LlmStrategyDto dtoItem1 : messages1){
                    messages.add(Message.builder()
                            .role(dtoItem1.getRole())
                            .content(dtoItem1.getContent())
                            .build());
                }
                break;
            if (StrUtil.equals(dtoItem.getRole(),AiTypeEnum.MESSAGES.getName())){
                bizParamsMap.put(bizParam_4, JSONUtil.toJsonStr(dtoItem.getMessages()));
            }
        }
        return messages;
        return bizParamsMap;
    }

    private String getQuery(List<LlmStrategyDto> dto) {
@@ -90,6 +77,17 @@
            if (StrUtil.equals(dtoItem.getRole(),Role.TOOL.getValue())){
                int code = Integer.parseInt(dtoItem.getContent());
                appId = LlmApplicationAppIdEnum.HIGH_LIGHT.getAppIdByCode(code);
                break;
            }
        }
        return appId;
    }

    private String getAppIdV2(List<LlmStrategyDto> dto) {
        String appId = null;
        for (LlmStrategyDto dtoItem : dto){
            if (StrUtil.equals(dtoItem.getRole(),Role.TOOL.getValue())){
                appId = dtoItem.getContent();
                break;
            }
        }
@@ -178,9 +176,9 @@
            throw new FebsException("百炼工作流初始化异常");
        }
        HashMap prompt = getPrompt(dto);
        List<Message> messages = getMessages(dto);
        String query = getQuery(dto);
        String appId = getAppId(dto);
//        String appId = getAppId(dto);
        String appId = getAppIdV2(dto);
        if (prompt == null || prompt.size() == 0){
            throw new FebsException("百炼工作流初始化异常");
        }
@@ -197,7 +195,6 @@
                .appId(appId) //替换为实际的应用 ID
                .flowStreamMode(FlowStreamMode.MESSAGE_FORMAT)
                .prompt(query)
                .messages( messages)
                .bizParams(JsonUtils.toJsonObject( prompt))
                .build();