Use validator's API_KEY and API_BASE_URL from environment

inference.py

@@ -18,10 +18,12 @@ except ImportError:
 from environment.env import EmailTriageEnv
 from environment.types import Action, EmailCategory, Team
 
-# Environment variables
+# Environment variables - check both formats
+# Validator provides: API_KEY and API_BASE_URL
+# Local usage: OPENAI_API_KEY
+API_KEY = os.getenv("API_KEY") or os.getenv("OPENAI_API_KEY")
 API_BASE_URL = os.getenv("API_BASE_URL", "https://api.openai.com/v1")
 MODEL_NAME = os.getenv("MODEL_NAME", "gpt-4o-mini")
-API_KEY = os.getenv("OPENAI_API_KEY")
 
 # Configuration
 MAX_STEPS = 50
@@ -200,9 +202,21 @@ def main() -> None:
     tasks = ["spam_detection", "multi_class_routing", "context_aware_triage"]
     all_scores = []
 
-    if not API_KEY or not OPENAI_AVAILABLE:
-        # Demo/Validation mode: No API key provided
-        print("[WARNING] OPENAI_API_KEY not set. Running in validation mode.",
+    # Try to initialize OpenAI client if API key is available
+    client = None
+    if API_KEY and OPENAI_AVAILABLE:
+        try:
+            # Initialize with validator's provided API_BASE_URL and API_KEY
+            client = OpenAI(api_key=API_KEY, base_url=API_BASE_URL)
+            print(f"[INFO] Using API endpoint: {API_BASE_URL}", flush=True)
+        except Exception as e:
+            print(f"[WARNING] Failed to initialize OpenAI client: {e}",
+                  file=sys.stderr, flush=True)
+            client = None
+
+    if client is None:
+        # Demo/Validation mode: No API key or OpenAI not available
+        print("[WARNING] No API credentials available. Running in validation mode.",
               flush=True)
 
         for task in tasks:
@@ -289,25 +303,16 @@ def main() -> None:
                   flush=True)
 
     else:
-        # Normal mode: Use OpenAI API
-        try:
-            client = OpenAI(api_key=API_KEY, base_url=API_BASE_URL if API_BASE_URL
-                            else None)
-        except Exception as e:
-            print(f"[ERROR] Failed to initialize OpenAI client: {e}",
-                  file=sys.stderr, flush=True)
-            all_scores = [0.0, 0.0, 0.0]
-
-        if client is not None:
-            for task in tasks:
-                try:
-                    success, steps, score, rewards = run_task(client, task)
-                    all_scores.append(score)
-                    print(f"[TASK_SUMMARY] {task}: score={score:.3f} steps={steps}",
-                          flush=True)
-                except Exception as e:
-                    print(f"[TASK_ERROR] {task}: {e}", file=sys.stderr, flush=True)
-                    all_scores.append(0.0)
+        # Normal mode: Use OpenAI API (through validator's proxy if available)
+        for task in tasks:
+            try:
+                success, steps, score, rewards = run_task(client, task)
+                all_scores.append(score)
+                print(f"[TASK_SUMMARY] {task}: score={score:.3f} steps={steps}",
+                      flush=True)
+            except Exception as e:
+                print(f"[TASK_ERROR] {task}: {e}", file=sys.stderr, flush=True)
+                all_scores.append(0.0)
 
     # Final summary
     avg_score = sum(all_scores) / len(all_scores) if all_scores else 0.0