Fix OpenAI key env handling and pydantic model_dump

app/env.py

@@ -53,8 +53,8 @@ class SupportDeskEnv:
             self.state_data.history.append({"role": "user", "content": self.state_data.user_message})
 
         return {
-            "observation": self.state_data.dict(),
-            "reward": reward.dict(),
+            "observation": self.state_data.model_dump(),
+            "reward": reward.model_dump(),
             "done": done,
             "info": {"expected_category": self.expected_category, "task_score": task_score}
         }

inference.py

@@ -7,7 +7,8 @@ from openai import OpenAI
 # ENV VARIABLES (STRICT)
 # =============================
 API_BASE_URL = os.environ.get("API_BASE_URL")
-API_KEY = os.environ.get("API_KEY")
+# Prefer OPENAI_API_KEY, but keep API_KEY as a backward-compatible fallback.
+API_KEY = os.environ.get("OPENAI_API_KEY") or os.environ.get("API_KEY")
 MODEL_NAME = os.environ.get("MODEL_NAME", "gpt-4o-mini")
 
 ENV_URL = "http://localhost:7860"
@@ -16,10 +17,11 @@ ENV_URL = "http://localhost:7860"
 # 🔥 GLOBAL LLM CLIENT
 # =============================
 try:
-    client = OpenAI(
-        base_url=API_BASE_URL,
-        api_key=API_KEY
-    )
+    client_kwargs = {"api_key": API_KEY}
+    if API_BASE_URL:
+        client_kwargs["base_url"] = API_BASE_URL
+
+    client = OpenAI(**client_kwargs)
     print("✅ LLM CLIENT INITIALIZED", flush=True)
 except Exception as e:
     print(f"❌ CLIENT INIT FAILED: {e}", flush=True)

server.py

@@ -19,13 +19,13 @@ def root():
 async def reset(req: dict = {}):
     level = req.get("level", "medium")
     obs = await env.reset(level)
-    return {"observation": obs.dict()}
+    return {"observation": obs.model_dump()}
 
 # Optional GET (for browser testing)
 @app.get("/reset")
 async def reset_get():
     obs = await env.reset("medium")
-    return {"observation": obs.dict()}
+    return {"observation": obs.model_dump()}
 
 # -----------------------------
 # STEP
@@ -41,7 +41,7 @@ async def step(action: Action):
 @app.get("/state")
 async def state():
     obs = await env.state()
-    return {"observation": obs.dict()}
+    return {"observation": obs.model_dump()}
 
 # -----------------------------
 # RUN SERVER