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 |
|