Update kadi_apy_bot.py

kadi_apy_bot.py

@@ -27,8 +27,14 @@ class KadiAPYBot:
     #    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})
 
+
+        # Vanilla
         doc_contexts = self.retrieve_contexts(query, k=3, filter={"folder": "doc/"})
-        code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"folder": "kadi_apy/"})
+        code_contexts = self.retrieve_contexts(query, k=5, filter={"folder": "kadi_apy/"})
+
+    #    doc_contexts = self.retrieve_contexts(query, k=3, filter={"folder": "doc/"})
+    #    code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"folder": "kadi_apy/"})
+        
         
         # Format contexts
     #   formatted_doc_contexts = self.format_documents(doc_contexts)