feat: add inference.py for OpenEnv compliance

inference.py

@@ -0,0 +1,55 @@
+import os
+import json
+from env import UIEnv, Observation, Action
+from baseline import agent_policy
+from openai import OpenAI
+
+def run_inference(task_id="easy"):
+    """
+    Standard OpenEnv inference entry point.
+    """
+    # 1. Setup Environment
+    env = UIEnv(seed=42, task=task_id)
+    obs = env.reset()
+    
+    # 2. Setup Client (OpenAI or HF Router)
+    openai_key = os.getenv("OPENAI_API_KEY")
+    hf_token = os.getenv("HF_TOKEN")
+    
+    if openai_key:
+        client = OpenAI(api_key=openai_key)
+        model_name = os.getenv("OPENAI_MODEL", "gpt-4o-mini")
+    elif hf_token:
+        client = OpenAI(
+            base_url="https://router.huggingface.co/v1",
+            api_key=hf_token
+        )
+        model_name = "katanemo/Arch-Router-1.5B"
+    else:
+        # Fallback to no-client (heuristic only)
+        client = None
+        model_name = None
+
+    # 3. Perform Inference Step
+    if client:
+        action = agent_policy(client, obs, model_name)
+    else:
+        # Fallback to heuristic if no API key is provided
+        from baseline import heuristic_policy
+        action = heuristic_policy(obs)
+    
+    # 4. Step Environment
+    new_obs, reward, done, info = env.step(action)
+    
+    result = {
+        "action": action.type,
+        "reward": reward,
+        "done": done,
+        "outcome": info.get("outcome")
+    }
+    
+    print(json.dumps(result, indent=2))
+    return result
+
+if __name__ == "__main__":
+    run_inference()