UjjwalPardeshi commited on
Commit
00f62eb
·
1 Parent(s): a72d669

fix: use os.environ for API_KEY and API_BASE_URL, remove dummy fallback

Browse files
Files changed (1) hide show
  1. inference.py +6 -6
inference.py CHANGED
@@ -21,13 +21,13 @@ from openai import OpenAI
21
  from openenv.core import GenericAction, GenericEnvClient
22
 
23
  # ---------------------------------------------------------------------------
24
- # Configuration
25
  # ---------------------------------------------------------------------------
26
  IMAGE_NAME = os.getenv("IMAGE_NAME") or os.getenv("LOCAL_IMAGE_NAME")
27
- API_KEY = os.getenv("API_KEY") or os.getenv("HF_TOKEN")
28
- API_BASE_URL = os.getenv("API_BASE_URL") or "https://api.openai.com/v1"
29
- MODEL_NAME = os.getenv("MODEL_NAME") or "gpt-4o"
30
- ENV_URL = os.getenv("ENV_URL", "https://ujjwalpardeshi-pytorch-training-debugger.hf.space")
31
  BENCHMARK = "pytorch-training-debugger"
32
 
33
  MAX_STEPS = 25
@@ -250,7 +250,7 @@ async def main() -> None:
250
  target_task = os.getenv("TASK_NAME")
251
  tasks_to_run = [target_task] if target_task else ALL_TASK_IDS
252
 
253
- client = OpenAI(base_url=API_BASE_URL, api_key=API_KEY or "dummy")
254
 
255
  print(f"[DEBUG] API_BASE_URL={API_BASE_URL}", flush=True)
256
  print(f"[DEBUG] API_KEY={'set' if API_KEY else 'NOT SET'} (source={'API_KEY' if os.getenv('API_KEY') else 'HF_TOKEN' if os.getenv('HF_TOKEN') else 'NONE'})", flush=True)
 
21
  from openenv.core import GenericAction, GenericEnvClient
22
 
23
  # ---------------------------------------------------------------------------
24
+ # Configuration — use evaluator-injected env vars EXACTLY as documented
25
  # ---------------------------------------------------------------------------
26
  IMAGE_NAME = os.getenv("IMAGE_NAME") or os.getenv("LOCAL_IMAGE_NAME")
27
+ API_KEY = os.environ.get("API_KEY") or os.environ.get("HF_TOKEN", "")
28
+ API_BASE_URL = os.environ.get("API_BASE_URL", "https://api.openai.com/v1")
29
+ MODEL_NAME = os.environ.get("MODEL_NAME", "gpt-4o")
30
+ ENV_URL = os.environ.get("ENV_URL", "https://ujjwalpardeshi-pytorch-training-debugger.hf.space")
31
  BENCHMARK = "pytorch-training-debugger"
32
 
33
  MAX_STEPS = 25
 
250
  target_task = os.getenv("TASK_NAME")
251
  tasks_to_run = [target_task] if target_task else ALL_TASK_IDS
252
 
253
+ client = OpenAI(base_url=API_BASE_URL, api_key=API_KEY)
254
 
255
  print(f"[DEBUG] API_BASE_URL={API_BASE_URL}", flush=True)
256
  print(f"[DEBUG] API_KEY={'set' if API_KEY else 'NOT SET'} (source={'API_KEY' if os.getenv('API_KEY') else 'HF_TOKEN' if os.getenv('HF_TOKEN') else 'NONE'})", flush=True)