Update config.py

config.py

@@ -9,16 +9,26 @@ import asyncio
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
 logger = logging.getLogger("TxAgentAPI")
 
-# Globals
-agent = None
-patients_collection = None
-analysis_collection = None
-alerts_collection = None
-users_collection = None
-
-# JWT settings (must match CPS-API)
-SECRET_KEY = os.getenv("SECRET_KEY", "your-secret-key")
-ALGORITHM = "HS256"
+# Initialize agent synchronously with error handling
+try:
+    agent = TxAgent(
+        model_name="mims-harvard/TxAgent-T1-Llama-3.1-8B",
+        rag_model_name="mims-harvard/ToolRAG-T1-GTE-Qwen2-1.5B",
+        enable_finish=True,
+        enable_rag=False,
+        force_finish=True,
+        enable_checker=True,
+        step_rag_num=4,
+        seed=42
+    )
+    agent.init_model()
+    agent.chat_prompt = (
+        "You are a clinical assistant AI. Analyze the patient's data and provide clear clinical recommendations."
+    )
+    logger.info("✅ TxAgent initialized synchronously with chat_prompt: %s", agent.chat_prompt)
+except Exception as e:
+    logger.error("❌ Failed to initialize TxAgent: %s", str(e))
+    raise
 
 # Initialize collections synchronously
 db = get_mongo_client()["cps_db"]
@@ -28,28 +38,13 @@ analysis_collection = db["patient_analysis_results"]
 alerts_collection = db["clinical_alerts"]
 logger.info("📡 Connected to MongoDB synchronously")
 
-def setup_app(app: FastAPI):
-    global agent
+# JWT settings (must match CPS-API)
+SECRET_KEY = os.getenv("SECRET_KEY", "your-secret-key")
+ALGORITHM = "HS256"
 
+def setup_app(app: FastAPI):
     @app.on_event("startup")
     async def startup_event():
-        global agent
-        agent = TxAgent(
-            model_name="mims-harvard/TxAgent-T1-Llama-3.1-8B",
-            rag_model_name="mims-harvard/ToolRAG-T1-GTE-Qwen2-1.5B",
-            enable_finish=True,
-            enable_rag=False,
-            force_finish=True,
-            enable_checker=True,
-            step_rag_num=4,
-            seed=42
-        )
-        agent.chat_prompt = (
-            "You are a clinical assistant AI. Analyze the patient's data and provide clear clinical recommendations."
-        )
-        agent.init_model()
-        logger.info("✅ TxAgent initialized")
-
         # Run async analysis task
         asyncio.create_task(analyze_all_patients())