Spaces:

bstraehle
/

advanced-rag

Running

bstraehle commited on Sep 6, 2024

Commit

5b4d275

verified ·

1 Parent(s): 8acfede

Update custom_utils.py

Files changed (1) hide show

custom_utils.py CHANGED Viewed

@@ -134,8 +134,8 @@ def vector_search_naive(openai_api_key,
             "index": vector_index,
             "queryVector": query_embedding,
             "path": "description_embedding",
-            "numCandidates": 150,
-            "limit": 25,
         }
     }
@@ -161,8 +161,8 @@ def vector_search_advanced(openai_api_key,
             "index": vector_index,
             "queryVector": query_embedding,
             "path": "description_embedding",
-            "numCandidates": 150,
-            "limit": 25,
             "filter": {
                 "$and": [
                     {"accommodates": {"$eq": accommodates}},

             "index": vector_index,
             "queryVector": query_embedding,
             "path": "description_embedding",
+            "numCandidates": 25,
+            "limit": 3,
         }
     }
             "index": vector_index,
             "queryVector": query_embedding,
             "path": "description_embedding",
+            "numCandidates": 25,
+            "limit": 3,
             "filter": {
                 "$and": [
                     {"accommodates": {"$eq": accommodates}},