omri374 commited on
Commit
e214413
·
verified ·
1 Parent(s): f98f63e

Update src/llm_calls.py

Browse files
Files changed (1) hide show
  1. src/llm_calls.py +1 -1
src/llm_calls.py CHANGED
@@ -121,7 +121,7 @@ def query_openai(full_messages, api_key: str) -> Generator[str, None, None]:
121
  openai_client = OpenAI(api_key=api_key)
122
 
123
  response = openai_client.chat.completions.create(
124
- model="gpt-4-turbo",
125
  messages=full_messages,
126
  stream=True, # Enable streaming
127
  )
 
121
  openai_client = OpenAI(api_key=api_key)
122
 
123
  response = openai_client.chat.completions.create(
124
+ model="gpt-4o",
125
  messages=full_messages,
126
  stream=True, # Enable streaming
127
  )