Update app.py

app.py

@@ -121,7 +121,9 @@ transport = SseServerTransport("/airtable/mcp")
 @app.get("/airtable/mcp")
 async def handle_sse(request: Request):
     logger.debug("Handling SSE connection request")
+    session_id = None  # We'll extract this from the endpoint event
     async def sse_writer():
+        nonlocal session_id
         logger.debug("Starting SSE writer")
         async with sse_stream_writer, write_stream_reader:
             endpoint_data = "/airtable/mcp?session_id={session_id}"
@@ -132,20 +134,20 @@ async def handle_sse(request: Request):
             async for session_message in write_stream_reader:
                 logger.debug(f"Sending message via SSE: {session_message}")
                 message_data = session_message.message.model_dump_json(by_alias=True, exclude_none=True)
-                await sse_stream_writer.send(
-                    {
-                        "event": "message",
-                        "data": message_data
-                    }
-                )
-                # Extract session_id from the endpoint event
                 message = json.loads(message_data)
+                # Extract session_id from the endpoint event
                 if message.get("event") == "endpoint":
                     endpoint_url = message.get("data", "")
                     if "session_id=" in endpoint_url:
                         session_id = endpoint_url.split("session_id=")[1]
                         write_streams[session_id] = write_stream
                         logger.debug(f"Extracted session_id: {session_id}")
+                await sse_stream_writer.send(
+                    {
+                        "event": "message",
+                        "data": message_data
+                    }
+                )
 
     sse_stream_writer, sse_stream_reader = anyio.create_memory_object_stream(0)
     try: