bupa1018 commited on
Commit
b1c79ae
·
1 Parent(s): da2d7a8

Update kadiApy_ragchain.py

Browse files
Files changed (1) hide show
  1. kadiApy_ragchain.py +5 -5
kadiApy_ragchain.py CHANGED
@@ -23,16 +23,16 @@ class KadiApyRagchain:
23
  code_library_usage_prediction = self.predict_library_usage(query)
24
 
25
  # Retrieve contexts
26
- doc_contexts = self.retrieve_contexts(query, k=3, filter={"directory": "doc/"})
27
  code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"usage": code_library_usage_prediction})
28
 
29
 
30
  # Vanilla
31
- #doc_contexts = self.retrieve_contexts(query, k=3, filter={"directory": "doc/"})
32
- #code_contexts = self.retrieve_contexts(query, k=5, filter={"directory": "kadi_apy/"})
33
 
34
- # doc_contexts = self.retrieve_contexts(query, k=3, filter={"directory": "doc/"})
35
- # code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"directory": "kadi_apy/"})
36
 
37
 
38
  # Format contexts
 
23
  code_library_usage_prediction = self.predict_library_usage(query)
24
 
25
  # Retrieve contexts
26
+ doc_contexts = self.retrieve_contexts(query, k=3, filter={"dataset_category": "kadi_apy_docs"})
27
  code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"usage": code_library_usage_prediction})
28
 
29
 
30
  # Vanilla
31
+ #doc_contexts = self.retrieve_contexts(query, k=3, filter={"dataset_category": "kadi_apy_docs"})
32
+ #code_contexts = self.retrieve_contexts(query, k=5, filter={"dataset_category": "kadi_apy_source_code"})
33
 
34
+ # doc_contexts = self.retrieve_contexts(query, k=3, filter={"dataset_category": "kadi_apy_docs"})
35
+ # code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"dataset_category": "kadi_apy_source_code"})
36
 
37
 
38
  # Format contexts