Spaces:
Running
Running
Update custom_utils.py
Browse files- 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 |
|