Update kadi_apy_bot.py

kadi_apy_bot.py

@@ -24,13 +24,13 @@ 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/"})
       #  code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"directory": "kadi_apy/"})