mirror of
https://github.com/aljazceru/mcp-python-sdk.git
synced 2025-12-19 14:54:24 +01:00
51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
import importlib.metadata
|
|
import logging
|
|
import sys
|
|
|
|
import anyio
|
|
|
|
from mcp_python.server.session import ServerSession
|
|
from mcp_python.server.stdio import stdio_server
|
|
from mcp_python.server.types import InitializationOptions
|
|
from mcp_python.types import ServerCapabilities
|
|
|
|
if not sys.warnoptions:
|
|
import warnings
|
|
|
|
warnings.simplefilter("ignore")
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
logger = logging.getLogger("server")
|
|
|
|
|
|
async def receive_loop(session: ServerSession):
|
|
logger.info("Starting receive loop")
|
|
async for message in session.incoming_messages:
|
|
if isinstance(message, Exception):
|
|
logger.error("Error: %s", message)
|
|
continue
|
|
|
|
logger.info("Received message from client: %s", message)
|
|
|
|
|
|
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,
|
|
InitializationOptions(
|
|
server_name="mcp_python",
|
|
server_version=version,
|
|
capabilities=ServerCapabilities(),
|
|
),
|
|
) as session,
|
|
write_stream,
|
|
):
|
|
await receive_loop(session)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
anyio.run(main, backend="trio")
|