krishnachoudhary-hclguvi commited on
Commit
1d48da1
·
unverified ·
1 Parent(s): 3d38d37

Update default MODEL_NAME to gpt-4.1-mini and use ValueError for missing HF_TOKEN

Browse files
Files changed (1) hide show
  1. inference.py +3 -8
inference.py CHANGED
@@ -8,21 +8,16 @@ from code_review_env import CodeReviewEnv
8
  # Configuration & Environment Variables
9
  # -------------------------------------------------------------------
10
  API_BASE_URL = os.getenv("API_BASE_URL", "https://api.openai.com/v1")
11
- MODEL_NAME = os.getenv("MODEL_NAME", "gpt-3.5-turbo")
12
  HF_TOKEN = os.getenv("HF_TOKEN")
13
 
14
- def validate_environment():
15
- """Ensure required environment variables like HF_TOKEN are present."""
16
- if not HF_TOKEN:
17
- print("[STEP] step=0 action=init reward=0.00 done=true error=HF_TOKEN_missing")
18
- print("[END] success=false steps=0 rewards=")
19
- sys.exit(1)
20
 
21
  # -------------------------------------------------------------------
22
  # Main Inference Loop
23
  # -------------------------------------------------------------------
24
  def main():
25
- validate_environment()
26
 
27
  # Initialize OpenAI Client
28
  client = OpenAI(
 
8
  # Configuration & Environment Variables
9
  # -------------------------------------------------------------------
10
  API_BASE_URL = os.getenv("API_BASE_URL", "https://api.openai.com/v1")
11
+ MODEL_NAME = os.getenv("MODEL_NAME", "gpt-4.1-mini")
12
  HF_TOKEN = os.getenv("HF_TOKEN")
13
 
14
+ if HF_TOKEN is None:
15
+ raise ValueError("HF_TOKEN environment variable is required")
 
 
 
 
16
 
17
  # -------------------------------------------------------------------
18
  # Main Inference Loop
19
  # -------------------------------------------------------------------
20
  def main():
 
21
 
22
  # Initialize OpenAI Client
23
  client = OpenAI(