Spaces:
Sleeping
Sleeping
Update kadi_apy_bot.py
Browse files- 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)
|