bupa1018 commited on
Commit
55442d3
·
1 Parent(s): d37f774

Update kadi_apy_bot.py

Browse files
Files changed (1) hide show
  1. kadi_apy_bot.py +2 -2
kadi_apy_bot.py CHANGED
@@ -27,8 +27,8 @@ class KadiAPYBot:
27
  # doc_contexts = self.retrieve_contexts(query, k=3, filter={"usage": "doc"})
28
  # code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"usage": code_library_usage_prediction})
29
 
30
- doc_contexts = self.retrieve_contexts(query, k=3, filter={"usage": "doc"})
31
- code_contexts = self.retrieve_contexts(rewritten_query, k=5)
32
 
33
  # Format contexts
34
  # formatted_doc_contexts = self.format_documents(doc_contexts)
 
27
  # doc_contexts = self.retrieve_contexts(query, k=3, filter={"usage": "doc"})
28
  # code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"usage": code_library_usage_prediction})
29
 
30
+ doc_contexts = self.retrieve_contexts(query, k=3, filter={"usage": "doc/"})
31
+ code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"usage": code_library_usage_prediction})
32
 
33
  # Format contexts
34
  # formatted_doc_contexts = self.format_documents(doc_contexts)