2026-04-09 10:01:23 +08:00
|
|
|
package mcps
|
|
|
|
|
|
|
|
|
|
import (
|
2026-08-21 00:41:07 +08:00
|
|
|
"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
|
|
|
|
|
}
|