wehub-resource-sync
66 lines
2.1 KiB
Go
66 lines
2.1 KiB
Go
package localaitools
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/modelcontextprotocol/go-sdk/mcp"
|
|
)
|
|
|
|
func registerBackendTools(s *mcp.Server, client LocalAIClient, opts Options) {
|
|
mcp.AddTool(s, &mcp.Tool{
|
|
Name: ToolListBackends,
|
|
Description: "List installed backends.",
|
|
}, func(ctx context.Context, _ *mcp.CallToolRequest, _ struct{}) (*mcp.CallToolResult, any, error) {
|
|
backends, err := client.ListBackends(ctx)
|
|
if err != nil {
|
|
return errorResult(err), nil, nil
|
|
}
|
|
return jsonResult(backends), nil, nil
|
|
})
|
|
|
|
mcp.AddTool(s, &mcp.Tool{
|
|
Name: ToolListKnownBackends,
|
|
Description: "List backends available to install from configured backend galleries.",
|
|
}, func(ctx context.Context, _ *mcp.CallToolRequest, _ struct{}) (*mcp.CallToolResult, any, error) {
|
|
backends, err := client.ListKnownBackends(ctx)
|
|
if err != nil {
|
|
return errorResult(err), nil, nil
|
|
}
|
|
return jsonResult(backends), nil, nil
|
|
})
|
|
|
|
if opts.DisableMutating {
|
|
return
|
|
}
|
|
|
|
mcp.AddTool(s, &mcp.Tool{
|
|
Name: ToolInstallBackend,
|
|
Description: "Install a backend from a backend gallery. Requires user confirmation per safety rule 1.",
|
|
}, func(ctx context.Context, _ *mcp.CallToolRequest, args InstallBackendRequest) (*mcp.CallToolResult, any, error) {
|
|
if args.BackendName == "" {
|
|
return errorResultf("backend_name is required"), nil, nil
|
|
}
|
|
jobID, err := client.InstallBackend(ctx, args)
|
|
if err != nil {
|
|
return errorResult(err), nil, nil
|
|
}
|
|
return jsonResult(map[string]any{"job_id": jobID}), nil, nil
|
|
})
|
|
|
|
mcp.AddTool(s, &mcp.Tool{
|
|
Name: ToolUpgradeBackend,
|
|
Description: "Upgrade an installed backend by name. Requires user confirmation per safety rule 1.",
|
|
}, func(ctx context.Context, _ *mcp.CallToolRequest, args struct {
|
|
Name string `json:"name" jsonschema:"The installed backend name."`
|
|
}) (*mcp.CallToolResult, any, error) {
|
|
if args.Name == "" {
|
|
return errorResultf("name is required"), nil, nil
|
|
}
|
|
jobID, err := client.UpgradeBackend(ctx, args.Name)
|
|
if err != nil {
|
|
return errorResult(err), nil, nil
|
|
}
|
|
return jsonResult(map[string]any{"job_id": jobID}), nil, nil
|
|
})
|
|
}
|