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

Move lazy imports to module scope

Browse source
This commit is contained in:
Chris McDonough 2026-06-22 13:48:09 -04:00
parent 97748c2fee
commit d160080ff8
1 changed files with 2 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
haiku_rag_slim/haiku/rag/client/rebuild.py
View file

@ -5,6 +5,7 @@ from collections.abc import AsyncGenerator
from datetime import datetime from datetime import datetime
from typing import TYPE_CHECKING from typing import TYPE_CHECKING
from docling_core.types.doc.document import DescriptionMetaField, PictureMeta
from lancedb.pydantic import LanceModel from lancedb.pydantic import LanceModel
from haiku.rag.client.documents import check_source_accessible from haiku.rag.client.documents import check_source_accessible
@ -12,6 +13,7 @@ from haiku.rag.converters import get_converter
from haiku.rag.store.engine import ChunkRecordBase from haiku.rag.store.engine import ChunkRecordBase
from haiku.rag.store.models.chunk import Chunk from haiku.rag.store.models.chunk import Chunk
from haiku.rag.store.models.document import Document from haiku.rag.store.models.document import Document
from haiku.rag.store.compression import compress_docling_split
from haiku.rag.store.models.document_item import extract_items from haiku.rag.store.models.document_item import extract_items
from haiku.rag.store.repositories.settings import SettingsRepository from haiku.rag.store.repositories.settings import SettingsRepository
@ -629,10 +631,6 @@ def _apply_descriptions_sync(docling_doc, doc, descriptions):
That would silently destroy page rasters for every doc with at That would silently destroy page rasters for every doc with at
least one undescribed picture. least one undescribed picture.
""" """
from docling_core.types.doc.document import DescriptionMetaField, PictureMeta
from haiku.rag.store.compression import compress_docling_split
for pic in docling_doc.pictures: for pic in docling_doc.pictures:
text = descriptions.get(pic.self_ref) text = descriptions.get(pic.self_ref)
if not text: if not text: