bupa1018 commited on
Commit
ddc35e2
·
verified ·
1 Parent(s): 0edf437

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=3, filter={"usage": code_library_usage_prediction})
29
  code_contexts = self.retrieve_contexts(rewritten_query, k=3, filter={"dataset_category": "kadi_apy_source_code"})
30
  doc_contexts = self.retrieve_contexts(query, k=2, filter={"dataset_category": "kadi_apy_docs"})
31
 
 
25
  print("Start retrieving:")
26
 
27
 
28
+ #code_contexts = self.retrieve_contexts(rewritten_query, k=3, filter={"usage": code_library_usage_prediction})
29
  code_contexts = self.retrieve_contexts(rewritten_query, k=3, filter={"dataset_category": "kadi_apy_source_code"})
30
  doc_contexts = self.retrieve_contexts(query, k=2, filter={"dataset_category": "kadi_apy_docs"})
31