Update custom_utils.py

custom_utils.py

@@ -19,10 +19,12 @@ def rag_ingestion(collection):
     collection.insert_many(dataset)
     return "Manually create a vector search index (in free tier, this feature is not available via SDK)"
 
-def rag_retrieval_naive(openai_api_key, 
-                        prompt, 
-                        db, 
-                        collection, 
+def rag_retrieval_naive(openai_api_key,
+                        prompt,
+                        accomodates,
+                        bedrooms,
+                        db,
+                        collection,
                         vector_index="vector_index"):
     # Naive RAG: Semantic search