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