When checking for updates, use haiku.rag-slim as the package, since it is always installed

haiku_rag_slim/haiku/rag/app.py

@@ -4,10 +4,6 @@ import logging
 from importlib.metadata import version as pkg_version
 from pathlib import Path
 
-from rich.console import Console
-from rich.markdown import Markdown
-from rich.progress import Progress
-
 from haiku.rag.client import HaikuRAG
 from haiku.rag.config import Config
 from haiku.rag.mcp import create_mcp_server
@@ -22,6 +18,9 @@ from haiku.rag.research.graph import (
 from haiku.rag.research.stream import stream_research_graph
 from haiku.rag.store.models.chunk import Chunk
 from haiku.rag.store.models.document import Document
+from rich.console import Console
+from rich.markdown import Markdown
+from rich.progress import Progress
 
 logger = logging.getLogger(__name__)
 
@@ -59,7 +58,7 @@ class HaikuRAGApp:
         except Exception:
             ldb_version = "unknown"
         try:
-            hr_version = pkg_version("haiku.rag")
+            hr_version = pkg_version("haiku.rag-slim")
         except Exception:
             hr_version = "unknown"
         try:
@@ -211,12 +210,11 @@ class HaikuRAGApp:
         async with HaikuRAG(db_path=self.db_path) as self.client:
             try:
                 if deep:
-                    from rich.console import Console
-
                     from haiku.rag.qa.deep.dependencies import DeepQAContext
                     from haiku.rag.qa.deep.graph import build_deep_qa_graph
                     from haiku.rag.qa.deep.nodes import DeepQAPlanNode
                     from haiku.rag.qa.deep.state import DeepQADeps, DeepQAState
+                    from rich.console import Console
 
                     graph = build_deep_qa_graph()
                     context = DeepQAContext(

haiku_rag_slim/haiku/rag/cli.py

@@ -36,7 +36,7 @@ async def check_version():
 
 def version_callback(value: bool):
     if value:
-        v = version("haiku.rag")
+        v = version("haiku.rag-slim")
         typer.echo(f"haiku.rag version {v}")
         raise typer.Exit()
 
@@ -353,7 +353,6 @@ def init_config(
 ):
     """Generate a YAML configuration file with defaults or from .env."""
     import yaml
-
     from haiku.rag.config.loader import generate_default_config, load_config_from_env
 
     if output.exists():

haiku_rag_slim/haiku/rag/store/engine.py

@@ -7,11 +7,10 @@ from pathlib import Path
 from uuid import uuid4
 
 import lancedb
-from lancedb.pydantic import LanceModel, Vector
-from pydantic import Field
-
 from haiku.rag.config import AppConfig, Config
 from haiku.rag.embeddings import get_embedder
+from lancedb.pydantic import LanceModel, Vector
+from pydantic import Field
 
 logger = logging.getLogger(__name__)
 
@@ -189,7 +188,7 @@ class Store:
         try:
             from haiku.rag.store.upgrades import run_pending_upgrades
 
-            current_version = metadata.version("haiku.rag")
+            current_version = metadata.version("haiku.rag-slim")
             db_version = self.get_haiku_version()
 
             if db_version != "0.0.0":
@@ -199,9 +198,8 @@ class Store:
             # to the greater of the installed package version and the
             # highest available upgrade step version in code.
             try:
-                from packaging.version import parse as _v
-
                 from haiku.rag.store.upgrades import upgrades as _steps
+                from packaging.version import parse as _v
 
                 highest_step = max((_v(u.version) for u in _steps), default=None)
                 effective_version = (
@@ -218,7 +216,7 @@ class Store:
             logger.warning(
                 "Skipping upgrade due to error (db=%s -> pkg=%s): %s",
                 self.get_haiku_version(),
-                metadata.version("haiku.rag") if hasattr(metadata, "version") else "",
+                metadata.version("haiku.rag-slim"),
                 e,
             )
 

haiku_rag_slim/haiku/rag/utils.py

@@ -82,7 +82,7 @@ async def is_up_to_date() -> tuple[bool, Version, Version]:
     import httpx
 
     async with httpx.AsyncClient() as client:
-        running_version = parse(metadata.version("haiku.rag"))
+        running_version = parse(metadata.version("haiku.rag-slim"))
         try:
             response = await client.get("https://pypi.org/pypi/haiku.rag/json")
             data = response.json()