Update wellbe_agent.py

wellbe_agent.py

@@ -14,7 +14,7 @@ import os
 import openai
 from agents import Agent, Runner
 from agents.extensions.models.litellm_model import LitellmModel
-from agents.mcp.server import MCPServerStdio, MCPServerStreamableHttp, MCPServerSse
+from agents.mcp.server import MCPServerStreamableHttp, MCPServerSse
 
 from config import PROMPT_TEMPLATE, MCP_CONFIGS
 
@@ -26,7 +26,7 @@ __all__ = [
 ]
 
 # Global variables to store initialized servers
-_healthcare_server: Optional[MCPServerStreamableHttp] = None
+_healthcare_server: Optional[Union[MCPServerSse, str]] = None
 _whoop_server: Optional[Union[MCPServerSse, str]] = None
 
 
@@ -38,10 +38,13 @@ def initialize_mcp_servers(whoop_email: str, whoop_password: str) -> None:
         os.environ["WHOOP_EMAIL"] = whoop_email
         os.environ["WHOOP_PASSWORD"] = whoop_password
         
-        _healthcare_server = MCPServerStreamableHttp(
-            params=cfg["healthcare-mcp-public"], name="Healthcare MCP Server"
+        _healthcare_server = MCPServerSse(
+            params={
+                "url": "https://agents-mcp-hackathon-healthcare-mcp-public.hf.space/gradio_api/mcp/sse"
+            },
+            name="Healthcare MCP Server"
         )
-        # _whoop_server = MCPServerStdio(params=cfg["whoop"], name="Whoop MCP Server")
+
         _whoop_server = MCPServerSse(
             params={
                 "url": "https://agents-mcp-hackathon-whoop-mcp-server.hf.space/gradio_api/mcp/sse"