Update app.py

app.py

@@ -342,6 +342,8 @@ def rag_workflow(query):
     )
     metadata_prediction = llm.predict(prompt)
 
+    print("METADATA PREDICTION -------------------------:", metadata_prediction)
+    print(metadata_prediction)
 
 
     rewrite_prompt = (
@@ -370,14 +372,12 @@ def rag_workflow(query):
         """
     )
 
-    rewritten_query_response = llm.invoke("Hello")
+    rewritten_query_response = llm.invoke(rewrite_prompt)
     rewritten_query = rewritten_query_response.content.strip()
     print("AAAAArewritten_query -------------------------:", metadata_prediction)
     print(rewritten_query)
 
 
-    print("METADATA PREDICTION -------------------------:", metadata_prediction)
-    print(metadata_prediction)
     
     kadi_apy_docs = retrieve_within_kadiApy_docs (vector_store, query, k = 5)
     kadi_apy_library_docs = vector_store.similarity_search(query, filter={"usage": metadata_prediction})