bupa1018 commited on
Commit
ddab6e1
·
1 Parent(s): 903fab4

Update kadi_apy_bot.py

Browse files
Files changed (1) hide show
  1. kadi_apy_bot.py +2 -2
kadi_apy_bot.py CHANGED
@@ -29,8 +29,8 @@ class KadiAPYBot:
29
 
30
 
31
  # Vanilla
32
- doc_contexts = self.retrieve_contexts(query, k=3, filter={"folder": "doc/"})
33
- code_contexts = self.retrieve_contexts(query, k=5, filter={"folder": "kadi_apy/"})
34
 
35
  # doc_contexts = self.retrieve_contexts(query, k=3, filter={"folder": "doc/"})
36
  # code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"folder": "kadi_apy/"})
 
29
 
30
 
31
  # Vanilla
32
+ doc_contexts = self.retrieve_contexts(query, k=3, filter={"directory": "doc/"})
33
+ code_contexts = self.retrieve_contexts(query, k=5, filter={"directory": "kadi_apy/"})
34
 
35
  # doc_contexts = self.retrieve_contexts(query, k=3, filter={"folder": "doc/"})
36
  # code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"folder": "kadi_apy/"})