LOUIS SANNA commited on
Commit
c6c35dc
1 Parent(s): d98ba57

feat(retriever)

Browse files
Files changed (1) hide show
  1. 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
- # namespace=self.namespace,
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
- # namespace=self.namespace,
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
  )