Update inference.py

inference.py

@@ -9,7 +9,7 @@ from app import smart_agent_logic
 
 API_BASE_URL = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
 MODEL_NAME = os.getenv("MODEL_NAME", "Qwen/Qwen2.5-72B-Instruct")
-API_KEY = os.getenv("HF_TOKEN") or os.getenv("API_KEY")
+API_KEY = os.getenv("HF_TOKEN") or os.getenv("API_KEY") or os.getenv("OPENAI_API_KEY")
 
 TASK_NAME = os.getenv("MY_ENV_V4_TASK", "easy")
 BENCHMARK = "email_triage_env"
@@ -42,7 +42,15 @@ def log_end(success: bool, steps: int, score: float, rewards: List[float]) -> No
 
 
 def main():
-    client = OpenAI(base_url=API_BASE_URL, api_key=API_KEY)
+    # ✅ SAFE OpenAI initialization (FIX)
+    client = None
+    try:
+        if API_KEY:
+            client = OpenAI(base_url=API_BASE_URL, api_key=API_KEY)
+        else:
+            print("[DEBUG] No API key found, running without OpenAI client", flush=True)
+    except Exception as e:
+        print(f"[DEBUG] OpenAI init failed: {e}", flush=True)
 
     env = EmailTriageEnv(task=TASK_NAME)