Spaces:
Sleeping
Sleeping
ajaxwin commited on
Commit Β·
409c8b7
1
Parent(s): 277ec6e
fix: Update environment variable loading to use load_dotenv and raise error for missing HF_TOKEN
Browse files- inference.py +6 -7
inference.py
CHANGED
|
@@ -35,20 +35,19 @@ from openai import OpenAI
|
|
| 35 |
from server import Task1Environment, Task2Environment, Task3Environment
|
| 36 |
from env.schemas import Action, ActionType
|
| 37 |
from utils import T1_SYSTEM, T2_SYSTEM, T3_SYSTEM
|
| 38 |
-
from dotenv import
|
| 39 |
|
| 40 |
# βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 41 |
# Configuration
|
| 42 |
# βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 43 |
|
| 44 |
-
|
| 45 |
-
API_BASE_URL =
|
| 46 |
-
MODEL_NAME =
|
| 47 |
-
HF_TOKEN =
|
| 48 |
|
| 49 |
if not HF_TOKEN:
|
| 50 |
-
|
| 51 |
-
exit(1)
|
| 52 |
|
| 53 |
# Benchmark / environment identifier (constant for this env)
|
| 54 |
ENV_BENCHMARK = "smart-contract-audit"
|
|
|
|
| 35 |
from server import Task1Environment, Task2Environment, Task3Environment
|
| 36 |
from env.schemas import Action, ActionType
|
| 37 |
from utils import T1_SYSTEM, T2_SYSTEM, T3_SYSTEM
|
| 38 |
+
from dotenv import load_dotenv
|
| 39 |
|
| 40 |
# βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 41 |
# Configuration
|
| 42 |
# βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 43 |
|
| 44 |
+
load_dotenv()
|
| 45 |
+
API_BASE_URL = os.getenv("API_BASE_URL", "https://api.openai.com/v1")
|
| 46 |
+
MODEL_NAME = os.getenv("MODEL_NAME", "gpt-4o")
|
| 47 |
+
HF_TOKEN = os.getenv("HF_TOKEN", "")
|
| 48 |
|
| 49 |
if not HF_TOKEN:
|
| 50 |
+
raise RuntimeError("HF_TOKEN environment variable not set")
|
|
|
|
| 51 |
|
| 52 |
# Benchmark / environment identifier (constant for this env)
|
| 53 |
ENV_BENCHMARK = "smart-contract-audit"
|