lanyuanxiaoyao/hudi-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(chat): 优化提示词，时间参数直接放在提示词中

Browse Source
This commit is contained in:
v-zhangjc9
2025-06-04 17:42:46 +08:00
parent 6d4dedc3f4
commit 4124a8a851
6 changed files with 110 additions and 75 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
service-ai/service-ai-chat/src/main/java/com/lanyuanxiaoyao/service/ai/chat/controller/ChatController.java
View File

@@ -3,9 +3,10 @@ package com.lanyuanxiaoyao.service.ai.chat.controller;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import com.lanyuanxiaoyao.service.ai.chat.entity.MessageVO;
import com.lanyuanxiaoyao.service.ai.chat.tools.DatetimeTools;
import com.lanyuanxiaoyao.service.forest.service.KnowledgeService;
import java.io.IOException;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.Optional;
import org.eclipse.collections.api.list.ImmutableList;
import org.slf4j.Logger;
@@ -35,6 +36,7 @@ import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
@RequestMapping("chat")
public class ChatController {
private static final Logger logger = LoggerFactory.getLogger(ChatController.class);
private final static DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
private static final String ROLE_ASSISTANT = "assistant";
private static final String ROLE_USER = "user";
@@ -48,26 +50,29 @@ public class ChatController {
}
private ChatClient.ChatClientRequestSpec buildRequest(Long knowledgeId, ImmutableList<MessageVO> messages) {
var systemPromptBuilder = new StringBuilder();
systemPromptBuilder.append("""
你是一名专业的AI运维助手，负责“Hudi数据同步服务平台”的运维工作；
你将会友好地帮助用户解答关于该平台运维工作的问题，你会尽可能通过各种方式获取知识和数据来解答；
对于无法通过已有知识回答的问题，你会提示用户你无法解答该问题，而不是虚构不存在的数据或答案；
对于与该平台无关的问题，你会委婉地拒绝用户，并提示无法回答；
你将始终在中文语境下进行对话。
""");
var builder = StrUtil.builder()
.append(StrUtil.format("""
你是一名专业的AI运维助手，负责“Hudi数据同步服务平台”的运维工作；
你将会友好地帮助用户解答关于该平台运维工作的问题，你会尽可能通过各种方式获取知识和数据来解答；
对于无法通过已有知识回答的问题，你会提示用户你无法解答该问题，而不是虚构不存在的数据或答案；
对于与该平台无关的问题，你会委婉地拒绝用户，并提示无法回答；
你将始终在中文语境下进行对话。
当前时间为：{}
""", LocalDateTime.now().format(formatter)));
if (ObjectUtil.isNotNull(knowledgeId)) {
var vo = messages.select(message -> StrUtil.equals(message.getRole(), "user")).getLastOptional().orElseThrow();
var documents = knowledgeService.query(knowledgeId, vo.getContent());
logger.info("Knowledge id:{}, content:{}", knowledgeId, vo.getContent());
var documents = knowledgeService.query(knowledgeId, vo.getContent(), 0.5);
if (ObjectUtil.isNotEmpty(documents)) {
systemPromptBuilder.append("以下是与用户问题有关的外部知识，优先利用该知识回答用户的提问：\n");
systemPromptBuilder.append(documents.makeString("\n"));
builder.append(StrUtil.format("""
以下是与用户问题有关的外部知识，优先利用该知识回答用户的提问：
{}
""", documents.makeString("\n")));
}
}
return chatClient.prompt()
.system(systemPromptBuilder.toString())
.tools(new DatetimeTools())
.system(builder.toString())
.messages(
messages
.collect(message -> StrUtil.equals(message.getRole(), ROLE_ASSISTANT)

18
service-ai/service-ai-chat/src/main/java/com/lanyuanxiaoyao/service/ai/chat/tools/DatetimeTools.java
View File

@@ -1,18 +0,0 @@
package com.lanyuanxiaoyao.service.ai.chat.tools;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import org.springframework.ai.tool.annotation.Tool;
/**
* @author lanyuanxiaoyao
* @version 20250516
*/
public class DatetimeTools {
private final static DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
@Tool(description = "获取当前日期和时间")
public String getCurrentDateTime() {
return LocalDateTime.now().format(formatter);
}
}