jim/morphik-core
1
0
Fork 0
mirror of https://github.com/james-m-jordan/morphik-core.git synced 2025-05-09 19:32:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

colpali hotfix

Browse Source
This commit is contained in:
Arnav Agrawal 2025-03-16 17:08:38 -04:00
parent 32a5d787fe
commit 989d25d8c7
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
core/services/document_service.py
View File

@ -104,12 +104,12 @@ class DocumentService:
query_embedding_regular, k=10 * k if should_rerank else k, doc_ids=doc_ids 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 = ( chunks_multivector = (
await self.colpali_vector_store.query_similar( await self.colpali_vector_store.query_similar(
query_embedding_multivector, k=k, doc_ids=doc_ids query_embedding_multivector, k=k, doc_ids=doc_ids
) ) if search_multi else []
if (use_colpali and self.colpali_vector_store and query_embedding_multivector)
else []
) )
logger.info(f"Found {len(chunks)} similar chunks via regular embedding") logger.info(f"Found {len(chunks)} similar chunks via regular embedding")