From 141e288b834781d9fa9c5a3d71727174b290ad3f Mon Sep 17 00:00:00 2001 From: Yiorgis Gozadinos Date: Tue, 23 Jun 2026 08:56:51 +0300 Subject: [PATCH] Tighten _merge_picture_chunks type annotations 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) --- haiku_rag_slim/haiku/rag/client/processing.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/haiku_rag_slim/haiku/rag/client/processing.py b/haiku_rag_slim/haiku/rag/client/processing.py index 75fed8db..198f7736 100644 --- a/haiku_rag_slim/haiku/rag/client/processing.py +++ b/haiku_rag_slim/haiku/rag/client/processing.py @@ -169,11 +169,11 @@ async def convert( def _merge_picture_chunks( - docling_document, - text_chunks: list, + docling_document: "DoclingDocument", + text_chunks: list[Chunk], document_id: str | None, - existing_picture_data: dict | None, -) -> list: + existing_picture_data: dict[str, bytes] | None, +) -> list[Chunk]: picture_chunks = build_picture_chunks( docling_document, document_id=document_id, @@ -190,7 +190,7 @@ def _merge_picture_chunks( 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 [] return positions.get(refs[0], len(positions)) if refs else len(positions)