Spaces:
Running
Running
Align inference env var handling with submission checklist.
Browse filesRequire HF_TOKEN at runtime, keep defaults only for API_BASE_URL and MODEL_NAME, and wire OpenAI client auth directly to HF_TOKEN.
- inference.py +7 -3
inference.py
CHANGED
|
@@ -17,8 +17,12 @@ import httpx
|
|
| 17 |
# ββ Configuration from environment variables ββββββββββββββββββββββββββββββββ
|
| 18 |
API_BASE_URL = os.environ.get("API_BASE_URL", "https://api.openai.com/v1")
|
| 19 |
MODEL_NAME = os.environ.get("MODEL_NAME", "gpt-4o-mini")
|
| 20 |
-
HF_TOKEN = os.environ.get("HF_TOKEN"
|
| 21 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 22 |
|
| 23 |
# ββ Environment config βββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 24 |
ENV_BASE_URL = os.environ.get("ENV_BASE_URL", "http://localhost:7860")
|
|
@@ -278,7 +282,7 @@ def run_task(
|
|
| 278 |
|
| 279 |
def main():
|
| 280 |
"""Run baseline agent across all 3 tasks."""
|
| 281 |
-
client = OpenAI(base_url=API_BASE_URL, api_key=
|
| 282 |
|
| 283 |
print(f"[DEBUG] Starting SQL Debug Env baseline", flush=True)
|
| 284 |
print(f"[DEBUG] Model: {MODEL_NAME}", flush=True)
|
|
|
|
| 17 |
# ββ Configuration from environment variables ββββββββββββββββββββββββββββββββ
|
| 18 |
API_BASE_URL = os.environ.get("API_BASE_URL", "https://api.openai.com/v1")
|
| 19 |
MODEL_NAME = os.environ.get("MODEL_NAME", "gpt-4o-mini")
|
| 20 |
+
HF_TOKEN = os.environ.get("HF_TOKEN")
|
| 21 |
+
# Optional: used only when running environments via from_docker_image() flows.
|
| 22 |
+
LOCAL_IMAGE_NAME = os.environ.get("LOCAL_IMAGE_NAME")
|
| 23 |
+
|
| 24 |
+
if not HF_TOKEN:
|
| 25 |
+
raise RuntimeError("HF_TOKEN is required for inference.py")
|
| 26 |
|
| 27 |
# ββ Environment config βββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 28 |
ENV_BASE_URL = os.environ.get("ENV_BASE_URL", "http://localhost:7860")
|
|
|
|
| 282 |
|
| 283 |
def main():
|
| 284 |
"""Run baseline agent across all 3 tasks."""
|
| 285 |
+
client = OpenAI(base_url=API_BASE_URL, api_key=HF_TOKEN)
|
| 286 |
|
| 287 |
print(f"[DEBUG] Starting SQL Debug Env baseline", flush=True)
|
| 288 |
print(f"[DEBUG] Model: {MODEL_NAME}", flush=True)
|