TheoLvs commited on
Commit
9a9100e
1 Parent(s): e1c50bf

Update retriever.py

Browse files
Files changed (1) hide show
  1. climateqa/engine/retriever.py +2 -2
climateqa/engine/retriever.py CHANGED
@@ -39,7 +39,7 @@ class ClimateQARetriever(BaseRetriever):
39
  # Search for k_summary documents in the summaries dataset
40
  filters_summaries = {
41
  **filters,
42
- "report_type": { "$in":["SPM","TS"]},
43
  }
44
 
45
  docs_summaries = self.vectorstore.similarity_search_with_score(query=query,filter = filters_summaries,k = self.k_summary)
@@ -48,7 +48,7 @@ class ClimateQARetriever(BaseRetriever):
48
  # Search for k_total - k_summary documents in the full reports dataset
49
  filters_full = {
50
  **filters,
51
- "report_type": { "$nin":["SPM","TS"]},
52
  }
53
  k_full = self.k_total - len(docs_summaries)
54
  docs_full = self.vectorstore.similarity_search_with_score(query=query,filter = filters_full,k = k_full)
 
39
  # Search for k_summary documents in the summaries dataset
40
  filters_summaries = {
41
  **filters,
42
+ "report_type": { "$in":["SPM"]},
43
  }
44
 
45
  docs_summaries = self.vectorstore.similarity_search_with_score(query=query,filter = filters_summaries,k = self.k_summary)
 
48
  # Search for k_total - k_summary documents in the full reports dataset
49
  filters_full = {
50
  **filters,
51
+ "report_type": { "$nin":["SPM"]},
52
  }
53
  k_full = self.k_total - len(docs_summaries)
54
  docs_full = self.vectorstore.similarity_search_with_score(query=query,filter = filters_full,k = k_full)