Update custom_utils.py

custom_utils.py

@@ -134,12 +134,16 @@ def vector_search_naive(openai_api_key,
             "index": vector_index,
             "queryVector": query_embedding,
             "path": "description_embedding",
-            "numCandidates": 25,
-            "limit": 3,
+            "numCandidates": 150,
+            "limit": 25,
         }
     }
 
-    pipeline = [vector_search_stage, get_stage_include_fields()]
+    pipeline = [
+        vector_search_stage, 
+        get_stage_include_fields(), 
+        get_stage_filter_result(accomodates, bedrooms)
+    ]
 
     return invoke_search(db, collection, pipeline)
 
@@ -161,8 +165,8 @@ def vector_search_advanced(openai_api_key,
             "index": vector_index,
             "queryVector": query_embedding,
             "path": "description_embedding",
-            "numCandidates": 25,
-            "limit": 3,
+            "numCandidates": 150,
+            "limit": 25,
             "filter": {
                 "$and": [
                     {"accommodates": {"$eq": accommodates}}, 
@@ -172,7 +176,10 @@ def vector_search_advanced(openai_api_key,
         }
     }
   
-    pipeline = [vector_search_and_filter_stage, get_stage_include_fields()] + additional_stages
+    pipeline = [
+        vector_search_and_filter_stage, 
+        get_stage_include_fields()
+    ] + additional_stages
 
     return invoke_search(db, collection, pipeline)
 
@@ -258,11 +265,11 @@ def get_stage_include_fields():
         }
     }
 
-def get_stage_filter_result():
+def get_stage_filter_result(accomodates, bedrooms):
     return {
         "$match": {
-            "accommodates": { "$eq": 2},
-            "bedrooms": { "$eq": 1}
+            "accommodates": { "$eq": accomodates},
+            "bedrooms": { "$eq": bedrooms}
         }
     }