chore: fix lint/typecheck

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

haiku_rag_slim/haiku/rag/embeddings/multimodal.py

@@ -1,5 +1,5 @@
-import base64
 import asyncio
+import base64
 from dataclasses import dataclass
 
 import httpx

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
 

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

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

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"),
+                    asset_id=str(asset_id),
-                    document_id=row.get("document_id"),
+                    document_id=str(document_id),
                     score=score,
                     doc_item_ref=row.get("doc_item_ref", ""),
                     item_index=row.get("item_index"),