danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Cover the detached-chunk guard in the multimodal rerank search test

Browse source
This commit is contained in:
Yiorgis Gozadinos 2026-07-24 13:07:39 +03:00
parent f31060e741
commit 35b8b046ce
No known key found for this signature in database
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/test_search.py
View file

@ -456,9 +456,13 @@ async def test_search_attaches_picture_bytes_for_multimodal_reranker(
document_id="doc-1",
metadata={"doc_item_refs": ["#/pictures/0"], "labels": ["picture"]},
)
detached_chunk = Chunk(
content="no parent document",
metadata={"doc_item_refs": ["#/pictures/1"], "labels": ["picture"]},
)
async def fake_chunk_search(query, limit, search_type, filter):
return [(text_chunk, 0.9), (picture_chunk, 0.8)]
return [(text_chunk, 0.9), (picture_chunk, 0.8), (detached_chunk, 0.7)]
async with HaikuRAG(temp_db_path, create=True) as rag:
await rag.document_item_repository.create_items(
@ -480,8 +484,9 @@ async def test_search_attaches_picture_bytes_for_multimodal_reranker(
await rag.search("totals", include_images=False)
reranked_text, reranked_picture = captured["chunks"]
reranked_text, reranked_picture, reranked_detached = captured["chunks"]
assert reranked_text._picture_data is None
assert reranked_detached._picture_data is None
if multimodal:
assert reranked_picture._picture_data == b"picture-bytes"
else: