Skip to content

Add action interceptor for config-driven MCP routing #9

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
jpthor merged 4 commits into from
Feb 26, 2026
Merged

Add action interceptor for config-driven MCP routing #9

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
1c7f3f1
add action interceptor for config-driven MCP routing
jpthor Feb 25, 2026
35e4dc2
emit mcp_status action to app when interceptor handles an action
jpthor Feb 25, 2026
032591c
fix: strip sign_tx when co-emitted with build action
jpthor Feb 25, 2026
4a3dfe2
feat: dynamic tool manifest, MCP init retry, and auto_execute removal
jpthor Feb 26, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
47 changes: 27 additions & 20 deletions cmd/server/main.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,6 @@ import (
"github.com/vultisig/agent-backend/internal/config"
"github.com/vultisig/agent-backend/internal/mcp"
"github.com/vultisig/agent-backend/internal/service/agent"
mcpclient "github.com/vultisig/agent-backend/internal/service/mcp"
"github.com/vultisig/agent-backend/internal/service/plugin"
"github.com/vultisig/agent-backend/internal/service/verifier"
"github.com/vultisig/agent-backend/internal/storage/postgres"
Expand Down Expand Up @@ -79,21 +78,35 @@ func main() {
cacheTTL := time.Duration(cfg.MCP.ToolCacheTTLSec) * time.Second
mcpClient := mcp.NewClient(cfg.MCP.ServerURL, cacheTTL, logger)

const mcpMaxRetries = 10
const mcpRetryInterval = 3 * time.Second
for attempt := 1; attempt <= mcpMaxRetries; attempt++ {
mcpCtx, mcpCancel := context.WithTimeout(ctx, 10*time.Second)
err := mcpClient.Initialize(mcpCtx)
mcpCancel()
if err == nil {
break
}
logger.WithError(err).WithField("attempt", attempt).Warn("failed to initialize mcp client")
if attempt == mcpMaxRetries {
logger.Warn("mcp client init exhausted retries, continuing without mcp tools")
} else {
time.Sleep(mcpRetryInterval)
}
}

mcpCtx, mcpCancel := context.WithTimeout(ctx, 10*time.Second)
defer mcpCancel()

if err := mcpClient.Initialize(mcpCtx); err != nil {
logger.WithError(err).Warn("failed to initialize mcp client, continuing without mcp tools")
tools, err := mcpClient.ListTools(mcpCtx)
if err != nil {
logger.WithError(err).Warn("failed to list mcp tools, continuing without mcp tools")
} else {
tools, err := mcpClient.ListTools(mcpCtx)
if err != nil {
logger.WithError(err).Warn("failed to list mcp tools, continuing without mcp tools")
} else {
logger.WithField("tool_count", len(tools)).Info("mcp tools loaded")
mcpProvider = mcpClient
}
logger.WithField("tool_count", len(tools)).Info("mcp tools loaded")
mcpProvider = mcpClient
}

// Pre-warm skill cache (non-fatal)
if mcpProvider != nil {
skills, err := mcpClient.ListSkills(mcpCtx)
if err != nil {
logger.WithError(err).Warn("failed to list mcp skills, continuing without skills")
Expand All @@ -103,16 +116,11 @@ func main() {
}
}

// Initialize MCP swap tx builder (optional)
var swapTxBuilder agent.SwapTxBuilder
if cfg.MCP.URL != "" {
mcpCl := mcpclient.NewClient(cfg.MCP.URL)
swapTxBuilder = mcpclient.NewSwapTxAdapter(mcpCl)
logger.WithField("url", cfg.MCP.URL).Info("MCP swap tx builder enabled")
}
// Initialize action router for intercepting actions (e.g. search_token → MCP)
actionRouter := agent.ParseActionRoutes(cfg.ActionRoutes, mcpProvider, logger)

// Initialize agent service
agentService := agent.NewAgentService(aiClient, msgRepo, convRepo, memRepo, redisClient, verifierClient, pluginService, mcpProvider, swapTxBuilder, logger, cfg.AI.SummaryModel, cfg.Context)
agentService := agent.NewAgentService(aiClient, msgRepo, convRepo, memRepo, redisClient, verifierClient, pluginService, mcpProvider, actionRouter, logger, cfg.AI.SummaryModel, cfg.Context)

// Initialize API server
server := api.NewServer(
Expand Down Expand Up @@ -174,7 +182,6 @@ func main() {
agentGroup.POST("/conversations/:id", server.GetConversation)
agentGroup.DELETE("/conversations/:id", server.DeleteConversation)
agentGroup.POST("/conversations/:id/messages", server.SendMessage, api.RateLimitMiddleware(redisClient))
agentGroup.POST("/conversations/:id/tx/build", server.BuildTx)
agentGroup.POST("/starters", server.GetStarters)

// Start server
Expand Down
37 changes: 0 additions & 37 deletions internal/api/swap.go
View file
Open in desktop

This file was deleted.

19 changes: 10 additions & 9 deletions internal/config/config.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,15 +6,16 @@ import (

// Config holds all configuration for the agent-backend service.
type Config struct {
LogFormat string `envconfig:"LOG_FORMAT" default:"json"`
Server ServerConfig
Database DatabaseConfig
Redis RedisConfig
AuthCache AuthCacheConfig
AI AIConfig
Context ContextConfig
Verifier VerifierConfig
MCP MCPConfig
LogFormat string `envconfig:"LOG_FORMAT" default:"json"`
ActionRoutes string `envconfig:"ACTION_ROUTES" default:""`
Server ServerConfig
Database DatabaseConfig
Redis RedisConfig
AuthCache AuthCacheConfig
AI AIConfig
Context ContextConfig
Verifier VerifierConfig
MCP MCPConfig
}

// ServerConfig holds HTTP server configuration.
Expand Down
Loading