Update app.py

app.py

@@ -28,18 +28,18 @@ def invoke(openai_api_key, prompt, rag_option):
         if (RAG_INGESTION):
             return rag_ingestion(collection)
         else:
-            """
+            ###
             match_stage = {
                 "$match": {
-                   "address.country": re.compile(r"United States"),
+                   "address.market": re.compile(r"San Francisco"),
                    "accommodates": { "$gt": 1, "$lt": 3}
                 }
             }
     
             additional_stages = [match_stage]
-            """
-    
-            search_results = rag_retrieval(openai_api_key, prompt, db, collection)
+            ###
+            
+            search_results = rag_retrieval(openai_api_key, prompt, db, collection, additional_stages)
             return rag_inference(openai_api_key, prompt, search_results)
 
 gr.close_all()