Allow setting a version in McpService and McpServer

Author: adwsingh

mcp/mcp-server/src/main/java/software/amazon/smithy/java/mcp/server/McpServerBuilder.java

@@ -25,6 +25,7 @@ public final class McpServerBuilder {
     Map<String, Service> services = new HashMap<>();
     List<McpServerProxy> proxyList = new ArrayList<>();
     String name;
+    String version;
     ToolFilter toolFilter = (server, tool) -> true;
     McpService mcpService;
 
@@ -51,15 +52,25 @@ public McpServerBuilder name(String name) {
         return this;
     }
 
+    public McpServerBuilder version(String version) {
+        this.version = version;
+        return this;
+    }
+
     public Server build() {
         validate();
         // Create McpService before building McpServer
-        this.mcpService = McpService.builder()
+        var builder = McpService.builder()
                 .services(services)
                 .proxyList(proxyList)
                 .name(name != null ? name : "mcp-server")
-                .toolFilter(toolFilter)
-                .build();
+                .toolFilter(toolFilter);
+
+        if (version != null) {
+            builder.version(version);
+        }
+
+        this.mcpService = builder.build();
         return new McpServer(this);
     }
 

mcp/mcp-server/src/main/java/software/amazon/smithy/java/mcp/server/McpService.java

@@ -74,17 +74,25 @@ public final class McpService {
     private final Map<String, Prompt> prompts;
     private final PromptProcessor promptProcessor;
     private final String name;
+    private final String version;
     private final Map<String, McpServerProxy> proxies;
     private final Map<String, Service> services;
     private final AtomicReference<JsonRpcRequest> initializeRequest = new AtomicReference<>();
     private final ToolFilter toolFilter;
 
-    McpService(Map<String, Service> services, List<McpServerProxy> proxyList, String name, ToolFilter toolFilter) {
+    McpService(
+            Map<String, Service> services,
+            List<McpServerProxy> proxyList,
+            String name,
+            String version,
+            ToolFilter toolFilter
+    ) {
         this.services = services;
         this.tools = createTools(services);
         this.prompts = PromptLoader.loadPrompts(services.values());
         this.promptProcessor = new PromptProcessor();
         this.name = name;
+        this.version = version;
         this.proxies = proxyList.stream().collect(Collectors.toMap(McpServerProxy::name, p -> p));
         this.toolFilter = toolFilter;
     }
@@ -142,7 +150,7 @@ private JsonRpcResponse handleInitialize(JsonRpcRequest req) {
                         .build())
                 .serverInfo(ServerInfo.builder()
                         .name(name)
-                        .version("1.0.0")
+                        .version(version)
                         .build())
                 .build();
 
@@ -594,6 +602,7 @@ public static class Builder {
         private Map<String, Service> services = new HashMap<>();
         private List<McpServerProxy> proxyList = new ArrayList<>();
         private String name = "mcp-server";
+        private String version = "1.0.0";
         private ToolFilter toolFilter = (serverId, toolName) -> true;
 
         public Builder services(Map<String, Service> services) {
@@ -611,13 +620,18 @@ public Builder name(String name) {
             return this;
         }
 
+        public Builder version(String version) {
+            this.version = version;
+            return this;
+        }
+
         public Builder toolFilter(ToolFilter toolFilter) {
             this.toolFilter = toolFilter;
             return this;
         }
 
         public McpService build() {
-            return new McpService(services, proxyList, name, toolFilter);
+            return new McpService(services, proxyList, name, version, toolFilter);
         }
     }
 }