alimamaTech commited on
Commit
82da82c
·
verified ·
1 Parent(s): 80cb61f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -77,7 +77,7 @@ class SpyAgent(BasicAgent):
77
  api_key=os.getenv('API_KEY')
78
  )
79
  completion = client.chat.completions.create(
80
- model="gpt-4o-mini",
81
  messages=[
82
  {'role': 'system', 'content': 'You are a helpful assistant.'},
83
  {'role': 'user', 'content': prompt}
 
77
  api_key=os.getenv('API_KEY')
78
  )
79
  completion = client.chat.completions.create(
80
+ model=os.getenv('MODEL_NAME'),
81
  messages=[
82
  {'role': 'system', 'content': 'You are a helpful assistant.'},
83
  {'role': 'user', 'content': prompt}