Administrator
2 days ago 89c8099e23ef5260ce6b3e46339064c559e9cc0f
src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliApplicationLlmStrategyServiceImpl.java
@@ -1,6 +1,7 @@
package cc.mrbird.febs.ai.strategy.Impl;
import cc.mrbird.febs.ai.enumerates.AiTypeEnum;
import cc.mrbird.febs.ai.req.agent.AiRequestDto;
import cc.mrbird.febs.ai.strategy.LlmStrategyService;
import cc.mrbird.febs.ai.strategy.enumerates.LlmApplicationAppIdEnum;
import cc.mrbird.febs.ai.strategy.enumerates.LlmStrategyContextEnum;
@@ -58,6 +59,36 @@
            }
        }
        return bizParamsMap;
    }
    private List<Message> getMessages(List<LlmStrategyDto> dto) {
        List<Message> messages = new ArrayList<>();
        for (LlmStrategyDto item : dto){
            if (StrUtil.equals(item.getRole(),AiTypeEnum.MESSAGES.getName())){
                List<LlmStrategyDto> messagesList = item.getMessages();
                for (LlmStrategyDto dtoItem : messagesList){
                    if (StrUtil.equals(dtoItem.getRole(),Role.SYSTEM.getValue())){
                        messages.add(Message.builder()
                                .role(Role.SYSTEM.getValue())
                                .content(dtoItem.getContent())
                                .build());
                    }
                    if (StrUtil.equals(dtoItem.getRole(),Role.USER.getValue())){
                        messages.add(Message.builder()
                                .role(Role.USER.getValue())
                                .content(dtoItem.getContent())
                                .build());
                    }
                    if (StrUtil.equals(dtoItem.getRole(),Role.ASSISTANT.getValue())){
                        messages.add(Message.builder()
                                .role(Role.ASSISTANT.getValue())
                                .content(dtoItem.getContent())
                                .build());
                    }
                }
            }
        }
        return messages;
    }
    private String getQuery(List<LlmStrategyDto> dto) {
@@ -188,6 +219,8 @@
        if (appId == null){
            throw new FebsException("百炼工作流初始化异常");
        }
        List<Message> messages = getMessages(dto);
        long startTime = System.currentTimeMillis();
        ApplicationParam param = ApplicationParam.builder()
                // 若没有配置环境变量,可用百炼API Key将下行替换为:.apiKey("sk-xxx")。但不建议在生产环境中直接将API Key硬编码到代码中,以减少API Key泄露风险。
@@ -195,6 +228,7 @@
                .appId(appId) //替换为实际的应用 ID
                .flowStreamMode(FlowStreamMode.MESSAGE_FORMAT)
                .prompt(query)
                .messages( messages)
                .bizParams(JsonUtils.toJsonObject( prompt))
                .build();
@@ -222,4 +256,9 @@
                    throw new FebsException(StrUtil.format("百炼工作流输出失败:{}",error));
                });
    }
    @Override
    public Flux<FebsResponse> llmInvokeStreamingNoThink(AiRequestDto aiRequestDto) {
        return null;
    }
}