From 4d8088efa8fb4024f47c168f196d31a1d8892c90 Mon Sep 17 00:00:00 2001
From: Yiorgis Gozadinos <ggozadinos@gmail.com>
Date: Wed, 16 Jul 2025 18:32:58 +0300
Subject: [PATCH] Log db upgrades

---
 src/haiku/rag/store/engine.py          | 5 +++++
 src/haiku/rag/store/upgrades/v0_3_4.py | 2 +-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/haiku/rag/store/engine.py b/src/haiku/rag/store/engine.py
index 72f17402..7e746bc7 100644
--- a/src/haiku/rag/store/engine.py
+++ b/src/haiku/rag/store/engine.py
@@ -6,6 +6,7 @@ from typing import Literal
 
 import sqlite_vec
 from packaging.version import parse
+from rich.console import Console
 
 from haiku.rag.config import Config
 from haiku.rag.embeddings import get_embedder
@@ -47,6 +48,7 @@ class Store:
         # If we have a db already, perform upgrades and return
         if self.db_path != ":memory:" and "documents" in existing_tables:
             # Upgrade database
+            console = Console()
             db_version = self.get_user_version()
             for version, steps in upgrades:
                 if parse(current_version) >= parse(version) and parse(version) > parse(
@@ -54,6 +56,9 @@ class Store:
                 ):
                     for step in steps:
                         step(db)
+                        console.print(
+                            f"[green][b]DB Upgrade: [/b]{step.__doc__}[/green]"
+                        )
             return
 
         # Create documents table
diff --git a/src/haiku/rag/store/upgrades/v0_3_4.py b/src/haiku/rag/store/upgrades/v0_3_4.py
index 9ba912a1..56e73f00 100644
--- a/src/haiku/rag/store/upgrades/v0_3_4.py
+++ b/src/haiku/rag/store/upgrades/v0_3_4.py
@@ -5,7 +5,7 @@ from haiku.rag.config import Config
 
 
 def add_settings_table(db: Connection) -> None:
-    # Create settings table for storing current configuration
+    """Create settings table for storing current configuration"""
     db.execute("""
         CREATE TABLE settings (
             id INTEGER PRIMARY KEY DEFAULT 1,