Update server.py

server.py

@@ -1,8 +1,11 @@
 import os
-import anyio
+from fastapi import FastAPI, Request
+from sse_starlette import EventSourceResponse
 from mcp.server.lowlevel import Server
 from mcp.server.sse import SseServerTransport
 
+app = FastAPI()
+
 # Define the MCP server
 server = Server(name="airtable-mcp")
 transport = SseServerTransport("/airtable/mcp")
@@ -13,12 +16,21 @@ base_id = os.getenv("AIRTABLE_BASE_ID")
 print(f"Using Airtable token: {token}")
 print(f"Using Airtable base ID: {base_id}")
 
-# Start the server with the transport's streams
-async def main():
-    # Use a dummy scope and empty receive/send functions as placeholders
-    scope = {"type": "http", "method": "GET", "path": "/airtable/mcp"}
-    async with transport.connect_sse(scope, lambda: None, lambda x: None) as (read_stream, write_stream):
+@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})
+
+    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)
 
 if __name__ == "__main__":
-    anyio.run(main, backend="asyncio")
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)