Update kadi_apy_bot.py

kadi_apy_bot.py

@@ -21,15 +21,18 @@ class KadiAPYBot:
         rewritten_query = self.rewrite_query(query)
 
         # Predict library usage
-        code_library_usage_prediction = self.predict_library_usage(query)
+    #    code_library_usage_prediction = self.predict_library_usage(query)
         
         # Retrieve contexts
-        doc_contexts = self.retrieve_contexts(query, k=5, filter={"usage": "doc"})
-        code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"usage": code_library_usage_prediction})
+    #    doc_contexts = self.retrieve_contexts(query, k=3, filter={"usage": "doc"})
+    #    code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"usage": code_library_usage_prediction})
+
+        doc_contexts = self.retrieve_contexts(query, k=3, filter={"usage": "doc"})
+        code_contexts = self.retrieve_contexts(rewritten_query, k=5)
         
         # Format contexts
-        formatted_doc_contexts = self.format_documents(doc_contexts)
-        formatted_code_contexts = self.format_documents(code_contexts)
+    #   formatted_doc_contexts = self.format_documents(doc_contexts)
+    #    formatted_code_contexts = self.format_documents(code_contexts)
         
         # Generate response
         response = self.generate_response(query, formatted_doc_contexts, formatted_code_contexts)