From d295fbd2b18e16e9eba5af62075817c1fa07a7c8 Mon Sep 17 00:00:00 2001
From: jvdrag <joca.dragone@gmail.com>
Date: Sun, 11 Jan 2026 07:47:44 -0300
Subject: [PATCH] chore: fix lint/typecheck

---
 haiku_rag_slim/haiku/rag/app.py                        |  2 --
 haiku_rag_slim/haiku/rag/embeddings/multimodal.py      |  2 +-
 .../haiku/rag/inspector/widgets/doc_index_modal.py     |  2 +-
 .../haiku/rag/inspector/widgets/mm_assets_modal.py     |  2 +-
 .../haiku/rag/inspector/widgets/visual_modal.py        | 10 +++++-----
 .../haiku/rag/store/repositories/mm_asset.py           |  8 ++++++--
 6 files changed, 14 insertions(+), 12 deletions(-)

diff --git a/haiku_rag_slim/haiku/rag/app.py b/haiku_rag_slim/haiku/rag/app.py
index cae1fff7..3a4f6878 100644
--- a/haiku_rag_slim/haiku/rag/app.py
+++ b/haiku_rag_slim/haiku/rag/app.py
@@ -405,7 +405,6 @@ class HaikuRAGApp:
 
     async def visualize_chunk(self, chunk_id: str):
         """Display visual grounding images for a chunk."""
-        import os
         import subprocess
         import sys
         import tempfile
@@ -475,7 +474,6 @@ class HaikuRAGApp:
         - crop: show the cropped image region (bbox + padding)
         - page: show the full page with bbox overlay (if available)
         """
-        import os
         import subprocess
         import sys
         import tempfile
diff --git a/haiku_rag_slim/haiku/rag/embeddings/multimodal.py b/haiku_rag_slim/haiku/rag/embeddings/multimodal.py
index d113fd5f..3ea8b26d 100644
--- a/haiku_rag_slim/haiku/rag/embeddings/multimodal.py
+++ b/haiku_rag_slim/haiku/rag/embeddings/multimodal.py
@@ -1,5 +1,5 @@
-import base64
 import asyncio
+import base64
 from dataclasses import dataclass
 
 import httpx
diff --git a/haiku_rag_slim/haiku/rag/inspector/widgets/doc_index_modal.py b/haiku_rag_slim/haiku/rag/inspector/widgets/doc_index_modal.py
index c1c987bb..fb91d559 100644
--- a/haiku_rag_slim/haiku/rag/inspector/widgets/doc_index_modal.py
+++ b/haiku_rag_slim/haiku/rag/inspector/widgets/doc_index_modal.py
@@ -1,9 +1,9 @@
 from __future__ import annotations
 
+import shutil
 import subprocess
 import sys
 import tempfile
-import shutil
 from copy import deepcopy
 from pathlib import Path
 
diff --git a/haiku_rag_slim/haiku/rag/inspector/widgets/mm_assets_modal.py b/haiku_rag_slim/haiku/rag/inspector/widgets/mm_assets_modal.py
index b7f5ead8..b7ac84f7 100644
--- a/haiku_rag_slim/haiku/rag/inspector/widgets/mm_assets_modal.py
+++ b/haiku_rag_slim/haiku/rag/inspector/widgets/mm_assets_modal.py
@@ -1,9 +1,9 @@
 from __future__ import annotations
 
+import shutil
 import subprocess
 import sys
 import tempfile
-import shutil
 from pathlib import Path
 
 from textual.app import ComposeResult
diff --git a/haiku_rag_slim/haiku/rag/inspector/widgets/visual_modal.py b/haiku_rag_slim/haiku/rag/inspector/widgets/visual_modal.py
index 377fbd83..99dbb359 100644
--- a/haiku_rag_slim/haiku/rag/inspector/widgets/visual_modal.py
+++ b/haiku_rag_slim/haiku/rag/inspector/widgets/visual_modal.py
@@ -1,3 +1,8 @@
+import shutil
+import subprocess
+import sys
+import tempfile
+from pathlib import Path
 from typing import TYPE_CHECKING
 
 from textual.app import ComposeResult
@@ -7,11 +12,6 @@ from textual.screen import Screen
 from textual.widget import Widget
 from textual.widgets import Static
 from textual_image.widget import Image as TextualImage
-import shutil
-import tempfile
-from pathlib import Path
-import subprocess
-import sys
 
 if TYPE_CHECKING:
     from PIL.Image import Image as PILImage
diff --git a/haiku_rag_slim/haiku/rag/store/repositories/mm_asset.py b/haiku_rag_slim/haiku/rag/store/repositories/mm_asset.py
index 2a176b11..d3dfb263 100644
--- a/haiku_rag_slim/haiku/rag/store/repositories/mm_asset.py
+++ b/haiku_rag_slim/haiku/rag/store/repositories/mm_asset.py
@@ -105,6 +105,10 @@ class MMAssetRepository:
         rows = q.to_list()
         results: list[MMSearchResult] = []
         for row in rows:
+            asset_id = row.get("id")
+            document_id = row.get("document_id")
+            if not asset_id or not document_id:
+                continue
             distance = float(row.get("_distance", 0.0))
             score = 1.0 / (1.0 + distance)
             bbox = None
@@ -114,8 +118,8 @@ class MMAssetRepository:
                 bbox = None
             results.append(
                 MMSearchResult(
-                    asset_id=row.get("id"),
-                    document_id=row.get("document_id"),
+                    asset_id=str(asset_id),
+                    document_id=str(document_id),
                     score=score,
                     doc_item_ref=row.get("doc_item_ref", ""),
                     item_index=row.get("item_index"),