Spaces:
Runtime error
Runtime error
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- 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.
|
| 28 |
-
API_BASE_URL = os.
|
| 29 |
-
MODEL_NAME = os.
|
| 30 |
-
ENV_URL = os.
|
| 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
|
| 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)
|