Spaces:
Sleeping
Sleeping
Update kadiApy_ragchain.py
Browse files- kadiApy_ragchain.py +1 -1
kadiApy_ragchain.py
CHANGED
@@ -25,7 +25,7 @@ class KadiApyRagchain:
|
|
25 |
print("Start retrieving:")
|
26 |
|
27 |
|
28 |
-
code_contexts = self.retrieve_contexts(rewritten_query, k=3, filter={"usage": code_library_usage_prediction})
|
29 |
code_contexts = self.retrieve_contexts(rewritten_query, k=3, filter={"dataset_category": "kadi_apy_source_code"})
|
30 |
doc_contexts = self.retrieve_contexts(query, k=2, filter={"dataset_category": "kadi_apy_docs"})
|
31 |
|
|
|
25 |
print("Start retrieving:")
|
26 |
|
27 |
|
28 |
+
#code_contexts = self.retrieve_contexts(rewritten_query, k=3, filter={"usage": code_library_usage_prediction})
|
29 |
code_contexts = self.retrieve_contexts(rewritten_query, k=3, filter={"dataset_category": "kadi_apy_source_code"})
|
30 |
doc_contexts = self.retrieve_contexts(query, k=2, filter={"dataset_category": "kadi_apy_docs"})
|
31 |
|