Update custom_utils.py

custom_utils.py

@@ -100,7 +100,7 @@ def process_records(data_frame):
         print("Validation error:", e)
         return []
 
-def vector_search(openai_api_key, user_query, db, collection, additional_stages=[], vector_index="vector_index_text"):
+def vector_search(openai_api_key, user_query, db, collection, additional_stages=[], vector_index="vector_index"):
     """
     Perform a vector search in the MongoDB collection based on the user query.
 
@@ -130,12 +130,12 @@ def vector_search(openai_api_key, user_query, db, collection, additional_stages=
             "limit": 20, # return top 20 matches
             "filter": {
                 "$and": [
+                    {"accommodates": {"$eq": 2}}, 
                     {"bedrooms": {"$eq": 1}}
                 ]
             },
         }
     }
-#                    {"accommodates": {"$eq": 2}}, 
     
     # Define the aggregate pipeline with the vector search stage and additional stages
     pipeline = [vector_search_stage] + additional_stages