From 634a0506c29607586456ea0a4ba7ec4b5c57f45c Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 02 Sep 2025 17:20:09 +0800
Subject: [PATCH] refactor(ai): 优化 AI交流输出类型的描述
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiServiceImpl.java | 18 ++++++++++++++++++
1 files changed, 18 insertions(+), 0 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiServiceImpl.java
index 1ea2e73..11c6794 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiServiceImpl.java
@@ -13,7 +13,10 @@
import cc.mrbird.febs.ai.service.AiProductRoleService;
import cc.mrbird.febs.ai.service.AiService;
import cc.mrbird.febs.ai.service.AiTalkItemService;
+import cc.mrbird.febs.ai.strategy.enumerates.LlmStrategyContextEnum;
import cc.mrbird.febs.common.entity.FebsResponse;
+import cc.mrbird.febs.mall.entity.DataDictionaryCustom;
+import cc.mrbird.febs.mall.mapper.DataDictionaryCustomMapper;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSONUtil;
@@ -74,6 +77,7 @@
private final AiProductRoleService aiProductRoleService;
private final ObjectMapper objectMapper;
private final AiTalkItemService aiTalkItemService;
+ private final DataDictionaryCustomMapper dataDictionaryCustomMapper;
@Value("${ai.service.ak}")
private String ak;
@@ -109,6 +113,19 @@
if (service != null) {
service.shutdownExecutor();
}
+ }
+
+ @Override
+ public Integer getSystemSetAiType() {
+ Integer type = 2;
+ DataDictionaryCustom dataDictionaryCustom = dataDictionaryCustomMapper.selectDicDataByTypeAndCode(
+ LlmStrategyContextEnum.LLM_STRATEGY.getCode(),
+ LlmStrategyContextEnum.LLM_STRATEGY.getCode()
+ );
+ if (dataDictionaryCustom != null) {
+ type = Integer.parseInt(dataDictionaryCustom.getValue());
+ }
+ return type;
}
@Override
@@ -498,6 +515,7 @@
content = message.getContent().toString();
}
apiMemberTalkStreamVo.setContent(content);
+ System.out.print(content);
log.debug("Content: {}", content);
return new FebsResponse().success().data(apiMemberTalkStreamVo);
--
Gitblit v1.9.1