From 6477d7156272a6f1fe126c781958369bb10970c6 Mon Sep 17 00:00:00 2001
From: zhou zhou <3272660260@qq.com>
Date: 星期六, 21 三月 2026 11:15:50 +0800
Subject: [PATCH] #ai 思维链
---
rsf-server/src/main/java/com/vincent/rsf/server/ai/service/impl/BuiltinMcpToolRegistryImpl.java | 41 +++++++++++++++++++++++++++++------------
1 files changed, 29 insertions(+), 12 deletions(-)
diff --git a/rsf-server/src/main/java/com/vincent/rsf/server/ai/service/impl/BuiltinMcpToolRegistryImpl.java b/rsf-server/src/main/java/com/vincent/rsf/server/ai/service/impl/BuiltinMcpToolRegistryImpl.java
index 2c4b725..c04d84b 100644
--- a/rsf-server/src/main/java/com/vincent/rsf/server/ai/service/impl/BuiltinMcpToolRegistryImpl.java
+++ b/rsf-server/src/main/java/com/vincent/rsf/server/ai/service/impl/BuiltinMcpToolRegistryImpl.java
@@ -28,6 +28,11 @@
private final RsfWmsTaskTools rsfWmsTaskTools;
private final RsfWmsBaseTools rsfWmsBaseTools;
+ /**
+ * 鏍￠獙鍐呯疆 MCP 缂栫爜鏄惁鍚堟硶銆�
+ * 褰撳墠鐗堟湰鍙厑璁镐娇鐢ㄦ樉寮忕櫥璁板湪娉ㄥ唽琛ㄤ腑鐨勭紪鐮侊紝鏈煡缂栫爜鐩存帴鎷掔粷锛�
+ * 杩欐牱鍙互纭繚鈥滈〉闈㈠彲閫夐」鈥濆拰鈥滆繍琛屾椂鍙寕杞介」鈥濆缁堜竴鑷淬��
+ */
@Override
public void validateBuiltinCode(String builtinCode) {
if (!StringUtils.hasText(builtinCode)) {
@@ -38,6 +43,10 @@
}
}
+ /**
+ * 鏍规嵁鎸傝浇璁板綍鍒涘缓鍐呯疆宸ュ叿鍥炶皟銆�
+ * 杩欓噷涓嶄細鍋氫换浣曞姩鎬佸彂鐜帮紝鎵�鏈夊伐鍏烽兘蹇呴』缁忚繃鏄惧紡娉ㄥ唽鍜屾不鐞嗙洰褰曟牎楠屽悗鎵嶈兘鏆撮湶缁欐ā鍨嬨��
+ */
@Override
public List<ToolCallback> createToolCallbacks(AiMcpMount mount, Long userId) {
String builtinCode = mount.getBuiltinCode();
@@ -52,6 +61,10 @@
throw new CoolException("涓嶆敮鎸佺殑鍐呯疆 MCP 缂栫爜: " + builtinCode);
}
+ /**
+ * 杩斿洖鏌愪釜鍐呯疆缂栫爜涓嬪彲棰勮鐨勫伐鍏风洰褰曚俊鎭��
+ * 璇ョ洰褰曟瘮杩愯鏃跺洖璋冨浜嗗伐鍏风敤閫斻�佹煡璇㈣竟鐣屽拰绀轰緥鎻愰棶锛屼緵绠＄悊椤靛睍绀恒��
+ */
@Override
public List<AiMcpToolPreviewDto> listBuiltinToolCatalog(String builtinCode) {
validateBuiltinCode(builtinCode);
@@ -62,6 +75,11 @@
}
private List<ToolCallback> createValidatedCallbacks(Object toolBean, String builtinCode) {
+ /**
+ * 鎶� `@Tool` Bean 杞垚 Spring AI ToolCallback锛屽苟寮哄埗鏍￠獙锛�
+ * 1. 宸ュ叿鍚嶅繀椤荤鍚堝懡鍚嶈鑼�
+ * 2. 姣忎釜宸ュ叿閮藉繀椤诲嚭鐜板湪娌荤悊鐩綍閲�
+ */
List<ToolCallback> callbacks = Arrays.asList(ToolCallbacks.from(toolBean));
Map<String, AiMcpToolPreviewDto> catalog = catalogByBuiltinCode(builtinCode);
for (ToolCallback callback : callbacks) {
@@ -80,10 +98,15 @@
}
private List<String> supportedBuiltinCodes() {
+ /** 褰撳墠鐗堟湰鍏佽鎸傝浇鐨勫叏閮ㄥ唴缃� MCP 缂栫爜銆� */
return List.of(AiDefaults.MCP_BUILTIN_RSF_WMS);
}
private Map<String, AiMcpToolPreviewDto> catalogByBuiltinCode(String builtinCode) {
+ /**
+ * 鏋勯�犲唴缃伐鍏锋不鐞嗙洰褰曘��
+ * 杩欓噷鐨勭洰褰曟槸杩愯鏃舵牎楠屽拰绠＄悊绔瑙堢殑鍏卞悓浜嬪疄鏉ユ簮锛屼笉鑳戒笌宸ュ叿瀹炵幇鑴辫妭銆�
+ */
if (AiDefaults.MCP_BUILTIN_RSF_WMS.equals(builtinCode)) {
Map<String, AiMcpToolPreviewDto> catalog = new LinkedHashMap<>();
catalog.put("rsf_query_available_inventory", buildCatalogItem(
@@ -100,19 +123,12 @@
"蹇呴』鎻愪緵绔欑偣绫诲瀷鍒楄〃锛岀被鍨嬫暟閲忔渶澶� 10 涓紝鏈�澶氳繑鍥� 50 涓珯鐐广��",
List.of("鏌ヨ鍏ュ簱鍜屽嚭搴撲綔涓氬彲鐢ㄧ珯鐐�", "鍒楀嚭 AGV_PICK 绫诲瀷鐨勪綔涓氱珯鐐�")
));
- catalog.put("rsf_query_task_list", buildCatalogItem(
- "rsf_query_task_list",
+ catalog.put("rsf_query_task", buildCatalogItem(
+ "rsf_query_task",
"浠诲姟鏌ヨ",
- "鎸変换鍔″彿銆佺姸鎬併�佺被鍨嬫垨绔欑偣鏉′欢鏌ヨ浠诲姟鍒楄〃銆�",
- "鑷冲皯鎻愪緵涓�涓繃婊ゆ潯浠讹紝鏈�澶氳繑鍥� 50 鏉′换鍔¤褰曪紝涓嶆敮鎸佸叏琛ㄦ壂鎻忋��",
- List.of("鏌ヨ鏈�杩� 10 鏉＄姸鎬佷负鎵ц涓殑浠诲姟", "鎸変换鍔″彿鍏抽敭瀛楁煡璇换鍔″垪琛�")
- ));
- catalog.put("rsf_query_task_detail", buildCatalogItem(
- "rsf_query_task_detail",
- "浠诲姟鏌ヨ",
- "鎸変换鍔� ID 鎴栦换鍔″彿鏌ヨ鍗曚釜浠诲姟璇︽儏銆�",
- "蹇呴』鎻愪緵浠诲姟 ID 鎴栦换鍔″彿涔嬩竴锛屽彧杩斿洖鍗曚釜浠诲姟銆�",
- List.of("鏌ヨ浠诲姟 12345 鐨勮鎯�", "鏍规嵁浠诲姟鍙� TASK24001 鏌ョ湅鎵ц鏄庣粏")
+ "鎸変换鍔″彿銆佺姸鎬併�佺被鍨嬫垨绔欑偣鏉′欢鏌ヨ浠诲姟锛涙敮鎸佷粠鑷劧璇█涓嚜鍔ㄦ彁鍙栦换鍔″彿锛岀簿纭懡涓椂杩斿洖浠诲姟鏄庣粏銆�",
+ "杩囨护鏉′欢鍧囦负鍙�夛紝涓嶄紶杩囨护鏉′欢鏃堕粯璁よ繑鍥炴渶杩戜换鍔★紝鏈�澶氳繑鍥� 50 鏉¤褰曘��",
+ List.of("鏌ヨ鏈�杩� 10 鏉′换鍔�", "鏌ヨ浠诲姟鍙� TASK24001 鐨勮鎯�")
));
catalog.put("rsf_query_warehouses", buildCatalogItem(
"rsf_query_warehouses",
@@ -142,6 +158,7 @@
private AiMcpToolPreviewDto buildCatalogItem(String name, String toolGroup, String toolPurpose,
String queryBoundary, List<String> exampleQuestions) {
+ /** 缁熶竴鍒涘缓宸ュ叿鐩綍鏉＄洰锛岄伩鍏嶄笉鍚屽伐鍏风粍鍑虹幇瀛楁椋庢牸涓嶄竴鑷淬�� */
return AiMcpToolPreviewDto.builder()
.name(name)
.toolGroup(toolGroup)
--
Gitblit v1.9.1