From 574bc2e8dba9d1d11be124f2054ba2243fca125c Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Wed, 03 Sep 2025 13:43:21 +0800
Subject: [PATCH] feat(ai): 优化 prompt 构建逻辑并添加思考内容输出- 在 AliLlmStrategyServiceImpl 中添加了对思考内容的输出- 在 ApiMemberTalkStreamServiceImpl 中优化了 prompt 的构建逻辑 - 在 TestController 中调整了测试用例,使用枚举生成 prompt- 在 TestController 中添加了对思考内容的输出

---
 src/main/java/cc/mrbird/febs/ai/service/impl/ApiMemberTalkStreamServiceImpl.java |    6 +++++-
 1 files changed, 5 insertions(+), 1 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 d0bd09f..492bdf5 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
@@ -165,12 +165,16 @@
         memberTalkItemQuery.orderByDesc(AiMemberTalkItem::getCreatedTime);
         memberTalkItemQuery.last("limit 1");
         AiMemberTalkItem aiMemberTalkItem = aiMemberTalkItemService.getByQuery(memberTalkItemQuery);
-        String prompt = this.buildPrompt(aiMemberTalkItem.getContext(),reqContext,aiProductRole.getPromptHead(), aiProductRole.getPromptTemplate(), type);
+        String question = aiMemberTalkItem.getContext();
+
+        String prompt = this.buildPrompt(question,reqContext,aiProductRole.getPromptHead(), aiProductRole.getPromptTemplate(), type);
 
 
         List<LlmStrategyDto> llmStrategyDtoList = new ArrayList<>();
         LlmStrategyDto llmStrategyDto = this.buildLlmStrategyDtoList(prompt, 1);
         llmStrategyDtoList.add(llmStrategyDto);
+        llmStrategyDto = this.buildLlmStrategyDtoList(question, 3);
+        llmStrategyDtoList.add(llmStrategyDto);
         llmStrategyDto = this.buildLlmStrategyDtoList(reqContext, 2);
         llmStrategyDtoList.add(llmStrategyDto);
         String modelName = LlmStrategyEnum.getName(aiService.getSystemSetAiType());

--
Gitblit v1.9.1