From 8a7b666f0bed7c1dc03767166c7ecd602552a5c8 Mon Sep 17 00:00:00 2001
From: liusuyi <1951119284@qq.com>
Date: Mon, 18 May 2026 15:24:46 +0800
Subject: [PATCH] 优化
---
ard-modules/ard-modules-agent/src/main/java/com/ard/agent/service/AIChatService.java | 143 ++++++++++-------------------------------------
1 files changed, 30 insertions(+), 113 deletions(-)
diff --git a/ard-modules/ard-modules-agent/src/main/java/com/ard/agent/service/AIChatService.java b/ard-modules/ard-modules-agent/src/main/java/com/ard/agent/service/AIChatService.java
index e03a52a..0d09721 100644
--- a/ard-modules/ard-modules-agent/src/main/java/com/ard/agent/service/AIChatService.java
+++ b/ard-modules/ard-modules-agent/src/main/java/com/ard/agent/service/AIChatService.java
@@ -2,24 +2,12 @@
import com.alibaba.cloud.ai.graph.RunnableConfig;
import com.alibaba.cloud.ai.graph.agent.ReactAgent;
-import com.alibaba.cloud.ai.graph.checkpoint.savers.MemorySaver;
import com.alibaba.cloud.ai.graph.exception.GraphRunnerException;
-import com.ard.agent.tool.KnowledgeTools;
-import com.ard.agent.tool.MonitorTools;
-import com.ard.agent.tool.SystemTools;
import com.ard.common.security.utils.SecurityUtils;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
-import org.springframework.ai.chat.client.ChatClient;
-import org.springframework.ai.chat.client.advisor.MessageChatMemoryAdvisor;
import org.springframework.ai.chat.messages.AssistantMessage;
-import org.springframework.ai.chat.messages.UserMessage;
-import org.springframework.ai.chat.model.ChatModel;
-import org.springframework.ai.chat.model.ChatResponse;
-import org.springframework.ai.chat.prompt.Prompt;
import org.springframework.ai.content.Content;
-import org.springframework.ai.support.ToolCallbacks;
-import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.stereotype.Service;
import reactor.core.publisher.Flux;
@@ -27,110 +15,15 @@
@Service
public class AIChatService {
- @Resource
- private ChatClient chatClient;
- @Resource
- @Qualifier("ollamaChatModel")
- private ChatModel chatModel;
- @Resource
- private SystemTools systemTools; // 注入工具类
- @Resource
- private MonitorTools monitorTools; // 注入工具类
- @Resource
- private KnowledgeTools knowledgeTools;
- @Resource
- private ReactAgent cameraAgent; // 注入配置好的 Agent
- private static final String SYSTEM_PROMPT = """
- 你是摄像头管理助手。请严格遵守以下规则：
-
- 1. **工具优先原则**：当用户问题涉及摄像机的实时状态、修改操作、查询具体设备信息时，
- 必须调用对应的工具（getCameraDetail/searchCameras/updateCameraInfo），
- 绝对不要使用自己的记忆或上下文中的静态文本来回答。
-
- 2. **知识库使用场景**：只有当用户询问操作指南、故障排查手册、产品说明等静态知识时，
- 才使用知识库中的内容。
-
- 3. **冲突处理**：如果知识库中提到了某类操作，但用户要求执行具体操作，
- 必须以工具调用的结果为准。
-
- ========== 输出格式 ==========
- 使用 Markdown 格式输出：
- - ## 标题
- - - 列表项
- - **加粗** 强调
- - 表格用于数据展示
- """;
+ @Resource(name = "unifiedAgent")
+ private ReactAgent unifiedAgent;
+
/**
- * 流式对话
+ * 唯一流式聊天入口
*/
public Flux<String> streamChat(String userMessage) {
- Long userId = SecurityUtils.getUserId();
- return chatClient.prompt()
- .system(SYSTEM_PROMPT)
- .user(userMessage)
- .advisors(advisor -> advisor.param("chat_memory_conversation_id", userId))
- .tools(systemTools,monitorTools, knowledgeTools)
- .stream()
- .content();
- }
-
- /**
- * 对话服务入口
- */
- public String chat(String userMessage) throws GraphRunnerException {
- Long userId = SecurityUtils.getUserId();
- // 1. 为每个用户创建唯一的 threadId
- RunnableConfig config = RunnableConfig.builder()
- .threadId(userId.toString()) // 比如从 SecurityUtils.getUserId() 获取
- .build();
- // 3. 调用 Agent 的 call 方法并执行
- AssistantMessage response = cameraAgent.call(userMessage,config);
- // 4. 从响应中提取文本内容
- return response.getText();
- }
-
- private static final String SYSTEM_PROMPT1 = """
- 你是智能管理助手，拥有以下能力：
-
- ## 可用工具
- ### 摄像头管理
- - getCameraDetail: 查询摄像机详情
- - searchCameras: 搜索摄像机
- - updateCameraInfo: 修改摄像机信息
-
- ### 系统管理
- - searchUsers: 搜索用户
- - searchDepts: 搜索部门
- - searchRoles: 搜索角色
- - getUserRoles: 查询用户角色
-
- ### 知识库
- - searchKnowledgeBase: 查询知识文档
-
- ## 原则
- 1. 根据问题自动选择工具
- 2. 数据必须来自工具调用
- 3. 用 Markdown 格式回复
- """;
-
- /**
- * 流式聊天
- */
- public Flux<String> streamChat1(String userMessage) {
- Long userId = SecurityUtils.getUserId();
- RunnableConfig config = RunnableConfig.builder()
- .threadId(userId.toString())
- .build();
- ReactAgent agent = ReactAgent.builder()
- .name("UnifiedAgent")
- .model(chatModel)
- .instruction(SYSTEM_PROMPT1)
- .tools(ToolCallbacks.from(monitorTools, systemTools, knowledgeTools))
- .saver(new MemorySaver())
- .build();
-
try {
- return agent.streamMessages(userMessage, config)
+ return unifiedAgent.streamMessages(userMessage, runnableConfig())
.filter(msg -> msg instanceof AssistantMessage)
.map(Content::getText)
.doOnError(error -> log.error("Agent执行出错: {}", error.getMessage()));
@@ -138,4 +31,28 @@
throw new RuntimeException(e);
}
}
-}
\ No newline at end of file
+
+ /**
+ * 唯一同步聊天入口
+ */
+ public String chat(String userMessage) throws GraphRunnerException {
+ AssistantMessage response = unifiedAgent.call(userMessage, runnableConfig());
+ return response.getText();
+ }
+
+ private RunnableConfig runnableConfig() {
+ return RunnableConfig.builder()
+ .threadId(currentThreadId())
+ .build();
+ }
+
+ private String currentThreadId() {
+ Long userId;
+ try {
+ userId = SecurityUtils.getUserId();
+ } catch (Exception e) {
+ userId = null;
+ }
+ return userId == null ? "anonymous" : userId.toString();
+ }
+}
--
Gitblit v1.9.3