From 989d25d8c7e6aba2481d307ef12a928efd46d50e Mon Sep 17 00:00:00 2001 From: Arnav Agrawal Date: Sun, 16 Mar 2025 17:08:38 -0400 Subject: [PATCH] colpali hotfix --- core/services/document_service.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/core/services/document_service.py b/core/services/document_service.py index 17eea8d..8eda1e3 100644 --- a/core/services/document_service.py +++ b/core/services/document_service.py @@ -104,12 +104,12 @@ class DocumentService: query_embedding_regular, k=10 * k if should_rerank else k, doc_ids=doc_ids ) + search_multi = use_colpali and self.colpali_vector_store and query_embedding_multivector is not None + chunks_multivector = ( await self.colpali_vector_store.query_similar( query_embedding_multivector, k=k, doc_ids=doc_ids - ) - if (use_colpali and self.colpali_vector_store and query_embedding_multivector) - else [] + ) if search_multi else [] ) logger.info(f"Found {len(chunks)} similar chunks via regular embedding")