mlogclub
92490d7faf
- Implement AnyCatalog method in MCPController to list MCP tools with detailed information. - Introduce MCPToolCatalogResponse struct for tool catalog responses. - Create ToolCatalogService to handle tool listing and validation of tool codes. - Enhance skill definition validation to include tool code validation against the catalog.
84 lines
2.1 KiB
Go
84 lines
2.1 KiB
Go
package services
|
|
|
|
import (
|
|
"context"
|
|
"slices"
|
|
"strings"
|
|
|
|
"cs-agent/internal/ai/mcps"
|
|
"cs-agent/internal/pkg/config"
|
|
"cs-agent/internal/pkg/errorsx"
|
|
"cs-agent/internal/pkg/toolx"
|
|
)
|
|
|
|
var ToolCatalogService = newToolCatalogService()
|
|
|
|
func newToolCatalogService() *toolCatalogService {
|
|
return &toolCatalogService{}
|
|
}
|
|
|
|
type toolCatalogService struct{}
|
|
|
|
type MCPToolCatalogItem struct {
|
|
ToolCode string
|
|
ServerCode string
|
|
ToolName string
|
|
Title string
|
|
Description string
|
|
InputSchema any
|
|
OutputSchema any
|
|
}
|
|
|
|
func (s *toolCatalogService) ListMCPTools(ctx context.Context) ([]MCPToolCatalogItem, error) {
|
|
cfg := config.Current()
|
|
if !cfg.MCP.Enabled {
|
|
return nil, errorsx.InvalidParam("MCP未启用")
|
|
}
|
|
if len(cfg.MCP.Servers) == 0 {
|
|
return nil, nil
|
|
}
|
|
serverCodes := make([]string, 0, len(cfg.MCP.Servers))
|
|
for serverCode, server := range cfg.MCP.Servers {
|
|
if !server.Enabled {
|
|
continue
|
|
}
|
|
serverCodes = append(serverCodes, serverCode)
|
|
}
|
|
slices.Sort(serverCodes)
|
|
ret := make([]MCPToolCatalogItem, 0)
|
|
for _, serverCode := range serverCodes {
|
|
tools, err := mcps.Runtime.ListTools(ctx, serverCode)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, item := range tools {
|
|
ret = append(ret, MCPToolCatalogItem{
|
|
ToolCode: toolx.BuildMCPToolCode(serverCode, item.Name),
|
|
ServerCode: serverCode,
|
|
ToolName: strings.TrimSpace(item.Name),
|
|
Title: strings.TrimSpace(item.Title),
|
|
Description: strings.TrimSpace(item.Description),
|
|
InputSchema: item.InputSchema,
|
|
OutputSchema: item.OutputSchema,
|
|
})
|
|
}
|
|
}
|
|
return ret, nil
|
|
}
|
|
|
|
func (s *toolCatalogService) ValidateMCPToolCode(toolCode string) error {
|
|
cfg := config.Current()
|
|
if !cfg.MCP.Enabled {
|
|
return errorsx.InvalidParam("MCP未启用")
|
|
}
|
|
serverCode, toolName := toolx.SplitMCPToolCode(toolCode)
|
|
if serverCode == "" || toolName == "" {
|
|
return errorsx.InvalidParam("toolCode格式不合法")
|
|
}
|
|
server, ok := cfg.MCP.Servers[serverCode]
|
|
if !ok || !server.Enabled {
|
|
return errorsx.InvalidParam("toolCode 绑定的 MCP 服务不存在或未启用")
|
|
}
|
|
return nil
|
|
}
|