vikingowl
d7b524664d
- Fix replace_default positional bug: []string → map[string]string for explicit MCP tool → built-in name mapping - Improve error messages for missing API keys (3 actionable options) and unknown providers (early validation with available list) - Remove python3 dependency from MCP tests (pure bash grep/sed parsing) - Add router benchmark scaffold (6 benchmarks in bench_test.go + docs) - Add .goreleaser.yml for cross-platform binary releases with ldflags - Add launch-ready README with quickstart, extensibility docs, GIF placeholder - Add CONTRIBUTING.md and Gitea issue templates (bug report, feature request)
104 lines
2.6 KiB
Go
104 lines
2.6 KiB
Go
package mcp
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log/slog"
|
|
|
|
"somegit.dev/Owlibou/gnoma/internal/tool"
|
|
)
|
|
|
|
// Manager coordinates multiple MCP server lifecycles and tool registration.
|
|
type Manager struct {
|
|
clients map[string]*Client
|
|
logger *slog.Logger
|
|
}
|
|
|
|
// NewManager creates an MCP manager.
|
|
func NewManager(logger *slog.Logger) *Manager {
|
|
return &Manager{
|
|
clients: make(map[string]*Client),
|
|
logger: logger,
|
|
}
|
|
}
|
|
|
|
// StartAll starts all configured MCP servers, discovers tools, and registers
|
|
// them in the tool registry. Servers start sequentially to simplify error handling.
|
|
func (m *Manager) StartAll(ctx context.Context, servers []ServerConfig, registry *tool.Registry) error {
|
|
for _, srv := range servers {
|
|
client, err := m.startServer(ctx, srv)
|
|
if err != nil {
|
|
m.Shutdown() // clean up already-started servers
|
|
return fmt.Errorf("mcp server %q: %w", srv.Name, err)
|
|
}
|
|
|
|
tools, err := client.ListTools(ctx)
|
|
if err != nil {
|
|
m.Shutdown()
|
|
return fmt.Errorf("mcp server %q: list tools: %w", srv.Name, err)
|
|
}
|
|
|
|
m.registerTools(srv, tools, client, registry)
|
|
m.clients[srv.Name] = client
|
|
|
|
m.logger.Info("mcp server started",
|
|
"name", srv.Name,
|
|
"tools", len(tools),
|
|
"replace", srv.ReplaceDefault,
|
|
)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Shutdown gracefully stops all MCP server processes.
|
|
func (m *Manager) Shutdown() error {
|
|
var firstErr error
|
|
for name, client := range m.clients {
|
|
if err := client.Close(); err != nil && firstErr == nil {
|
|
firstErr = fmt.Errorf("mcp shutdown %q: %w", name, err)
|
|
}
|
|
}
|
|
m.clients = make(map[string]*Client)
|
|
return firstErr
|
|
}
|
|
|
|
func (m *Manager) startServer(ctx context.Context, srv ServerConfig) (*Client, error) {
|
|
tr := NewTransport(srv.Command, srv.Args, srv.Env, m.logger)
|
|
|
|
if err := tr.Start(ctx); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
client := NewClient(tr, m.logger)
|
|
|
|
initCtx, cancel := context.WithTimeout(ctx, srv.Timeout)
|
|
defer cancel()
|
|
|
|
if err := client.Initialize(initCtx); err != nil {
|
|
tr.Close()
|
|
return nil, err
|
|
}
|
|
|
|
return client, nil
|
|
}
|
|
|
|
func (m *Manager) registerTools(srv ServerConfig, tools []MCPTool, client *Client, registry *tool.Registry) {
|
|
for _, mt := range tools {
|
|
adapter := NewAdapter(srv.Name, mt, client)
|
|
|
|
// Explicit mapping: if this MCP tool name has a replace_default entry,
|
|
// register it under the built-in's name instead of mcp__{server}__{tool}.
|
|
if builtinName, ok := srv.ReplaceDefault[mt.Name]; ok {
|
|
adapter.SetOverrideName(builtinName)
|
|
}
|
|
|
|
registry.Register(adapter)
|
|
m.logger.Debug("mcp tool registered",
|
|
"name", adapter.Name(),
|
|
"server", srv.Name,
|
|
"mcp_name", mt.Name,
|
|
)
|
|
}
|
|
}
|