Update app.py

app.py

@@ -111,14 +111,14 @@ def rag_workflow(query):
     code_library_usage_prediction = rag_chain.predict_library_usage(query)
     print(f"Predicted library usage: {code_library_usage_prediction}")
     
-    rewritten query = rag_chain.rewrite_query(query)
+    rewritten_query = rag_chain.rewrite_query(query)
     print(f"Rewritten query: {code_library_usage_prediction}")
 
     """
     Retrieval-Stage
     """  
-    kadiAPY_doc_documents = rag_chain.retrieve_contexts(query, k = 3, {"usage: doc"})  
-    kadiAPY_code_documents =rag_chain.retrieve_contexts(rewritten_query, k = 3, {"usage": library_usage_prediction} )
+    kadiAPY_doc_documents = rag_chain.retrieve_contexts(query, k = 5, {"usage: doc"})  
+    kadiAPY_code_documents =rag_chain.retrieve_contexts(rewritten_query, k = 5, {"usage": library_usage_prediction} )
     
     print("Retrieved Document Contexts:", kadiAPY_doc_documents)
     print("Retrieved Code Contexts:", kadiAPY_code_documents)