Update server.py

server.py

@@ -1,9 +1,14 @@
 import os
+import logging
 from fastapi import FastAPI, Request
 from sse_starlette import EventSourceResponse
 from mcp.server.lowlevel import Server
 from mcp.server.sse import SseServerTransport
 
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
 app = FastAPI()
 
 # Define the MCP server
@@ -13,23 +18,29 @@ transport = SseServerTransport("/airtable/mcp")
 # Configure environment variables (for logging purposes)
 token = os.getenv("AIRTABLE_API_TOKEN")
 base_id = os.getenv("AIRTABLE_BASE_ID")
-print(f"Using Airtable token: {token}")
-print(f"Using Airtable base ID: {base_id}")
+logger.info(f"Using Airtable token: {token}")
+logger.info(f"Using Airtable base ID: {base_id}")
 
 @app.get("/airtable/mcp")
 async def handle_sse(request: Request):
-    async def sse_writer():
-        async with sse_stream_writer:
-            await sse_stream_writer.send({"event": "endpoint", "data": "/airtable/mcp"})
-            async for session_message in write_stream_reader:
-                message_data = session_message.message.model_dump_json(by_alias=True, exclude_none=True)
-                await sse_stream_writer.send({"event": "message", "data": message_data})
+    try:
+        async def sse_writer():
+            async with sse_stream_writer:
+                await sse_stream_writer.send({"event": "endpoint", "data": "/airtable/mcp"})
+                async for session_message in write_stream_reader:
+                    message_data = session_message.message.model_dump_json(by_alias=True, exclude_none=True)
+                    await sse_stream_writer.send({"event": "message", "data": message_data})
 
-    sse_stream_writer, sse_stream_reader = anyio.create_memory_object_stream(0)
-    async with transport.connect_sse(request.scope, request.receive, request._send) as (read_stream, write_stream):
-        write_stream_reader = write_stream
-        await server.run(read_stream, write_stream, server.create_initialization_options())
-    return EventSourceResponse(sse_stream_reader, data_sender_callable=sse_writer)
+        sse_stream_writer, sse_stream_reader = anyio.create_memory_object_stream(0)
+        async with transport.connect_sse(request.scope, request.receive, request._send) as (read_stream, write_stream):
+            write_stream_reader = write_stream
+            logger.info("Starting MCP server with streams")
+            await server.run(read_stream, write_stream, server.create_initialization_options())
+            logger.info("MCP server running")
+        return EventSourceResponse(sse_stream_reader, data_sender_callable=sse_writer)
+    except Exception as e:
+        logger.error(f"Error in handle_sse: {str(e)}")
+        raise
 
 if __name__ == "__main__":
     import uvicorn