Spaces:
Runtime error
Runtime error
LOUIS SANNA
commited on
Commit
·
c6c35dc
1
Parent(s):
d98ba57
feat(retriever)
Browse files- climateqa/retriever.py +2 -2
climateqa/retriever.py
CHANGED
@@ -39,7 +39,7 @@ class QARetriever(BaseRetriever):
|
|
39 |
}
|
40 |
docs_summaries = self.vectorstore.similarity_search_with_score(
|
41 |
query=query,
|
42 |
-
|
43 |
filter=self.format_filter(filters_summaries),
|
44 |
k=self.k_summary,
|
45 |
)
|
@@ -55,7 +55,7 @@ class QARetriever(BaseRetriever):
|
|
55 |
k_full = self.k_total - len(docs_summaries)
|
56 |
docs_full = self.vectorstore.similarity_search_with_score(
|
57 |
query=query,
|
58 |
-
|
59 |
filter=self.format_filter(filters_full),
|
60 |
k=k_full,
|
61 |
)
|
|
|
39 |
}
|
40 |
docs_summaries = self.vectorstore.similarity_search_with_score(
|
41 |
query=query,
|
42 |
+
namespace=self.namespace,
|
43 |
filter=self.format_filter(filters_summaries),
|
44 |
k=self.k_summary,
|
45 |
)
|
|
|
55 |
k_full = self.k_total - len(docs_summaries)
|
56 |
docs_full = self.vectorstore.similarity_search_with_score(
|
57 |
query=query,
|
58 |
+
namespace=self.namespace,
|
59 |
filter=self.format_filter(filters_full),
|
60 |
k=k_full,
|
61 |
)
|