Spaces:
Sleeping
Sleeping
Update kadiApy_ragchain.py
Browse files- kadiApy_ragchain.py +5 -5
kadiApy_ragchain.py
CHANGED
@@ -23,16 +23,16 @@ class KadiApyRagchain:
|
|
23 |
code_library_usage_prediction = self.predict_library_usage(query)
|
24 |
|
25 |
# Retrieve contexts
|
26 |
-
doc_contexts = self.retrieve_contexts(query, k=3, filter={"
|
27 |
code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"usage": code_library_usage_prediction})
|
28 |
|
29 |
|
30 |
# Vanilla
|
31 |
-
#doc_contexts = self.retrieve_contexts(query, k=3, filter={"
|
32 |
-
#code_contexts = self.retrieve_contexts(query, k=5, filter={"
|
33 |
|
34 |
-
# doc_contexts = self.retrieve_contexts(query, k=3, filter={"
|
35 |
-
# code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"
|
36 |
|
37 |
|
38 |
# Format contexts
|
|
|
23 |
code_library_usage_prediction = self.predict_library_usage(query)
|
24 |
|
25 |
# Retrieve contexts
|
26 |
+
doc_contexts = self.retrieve_contexts(query, k=3, filter={"dataset_category": "kadi_apy_docs"})
|
27 |
code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"usage": code_library_usage_prediction})
|
28 |
|
29 |
|
30 |
# Vanilla
|
31 |
+
#doc_contexts = self.retrieve_contexts(query, k=3, filter={"dataset_category": "kadi_apy_docs"})
|
32 |
+
#code_contexts = self.retrieve_contexts(query, k=5, filter={"dataset_category": "kadi_apy_source_code"})
|
33 |
|
34 |
+
# doc_contexts = self.retrieve_contexts(query, k=3, filter={"dataset_category": "kadi_apy_docs"})
|
35 |
+
# code_contexts = self.retrieve_contexts(rewritten_query, k=5, filter={"dataset_category": "kadi_apy_source_code"})
|
36 |
|
37 |
|
38 |
# Format contexts
|