modelcontextprotocol/python-sdk
Watch
1
0
Fork
You've already forked python-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: add version string parameter to Server constructor

Browse source
This commit is contained in:
restlessronin 2024-12-10 20:54:52 +05:30 • committed by David Soria Parra
commit 3de4dc1f13
No known key found for this signature in database
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/mcp/server/__init__.py
View file

@ -101,8 +101,9 @@ class NotificationOptions:
class Server: class Server:
def __init__(self, name: str): def __init__(self, name: str, version: str | None = None):
self.name = name self.name = name
self.version = version
self.request_handlers: dict[ self.request_handlers: dict[
type, Callable[..., Awaitable[types.ServerResult]] type, Callable[..., Awaitable[types.ServerResult]]
] = { ] = {
@ -114,7 +115,6 @@ class Server:
def create_initialization_options( def create_initialization_options(
self, self,
version: str | None = None,
notification_options: NotificationOptions | None = None, notification_options: NotificationOptions | None = None,
experimental_capabilities: dict[str, dict[str, Any]] | None = None, experimental_capabilities: dict[str, dict[str, Any]] | None = None,
) -> InitializationOptions: ) -> InitializationOptions:
@ -134,7 +134,7 @@ class Server:
return InitializationOptions( return InitializationOptions(
server_name=self.name, 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( capabilities=self.get_capabilities(
notification_options or NotificationOptions(), notification_options or NotificationOptions(),
experimental_capabilities or {}, experimental_capabilities or {},