Filter tools

Author: MelbourneDeveloper

RestClient.Net.McpGenerator.Cli/Program.cs

@@ -41,6 +41,9 @@ static void PrintUsage()
     Console.WriteLine(
         "  --ext-class <class>           Extensions class name (default: 'ApiExtensions')"
     );
+    Console.WriteLine(
+        "  -t, --tags <tag1,tag2>        Comma-separated list of OpenAPI tags to include (optional)"
+    );
     Console.WriteLine("  -h, --help                    Show this help message");
 }
 
@@ -52,6 +55,7 @@ static void PrintUsage()
     var serverName = "ApiMcp";
     var extensionsNamespace = "Generated";
     var extensionsClass = "ApiExtensions";
+    string? tagsFilter = null;
 
     for (var i = 0; i < args.Length; i++)
     {
@@ -79,6 +83,10 @@ static void PrintUsage()
             case "--ext-class":
                 extensionsClass = GetNextArg(args, i++, "ext-class") ?? extensionsClass;
                 break;
+            case "-t"
+            or "--tags":
+                tagsFilter = GetNextArg(args, i++, "tags");
+                break;
             default:
                 break;
         }
@@ -104,7 +112,8 @@ static void PrintUsage()
         namespaceName,
         serverName,
         extensionsNamespace,
-        extensionsClass
+        extensionsClass,
+        tagsFilter
     );
 }
 
@@ -154,13 +163,26 @@ static async Task GenerateCode(Config config)
     }
 
     Console.WriteLine($"Read {openApiSpec.Length} characters\n");
+
+    // Parse tags filter if provided
+    ISet<string>? includeTags = null;
+    if (!string.IsNullOrWhiteSpace(config.TagsFilter))
+    {
+        includeTags = new HashSet<string>(
+            config.TagsFilter.Split(',', StringSplitOptions.RemoveEmptyEntries | StringSplitOptions.TrimEntries),
+            StringComparer.OrdinalIgnoreCase
+        );
+        Console.WriteLine($"Filtering to tags: {string.Join(", ", includeTags)}");
+    }
+
     Console.WriteLine("Generating MCP tools code...");
 
     var result = McpServerGenerator.Generate(
         openApiSpec,
         @namespace: config.Namespace,
         serverName: config.ServerName,
-        extensionsNamespace: config.ExtensionsNamespace
+        extensionsNamespace: config.ExtensionsNamespace,
+        includeTags: includeTags
     );
 
 #pragma warning disable IDE0010
@@ -186,5 +208,6 @@ internal sealed record Config(
     string Namespace,
     string ServerName,
     string ExtensionsNamespace,
-    string ExtensionsClass
+    string ExtensionsClass,
+    string? TagsFilter
 );

RestClient.Net.McpGenerator/McpServerGenerator.cs

@@ -12,13 +12,15 @@ public static class McpServerGenerator
     /// <param name="namespace">The namespace for generated MCP tools.</param>
     /// <param name="serverName">The MCP server name.</param>
     /// <param name="extensionsNamespace">The namespace of the pre-generated extensions.</param>
+    /// <param name="includeTags">Optional set of tags to include. If specified, only operations with these tags are generated.</param>
     /// <returns>A Result containing the generated C# code or error message.</returns>
 #pragma warning disable CA1054
     public static Result<string, string> Generate(
         string openApiContent,
         string @namespace,
         string serverName,
-        string extensionsNamespace
+        string extensionsNamespace,
+        ISet<string>? includeTags = null
     )
 #pragma warning restore CA1054
     {
@@ -43,7 +45,8 @@ string extensionsNamespace
                     document,
                     @namespace,
                     serverName,
-                    extensionsNamespace
+                    extensionsNamespace,
+                    includeTags
                 )
             );
         }

RestClient.Net.McpGenerator/McpToolGenerator.cs

@@ -18,12 +18,14 @@ internal static class McpToolGenerator
     /// <param name="namespace">The namespace for the MCP server.</param>
     /// <param name="serverName">The MCP server name.</param>
     /// <param name="extensionsNamespace">The namespace of the extensions.</param>
