Update custom_utils.py

custom_utils.py

@@ -29,10 +29,12 @@ def rag_retrieval_naive(openai_api_key,
     # Naive RAG: Semantic search
     
     retrieval_result = vector_search_naive(
-        openai_api_key, 
-        prompt, 
-        db, 
-        collection, 
+        openai_api_key,
+        prompt,
+        accomodates,
+        bedrooms,
+        db,
+        collection,
         vector_index
     )
 
@@ -122,11 +124,13 @@ def invoke_llm(openai_api_key, content):
     return completion.choices[0].message.content
     
 def vector_search_naive(openai_api_key, 
-                        user_query, 
+                        prompt,
+                        accomodates,
+                        bedrooms,
                         db, 
                         collection, 
                         vector_index="vector_index"):
-    query_embedding = get_text_embedding(openai_api_key, user_query)
+    query_embedding = get_text_embedding(openai_api_key, prompt)
 
     if query_embedding is None:
         return "Invalid query or embedding generation failed."
@@ -150,14 +154,14 @@ def vector_search_naive(openai_api_key,
     return invoke_search(db, collection, pipeline)
 
 def vector_search_advanced(openai_api_key, 
-                           user_query, 
+                           prompt, 
                            accommodates, 
                            bedrooms, 
                            db, 
                            collection, 
                            additional_stages=[], 
                            vector_index="vector_index"):
-    query_embedding = get_text_embedding(openai_api_key, user_query)
+    query_embedding = get_text_embedding(openai_api_key, prompt)
 
     if query_embedding is None:
         return "Invalid query or embedding generation failed."