Upload mcp_server_fastmcp.py

mcp_server_fastmcp.py

@@ -200,16 +200,21 @@ def advanced_search_company(company_input: str) -> dict:
 # For production deployment
 if __name__ == "__main__":
     import os
-    import uvicorn
     
-    # Get port from environment or use 7860 for HF Space
+    # Set port from environment (HF Space sets PORT=7860)
     port = int(os.getenv("PORT", "7860"))
     host = os.getenv("HOST", "0.0.0.0")
     
-    # Run with uvicorn manually since mcp.run() defaults to port 8000
-    # We create a simple ASGI app wrapper
-    async def asgi_app(scope, receive, send):
-        # Delegate to FastMCP's SSE handler
-        await mcp._handle_sse_message(scope, receive, send)
+    # Monkeypatch uvicorn.Config to use our port
+    import uvicorn
+    original_config_init = uvicorn.Config.__init__
+    
+    def patched_init(self, *args, **kwargs):
+        kwargs['host'] = host
+        kwargs['port'] = port
+        return original_config_init(self, *args, **kwargs)
+    
+    uvicorn.Config.__init__ = patched_init
     
-    uvicorn.run(asgi_app, host=host, port=port)
+    # Run FastMCP with SSE transport (it will use our patched config)
+    mcp.run(transport="sse")