md896 commited on
Commit
e7c61ad
Β·
1 Parent(s): 74e3e43

Align inference env var handling with submission checklist.

Browse files

Require HF_TOKEN at runtime, keep defaults only for API_BASE_URL and MODEL_NAME, and wire OpenAI client auth directly to HF_TOKEN.

Files changed (1) hide show
  1. inference.py +7 -3
inference.py CHANGED
@@ -17,8 +17,12 @@ import httpx
17
  # ── Configuration from environment variables ────────────────────────────────
18
  API_BASE_URL = os.environ.get("API_BASE_URL", "https://api.openai.com/v1")
19
  MODEL_NAME = os.environ.get("MODEL_NAME", "gpt-4o-mini")
20
- HF_TOKEN = os.environ.get("HF_TOKEN", "")
21
- API_KEY = os.environ.get("OPENAI_API_KEY", HF_TOKEN or "sk-placeholder")
 
 
 
 
22
 
23
  # ── Environment config ───────────────────────────────────────────────────────
24
  ENV_BASE_URL = os.environ.get("ENV_BASE_URL", "http://localhost:7860")
@@ -278,7 +282,7 @@ def run_task(
278
 
279
  def main():
280
  """Run baseline agent across all 3 tasks."""
281
- client = OpenAI(base_url=API_BASE_URL, api_key=API_KEY)
282
 
283
  print(f"[DEBUG] Starting SQL Debug Env baseline", flush=True)
284
  print(f"[DEBUG] Model: {MODEL_NAME}", flush=True)
 
17
  # ── Configuration from environment variables ────────────────────────────────
18
  API_BASE_URL = os.environ.get("API_BASE_URL", "https://api.openai.com/v1")
19
  MODEL_NAME = os.environ.get("MODEL_NAME", "gpt-4o-mini")
20
+ HF_TOKEN = os.environ.get("HF_TOKEN")
21
+ # Optional: used only when running environments via from_docker_image() flows.
22
+ LOCAL_IMAGE_NAME = os.environ.get("LOCAL_IMAGE_NAME")
23
+
24
+ if not HF_TOKEN:
25
+ raise RuntimeError("HF_TOKEN is required for inference.py")
26
 
27
  # ── Environment config ───────────────────────────────────────────────────────
28
  ENV_BASE_URL = os.environ.get("ENV_BASE_URL", "http://localhost:7860")
 
282
 
283
  def main():
284
  """Run baseline agent across all 3 tasks."""
285
+ client = OpenAI(base_url=API_BASE_URL, api_key=HF_TOKEN)
286
 
287
  print(f"[DEBUG] Starting SQL Debug Env baseline", flush=True)
288
  print(f"[DEBUG] Model: {MODEL_NAME}", flush=True)