Update kadi_apy_bot.py

kadi_apy_bot.py

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