Update app.py

app.py

@@ -112,13 +112,13 @@ def rag_workflow(query):
     print(f"Predicted library usage: {code_library_usage_prediction}")
     
     rewritten_query = rag_chain.rewrite_query(query)
-    print(f"Rewritten query: {code_library_usage_prediction}")
+    print(f"Rewritten query: {rewritten_query}")
 
     """
     Retrieval-Stage
     """  
     kadiAPY_doc_documents = rag_chain.retrieve_contexts(query, k=5, filter={"usage": "doc"})
-    kadiAPY_code_documents = rag_chain.retrieve_contexts(rewritten_query, k=3, filter={"usage": code_library_usage_prediction})
+    kadiAPY_code_documents = rag_chain.retrieve_contexts(str(rewritten_query.content), k=3, filter={"usage": code_library_usage_prediction})
     
     print("Retrieved Document Contexts:", kadiAPY_doc_documents)
     print("Retrieved Code Contexts:", kadiAPY_code_documents)