Update server.py

server.py

@@ -2,6 +2,7 @@ import os
 import json
 import logging
 import anyio
+import asyncio
 from fastapi import FastAPI, Request
 from fastapi.responses import Response
 from sse_starlette import EventSourceResponse
@@ -53,6 +54,10 @@ async def handle_sse(request: Request):
                                 write_streams[session_id] = write_streams.pop(placeholder_id)
                                 logger.info(f"Updated placeholder {placeholder_id} to session_id {session_id}")
                     await sse_stream_writer.send({"event": event, "data": data})
+                # Keep-alive loop to maintain the SSE connection
+                while True:
+                    await sse_stream_writer.send({"event": "ping", "data": "keep-alive"})
+                    await asyncio.sleep(15)  # Send keep-alive every 15 seconds
 
         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):
@@ -86,6 +91,31 @@ async def handle_post(request: Request):
                     logger.info(f"Associated placeholder {sid} with session_id {session_id}")
                     break
 
+        if message.get("method") == "initialize" and write_stream:
+            response = {
+                "jsonrpc": "2.0",
+                "id": message.get("id"),
+                "result": {
+                    "protocolVersion": "2025-03-26",
+                    "capabilities": {
+                        "tools": {"listChanged": True},
+                        "prompts": {"listChanged": False},
+                        "resources": {"subscribe": False, "listChanged": False},
+                        "logging": {},
+                        "experimental": {}
+                    },
+                    "serverInfo": {
+                        "name": "airtable-mcp",
+                        "version": "1.0.0"
+                    },
+                    "instructions": "Airtable MCP server for listing and creating records."
+                }
+            }
+            response_data = json.dumps(response)
+            await write_stream.send({"event": "message", "data": response_data})
+            logger.info(f"Sent initialize response for session {session_id} via write_stream")
+            return Response(status_code=202)
+
         if message.get("method") == "tools/list" and write_stream:
             response = {
                 "jsonrpc": "2.0",