Skip to content

Commit 29d6b47

Browse files
Dumbrisclaude
andcommitted
fix: register code_execution tool in CallBuiltInTool and CallToolDirect methods
Add missing code_execution operation constant and register it in all tool routing methods to enable CLI command support. - Add operationCodeExecution constant - Add case in CallBuiltInTool switch statement - Add case in CallToolDirect switch statement - Add to proxyTools map in handleCallTool - Add to proxy tool routing switch statement This fixes the 'unknown built-in tool: code_execution' error when using the mcpproxy code exec CLI command. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <[email protected]>
1 parent 36899fa commit 29d6b47

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

internal/server/mcp.go

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,7 @@ const (
4242
operationQuarantineSec = "quarantine_security"
4343
operationRetrieveTools = "retrieve_tools"
4444
operationReadCache = "read_cache"
45+
operationCodeExecution = "code_execution"
4546
operationListRegistries = "list_registries"
4647
operationSearchServers = "search_servers"
4748

@@ -572,6 +573,7 @@ func (p *MCPProxyServer) handleCallTool(ctx context.Context, request mcp.CallToo
572573
operationRetrieveTools: true,
573574
operationCallTool: true,
574575
"read_cache": true,
576+
"code_execution": true,
575577
"list_registries": true,
576578
"search_servers": true,
577579
}
@@ -592,6 +594,8 @@ func (p *MCPProxyServer) handleCallTool(ctx context.Context, request mcp.CallToo
592594
return p.handleRetrieveTools(ctx, proxyRequest)
593595
case operationReadCache:
594596
return p.handleReadCache(ctx, proxyRequest)
597+
case operationCodeExecution:
598+
return p.handleCodeExecution(ctx, proxyRequest)
595599
case operationListRegistries:
596600
return p.handleListRegistries(ctx, proxyRequest)
597601
case operationSearchServers:
@@ -2306,6 +2310,8 @@ func (p *MCPProxyServer) CallBuiltInTool(ctx context.Context, toolName string, a
23062310
return p.handleRetrieveTools(ctx, request)
23072311
case operationReadCache:
23082312
return p.handleReadCache(ctx, request)
2313+
case operationCodeExecution:
2314+
return p.handleCodeExecution(ctx, request)
23092315
case operationListRegistries:
23102316
return p.handleListRegistries(ctx, request)
23112317
case operationSearchServers:
@@ -2406,6 +2412,8 @@ func (p *MCPProxyServer) CallToolDirect(ctx context.Context, request mcp.CallToo
24062412
result, err = p.handleRetrieveTools(ctx, request)
24072413
case "quarantine_security":
24082414
result, err = p.handleQuarantineSecurity(ctx, request)
2415+
case "code_execution":
2416+
result, err = p.handleCodeExecution(ctx, request)
24092417
case "list_registries":
24102418
result, err = p.handleListRegistries(ctx, request)
24112419
case "search_servers":

0 commit comments

Comments
 (0)