From 438b5375d320af6d18f3f34cb26b803200e1ca81 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 23 Sep 2025 14:51:32 +0800
Subject: [PATCH] feat(ai): 新增公司工作流配置查询功能
---
src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliApplicationLlmStrategyServiceImpl.java | 19 ++++++++++++++++++-
1 files changed, 18 insertions(+), 1 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliApplicationLlmStrategyServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliApplicationLlmStrategyServiceImpl.java
index bd1e116..95441da 100644
--- a/src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliApplicationLlmStrategyServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliApplicationLlmStrategyServiceImpl.java
@@ -1,5 +1,6 @@
package cc.mrbird.febs.ai.strategy.Impl;
+import cc.mrbird.febs.ai.enumerates.AiTypeEnum;
import cc.mrbird.febs.ai.strategy.LlmStrategyService;
import cc.mrbird.febs.ai.strategy.enumerates.LlmApplicationAppIdEnum;
import cc.mrbird.febs.ai.strategy.enumerates.LlmStrategyContextEnum;
@@ -38,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<>();
@@ -50,6 +52,9 @@
}
if (StrUtil.equals(dtoItem.getRole(),Role.USER.getValue())){
bizParamsMap.put(bizParam_3, dtoItem.getContent());
+ }
+ if (StrUtil.equals(dtoItem.getRole(),AiTypeEnum.MESSAGES.getName())){
+ bizParamsMap.put(bizParam_4, JSONUtil.toJsonStr(dtoItem.getMessages()));
}
}
return bizParamsMap;
@@ -72,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;
}
}
@@ -161,7 +177,8 @@
}
HashMap prompt = getPrompt(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("百炼工作流初始化异常");
}
--
Gitblit v1.9.1