Spaces:
Sleeping
Sleeping
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- inference.py +5 -4
inference.py
CHANGED
|
@@ -24,19 +24,20 @@ from openai import OpenAI
|
|
| 24 |
# Config
|
| 25 |
# ---------------------------------------------------------------------------
|
| 26 |
|
| 27 |
-
#
|
| 28 |
-
#
|
| 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("
|
|
|
|
| 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
|
| 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"
|