From 18fdf89338e84e1cab48c2795631777e22b05b81 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Wed, 03 Sep 2025 14:37:54 +0800
Subject: [PATCH] feat(ai): 添加灵通角色设定PROMPT功能
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiMemberTalkServiceImpl.java | 31 ++++++++++++++++++-------------
1 files changed, 18 insertions(+), 13 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 b921f93..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);
}
@@ -108,6 +110,7 @@
queryWrapper.eq(AiMemberTalkItem::getMemberTalkId,memberTalkId);
queryWrapper.eq(AiMemberTalkItem::getType,AiTypeEnum.QUESTION_ANSWER.getCode());
queryWrapper.orderByDesc(AiMemberTalkItem::getCreatedTime);
+ queryWrapper.last("limit 1");
AiMemberTalkItem byQuery = aiMemberTalkItemService.getByQuery(queryWrapper);
if (ObjectUtil.isNotNull(byQuery)){
apiMemberTalkReloadVo.setContext(byQuery.getContext());
@@ -237,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;
@@ -246,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
@@ -270,10 +276,9 @@
@Override
public FebsResponse historyPage(ApiMemberTalkItemPageDto dto) {
-
String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
dto.setMemberUuid(memberUuid);
- return aiMemberTalkItemService.historyPage(dto);
+ return new FebsResponse().success().data(aiMemberTalkItemService.historyPage(dto));
}
@Override
--
Gitblit v1.9.1