|
9 | 9 | "github.com/modelcontextprotocol/go-sdk/mcp" |
10 | 10 |
|
11 | 11 | "github.com/docker/mcp-gateway/pkg/log" |
12 | | - "github.com/docker/mcp-gateway/pkg/prompts" |
| 12 | + // "github.com/docker/mcp-gateway/pkg/prompts" |
13 | 13 | ) |
14 | 14 |
|
15 | 15 | func (g *Gateway) reloadConfiguration(ctx context.Context, configuration Configuration, serverNames []string, clientConfig *clientConfig) error { |
@@ -95,42 +95,43 @@ func (g *Gateway) reloadConfiguration(ctx context.Context, configuration Configu |
95 | 95 | g.mcpServer.AddTool(mcpRemoveTool.Tool, mcpRemoveTool.Handler) |
96 | 96 | g.toolRegistrations[mcpRemoveTool.Tool.Name] = *mcpRemoveTool |
97 | 97 |
|
98 | | - // Add mcp-registry-import tool |
99 | | - mcpRegistryImportTool := g.createMcpRegistryImportTool(configuration, clientConfig) |
100 | | - g.mcpServer.AddTool(mcpRegistryImportTool.Tool, mcpRegistryImportTool.Handler) |
101 | | - g.toolRegistrations[mcpRegistryImportTool.Tool.Name] = *mcpRegistryImportTool |
102 | | - |
103 | | - // Add mcp-config-set tool |
104 | | - mcpConfigSetTool := g.createMcpConfigSetTool(clientConfig) |
105 | | - g.mcpServer.AddTool(mcpConfigSetTool.Tool, mcpConfigSetTool.Handler) |
106 | | - g.toolRegistrations[mcpConfigSetTool.Tool.Name] = *mcpConfigSetTool |
107 | | - |
108 | | - // Add mcp-session-name tool |
109 | | - mcpSessionNameTool := g.createMcpSessionNameTool() |
110 | | - g.mcpServer.AddTool(mcpSessionNameTool.Tool, mcpSessionNameTool.Handler) |
111 | | - g.toolRegistrations[mcpSessionNameTool.Tool.Name] = *mcpSessionNameTool |
112 | | - |
113 | 98 | // Add codemode |
114 | 99 | codeModeTool := g.createCodeModeTool(clientConfig) |
115 | 100 | g.mcpServer.AddTool(codeModeTool.Tool, codeModeTool.Handler) |
116 | 101 | g.toolRegistrations[codeModeTool.Tool.Name] = *codeModeTool |
117 | 102 |
|
118 | | - // Add mcp-exec tool |
| 103 | + // Add mcp-exec tool only if client name contains "claude" |
119 | 104 | mcpExecTool := g.createMcpExecTool() |
120 | 105 | g.mcpServer.AddTool(mcpExecTool.Tool, mcpExecTool.Handler) |
121 | 106 | g.toolRegistrations[mcpExecTool.Tool.Name] = *mcpExecTool |
122 | 107 |
|
123 | | - prompts.AddDiscoverPrompt(g.mcpServer) |
| 108 | + // Add mcp-config-set tool |
| 109 | + mcpConfigSetTool := g.createMcpConfigSetTool(clientConfig) |
| 110 | + g.mcpServer.AddTool(mcpConfigSetTool.Tool, mcpConfigSetTool.Handler) |
| 111 | + g.toolRegistrations[mcpConfigSetTool.Tool.Name] = *mcpConfigSetTool |
124 | 112 |
|
125 | | - log.Log(" > mcp-discover: prompt for learning about dynamic server management") |
126 | 113 | log.Log(" > mcp-find: tool for finding MCP servers in the catalog") |
127 | 114 | log.Log(" > mcp-add: tool for adding MCP servers to the registry") |
128 | 115 | log.Log(" > mcp-remove: tool for removing MCP servers from the registry") |
129 | | - log.Log(" > mcp-registry-import: tool for importing servers from MCP registry URLs") |
130 | 116 | log.Log(" > mcp-config-set: tool for setting configuration values for MCP servers") |
131 | | - log.Log(" > mcp-session-name: tool for setting session name to persist configuration") |
132 | 117 | log.Log(" > code-mode: write code that calls other MCPs directly") |
133 | 118 | log.Log(" > mcp-exec: execute tools that exist in the current session") |
| 119 | + |
| 120 | + // Add mcp-registry-import tool |
| 121 | + // mcpRegistryImportTool := g.createMcpRegistryImportTool(configuration, clientConfig) |
| 122 | + // g.mcpServer.AddTool(mcpRegistryImportTool.Tool, mcpRegistryImportTool.Handler) |
| 123 | + // g.toolRegistrations[mcpRegistryImportTool.Tool.Name] = *mcpRegistryImportTool |
| 124 | + |
| 125 | + // Add mcp-session-name tool |
| 126 | + // mcpSessionNameTool := g.createMcpSessionNameTool() |
| 127 | + // g.mcpServer.AddTool(mcpSessionNameTool.Tool, mcpSessionNameTool.Handler) |
| 128 | + // g.toolRegistrations[mcpSessionNameTool.Tool.Name] = *mcpSessionNameTool |
| 129 | + // log.Log(" > mcp-registry-import: tool for importing servers from MCP registry URLs") |
| 130 | + // log.Log(" > mcp-session-name: tool for setting session name to persist configuration") |
| 131 | + |
| 132 | + // Add prompt |
| 133 | + // prompts.AddDiscoverPrompt(g.mcpServer) |
| 134 | + // log.Log(" > mcp-discover: prompt for learning about dynamic server management") |
134 | 135 | } |
135 | 136 |
|
136 | 137 | for _, prompt := range capabilities.Prompts { |
|
0 commit comments