ajaxwin commited on
Commit
0b06e9e
Β·
1 Parent(s): 6be844b

fix: Update environment variable loading to use load_dotenv and adjust episode count

Browse files
Files changed (1) hide show
  1. inference.py +6 -6
inference.py CHANGED
@@ -35,23 +35,23 @@ 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
  ENV_BENCHMARK = "smart-contract-audit"
54
- NUM_EPISODES = 1 # keep low on free tier; raise for full eval
55
  SEED_BASE = 42
56
 
57
  # Max LLM calls per episode (including the mandatory submit on last step).
 
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() # Load from .env if available; otherwise rely on actual env vars
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
  print("[WARN] HF_TOKEN not set β€” API calls may fail.", file=sys.stderr)
51
  exit(1)
52
 
53
  ENV_BENCHMARK = "smart-contract-audit"
54
+ NUM_EPISODES = 2 # keep low on free tier; raise for full eval
55
  SEED_BASE = 42
56
 
57
  # Max LLM calls per episode (including the mandatory submit on last step).