Update server.py

server.py

@@ -37,7 +37,11 @@ async def handle_sse(request: Request):
                 endpoint_data = f"/airtable/mcp?session_id={{session_id}}"
                 await sse_stream_writer.send({"event": "endpoint", "data": endpoint_data})
                 async for session_message in write_stream_reader:
-                    message_data = session_message.message.model_dump_json(by_alias=True, exclude_none=True)
+                    # Handle both dict and SessionMessage objects
+                    if isinstance(session_message, dict):
+                        message_data = json.dumps(session_message)
+                    else:
+                        message_data = session_message.message.model_dump_json(by_alias=True, exclude_none=True)
                     message = json.loads(message_data)
                     if not session_id and message.get("event") == "endpoint":
                         endpoint_url = message.get("data", "")