Allow passing initialization options to a session

We need a way for servers to pass initialization options to the session.
This is the beginning of this.
This commit is contained in:
David Soria Parra 2024-10-11 11:06:02 +01:00
commit cc342a05ad
No known key found for this signature in database

View file

@ -1,10 +1,12 @@
import logging
import sys
import importlib.metadata
import anyio
from mcp_python.server.session import ServerSession
from mcp_python.server.types import InitializationOptions
from mcp_python.server.stdio import stdio_server
from mcp_python.types import ServerCapabilities
if not sys.warnoptions:
import warnings
@ -26,8 +28,9 @@ async def receive_loop(session: ServerSession):
async def main():
version = importlib.metadata.version("mcp_python")
async with stdio_server() as (read_stream, write_stream):
async with ServerSession(read_stream, write_stream) as session, write_stream:
async with ServerSession(read_stream, write_stream, InitializationOptions(server_name="mcp_python", server_version=version, capabilities=ServerCapabilities())) as session, write_stream:
await receive_loop(session)