Update kadiApy_ragchain.py

kadiApy_ragchain.py

@@ -20,9 +20,11 @@ class KadiApyRagchain:
         rewritten_query = self.rewrite_query(query)
         print("RRRRRRRRRREEEEEEEEEEWRITEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEE: ",rewritten_query)
         # Predict library usage
+        print("Start prediction:")
         code_library_usage_prediction = self.predict_library_usage(query)
         
         # Retrieve contexts
+        print("Start retrieving:")
         doc_contexts = self.retrieve_contexts(query, k=3, filter={"dataset_category": "kadi_apy_docs"})
         code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"usage": code_library_usage_prediction})
 
@@ -36,10 +38,12 @@ class KadiApyRagchain:
         
         
         # Format contexts
+        print("Formatting docs:")
         formatted_doc_contexts = self.format_documents(doc_contexts)
         formatted_code_contexts = self.format_documents(code_contexts)
         
         # Generate response
+        print("Start generatin repsonsse:")
         response = self.generate_response(query, chat_history, formatted_doc_contexts, formatted_code_contexts)
         
         # Add the response to the existing query in the conversation history