fixed the timeout

proxy.py

@@ -78,7 +78,15 @@ async def proxy_handler(request):
         target_url += f"?{query_string}"
 
     # Forward request to FastMCP
-    async with ClientSession(timeout=ClientTimeout(total=300)) as session:
+    # For SSE connections: total=None disables overall timeout (keeps connection alive)
+    # Still use socket timeouts for safety (sock_connect, sock_read)
+    async with ClientSession(
+        timeout=ClientTimeout(
+            total=None,         # No total timeout for long-lived SSE connections
+            sock_connect=30,    # 30 seconds for initial connection
+            sock_read=300       # 5 minutes for individual socket reads
+        )
+    ) as session:
         try:
             # Copy headers
             headers = dict(request.headers)
@@ -98,11 +106,31 @@ async def proxy_handler(request):
                         )
                         await response.prepare(request)
 
-                        # Stream chunks from FastMCP to client
-                        async for chunk in resp.content.iter_any():
-                            await response.write(chunk)
+                        # Background task to send keep-alive pings (prevents timeout)
+                        async def send_keepalive():
+                            try:
+                                while True:
+                                    await asyncio.sleep(30)  # Send ping every 30 seconds
+                                    await response.write(b":\n\n")  # SSE comment (ignored by client)
+                            except asyncio.CancelledError:
+                                pass
+
+                        keepalive_task = asyncio.create_task(send_keepalive())
+
+                        try:
+                            # Stream chunks from FastMCP to client
+                            async for chunk in resp.content.iter_any():
+                                await response.write(chunk)
+
+                            await response.write_eof()
+                        finally:
+                            # Cancel keep-alive task when streaming completes
+                            keepalive_task.cancel()
+                            try:
+                                await keepalive_task
+                            except asyncio.CancelledError:
+                                pass
 
-                        await response.write_eof()
                         return response
                     else:
                         # For regular responses, read entire body

server.py

@@ -113,6 +113,8 @@ def extract_api_key_from_request():
 mcp = FastMCP(
     name="FleetMind Dispatch Coordinator",
     version="1.0.0"
+    # Note: request_timeout not supported in FastMCP 2.13.1
+    # Timeout handling is done in proxy.py instead
 )
 
 # ============================================================================