Update kadi_apy_bot.py

kadi_apy_bot.py

@@ -24,15 +24,15 @@ class KadiAPYBot:
         code_library_usage_prediction = self.predict_library_usage(query)
         
         # Retrieve contexts
-     #   doc_contexts = self.retrieve_contexts(query, k=3, filter={"directory": "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={"directory": "doc/"})
+        code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"usage": code_library_usage_prediction})
 
 
         # Vanilla
-        doc_contexts = self.retrieve_contexts(query, k=3, filter={"directory": "doc/"})
-        code_contexts = self.retrieve_contexts(query, k=5, filter={"directory": "kadi_apy/"})
+        #doc_contexts = self.retrieve_contexts(query, k=3, filter={"directory": "doc/"})
+        #code_contexts = self.retrieve_contexts(query, k=5, filter={"directory": "kadi_apy/"})
 
-     #   doc_contexts = self.retrieve_contexts(query, k=3, filter={"directory": "doc/"})
+      #  doc_contexts = self.retrieve_contexts(query, k=3, filter={"directory": "doc/"})
       #  code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"directory": "kadi_apy/"})