Update custom_utils.py

custom_utils.py

@@ -80,6 +80,10 @@ def rag_retrieval(openai_api_key, prompt, db, collection, stages=[], vector_inde
 
 def rag_inference(openai_api_key, prompt, search_results):
     openai.api_key = openai_api_key
+
+    print("###")
+    print(search_results)
+    print("###")
     
     content = f"Answer this user question: {prompt} with the following context:\n{search_results}"
     
@@ -110,7 +114,18 @@ def vector_search(openai_api_key, user_query, db, collection, additional_stages=
             "queryVector": query_embedding,
             "path": "description_embedding",
             "numCandidates": 150,
-            "limit": 20,
+            "limit": 3,
+        }
+    }
+
+    """ filter
+    vector_search_stage = {
+        "$vectorSearch": {
+            "index": vector_index,
+            "queryVector": query_embedding,
+            "path": "description_embedding",
+            "numCandidates": 150,
+            "limit": 3,
             "filter": {
                 "$and": [
                     {"accommodates": {"$eq": 2}}, 
@@ -119,6 +134,7 @@ def vector_search(openai_api_key, user_query, db, collection, additional_stages=
             },
         }
     }
+    """
     
     pipeline = [vector_search_stage] + additional_stages