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/controller/TestController.java | 30 ++++++++++++++++--------------
src/main/java/cc/mrbird/febs/ai/service/impl/ApiMemberTalkStreamServiceImpl.java | 6 +++++-
src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliLlmStrategyServiceImpl.java | 4 ++++
3 files changed, 25 insertions(+), 15 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/controller/TestController.java b/src/main/java/cc/mrbird/febs/ai/controller/TestController.java
index 811117b..a25deb7 100644
--- a/src/main/java/cc/mrbird/febs/ai/controller/TestController.java
+++ b/src/main/java/cc/mrbird/febs/ai/controller/TestController.java
@@ -1,6 +1,7 @@
package cc.mrbird.febs.ai.controller;
import cc.mrbird.febs.ai.entity.AiMember;
+import cc.mrbird.febs.ai.enumerates.AiPromptEnum;
import cc.mrbird.febs.ai.mapper.AiMemberMapper;
import cc.mrbird.febs.ai.req.talk.AiTalkAnswerStream;
import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkStreamVo;
@@ -221,6 +222,7 @@
.messages(Arrays.asList(systemMsg, userMsg))
// .resultFormat(GenerationParam.ResultFormat.TEXT)
.resultFormat(GenerationParam.ResultFormat.MESSAGE)
+ .enableThinking( true)
.incrementalOutput(true)
.build();
return gen.streamCall(param);
@@ -251,22 +253,22 @@
public static void main(String[] args) {
//定义一个开始时间为启动这个main方法的开始时间，用于计算运行时间
long startTime = System.currentTimeMillis();
- String question = "干的漂亮";
- String prompt = "{\"task\": \"作为专业的表扬能力评估分析师，分析question和用户输入内容，对用户输入内容进行评估与指导。\",\"question\":\"当资深骨干员工在突发场景中迅速响应并解决了持续一周的客户投诉达成客户满意度成就时，作为上级的你应如何表达赞赏？\",\"highlight\": \"识别符合表扬原则的积极策略，关注角色视角适配性、员工类型特点、具体行为描述、情感表达真诚度与个性化认可方式\"}";
- String prompt1 = "{\"task\": \"作为专业的表扬能力评估分析师，分析question和用户输入内容，对用户输入内容进行评估与指导。\",\"question\":\"当资深骨干员工在突发场景中迅速响应并解决了持续一周的客户投诉达成客户满意度成就时，作为上级的你应如何表达赞赏？\",\"suggestion\": \"提供针对性优化方案，涵盖场景要素匹配度提升（如正式场合需增加公开认可环节）、成就类型适配改进（如创新贡献应强调思维突破而非仅结果）、具体维度强化方向（如增加 STAR 法则中的 Result 部分描述）、跨文化 / 远程场景特殊策略（如虚拟团队可采用异步表扬 + 公开认可组合）\"}";
+ String question = "下属负责的项目高质量按期交付，除了表扬，你希望 下属要多注意和同事的沟通，应如何表达";
+ String prompt = AiPromptEnum.STREAM_NORMAL.getPrompt();
+ String prompt1 = "{ \"question\":\"下属负责的项目高质量按期交付，除了表扬，你希望 下属要多注意和同事的沟通，应如何表达\", \"task\": \"作为专业的表扬能力评估分析师，根据表扬场景题目及用户输入内容，实现对表扬能力的全方位评估与指导。\", \"rules\": [ \"用户输入内容需符合问题中的场景要素与角色设定，若无关则返回：[请回答提出的问题]\", \"生成四维评价体系[evaluation]，每部分不少于50字：\", \" - 亮点[highlight]：识别符合表扬原则的积极策略，关注角色视角适配性、员工类型特点、具体行为描述、情感表达真诚度与个性化认可方式\", \" - 建议[suggestion]：提供针对性优化方案，涵盖场景要素匹配度、成就类型适配、STAR法则强化、跨文化/远程场景策略\", \" - 参考答案[referenceAnswer]：遵循平级表扬'三明治法则'、危机处理三维度、长期贡献认可策略\", \" - 核心知识点[keyKnowledge]：包含心理学依据、组织行为学理论、标杆案例、文化适配分析\", \"生成五维评分模型[radarDataItems]，采用1-10分制：\", \" - 针对性(problemUnderstanding)：表扬内容与实际贡献匹配度\", \" - 具体性(fluency)：量化成果、行为细节、独特贡献描述\", \" - 情感温度(principleAdherence)：语言真诚度和情感连接强度\", \" - 激励性(logicality)：对后续工作动力和行为导向的积极影响\", \" - 文化契合度(knowledgeMastery)：表扬方式与组织价值观一致性\", \"评分标准：1-4分(需重大改进)、5-7分(基本合格)、8-10分(优秀示范)\", \"特殊场景处理：\", \" - 下级对上级表扬：评估谦逊度与事实依据，避免过度恭维\", \" - 跨部门协作场景：关注协作价值强调与横向影响力评价\", \" - 远程团队情境：评价异步沟通适应性与技术工具合理运用\" ], \"output_format\": { \"type\": \"json\", \"structure\": { \"evaluation\": { \"highlight\": \"字符串(≥50字)\", \"suggestion\": \"字符串(≥50字)\", \"referenceAnswer\": \"字符串\", \"keyKnowledge\": \"字符串(≥50字)\" }, \"radarDataItems\": [ { \"code\": \"problemUnderstanding\", \"score\": \"整数(1-10)\", \"name\": \"针对性\" }, { \"code\": \"fluency\", \"score\": \"整数(1-10)\", \"name\": \"具体性\" }, { \"code\": \"principleAdherence\", \"score\": \"整数(1-10)\", \"name\": \"情感温度\" }, { \"code\": \"logicality\", \"score\": \"整数(1-10)\", \"name\": \"激励性\" }, { \"code\": \"knowledgeMastery\", \"score\": \"整数(1-10)\", \"name\": \"文化契合度\" } ] } }}";
String prompt2 = "{\"task\": \"作为专业的表扬能力评估分析师，分析question和用户输入内容，对用户输入内容进行评估与指导。\",\"question\":\"当资深骨干员工在突发场景中迅速响应并解决了持续一周的客户投诉达成客户满意度成就时，作为上级的你应如何表达赞赏？\",\"reference_answer\": \"根据问题用户的回答，生成参考示例\"}";
String prompt3 = "{\"task\":\"假如你是一个表扬题目生成专家，你将根据多样化结构表扬场景题目的生成需求，来解决生成多样化结构的表扬场景题目的任务。示例1：作为跨部门同事，如何在视频会议中认可技术研发人员提前完成创新改进项目的贡献？示例2：当实习生在团队庆功宴的危机处理中成功化解客户投诉时，作为部门经理应如何表达赞赏？示例3：在年度颁奖典礼上，基层管理者发现远程办公团队成员凭借长期稳定的工作表现达成长期贡献，适合采取何种表扬方式？\",\"rules\":\"要求：1 输出一个符合上述规则生成的表扬场景题目。2 题目需包含角色视角、员工类型、成就类型、场景要素等关键信息，且符合三种结构中的一种，并带有具体细节描述。3 无论用户输入什么内容只生成 1 个独特题目。\",\"output\":\"纯文本输出\"}";
- GenerationResult result = null;
- try {
- result = callWithMessage(question, prompt);
- } catch (NoApiKeyException e) {
- e.printStackTrace();
- } catch (InputRequiredException e) {
- e.printStackTrace();
- }
- System.out.println(result.getOutput().getChoices().get(0).getMessage().getContent());
- long endTime = System.currentTimeMillis();
- System.out.println("运行时间：" + (endTime - startTime) + "毫秒");
+// GenerationResult result = null;
+// try {
+// result = callWithMessage(question, prompt);
+// } catch (NoApiKeyException e) {
+// e.printStackTrace();
+// } catch (InputRequiredException e) {
+// e.printStackTrace();
+// }
+// System.out.println(result.getOutput().getChoices().get(0).getMessage().getContent());
+// long endTime = System.currentTimeMillis();
+// System.out.println("运行时间：" + (endTime - startTime) + "毫秒");
long startTimeStream = System.currentTimeMillis();
Flowable<GenerationResult> resultStream = null;
try {
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());
diff --git a/src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliLlmStrategyServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliLlmStrategyServiceImpl.java
index f9ad98f..ec12f19 100644
--- a/src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliLlmStrategyServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliLlmStrategyServiceImpl.java
@@ -7,6 +7,7 @@
import cc.mrbird.febs.common.exception.FebsException;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
+import cn.hutool.json.JSONUtil;
import com.alibaba.dashscope.aigc.generation.Generation;
import com.alibaba.dashscope.aigc.generation.GenerationParam;
import com.alibaba.dashscope.aigc.generation.GenerationResult;
@@ -118,9 +119,12 @@
HashMap<String, String> stringStringHashMap = new HashMap<>();
if (StrUtil.isNotEmpty(message.getOutput().getChoices().get(0).getMessage().getReasoningContent())){
stringStringHashMap.put(LlmStrategyContextEnum.THINK.name(),message.getOutput().getChoices().get(0).getMessage().getReasoningContent());
+
+ System.out.print(message.getOutput().getChoices().get(0).getMessage().getReasoningContent());
}
if (StrUtil.isNotEmpty(message.getOutput().getChoices().get(0).getMessage().getContent())){
stringStringHashMap.put(LlmStrategyContextEnum.CONTENT.name(),message.getOutput().getChoices().get(0).getMessage().getContent());
+ System.out.print(message.getOutput().getChoices().get(0).getMessage().getContent());
}
return new FebsResponse().success().data(stringStringHashMap);
})
--
Gitblit v1.9.1