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/AiTalkServiceImpl.java |   67 +++++++++++++++++++++++++++++++++
 1 files changed, 66 insertions(+), 1 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiTalkServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiTalkServiceImpl.java
index 9410135..3d7be4e 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiTalkServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiTalkServiceImpl.java
@@ -1,21 +1,33 @@
 package cc.mrbird.febs.ai.service.impl;
 
+import cc.mrbird.febs.ai.entity.AiProductQuestion;
 import cc.mrbird.febs.ai.entity.AiTalk;
+import cc.mrbird.febs.ai.enumerates.AiPromptEnum;
 import cc.mrbird.febs.ai.mapper.AiTalkMapper;
+import cc.mrbird.febs.ai.req.talk.AiTalkAnswerStream;
 import cc.mrbird.febs.ai.req.talk.ApiTalkDto;
+import cc.mrbird.febs.ai.req.talk.ApiTalkItemPageDto;
 import cc.mrbird.febs.ai.req.talk.ApiTalkPageDto;
-import cc.mrbird.febs.ai.res.memberAnswer.ApiMemberProductWorkVo;
 import cc.mrbird.febs.ai.res.talk.ApiTalkPageVo;
+import cc.mrbird.febs.ai.res.talk.ApiTalkQuestionVo;
 import cc.mrbird.febs.ai.res.talk.ApiTalkVo;
+import cc.mrbird.febs.ai.service.AiProductQuestionService;
 import cc.mrbird.febs.ai.service.AiService;
 import cc.mrbird.febs.ai.service.AiTalkItemService;
 import cc.mrbird.febs.ai.service.AiTalkService;
+import cc.mrbird.febs.ai.strategy.LlmStrategyFactory;
+import cc.mrbird.febs.ai.strategy.enumerates.LlmStrategyEnum;
+import cc.mrbird.febs.ai.strategy.param.LlmStrategyDto;
 import cc.mrbird.febs.ai.utils.UUID;
 import cc.mrbird.febs.common.entity.FebsResponse;
 import cc.mrbird.febs.common.utils.LoginUserUtil;
+import cn.hutool.core.collection.CollUtil;
 import cn.hutool.core.date.DateTime;
 import cn.hutool.core.date.DateUtil;
 import cn.hutool.core.util.StrUtil;
+import com.alibaba.dashscope.common.Role;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import lombok.RequiredArgsConstructor;
@@ -23,7 +35,9 @@
 import org.springframework.stereotype.Service;
 import reactor.core.publisher.Flux;
 
+import java.util.ArrayList;
 import java.util.Date;
+import java.util.List;
 
 @Slf4j
 @Service
@@ -31,7 +45,27 @@
 public class AiTalkServiceImpl extends ServiceImpl<AiTalkMapper, AiTalk> implements AiTalkService {
 
     private final AiTalkItemService aiTalkItemService;
+    private final AiProductQuestionService aiProductQuestionService;
     private final AiService aiService;
+    private final LlmStrategyFactory llmStrategyFactory;
+
+    @Override
+    public FebsResponse questionList() {
+        List<ApiTalkQuestionVo> list = new ArrayList<>();
+        LambdaQueryWrapper<AiProductQuestion> queryWrapper = Wrappers.lambdaQuery(AiProductQuestion.class);
+        queryWrapper.orderByDesc(AiProductQuestion::getCreatedTime);
+        queryWrapper.last("limit 30");
+        List<AiProductQuestion> listByQuery = aiProductQuestionService.getListByQuery(queryWrapper);
+        if (CollUtil.isNotEmpty(listByQuery)){
+            for (AiProductQuestion aiProductQuestion : listByQuery){
+                ApiTalkQuestionVo apiTalkQuestionVo = new ApiTalkQuestionVo();
+                apiTalkQuestionVo.setContext(aiProductQuestion.getTitle());
+                list.add(apiTalkQuestionVo);
+            }
+        }
+        return new FebsResponse().success().data(list);
+    }
+
     @Override
     public FebsResponse talkOpen(ApiTalkDto dto) {
         String talkId = dto.getTalkId();
@@ -80,4 +114,35 @@
         return new FebsResponse().success().data(pageListByQuery);
     }
 
+    @Override
+    public FebsResponse historyPage(ApiTalkItemPageDto dto) {
+        String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
+        dto.setMemberUuid(memberUuid);
+        return aiTalkItemService.historyPage(dto);
+    }
+
+    @Override
+    public Flux<FebsResponse> answerStreamV2(AiTalkAnswerStream dto) {
+        return aiService.answerStreamV2(dto);
+    }
+
+    @Override
+    public Flux<FebsResponse> answerStreamV3(AiTalkAnswerStream dto) {
+        ArrayList<LlmStrategyDto> llmStrategyDtoList = new ArrayList<>();
+        if (dto.getPrompt() != null){
+            LlmStrategyDto llmStrategyDto = new LlmStrategyDto();
+            llmStrategyDto.setRole(Role.SYSTEM.getValue());
+            llmStrategyDto.setContent(AiPromptEnum.STREAM_NORMAL.getPrompt());
+            llmStrategyDtoList.add(llmStrategyDto);
+        }
+        if (dto.getQuestion() != null){
+            LlmStrategyDto llmStrategyDto = new LlmStrategyDto();
+            llmStrategyDto.setRole(Role.USER.getValue());
+            llmStrategyDto.setContent(dto.getQuestion());
+            llmStrategyDtoList.add(llmStrategyDto);
+        }
+        String modelName = LlmStrategyEnum.getName(aiService.getSystemSetAiType());
+        return llmStrategyFactory.getCalculationStrategyMap().get(modelName).llmInvokeStreamingWithThink(llmStrategyDtoList);
+    }
+
 }

--
Gitblit v1.9.1