bstraehle commited on
Commit
bbfd24e
·
verified ·
1 Parent(s): c067410

Update custom_utils.py

Browse files
Files changed (1) hide show
  1. custom_utils.py +2 -2
custom_utils.py CHANGED
@@ -109,7 +109,7 @@ def invoke_llm(openai_api_key, content):
109
  openai.api_key = openai_api_key
110
 
111
  completion = openai.chat.completions.create(
112
- model="gpt-4o",
113
  messages=[
114
  {
115
  "role": "system",
@@ -323,7 +323,7 @@ def get_stage_sorting():
323
  def invoke_search(db, collection, pipeline):
324
  results = collection.aggregate(pipeline)
325
 
326
- print(f"Vector search millis elapsed: {get_millis_elapsed(db, collection, pipeline)}")
327
 
328
  return list(results)
329
 
 
109
  openai.api_key = openai_api_key
110
 
111
  completion = openai.chat.completions.create(
112
+ model="gpt-4o-mini",
113
  messages=[
114
  {
115
  "role": "system",
 
323
  def invoke_search(db, collection, pipeline):
324
  results = collection.aggregate(pipeline)
325
 
326
+ #print(f"Vector search millis elapsed: {get_millis_elapsed(db, collection, pipeline)}")
327
 
328
  return list(results)
329