da4a803c11
- Added locale support in various components, including the Assistant's execution context and agent tools, to improve internationalization. - Updated the `pickNodeByFilter` function to include a fallback mechanism for host_exec nodes when no container nodes are available. - Introduced a new `agent_reference` tool for downloading agent source code for read-only study, alongside updates to the `agent_download` tool to restrict downloads to the 'smith' namespace. - Enhanced documentation and tests to reflect the new functionalities and ensure robust locale extraction and handling.
83 lines
2.6 KiB
Go
83 lines
2.6 KiB
Go
package tools
|
|
|
|
import (
|
|
_ "embed"
|
|
"encoding/json"
|
|
|
|
"github.com/yaoapp/gou/mcp"
|
|
mcpTypes "github.com/yaoapp/gou/mcp/types"
|
|
"github.com/yaoapp/gou/process"
|
|
"github.com/yaoapp/kun/log"
|
|
"github.com/yaoapp/yao/tools/agent"
|
|
"github.com/yaoapp/yao/tools/docs"
|
|
"github.com/yaoapp/yao/tools/image"
|
|
"github.com/yaoapp/yao/tools/proc"
|
|
"github.com/yaoapp/yao/tools/webfetch"
|
|
"github.com/yaoapp/yao/tools/websearch"
|
|
)
|
|
|
|
//go:embed mcps/web.json
|
|
var mcpWebDSL []byte
|
|
|
|
//go:embed mcps/process.json
|
|
var mcpProcessDSL []byte
|
|
|
|
//go:embed mcps/doc.json
|
|
var mcpDocDSL []byte
|
|
|
|
//go:embed mcps/image.json
|
|
var mcpImageDSL []byte
|
|
|
|
//go:embed mcps/agent.json
|
|
var mcpAgentDSL []byte
|
|
|
|
func init() {
|
|
process.RegisterGroup("tools", map[string]process.Handler{
|
|
"web_search": websearch.Handler,
|
|
"web_fetch": webfetch.Handler,
|
|
"process_call": proc.Handler,
|
|
"process_allowed": proc.AllowedHandler,
|
|
"doc_list": docs.ListHandler,
|
|
"doc_inspect": docs.InspectHandler,
|
|
"doc_validate": docs.ValidateHandler,
|
|
"image_read": image.ReadHandler,
|
|
"image_generate": image.GenerateHandler,
|
|
"image_providers": image.ProvidersHandler,
|
|
"agent_list": agent.ListHandler,
|
|
"agent_download": agent.DownloadHandler,
|
|
"agent_reference": agent.ReferenceHandler,
|
|
"agent_deploy": agent.DeployHandler,
|
|
"agent_connectors": agent.ConnectorsHandler,
|
|
})
|
|
|
|
registerMCPServer(mcpWebDSL, "yao-web",
|
|
websearch.SchemaJSON, webfetch.SchemaJSON)
|
|
registerMCPServer(mcpProcessDSL, "yao-process",
|
|
proc.SchemaJSON, proc.AllowedSchemaJSON)
|
|
registerMCPServer(mcpDocDSL, "yao-doc",
|
|
docs.ListSchemaJSON, docs.InspectSchemaJSON, docs.ValidateSchemaJSON)
|
|
registerMCPServer(mcpImageDSL, "yao-image",
|
|
image.ReadSchemaJSON, image.GenerateSchemaJSON, image.ProvidersSchemaJSON)
|
|
registerMCPServer(mcpAgentDSL, "yao-agent",
|
|
agent.ListSchemaJSON, agent.DownloadSchemaJSON, agent.ReferenceSchemaJSON,
|
|
agent.DeploySchemaJSON, agent.ConnectorsSchemaJSON)
|
|
}
|
|
|
|
func registerMCPServer(dsl []byte, id string, schemas ...[]byte) {
|
|
mapping := &mcpTypes.MappingData{
|
|
Tools: map[string]*mcpTypes.ToolSchema{},
|
|
Resources: map[string]*mcpTypes.ResourceSchema{},
|
|
Prompts: map[string]*mcpTypes.PromptSchema{},
|
|
}
|
|
for _, raw := range schemas {
|
|
var s mcpTypes.ToolSchema
|
|
if err := json.Unmarshal(raw, &s); err != nil {
|
|
log.Error("[tools] failed to parse schema: %s", err.Error())
|
|
continue
|
|
}
|
|
mapping.Tools[s.Name] = &s
|
|
}
|
|
if _, err := mcp.LoadClientSourceWithType(string(dsl), id, "", mapping); err != nil {
|
|
log.Error("[tools] failed to register MCP server %s: %s", id, err.Error())
|
|
}
|
|
}
|