| | |
| | | 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<>(); |
| | |
| | | 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) { |
| | |
| | | throw new FebsException("百炼工作流初始化异常"); |
| | | } |
| | | HashMap prompt = getPrompt(dto); |
| | | List<Message> messages = getMessages(dto); |
| | | String query = getQuery(dto); |
| | | String appId = getAppId(dto); |
| | | if (prompt == null || prompt.size() == 0){ |
| | |
| | | .appId(appId) //替换为实际的应用 ID |
| | | .flowStreamMode(FlowStreamMode.MESSAGE_FORMAT) |
| | | .prompt(query) |
| | | .messages( messages) |
| | | .bizParams(JsonUtils.toJsonObject( prompt)) |
| | | .build(); |
| | | |