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/ApiMemberTalkStreamServiceImpl.java | 25 +++++++++++++++++++------
1 files changed, 19 insertions(+), 6 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/ApiMemberTalkStreamServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/ApiMemberTalkStreamServiceImpl.java
index 3b1b327..d0bd09f 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/ApiMemberTalkStreamServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/ApiMemberTalkStreamServiceImpl.java
@@ -119,6 +119,20 @@
}
@Override
+ public FebsResponse saveMemberAnswer(ApiMemberTalkMemberAnswerSavaDto dto) {
+ String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
+ String memberTalkId = dto.getMemberTalkId();
+ String content = dto.getContent();
+
+ AiMemberTalk aiMemberTalk = this.getById(memberTalkId);
+ if (ObjectUtil.isNull(aiMemberTalk)){
+ throw new FebsException("对话不存在");
+ }
+ aiMemberTalkItemService.add(memberUuid,aiMemberTalk.getId(),2,content,new Date());
+ return new FebsResponse().success();
+ }
+
+ @Override
public Flux<FebsResponse> answer(AiTalkAnswerStreamDto dto) {
String memberUuid = LoginUserUtil.getLoginUser().getMemberUuid();
@@ -151,9 +165,7 @@
memberTalkItemQuery.orderByDesc(AiMemberTalkItem::getCreatedTime);
memberTalkItemQuery.last("limit 1");
AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.getByQuery(memberTalkItemQuery);
- aiMemberTalkItemService.add(memberUuid,aiMemberTalk.getId(),2,reqContext,new Date());
-
- String prompt = this.buildPrompt(aiProductRole.getPromptHead(), aiProductRole.getPromptTemplate(), type);
+ String prompt = this.buildPrompt(aiMemberTalkItem.getContext(),reqContext,aiProductRole.getPromptHead(), aiProductRole.getPromptTemplate(), type);
List<LlmStrategyDto> llmStrategyDtoList = new ArrayList<>();
@@ -166,10 +178,11 @@
return llmStrategyFactory.getCalculationStrategyMap().get(modelName).llmInvokeStreamingNoThink(llmStrategyDtoList);
}
- private String buildPrompt(String promptHead, String promptTemplate,Integer type){
+ private String buildPrompt(String question,String answer,String promptHead, String promptTemplate,Integer type){
AiPromptJsonReq aiPromptJsonReq = new AiPromptJsonReq();
- aiPromptJsonReq.setTask(promptHead);
- aiPromptJsonReq.setRule(promptTemplate);
+ aiPromptJsonReq.setQuestion( question);
+ aiPromptJsonReq.setTask( promptHead);
+ aiPromptJsonReq.setRule( promptTemplate);
String contentByCode = AiTalkOutputEnum.HIGH_LIGHT.getContentByType(type);
aiPromptJsonReq.setOutputFormat(contentByCode);
--
Gitblit v1.9.1