Update kadiApy_ragchain.py

kadiApy_ragchain.py

@@ -23,16 +23,16 @@ class KadiApyRagchain:
         code_library_usage_prediction = self.predict_library_usage(query)
         
         # Retrieve contexts
-        doc_contexts = self.retrieve_contexts(query, k=3, filter={"directory": "doc/"})
+        doc_contexts = self.retrieve_contexts(query, k=3, filter={"dataset_category": "kadi_apy_docs"})
         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={"dataset_category": "kadi_apy_docs"})
+        #code_contexts = self.retrieve_contexts(query, k=5, filter={"dataset_category": "kadi_apy_source_code"})
 
-      #  doc_contexts = self.retrieve_contexts(query, k=3, filter={"directory": "doc/"})
-      #  code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"directory": "kadi_apy/"})
+      #  doc_contexts = self.retrieve_contexts(query, k=3, filter={"dataset_category": "kadi_apy_docs"})
+      #  code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"dataset_category": "kadi_apy_source_code"})
         
         
         # Format contexts