Spaces:

Ekimetrics
/

climate-question-answering

Running

timeki commited on 15 days ago

Commit

2bb9320

1 Parent(s): e2f3eb3

fix retrieval

Files changed (1) hide show

climateqa/engine/chains/retrieve_documents.py CHANGED Viewed

@@ -232,8 +232,7 @@ async def get_IPCC_relevant_documents(
             "chunk_type":"text",
             "report_type": { "$nin":["SPM"]},
         }
-        k_full = k_total - len(docs_summaries)
-        docs_full = vectorstore.similarity_search_with_score(query=query,filter = filters_full,k = k_full)
         if search_figures:
             # Images

             "chunk_type":"text",
             "report_type": { "$nin":["SPM"]},
         }
+        docs_full = vectorstore.similarity_search_with_score(query=query,filter = filters_full,k = k_total)
         if search_figures:
             # Images