From 65f10033448a07535d610ad56a628d7b3f137480 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Wed, 27 Aug 2025 17:22:14 +0800
Subject: [PATCH] feat(ai): 优化对话项添加逻辑并增强接口返回值

---
 src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkServiceImpl.java |   27 ++++++++++++++++-----------
 1 files changed, 16 insertions(+), 11 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 ad3a0cc..4885a94 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
@@ -85,16 +85,18 @@
         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());
-        }else{
-            throw new FebsException(aiResponse.getDescription());
-        }
+
         ApiMemberTalkVo apiMemberTalkVo = new ApiMemberTalkVo();
         apiMemberTalkVo.setMemberTalkId(aiMemberTalk.getId());
         apiMemberTalkVo.setType(1);
         apiMemberTalkVo.setContext(aiResponse.getResContext());
+        if(aiResponse.getCode().equals("200")){
+            AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(), 1, aiResponse.getResContext(), nowTime);
+            apiMemberTalkVo.setMemberTalkItemId(aiMemberTalkItem.getId());
+            this.updateTimeUpdate(nowTime,aiMemberTalk.getId());
+        }else{
+            throw new FebsException(aiResponse.getDescription());
+        }
         return new FebsResponse().success().data(apiMemberTalkVo);
     }
 
@@ -238,6 +240,9 @@
 //        AiResponse aiResponse = aiService.start(aiProductRoleLink.getProductRoleId(), format);
 
 
+        ApiMemberTalkVo apiMemberTalkVo = new ApiMemberTalkVo();
+        apiMemberTalkVo.setMemberTalkId(aiMemberTalk.getId());
+        apiMemberTalkVo.setType(3);
         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;
@@ -247,15 +252,15 @@
             if ("null".equals( context)){
                 context = aiResponse.getResContext();
             }
-            aiMemberTalkItemService.add(memberUuid,aiMemberTalk.getId(),3, context,nowTime);
+            AiMemberTalkItem talkItem = aiMemberTalkItemService.add(memberUuid, aiMemberTalk.getId(), 3, context, nowTime);
             this.updateTimeUpdate(nowTime,aiMemberTalk.getId());
+
+
+            apiMemberTalkVo.setContext(context);
+            apiMemberTalkVo.setMemberTalkItemId(talkItem.getId());
         }else{
             throw new FebsException(aiResponse.getDescription());
         }
-        ApiMemberTalkVo apiMemberTalkVo = new ApiMemberTalkVo();
-        apiMemberTalkVo.setMemberTalkId(aiMemberTalk.getId());
-        apiMemberTalkVo.setType(3);
-        apiMemberTalkVo.setContext(context);
         return new FebsResponse().success().data(apiMemberTalkVo);
     }
     @Override

--
Gitblit v1.9.1