From b0728aba5c01842e24da3cff04e44be06c6bb655 Mon Sep 17 00:00:00 2001
From: zhou zhou <3272660260@qq.com>
Date: 星期四, 19 三月 2026 13:38:38 +0800
Subject: [PATCH] #AI.去除多余mcp
---
rsf-server/src/main/java/com/vincent/rsf/server/ai/service/impl/BuiltinMcpToolRegistryImpl.java | 38 ++++++--------------------------------
1 files changed, 6 insertions(+), 32 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 41f0d52..2c4b725 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
@@ -44,19 +44,10 @@
validateBuiltinCode(builtinCode);
if (AiDefaults.MCP_BUILTIN_RSF_WMS.equals(builtinCode)) {
List<ToolCallback> callbacks = new ArrayList<>();
- callbacks.addAll(createValidatedCallbacks(rsfWmsStockTools, AiDefaults.MCP_BUILTIN_RSF_WMS_STOCK));
- callbacks.addAll(createValidatedCallbacks(rsfWmsTaskTools, AiDefaults.MCP_BUILTIN_RSF_WMS_TASK));
- callbacks.addAll(createValidatedCallbacks(rsfWmsBaseTools, AiDefaults.MCP_BUILTIN_RSF_WMS_BASE));
+ callbacks.addAll(createValidatedCallbacks(rsfWmsStockTools, builtinCode));
+ callbacks.addAll(createValidatedCallbacks(rsfWmsTaskTools, builtinCode));
+ callbacks.addAll(createValidatedCallbacks(rsfWmsBaseTools, builtinCode));
return callbacks;
- }
- if (AiDefaults.MCP_BUILTIN_RSF_WMS_STOCK.equals(builtinCode)) {
- return createValidatedCallbacks(rsfWmsStockTools, builtinCode);
- }
- if (AiDefaults.MCP_BUILTIN_RSF_WMS_TASK.equals(builtinCode)) {
- return createValidatedCallbacks(rsfWmsTaskTools, builtinCode);
- }
- if (AiDefaults.MCP_BUILTIN_RSF_WMS_BASE.equals(builtinCode)) {
- return createValidatedCallbacks(rsfWmsBaseTools, builtinCode);
}
throw new CoolException("涓嶆敮鎸佺殑鍐呯疆 MCP 缂栫爜: " + builtinCode);
}
@@ -65,11 +56,7 @@
public List<AiMcpToolPreviewDto> listBuiltinToolCatalog(String builtinCode) {
validateBuiltinCode(builtinCode);
if (AiDefaults.MCP_BUILTIN_RSF_WMS.equals(builtinCode)) {
- List<AiMcpToolPreviewDto> catalog = new ArrayList<>();
- catalog.addAll(catalogByBuiltinCode(AiDefaults.MCP_BUILTIN_RSF_WMS_STOCK).values());
- catalog.addAll(catalogByBuiltinCode(AiDefaults.MCP_BUILTIN_RSF_WMS_TASK).values());
- catalog.addAll(catalogByBuiltinCode(AiDefaults.MCP_BUILTIN_RSF_WMS_BASE).values());
- return catalog;
+ return new ArrayList<>(catalogByBuiltinCode(builtinCode).values());
}
return new ArrayList<>(catalogByBuiltinCode(builtinCode).values());
}
@@ -93,16 +80,11 @@
}
private List<String> supportedBuiltinCodes() {
- return List.of(
- AiDefaults.MCP_BUILTIN_RSF_WMS,
- AiDefaults.MCP_BUILTIN_RSF_WMS_STOCK,
- AiDefaults.MCP_BUILTIN_RSF_WMS_TASK,
- AiDefaults.MCP_BUILTIN_RSF_WMS_BASE
- );
+ return List.of(AiDefaults.MCP_BUILTIN_RSF_WMS);
}
private Map<String, AiMcpToolPreviewDto> catalogByBuiltinCode(String builtinCode) {
- if (AiDefaults.MCP_BUILTIN_RSF_WMS_STOCK.equals(builtinCode)) {
+ if (AiDefaults.MCP_BUILTIN_RSF_WMS.equals(builtinCode)) {
Map<String, AiMcpToolPreviewDto> catalog = new LinkedHashMap<>();
catalog.put("rsf_query_available_inventory", buildCatalogItem(
"rsf_query_available_inventory",
@@ -118,10 +100,6 @@
"蹇呴』鎻愪緵绔欑偣绫诲瀷鍒楄〃锛岀被鍨嬫暟閲忔渶澶� 10 涓紝鏈�澶氳繑鍥� 50 涓珯鐐广��",
List.of("鏌ヨ鍏ュ簱鍜屽嚭搴撲綔涓氬彲鐢ㄧ珯鐐�", "鍒楀嚭 AGV_PICK 绫诲瀷鐨勪綔涓氱珯鐐�")
));
- return catalog;
- }
- if (AiDefaults.MCP_BUILTIN_RSF_WMS_TASK.equals(builtinCode)) {
- Map<String, AiMcpToolPreviewDto> catalog = new LinkedHashMap<>();
catalog.put("rsf_query_task_list", buildCatalogItem(
"rsf_query_task_list",
"浠诲姟鏌ヨ",
@@ -136,10 +114,6 @@
"蹇呴』鎻愪緵浠诲姟 ID 鎴栦换鍔″彿涔嬩竴锛屽彧杩斿洖鍗曚釜浠诲姟銆�",
List.of("鏌ヨ浠诲姟 12345 鐨勮鎯�", "鏍规嵁浠诲姟鍙� TASK24001 鏌ョ湅鎵ц鏄庣粏")
));
- return catalog;
- }
- if (AiDefaults.MCP_BUILTIN_RSF_WMS_BASE.equals(builtinCode)) {
- Map<String, AiMcpToolPreviewDto> catalog = new LinkedHashMap<>();
catalog.put("rsf_query_warehouses", buildCatalogItem(
"rsf_query_warehouses",
"鍩虹璧勬枡",
--
Gitblit v1.9.1