From 4efee316c1411c596a4d333740c3921742f08116 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Mon, 25 Aug 2025 16:41:04 +0800
Subject: [PATCH] feat(ai): 添加流式回答接口 V2 版本

---
 src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkServiceImpl.java |   33 +++++++++++++++++++++++++++------
 1 files changed, 27 insertions(+), 6 deletions(-)

diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkServiceImpl.java
index 96ceed5..05791c7 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkServiceImpl.java
@@ -1,17 +1,18 @@
 package cc.mrbird.febs.ai.service.impl;
 
+import cc.mrbird.febs.ai.enumerates.AiTypeEnum;
 import cc.mrbird.febs.ai.entity.AiMemberTalk;
 import cc.mrbird.febs.ai.entity.AiMemberTalkItem;
 import cc.mrbird.febs.ai.entity.AiProductRoleLink;
 import cc.mrbird.febs.ai.mapper.AiMemberTalkMapper;
+import cc.mrbird.febs.ai.req.ai.AiMessage;
 import cc.mrbird.febs.ai.req.ai.AiRequest;
 import cc.mrbird.febs.ai.req.memberTalk.ApiMemberTalkAnswerDto;
 import cc.mrbird.febs.ai.req.memberTalk.ApiMemberTalkDto;
 import cc.mrbird.febs.ai.req.memberTalk.ApiMemberTalkItemPageDto;
+import cc.mrbird.febs.ai.req.memberTalk.ApiMemberTalkStreamDto;
 import cc.mrbird.febs.ai.res.ai.AiResponse;
-import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkItemVo;
 import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkVo;
-import cc.mrbird.febs.ai.res.product.ApiProductVo;
 import cc.mrbird.febs.ai.service.AiMemberTalkItemService;
 import cc.mrbird.febs.ai.service.AiMemberTalkService;
 import cc.mrbird.febs.ai.service.AiProductRoleLinkService;
@@ -25,13 +26,15 @@
 import cn.hutool.json.JSONUtil;
 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;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+import reactor.core.publisher.Flux;
 
 import java.util.Date;
+import java.util.List;
 import java.util.function.Consumer;
 
 /**
@@ -57,6 +60,7 @@
     }
 
     @Override
+    @Transactional
     public FebsResponse start(ApiMemberTalkDto dto) {
 
         String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
@@ -80,7 +84,9 @@
             aiMemberTalk = this.add(memberUuid,productId,nowTime);
         }
 
-        AiResponse aiResponse = aiService.start(aiProductRoleLink.getProductRoleId(),"<strong>\"生成题目\"</strong>");
+        List<AiMessage> aiMessageDtoList = aiMemberTalkItemService.getQuestionUpDownContext(aiMemberTalk.getId(),AiTypeEnum.QUESTION.getCode());
+
+        AiResponse aiResponse = aiService.start(aiMessageDtoList,AiTypeEnum.QUESTION.getCode(),aiProductRoleLink.getProductRoleId(),AiTypeEnum.QUESTION.getName(), null);
         if(aiResponse.getCode().equals("200")){
             aiMemberTalkItemService.add(memberUuid,aiMemberTalk.getId(),1,aiResponse.getResContext(),nowTime);
             this.updateTimeUpdate(nowTime,aiMemberTalk.getId());
@@ -179,8 +185,9 @@
     }
 
 
-    public static final String ANSWER_FORMAT = "{}/n[回答]{}/n";
+    public static final String ANSWER_FORMAT = "###题目：{}###用户回答：{}";
     @Override
+    @Transactional
     public FebsResponse answer(ApiMemberTalkAnswerDto dto) {
         String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
         String memberTalkId = dto.getId();
@@ -211,11 +218,17 @@
         String format = StrUtil.format(ANSWER_FORMAT, aiMemberTalkItem.getContext(), reqContext);
         log.info("format:{}",format);
 //        AiResponse aiResponse = aiService.start(aiProductRoleLink.getProductRoleId(), format);
-        AiResponse aiResponse = aiService.start(aiProductRoleLink.getProductRoleId(), reqContext);
+
+
+        List<AiMessage> aiMessageDtoList = aiMemberTalkItemService.getQuestionUpDownContext(aiMemberTalk.getId(),AiTypeEnum.ANSWER.getCode());
+        AiResponse aiResponse = aiService.start(aiMessageDtoList,AiTypeEnum.ANSWER.getCode(),aiProductRoleLink.getProductRoleId(), reqContext,aiMemberTalkItem.getContext());
         String context = null;
         if(aiResponse.getCode().equals("200")){
             Date nowTime = new Date();
             context = String.valueOf(JSONUtil.parse(aiResponse.getReport()));
+            if ("null".equals( context)){
+                context = aiResponse.getResContext();
+            }
             aiMemberTalkItemService.add(memberUuid,aiMemberTalk.getId(),3, context,nowTime);
             this.updateTimeUpdate(nowTime,aiMemberTalk.getId());
         }else{
@@ -245,4 +258,12 @@
         dto.setMemberUuid(memberUuid);
         return aiMemberTalkItemService.historyPage(dto);
     }
+
+    @Override
+    public Flux<FebsResponse> answerStream(String question) {
+
+//        String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
+        return aiService.answerStream(question);
+    }
+
 }

--
Gitblit v1.9.1