Cursor fix

inference.py

@@ -30,9 +30,9 @@ from models import (
     dump_model,
 )
 
-API_BASE_URL = os.environ["API_BASE_URL"]
-MODEL_NAME = os.environ["MODEL_NAME"]
-HF_TOKEN = os.environ["HF_TOKEN"]
+API_BASE_URL = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
+MODEL_NAME = os.getenv("MODEL_NAME", "Qwen/Qwen3.5-397B-A17B")
+HF_TOKEN = os.getenv("HF_TOKEN")
 
 BENCHMARK = "store"
 
@@ -64,13 +64,7 @@ ENV_URL = os.environ.get("ENV_URL", "http://localhost:8000")
 
 
 def log_start(*, task: str, env: str, model: str) -> None:
-    print(
-        json.dumps(
-            {"type": "START", "task": task, "env": env, "model": model},
-            separators=(",", ":"),
-        ),
-        flush=True,
-    )
+    print(f"[START] task={task} env={env} model={model}", flush=True)
 
 
 def log_step(
@@ -81,18 +75,11 @@ def log_step(
     done: bool,
     error: str | None,
 ) -> None:
+    done_str = "true" if done else "false"
+    error_str = error if error else "null"
     print(
-        json.dumps(
-            {
-                "type": "STEP",
-                "step": step,
-                "action": action,
-                "reward": reward,
-                "done": done,
-                "error": error,
-            },
-            separators=(",", ":"),
-        ),
+        f"[STEP] step={step} action={action} reward={reward:.2f} "
+        f"done={done_str} error={error_str}",
         flush=True,
     )
 
@@ -104,17 +91,11 @@ def log_end(
     score: float,
     rewards: list[float],
 ) -> None:
+    success_str = "true" if success else "false"
+    rewards_str = ",".join(f"{r:.2f}" for r in rewards)
     print(
-        json.dumps(
-            {
-                "type": "END",
-                "success": success,
-                "steps": steps,
-                "score": score,
-                "rewards": rewards,
-            },
-            separators=(",", ":"),
-        ),
+        f"[END] success={success_str} steps={steps} score={score:.2f} "
+        f"rewards={rewards_str}",
         flush=True,
     )