7da06a4ae1
- Updated the builtinSearch function to delegate search operations to the websearch tool, improving modularity and maintainability. - Enhanced context handling by passing the agent context to builtinSearch, allowing for user and team identification during searches. - Added a new DecryptValue function in cloud.go to streamline value decryption, delegating to the existing config.DecryptValue method. - Updated .gitignore to include tools/README.md for better project organization.
61 lines
1.7 KiB
Go
61 lines
1.7 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/docs"
|
|
"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
|
|
|
|
func init() {
|
|
process.RegisterGroup("tools", map[string]process.Handler{
|
|
"websearch": websearch.Handler,
|
|
"webfetch": webfetch.Handler,
|
|
"processcall": proc.Handler,
|
|
"doclist": docs.ListHandler,
|
|
"docinspect": docs.InspectHandler,
|
|
"docvalidate": docs.ValidateHandler,
|
|
})
|
|
|
|
registerMCPServer(mcpWebDSL, "yao-web",
|
|
websearch.SchemaJSON, webfetch.SchemaJSON)
|
|
registerMCPServer(mcpProcessDSL, "yao-process",
|
|
proc.SchemaJSON)
|
|
registerMCPServer(mcpDocDSL, "yao-doc",
|
|
docs.ListSchemaJSON, docs.InspectSchemaJSON, docs.ValidateSchemaJSON)
|
|
}
|
|
|
|
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())
|
|
}
|
|
}
|