Update wellbe_agent.py

wellbe_agent.py

@@ -10,7 +10,7 @@ from __future__ import annotations
 import asyncio
 from copy import deepcopy
 from typing import Tuple, Union, Optional
-
+import os
 from agents import Agent, Runner
 from agents.extensions.models.litellm_model import LitellmModel
 from agents.mcp.server import MCPServerStdio, MCPServerStreamableHttp
@@ -74,6 +74,9 @@ async def answer_question(
 
 def answer_sync(question: str, openai_key: str, email: str, password: str) -> str:
     """Blocking wrapper around :func:`answer_question`."""
+    api_key = openai_key or os.getenv("OPENAI_API_KEY")
+    openai.api_key = api_key
+    
     if not question.strip():
         return "Please enter a question."
     try: