Ajjack4 commited on
Commit
2f0f7b7
·
1 Parent(s): ed395b6

read API_KEY env var for validator LiteLLM proxy, keep HF_TOKEN/OPENAI_API_KEY as fallbacks

Browse files
Files changed (1) hide show
  1. inference.py +5 -4
inference.py CHANGED
@@ -24,19 +24,20 @@ from openai import OpenAI
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", "https://ajjack404-dev-env-debugger.hf.space")
37
 
38
  if not HF_TOKEN:
39
- print("[DEBUG] Warning: no API key found (HF_TOKEN / OPENAI_API_KEY). LLM calls will fail.", flush=True)
40
 
41
  TASK_IDS = ["task1", "task2", "task3"]
42
  BENCHMARK = "dev-env-debugger"
 
24
  # Config
25
  # ---------------------------------------------------------------------------
26
 
27
+ # Validator injects API_BASE_URL and API_KEY via LiteLLM proxy.
28
+ # Also accept HF_TOKEN / OPENAI_API_KEY for local testing.
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("API_KEY") # validator-injected key (LiteLLM proxy)
33
+ or os.getenv("HF_TOKEN")
34
  or os.getenv("OPENAI_API_KEY")
35
  or ""
36
  )
37
  ENV_BASE_URL: str = os.getenv("ENV_BASE_URL", "https://ajjack404-dev-env-debugger.hf.space")
38
 
39
  if not HF_TOKEN:
40
+ print("[DEBUG] Warning: no API key found. LLM calls will fail.", flush=True)
41
 
42
  TASK_IDS = ["task1", "task2", "task3"]
43
  BENCHMARK = "dev-env-debugger"