Update app.py

app.py

@@ -110,12 +110,13 @@ mcp_server = Server(name="airtable-mcp")
 mcp_server.tool_handlers = tool_handlers  # Set as attribute
 mcp_server.tools = tools  # Set tools as attribute for Deep Agent to discover
 
-# Initialize SseServerTransport
-transport = SseServerTransport("/airtable/mcp")
+# Store write_stream globally to access in POST handler
+write_stream_global = None
 
 # SSE endpoint for GET requests
 @app.get("/airtable/mcp")
 async def handle_sse(request: Request):
+    global write_stream_global
     logger.debug("Handling SSE connection request")
     async def sse_writer():
         logger.debug("Starting SSE writer")
@@ -140,8 +141,9 @@ async def handle_sse(request: Request):
     async with transport.connect_sse(request.scope, request.receive, request._send) as streams:
         read_stream, write_stream = streams
         write_stream_reader = write_stream  # Since streams are MemoryObject streams
+        write_stream_global = write_stream  # Store write_stream for POST handler
         logger.debug("Running MCP server with streams")
-        await mcp_server.run(read_stream, write_stream, mcp_server.create_initialization_options())
+        await mcp_server.run(read_stream, write_stream, mcp_server.create_initialization_options(), tools=tools)
     return EventSourceResponse(sse_stream_reader, data_sender_callable=sse_writer)
 
 # Message handling endpoint for POST requests
@@ -152,9 +154,36 @@ async def handle_post_message(request: Request):
     logger.debug(f"Received POST message body: {body}")
     try:
         message = json.loads(body.decode())
-        if message.get("method") == "tools/list":
+        # Handle initialize request manually to ensure capabilities include tools
+        if message.get("method") == "initialize" and write_stream_global:
+            logger.debug("Handling initialize request manually")
+            response = {
+                "jsonrpc": "2.0",
+                "id": message.get("id"),
+                "result": {
+                    "protocolVersion": "2025-03-26",
+                    "capabilities": {
+                        "tools": {
+                            "listChanged": True
+                        }
+                    },
+                    "serverInfo": {
+                        "name": "airtable-mcp",
+                        "version": "1.0.0"
+                    },
+                    "instructions": "Airtable MCP server for listing and creating records."
+                }
+            }
+            logger.debug(f"Manual initialize response: {response}")
+            session_message = mcp_types.SessionMessage(
+                message=mcp_types.JSONRPCResponse(**response),
+                metadata=mcp_types.ServerMessageMetadata(request_context=request)
+            )
+            await write_stream_global.send(session_message)
+            return Response(status_code=202)
+        # Handle tools/list request manually
+        if message.get("method") == "tools/list" and write_stream_global:
             logger.debug("Handling tools/list request manually")
-            # Create a ListToolsResult response
             response = {
                 "jsonrpc": "2.0",
                 "id": message.get("id"),
@@ -162,11 +191,13 @@ async def handle_post_message(request: Request):
                     "tools": [tool.model_dump(by_alias=True, exclude_none=True) for tool in tools]
                 }
             }
-            # Send the response back through the SSE stream
-            # Since we don't have direct access to transport's send method, log for now
             logger.debug(f"Manual tools/list response: {response}")
-            # We need to find a way to send this response back through the SSE stream
-            # For now, rely on Server to handle it
+            session_message = mcp_types.SessionMessage(
+                message=mcp_types.JSONRPCResponse(**response),
+                metadata=mcp_types.ServerMessageMetadata(request_context=request)
+            )
+            await write_stream_global.send(session_message)
+            return Response(status_code=202)
         await transport.handle_post_message(request.scope, request.receive, request._send)
         logger.debug("POST message handled successfully")
     except Exception as e: