Spaces:
Runtime error
Runtime error
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=
|
| 29 |
doc_contexts = self.retrieve_contexts(query, k=2, filter={"dataset_category": "kadi_apy_docs"})
|
| 30 |
|
| 31 |
# Format contexts
|
|
|
|
| 25 |
print("Start retrieving:")
|
| 26 |
|
| 27 |
|
| 28 |
+
code_contexts = self.retrieve_contexts(rewritten_query, k=3, filter={"usage": code_library_usage_prediction})
|
| 29 |
doc_contexts = self.retrieve_contexts(query, k=2, filter={"dataset_category": "kadi_apy_docs"})
|
| 30 |
|
| 31 |
# Format contexts
|