From 506e28c9f7d2328650894a9c9b1dd61e40035352 Mon Sep 17 00:00:00 2001 From: v-zhangjc9 Date: Wed, 11 Jun 2025 19:44:25 +0800 Subject: [PATCH] =?UTF-8?q?feat(chat):=20=E4=BC=98=E5=8C=96=E6=8F=90?= =?UTF-8?q?=E7=A4=BA=E8=AF=8D=EF=BC=8C=E5=A2=9E=E5=8A=A0mermaid=E5=9B=BE?= =?UTF-8?q?=E8=A1=A8=E7=94=9F=E6=88=90?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../ai/chat/controller/ChatController.java | 4 +- .../service/ai/chat/tools/ChartTool.java | 66 +++++++++++++++++++ .../service/ai/chat/tools/TableTool.java | 8 ++- .../service/ai/chat/tools/YarnTool.java | 8 ++- 4 files changed, 81 insertions(+), 5 deletions(-) create mode 100644 service-ai/service-ai-chat/src/main/java/com/lanyuanxiaoyao/service/ai/chat/tools/ChartTool.java diff --git a/service-ai/service-ai-chat/src/main/java/com/lanyuanxiaoyao/service/ai/chat/controller/ChatController.java b/service-ai/service-ai-chat/src/main/java/com/lanyuanxiaoyao/service/ai/chat/controller/ChatController.java index 2e74515..6b3a3f6 100644 --- a/service-ai/service-ai-chat/src/main/java/com/lanyuanxiaoyao/service/ai/chat/controller/ChatController.java +++ b/service-ai/service-ai-chat/src/main/java/com/lanyuanxiaoyao/service/ai/chat/controller/ChatController.java @@ -4,6 +4,7 @@ import cn.hutool.core.util.ObjectUtil; import cn.hutool.core.util.StrUtil; import com.lanyuanxiaoyao.service.ai.chat.Prompts; import com.lanyuanxiaoyao.service.ai.chat.entity.MessageVO; +import com.lanyuanxiaoyao.service.ai.chat.tools.ChartTool; import com.lanyuanxiaoyao.service.ai.chat.tools.KnowledgeTool; import com.lanyuanxiaoyao.service.ai.chat.tools.TableTool; import com.lanyuanxiaoyao.service.ai.chat.tools.YarnTool; @@ -83,7 +84,8 @@ public class ChatController { } spec.tools( new TableTool(), - new YarnTool() + new YarnTool(), + new ChartTool() ); return spec; } diff --git a/service-ai/service-ai-chat/src/main/java/com/lanyuanxiaoyao/service/ai/chat/tools/ChartTool.java b/service-ai/service-ai-chat/src/main/java/com/lanyuanxiaoyao/service/ai/chat/tools/ChartTool.java new file mode 100644 index 0000000..66b4caa --- /dev/null +++ b/service-ai/service-ai-chat/src/main/java/com/lanyuanxiaoyao/service/ai/chat/tools/ChartTool.java @@ -0,0 +1,66 @@ +package com.lanyuanxiaoyao.service.ai.chat.tools; + +import com.lanyuanxiaoyao.service.ai.chat.AiChatApplication; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.ai.chat.client.ChatClient; +import org.springframework.ai.tool.annotation.Tool; +import org.springframework.ai.tool.annotation.ToolParam; + +/** + * 图表工具 + * + * @author lanyuanxiaoyao + * @version 20250611 + */ +public class ChartTool { + private static final Logger logger = LoggerFactory.getLogger(ChartTool.class); + + @Tool(description = """ + 根据需求生成mermaid图表代码 + """) + public String mermaid( + // language=TEXT + @ToolParam(description = """ + 请按以下结构描述图表需求： + 1.图表类型：[必填] 明确指定图表类型，例如：流程图(flowchart TD)、序列图(sequenceDiagram)、类图(classDiagram)、甘特图(gantt)、状态图(stateDiagram-v2)、用户旅程图(journey)、饼图(pie)、思维导图(mindmap)、实体关系图(erDiagram)等。 + 2.图表标题：[可选] 简述图表的核心主题，例如：“用户登录流程”，“订单处理系统时序交互”，“项目开发计划甘特图”。 + 3.核心内容：[必填] 描述图表逻辑： + 实体清单：列出所有节点/角色/对象（如“用户、支付系统、数据库”） + 示例 (流程图)：`用户, 登录界面, 认证服务, 数据库, 主界面, 错误提示` + 示例 (序列图)：`用户 (User), 客户端 (ClientApp), 认证服务器 (AuthServer), 数据库 (DB)` + 示例 (甘特图)：`需求分析, 设计, 开发, 测试, 部署` + 关系描述：说明实体间交互/流程顺序（如“用户提交订单 → 支付系统验证 → 数据库更新记录”） + 示例 (流程图)：`用户 输入 -> 登录界面 -> 提交 -> 认证服务 -> 验证通过？ --> 是 --> 主界面； 否 --> 错误提示` + 示例 (序列图)：`用户 -> 客户端：输入凭据； 客户端 -> 认证服务器：发送认证请求； 认证服务器 -> 数据库：查询用户信息； 数据库 --> 认证服务器：返回结果； 认证服务器 --> 客户端：返回认证结果； 客户端 --> 用户：显示结果` + 示例 (甘特图)：`需求分析 开始于 2023-10-01, 持续 5天； 设计 开始于 需求分析结束, 持续 7天； ...` + 示例（饼图）：`成功：15.25; 失败：20.22` + 关键分支：如有条件判断，说明条件及路径（如“验证失败 → 显示错误信息”） + 示例 (流程图)：`认证服务 --> |验证通过| 主界面； 认证服务 --> |验证失败| 错误提示` + 4.样式规范：[选填] 指定细节要求： + 主题：指定预定义主题，例如：default, forest, dark, neutral]` (默认是 default) + 颜色：实体颜色映射（如“用户节点用#3498db”） + 形状：特殊节点形状（如“决策节点用菱形”） + 布局：指定图表方向，例如：从上到下 (TB/TD)， 从左到右 (LR)， 从右到左 (RL)， 从下到上 (BT)]` (默认通常是 TB/TD) + 其他：箭头类型（实线/虚线）、注释文本等 + """) String request + ) { + logger.info("Enter method: mermaid[request]. request:{}", request); + ChatClient.Builder builder = AiChatApplication.getBean(ChatClient.Builder.class); + ChatClient client = builder.build(); + return client.prompt() + // language=TEXT + .system(""" + 你是一位专业的Mermaid图表专家，精通各类图表语法（流程图、序列图、类图、甘特图、状态图、饼图等），并能根据需求生成清晰、准确、符合最佳实践的Mermaid代码。 + 请根据我提供的详细描述，为我生成可直接用于渲染的Mermaid图表代码。确保代码语法正确、布局合理、易于理解。 + 输出要求： + 仅输出纯净的、可立即渲染的 Mermaid 代码。 + 不要包含任何解释性文字、Markdown 标记（除了代码块标识）或额外的说明。 + 代码格式清晰，使用适当的缩进（如果适用）。 + 确保代码语法完全符合 Mermaid 官方文档规范。 + """) + .user(request) + .call() + .content(); + } +} diff --git a/service-ai/service-ai-chat/src/main/java/com/lanyuanxiaoyao/service/ai/chat/tools/TableTool.java b/service-ai/service-ai-chat/src/main/java/com/lanyuanxiaoyao/service/ai/chat/tools/TableTool.java index d4b9972..d244a61 100644 --- a/service-ai/service-ai-chat/src/main/java/com/lanyuanxiaoyao/service/ai/chat/tools/TableTool.java +++ b/service-ai/service-ai-chat/src/main/java/com/lanyuanxiaoyao/service/ai/chat/tools/TableTool.java @@ -27,6 +27,7 @@ public class TableTool { 完整的MySQL查询语句，禁止使用除select外的任何语句。 """) String sql ) { + logger.info("Enter method: executeJdbc[sql]. sql:{}", sql); InfoService infoService = AiChatApplication.getBean(InfoService.class); String result = infoService.jdbc(sql) .collect(map -> map.valuesView().makeString(",")) @@ -44,6 +45,7 @@ public class TableTool { Hudi表：hudi Hive表：hive 逻辑表：logic + 一次调用只能传一个类型，不支持多个类型同时查询 """) String type ) { logger.info("Enter method: tableCount[type]. type:{}", type); @@ -69,13 +71,15 @@ public class TableTool { """) public String version( @ToolParam(description = """ - 查询指定日期的跨天情况，格式为yyyyMMdd（如2025年6月6日取值为20250606），如果参数为空则查询当天跨天情况 - """, required = false) + 查询指定日期的跨天情况，格式为yyyyMMdd（如2025年6月6日取值为20250606）。 + 一次调用只能传一个日期，不支持多个日期同时查询 + """) String date, @ToolParam(description = """ 表类型，取值如下 普通表：normal 重点表：focus + 一次调用只能传一个类型，不支持多个类型同时查询 """) String type ) { diff --git a/service-ai/service-ai-chat/src/main/java/com/lanyuanxiaoyao/service/ai/chat/tools/YarnTool.java b/service-ai/service-ai-chat/src/main/java/com/lanyuanxiaoyao/service/ai/chat/tools/YarnTool.java index 81cb8c4..6c6a24c 100644 --- a/service-ai/service-ai-chat/src/main/java/com/lanyuanxiaoyao/service/ai/chat/tools/YarnTool.java +++ b/service-ai/service-ai-chat/src/main/java/com/lanyuanxiaoyao/service/ai/chat/tools/YarnTool.java @@ -24,7 +24,8 @@ public class YarnTool { """) public Double yarnStatus( @ToolParam(description = """ - yarn集群名称，取值有b12、b1和a4 + yarn集群名称，取值有b12、b1和a4； + 一次调用只能查询一个集群，不支持多个集群同时查询 """) String cluster ) { logger.info("Enter method: yarnStatus[cluster]. cluster:{}", cluster); @@ -38,7 +39,8 @@ public class YarnTool { """) public Double yarnQueueStatus( @ToolParam(description = """ - yarn集群名称，取值有b12、b1和a4 + yarn集群名称，取值有b12、b1和a4； + 一次调用只能查询一个集群，不支持多个集群同时查询 """) String cluster, @ToolParam(description = """ yarn队列名称 @@ -56,11 +58,13 @@ public class YarnTool { public String yarnTaskStatus( @ToolParam(description = """ yarn集群名称，取值有b12、b1和a4 + 一次调用只能查询一个集群，不支持多个集群同时查询 """) String cluster, @ToolParam(description = """ 查询任务种类，取值如下 同步任务：Sync 压缩任务：Compaction + 一次调用只能传一个类型，不支持多个类型同时查询 """) String type ) { logger.info("Enter method: yarnTaskStatus[cluster, type]. cluster:{},type:{}", cluster, type);