Ajjack4 commited on
Commit
c1ff000
·
1 Parent(s): d6d0673

accept both HF_TOKEN and OPENAI_API_KEY, default API_BASE_URL and MODEL_NAME

Browse files
Files changed (1) hide show
  1. inference.py +13 -8
inference.py CHANGED
@@ -24,16 +24,21 @@ from openai import OpenAI
24
  # Config
25
  # ---------------------------------------------------------------------------
26
 
27
- _missing = [v for v in ("API_BASE_URL", "MODEL_NAME", "HF_TOKEN") if not os.getenv(v)]
28
- if _missing:
29
- print(f"[ERROR] Missing required environment variables: {', '.join(_missing)}", flush=True)
30
- sys.exit(1)
31
-
32
- API_BASE_URL: str = os.environ["API_BASE_URL"]
33
- MODEL_NAME: str = os.environ["MODEL_NAME"]
34
- HF_TOKEN: str = os.environ["HF_TOKEN"]
 
35
  ENV_BASE_URL: str = os.getenv("ENV_BASE_URL", "http://localhost:7860")
36
 
 
 
 
 
37
  TASK_IDS = ["task1", "task2", "task3"]
38
  BENCHMARK = "dev-env-debugger"
39
  MAX_STEPS = 20
 
24
  # Config
25
  # ---------------------------------------------------------------------------
26
 
27
+ # Accept HF_TOKEN or OPENAI_API_KEY (functional req says OPENAI_API_KEY,
28
+ # mandatory instructions say HF_TOKEN — support both so either works)
29
+ API_BASE_URL: str = os.getenv("API_BASE_URL", "https://api.openai.com/v1")
30
+ MODEL_NAME: str = os.getenv("MODEL_NAME", "gpt-4o-mini")
31
+ HF_TOKEN: str = (
32
+ os.getenv("HF_TOKEN")
33
+ or os.getenv("OPENAI_API_KEY")
34
+ or ""
35
+ )
36
  ENV_BASE_URL: str = os.getenv("ENV_BASE_URL", "http://localhost:7860")
37
 
38
+ if not HF_TOKEN:
39
+ print("[ERROR] Missing API key — set HF_TOKEN or OPENAI_API_KEY", flush=True)
40
+ sys.exit(1)
41
+
42
  TASK_IDS = ["task1", "task2", "task3"]
43
  BENCHMARK = "dev-env-debugger"
44
  MAX_STEPS = 20