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

Tighten _merge_picture_chunks type annotations

Browse source 
Restore the typing the merge logic carried inline before extraction:
text_chunks: list[Chunk], existing_picture_data: dict[str, bytes] | None,
-> list[Chunk], and first_pos(c: Chunk) -> int.

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
This commit is contained in:
Yiorgis Gozadinos 2026-06-23 08:56:51 +03:00
parent c59f88bcd6
commit 141e288b83
No known key found for this signature in database
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
haiku_rag_slim/haiku/rag/client/processing.py
View file

@ -169,11 +169,11 @@ async def convert(
def _merge_picture_chunks( def _merge_picture_chunks(
docling_document, docling_document: "DoclingDocument",
text_chunks: list, text_chunks: list[Chunk],
document_id: str | None, document_id: str | None,
existing_picture_data: dict | None, existing_picture_data: dict[str, bytes] | None,
) -> list: ) -> list[Chunk]:
picture_chunks = build_picture_chunks( picture_chunks = build_picture_chunks(
docling_document, docling_document,
document_id=document_id, document_id=document_id,
@ -190,7 +190,7 @@ def _merge_picture_chunks(
for pos, (item, _level) in enumerate(docling_document.iterate_items()) for pos, (item, _level) in enumerate(docling_document.iterate_items())
} }
def first_pos(c): def first_pos(c: Chunk) -> int:
refs = (c.metadata or {}).get("doc_item_refs") or [] refs = (c.metadata or {}).get("doc_item_refs") or []
return positions.get(refs[0], len(positions)) if refs else len(positions) return positions.get(refs[0], len(positions)) if refs else len(positions)