Update kadiApy_ragchain.py

kadiApy_ragchain.py

@@ -19,13 +19,14 @@ class KadiApyRagchain:
         print("Rewritten Query: ",rewritten_query)
         # Predict library usage
         print("Start prediction:")
-        code_library_usage_prediction = self.predict_library_usage(query)
+        #code_library_usage_prediction = self.predict_library_usage(query)
         
         # Retrieve contexts
         print("Start retrieving:")
 
         
         code_contexts = self.retrieve_contexts(rewritten_query, k=3, filter={"usage": code_library_usage_prediction})
+        code_contexts = self.retrieve_contexts(rewritten_query, k=3, filter={"dataset_category": "kadi_apy_source_code"})
         doc_contexts = self.retrieve_contexts(query, k=2, filter={"dataset_category": "kadi_apy_docs"})
      
         # Format contexts