7877797549
- Skipped tests related to KB/DB search functionality due to temporary unavailability. - Updated test cases in chat_test.go, search_auth_integration_test.go, search_auto_full_test.go, and others to reflect this change. - Adjusted search handling in search.go to limit search types to "web" only until KB/DB search is re-enabled.
62 lines
1.7 KiB
Go
62 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{
|
|
"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,
|
|
})
|
|
|
|
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)
|
|
}
|
|
|
|
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())
|
|
}
|
|
}
|