Files
mcp-python-sdk/mcp_python/server/__main__.py
David Soria Parra 4cbf815430 Initial import
2024-09-24 22:04:19 +01:00

36 lines
895 B
Python

import logging
import sys
import anyio
from mcp_python.server.session import ServerSession
from mcp_python.server.stdio import stdio_server
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():
async with stdio_server() as (read_stream, write_stream):
async with ServerSession(read_stream, write_stream) as session, write_stream:
await receive_loop(session)
if __name__ == "__main__":
anyio.run(main, backend="trio")