From d68e2f99592dc982a722d031219f1d0b4f87ed00 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 02 Sep 2025 10:11:05 +0800
Subject: [PATCH] feat(ai): 新增 AI 流式回答功能 V3 版本

---
 src/main/java/cc/mrbird/febs/ai/service/impl/AiServiceImpl.java |   17 +++++++++++++++++
 1 files changed, 17 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 73aa812..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;
@@ -112,6 +116,19 @@
     }
 
     @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
     public AiResponse start(List<AiMessage> aiMessageDtoList,Integer type,String productRoleId, String content, String question) {
         if (!StringUtils.hasText(productRoleId)) {
             log.warn("productRoleId 不能为空");

--
Gitblit v1.9.1