bupa1018 commited on
Commit
d8f8d74
·
verified ·
1 Parent(s): 23f4676

Update kadiApy_ragchain.py

Browse files
Files changed (1) hide show
  1. kadiApy_ragchain.py +1 -1
kadiApy_ragchain.py CHANGED
@@ -25,7 +25,7 @@ class KadiApyRagchain:
25
  print("Start retrieving:")
26
 
27
 
28
- code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"usage": code_library_usage_prediction})
29
  doc_contexts = self.retrieve_contexts(query, k=2, filter={"dataset_category": "kadi_apy_docs"})
30
 
31
  # Format contexts
 
25
  print("Start retrieving:")
26
 
27
 
28
+ code_contexts = self.retrieve_contexts(rewritten_query, k=3, filter={"usage": code_library_usage_prediction})
29
  doc_contexts = self.retrieve_contexts(query, k=2, filter={"dataset_category": "kadi_apy_docs"})
30
 
31
  # Format contexts