Update wellbe_agent.py

wellbe_agent.py

@@ -1,6 +1,5 @@
 """
 Core logic for the WellBe+ multi‑context AI assistant.
-
 This module handles MCP server instantiation, model selection, and single‑shot
 question answering. It is intentionally agnostic of any UI layer so that it can
 be re‑used from a command‑line script, a notebook, or the Gradio front‑end.
@@ -10,7 +9,7 @@ from __future__ import annotations
 
 import asyncio
 from copy import deepcopy
-from typing import Tuple, Union
+from typing import Tuple, Union, Optional
 
 from agents import Agent, Runner
 from agents.extensions.models.litellm_model import LitellmModel
@@ -19,33 +18,30 @@ from agents.mcp.server import MCPServerStdio, MCPServerStreamableHttp
 from config import PROMPT_TEMPLATE, MCP_CONFIGS
 
 __all__ = [
-    "build_mcp_servers",
+    "initialize_mcp_servers",
     "select_model",
     "answer_question",
+    "answer_sync",
 ]
 
+# Global variables to store initialized servers
+_healthcare_server: Optional[MCPServerStreamableHttp] = None
+_whoop_server: Optional[Union[MCPServerStdio, str]] = None
+
+
+def initialize_mcp_servers(whoop_email: str, whoop_password: str) -> None:
+    """Initialize and cache the MCP servers (once only)."""
+    global _healthcare_server, _whoop_server
+    if _healthcare_server is None:
+        cfg = deepcopy(MCP_CONFIGS)
+        cfg["whoop"].update({"username": whoop_email, "password": whoop_password})
+
+        _healthcare_server = MCPServerStreamableHttp(
+            params=cfg["healthcare-mcp-public"], name="Healthcare MCP Server"
+        )
+        # _whoop_server = MCPServerStdio(params=cfg["whoop"], name="Whoop MCP Server")
+        _whoop_server = "Whoop Tool Not available now"
 
-def build_mcp_servers(
-    whoop_email: str, whoop_password: str
-) -> Tuple[MCPServerStreamableHttp, MCPServerStdio]:
-    """Return configured Healthcare and Whoop MCP servers.
-
-    Parameters
-    ----------
-    whoop_email, whoop_password
-        User credentials for the private Whoop feed. The credentials are merged
-        into a *deep‑copied* variant of ``MCP_CONFIGS`` so that the global
-        settings remain untouched.
-    """
-    cfg = deepcopy(MCP_CONFIGS)
-    cfg["whoop"].update({"username": whoop_email, "password": whoop_password})
-
-    healthcare_server = MCPServerStreamableHttp(
-        params=cfg["healthcare-mcp-public"], name="Healthcare MCP Server"
-    )
-    # whoop_server = MCPServerStdio(params=cfg["whoop"], name="Whoop MCP Server")
-    whoop_server = "Whoop Tool Not available now"
-    return healthcare_server, whoop_server
 
 def select_model() -> Union[str, LitellmModel]:
     return "o3-mini"
@@ -58,16 +54,18 @@ async def answer_question(
     whoop_password: str,
 ) -> str:
     """Run the WellBe+ agent on a single question and return the assistant reply."""
-    healthcare_srv, whoop_srv = build_mcp_servers(whoop_email, whoop_password)
+    initialize_mcp_servers(whoop_email, whoop_password)
+    if _healthcare_server is None:
+        return "**Error:** MCP servers not initialized."
 
-    async with healthcare_srv as hserver: #, whoop_srv as wserver:
+    async with _healthcare_server as hserver:  # , _whoop_server as wserver:
         agent = Agent(
             name="WellBe+ Assistant",
             instructions=PROMPT_TEMPLATE,
             model=select_model(),
             mcp_servers=[
-                hserver 
-                #, wserver
+                hserver
+                # , wserver
             ],
         )
         result = await Runner.run(agent, question)