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
Files changed (1) hide show
  1. 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 dotenv_values
39
 
40
  # ─────────────────────────────────────────────────────────────────────────────
41
  # Configuration
42
  # ─────────────────────────────────────────────────────────────────────────────
43
 
44
- config = dotenv_values(".env")
45
- API_BASE_URL = config.get("API_BASE_URL", "https://api.openai.com/v1")
46
- MODEL_NAME = config.get("MODEL_NAME", "gpt-4o")
47
- HF_TOKEN = config.get("HF_TOKEN", "")
48
 
49
  if not HF_TOKEN:
50
- print("[WARN] HF_TOKEN not set β€” API calls may fail.", file=sys.stderr)
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"