bupa1018 commited on
Commit
10d6238
·
1 Parent(s): 143c033

Update kadi_apy_bot.py

Browse files
Files changed (1) hide show
  1. kadi_apy_bot.py +1 -1
kadi_apy_bot.py CHANGED
@@ -25,7 +25,7 @@ class KadiAPYBot:
25
 
26
  # Retrieve contexts
27
  doc_contexts = self.retrieve_contexts(query, k=3, filter={"directory": "doc/"})
28
- code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"directory": code_library_usage_prediction})
29
 
30
 
31
  # Vanilla
 
25
 
26
  # Retrieve contexts
27
  doc_contexts = self.retrieve_contexts(query, k=3, filter={"directory": "doc/"})
28
+ code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"usage": code_library_usage_prediction})
29
 
30
 
31
  # Vanilla