package com.vincent.rsf.server.ai.service.impl.mcp;
|
|
import com.vincent.rsf.framework.exception.CoolException;
|
import com.vincent.rsf.server.ai.config.AiDefaults;
|
import com.vincent.rsf.server.ai.dto.AiMcpToolPreviewDto;
|
import org.springframework.stereotype.Component;
|
|
import java.util.LinkedHashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
@Component
|
public class BuiltinMcpToolCatalogProvider {
|
|
public List<String> supportedBuiltinCodes() {
|
return List.of(AiDefaults.MCP_BUILTIN_RSF_WMS);
|
}
|
|
public Map<String, AiMcpToolPreviewDto> getCatalog(String 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",
|
"库存查询",
|
"查询指定物料当前可用于出库的库存明细。",
|
"必须提供物料编码或物料名称，并且最多返回 50 条库存记录。",
|
List.of("查询物料 MAT001 当前可出库库存", "按物料名称查询托盘库存明细")
|
));
|
catalog.put("rsf_query_station_list", buildCatalogItem(
|
"rsf_query_station_list",
|
"库存查询",
|
"查询指定作业类型可用的设备站点。",
|
"必须提供站点类型列表，类型数量最多 10 个，最多返回 50 个站点。",
|
List.of("查询入库和出库作业可用站点", "列出 AGV_PICK 类型的作业站点")
|
));
|
catalog.put("rsf_query_task", buildCatalogItem(
|
"rsf_query_task",
|
"任务查询",
|
"按任务号、状态、类型或站点条件查询任务；支持从自然语言中自动提取任务号，精确命中时返回任务明细。",
|
"过滤条件均为可选，不传过滤条件时默认返回最近任务，最多返回 50 条记录。",
|
List.of("查询最近 10 条任务", "查询任务号 TASK24001 的详情")
|
));
|
catalog.put("rsf_query_warehouses", buildCatalogItem(
|
"rsf_query_warehouses",
|
"基础资料",
|
"查询仓库基础信息。",
|
"至少提供仓库编码或名称，最多返回 50 条仓库记录。",
|
List.of("查询编码包含 WH 的仓库", "按仓库名称查询仓库地址")
|
));
|
catalog.put("rsf_query_bas_stations", buildCatalogItem(
|
"rsf_query_bas_stations",
|
"基础资料",
|
"查询基础站点信息。",
|
"至少提供站点编号、站点名称或使用状态之一，最多返回 50 条站点记录。",
|
List.of("查询使用中的基础站点", "按站点编号查询基础站点")
|
));
|
catalog.put("rsf_query_dict_data", buildCatalogItem(
|
"rsf_query_dict_data",
|
"基础资料",
|
"查询指定字典类型下的字典数据。",
|
"必须提供字典类型编码，最多返回 100 条字典记录。",
|
List.of("查询 task_status 字典", "按字典标签过滤 task_type 字典数据")
|
));
|
return catalog;
|
}
|
throw new CoolException("不支持的内置 MCP 编码: " + builtinCode);
|
}
|
|
private AiMcpToolPreviewDto buildCatalogItem(String name, String toolGroup, String toolPurpose,
|
String queryBoundary, List<String> exampleQuestions) {
|
return AiMcpToolPreviewDto.builder()
|
.name(name)
|
.toolGroup(toolGroup)
|
.toolPurpose(toolPurpose)
|
.queryBoundary(queryBoundary)
|
.exampleQuestions(exampleQuestions)
|
.build();
|
}
|
}
|
