Spaces:
Sleeping
Sleeping
Ajjack4 commited on
Commit ·
c1ff000
1
Parent(s): d6d0673
accept both HF_TOKEN and OPENAI_API_KEY, default API_BASE_URL and MODEL_NAME
Browse files- inference.py +13 -8
inference.py
CHANGED
|
@@ -24,16 +24,21 @@ from openai import OpenAI
|
|
| 24 |
# Config
|
| 25 |
# ---------------------------------------------------------------------------
|
| 26 |
|
| 27 |
-
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
|
| 31 |
-
|
| 32 |
-
|
| 33 |
-
|
| 34 |
-
|
|
|
|
| 35 |
ENV_BASE_URL: str = os.getenv("ENV_BASE_URL", "http://localhost:7860")
|
| 36 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
TASK_IDS = ["task1", "task2", "task3"]
|
| 38 |
BENCHMARK = "dev-env-debugger"
|
| 39 |
MAX_STEPS = 20
|
|
|
|
| 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", "http://localhost:7860")
|
| 37 |
|
| 38 |
+
if not HF_TOKEN:
|
| 39 |
+
print("[ERROR] Missing API key — set HF_TOKEN or OPENAI_API_KEY", flush=True)
|
| 40 |
+
sys.exit(1)
|
| 41 |
+
|
| 42 |
TASK_IDS = ["task1", "task2", "task3"]
|
| 43 |
BENCHMARK = "dev-env-debugger"
|
| 44 |
MAX_STEPS = 20
|