Update kadiApy_ragchain.py

kadiApy_ragchain.py

@@ -25,8 +25,8 @@ class KadiApyRagchain:
         
         # Retrieve contexts
         print("Start retrieving:")
-        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})
+        doc_contexts = self.retrieve_contexts(query, k=2, filter={"dataset_category": "kadi_apy_docs"})
+        code_contexts = self.retrieve_contexts(rewritten_query, k=3, filter={"usage": code_library_usage_prediction})
 
 
         # Vanilla