From 3de4dc1f13fc4800a8bd4f3e6a52f0d4219f30d0 Mon Sep 17 00:00:00 2001
From: restlessronin <88921269+restlessronin@users.noreply.github.com>
Date: Tue, 10 Dec 2024 20:54:52 +0530
Subject: [PATCH] feat: add version string parameter to Server constructor

---
 src/mcp/server/__init__.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/mcp/server/__init__.py b/src/mcp/server/__init__.py
index 212f011..4c5dc04 100644
--- a/src/mcp/server/__init__.py
+++ b/src/mcp/server/__init__.py
@@ -101,8 +101,9 @@ class NotificationOptions:
 
 
 class Server:
-    def __init__(self, name: str):
+    def __init__(self, name: str, version: str | None = None):
         self.name = name
+        self.version = version
         self.request_handlers: dict[
             type, Callable[..., Awaitable[types.ServerResult]]
         ] = {
@@ -114,7 +115,6 @@ class Server:
 
     def create_initialization_options(
         self,
-        version: str | None = None,
         notification_options: NotificationOptions | None = None,
         experimental_capabilities: dict[str, dict[str, Any]] | None = None,
     ) -> InitializationOptions:
@@ -134,7 +134,7 @@ class Server:
 
         return InitializationOptions(
             server_name=self.name,
-            server_version=version if version else pkg_version("mcp"),
+            server_version=self.version if self.version else pkg_version("mcp"),
             capabilities=self.get_capabilities(
                 notification_options or NotificationOptions(),
                 experimental_capabilities or {},