krishnachoudhary-hclguvi commited on
Commit
78c01fe
·
unverified ·
1 Parent(s): 386559c

Update OpenAI client to use HF_TOKEN as api_key instead of OPENAI_API_KEY

Browse files
Files changed (1) hide show
  1. inference.py +2 -2
inference.py CHANGED
@@ -24,10 +24,10 @@ def validate_environment():
24
  def main():
25
  validate_environment()
26
 
27
- # Initialize OpenAI Client (per requirements, use OpenAI Python client)
28
  client = OpenAI(
29
  base_url=API_BASE_URL,
30
- api_key=os.getenv("OPENAI_API_KEY", "dummy_if_not_needed_for_custom_endpoint")
31
  )
32
 
33
  env = CodeReviewEnv()
 
24
  def main():
25
  validate_environment()
26
 
27
+ # Initialize OpenAI Client
28
  client = OpenAI(
29
  base_url=API_BASE_URL,
30
+ api_key=HF_TOKEN
31
  )
32
 
33
  env = CodeReviewEnv()