+    /// <param name="includeTags">Optional set of tags to filter operations. If specified, only operations with these tags are generated.</param>
     /// <returns>The generated MCP tools code.</returns>
     public static string GenerateTools(
         OpenApiDocument document,
         string @namespace,
         string serverName,
-        string extensionsNamespace
+        string extensionsNamespace,
+        ISet<string>? includeTags = null
     )
     {
         var tools = new List<string>();
@@ -38,6 +40,21 @@ string extensionsNamespace
 
             foreach (var operation in path.Value.Operations)
             {
+                // Skip if tags filter is specified and operation doesn't match
+                if (includeTags != null && includeTags.Count > 0)
+                {
+                    var operationTags = operation.Value.Tags;
+                    if (
+                        operationTags == null
+                        || !operationTags.Any(tag =>
+                            includeTags.Contains(tag.Name, StringComparer.OrdinalIgnoreCase)
+                        )
+                    )
+                    {
+                        continue;
+                    }
+                }
+
                 var toolMethod = GenerateTool(
                     path.Key,
                     operation.Key,

Samples/NucliaDbClient.Demo/NucliaDbClient.Demo.csproj

@@ -30,10 +30,7 @@
 builder.Services.AddSingleton<NucliaDbTools>();
 
 // Add MCP server with stdio transport and tools from assembly
-builder.Services
-    .AddMcpServer()
-    .WithStdioServerTransport()
-    .WithToolsFromAssembly();
+builder.Services.AddMcpServer().WithStdioServerTransport().WithToolsFromAssembly();
 
 var host = builder.Build();
 await host.RunAsync();

Samples/NucliaDbClient.Demo/Program.cs

@@ -10,27 +10,20 @@ This is a Model Context Protocol (MCP) server that provides Claude Code with acc
 
 ## Quick Start
 
-### 1. Start NucliaDB and MCP Server
+### Run the setup script
 
 ```bash
 cd Samples/NucliaDbClient.McpServer
-./start-mcp-server.sh
+./run-for-claude.sh
 ```
 
-This will:
-- Start NucliaDB via docker-compose (PostgreSQL + NucliaDB containers)
-- Wait for NucliaDB to be ready
-- Build and run the MCP server
+This ONE script does everything:
+- Starts NucliaDB via docker-compose (if not already running)
+- Waits for NucliaDB to be ready
+- Builds the MCP server
+- Adds it to Claude Code configuration
 
-### 2. Alternative: Run MCP Server Only
-
-If NucliaDB is already running:
-
-```bash
-./run-mcp-server.sh
-```
-
-### 3. Stop NucliaDB
+### Stop NucliaDB
 
 ```bash
 ./stop-nucliadb.sh
@@ -94,16 +87,22 @@ Then edit the file to update paths for your system.
 
 ## Available Tools
 
-The MCP server provides access to all NucliaDB REST API operations, including:
+The MCP server is **filtered to Search operations only** (18 tools) to avoid flooding Claude Code:
 
-- **Knowledge Box Management**: Get, create, delete knowledge boxes
-- **Search**: Full-text search, semantic search, catalog search
 - **Ask**: Question-answering on knowledge bases
-- **Resources**: Create, read, update, delete resources
-- **Labels & Entities**: Manage labels and entity recognition
-- **Configuration**: Configure models and settings
+- **Search**: Full-text search, semantic search, catalog search
+- **List Resources**: Query and list knowledge box contents
+- **Suggest**: Get suggestions based on queries
 
-See the [generated MCP tools](../NucliaDbClient/Generated/NucliaDbMcpTools.g.cs) for the complete list.
+The full NucliaDB API has 110+ operations across these tags:
+- Search (18 tools) ✓ Currently enabled
+- Resources - Resource CRUD operations
+- Knowledge Boxes - KB management
+- Models - Model configuration
+- Knowledge Box Services - Advanced KB services
+- And more...
+
+See the [generated MCP tools](../NucliaDbClient/Generated/NucliaDbMcpTools.g.cs) for the current filtered list. To include different operations, see the **Tag Filtering** section in Development below.
 
 ## Environment Variables
 
@@ -166,14 +165,57 @@ dotnet run --project RestClient.Net.OpenApiGenerator.Cli/RestClient.Net.OpenApiG
   -n NucliaDB.Generated \
   -c NucliaDBApiExtensions
 
-# Regenerate the MCP tools
+# Regenerate the MCP tools with tag filtering (recommended to reduce tool count)
+# Only include Search-related tools (18 tools instead of 110)
+dotnet run --project RestClient.Net.McpGenerator.Cli/RestClient.Net.McpGenerator.Cli.csproj -- \
+  --openapi-url Samples/NucliaDbClient/api.yaml \
+  --output-file Samples/NucliaDbClient/Generated/NucliaDbMcpTools.g.cs \
+  --namespace NucliaDB.Mcp \
+  --server-name NucliaDb \
+  --ext-namespace NucliaDB.Generated \
+  --tags Search
+
+# Or generate all tools (not recommended - creates 110 tools)
 dotnet run --project RestClient.Net.McpGenerator.Cli/RestClient.Net.McpGenerator.Cli.csproj -- \
   --openapi-url Samples/NucliaDbClient/api.yaml \
   --output-file Samples/NucliaDbClient/Generated/NucliaDbMcpTools.g.cs \
   --namespace NucliaDB.Mcp \
   --server-name NucliaDb \
   --ext-namespace NucliaDB.Generated
 
+# Or include multiple tags (e.g., Search and Resources)
+dotnet run --project RestClient.Net.McpGenerator.Cli/RestClient.Net.McpGenerator.Cli.csproj -- \
+  --openapi-url Samples/NucliaDbClient/api.yaml \
+  --output-file Samples/NucliaDbClient/Generated/NucliaDbMcpTools.g.cs \
+  --namespace NucliaDB.Mcp \
+  --server-name NucliaDb \
+  --ext-namespace NucliaDB.Generated \
+  --tags "Search,Resources"
+
 # Rebuild the MCP server
 dotnet build Samples/NucliaDbClient.McpServer
 ```
+
+### Tag Filtering
+
+The NucliaDB OpenAPI spec has many operations organized by tags. To avoid flooding Claude Code with too many tools, use the `--tags` parameter to specify which sections you want:
+
+**Available tags:**
+- `Search` - Search and ask operations (18 tools)
+- `Resources` - Resource CRUD operations
+- `Knowledge Boxes` - Knowledge box management
+- `Models` - Model configuration
+- `Knowledge Box Services` - Advanced KB services
+- And more...
+
+**Examples:**
+```bash
+# Only search operations (recommended for most use cases)
+--tags Search
+
+# Search and resources only
+--tags "Search,Resources"
+
+# Multiple tags
+--tags "Search,Resources,Knowledge Boxes"
+```