Files
ai-agent/internal/ai/mcps/registry.go
T

24 lines
489 B
Go
Raw Normal View History

2026-04-09 10:01:23 +08:00
package mcps
import (
"code.tczkiot.com/wlw/ai-agent/internal/ai/mcps/providers"
2026-04-09 10:01:23 +08:00
"github.com/modelcontextprotocol/go-sdk/mcp"
)
func defaultProviders() []providers.ToolProvider {
return []providers.ToolProvider{
providers.NewSystemToolProvider(),
// 在这里注册其他的 ToolProvider
}
}
func registerProviders(server *mcp.Server) error {
for _, provider := range defaultProviders() {
if err := provider.Register(server); err != nil {
return err
}
}
return nil
}