From 170d6ea6f6bb2b4c6d64763a5ac874276074c690 Mon Sep 17 00:00:00 2001
From: Andriy Oblivantsev <eslider@gmail.com>
Date: Mon, 10 Aug 2026 21:59:25 +0100
Subject: [PATCH] fix(kblib): INSTALL FTS/VECTOR before LOAD for fresh runners

---
 tools/kblib.py | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/tools/kblib.py b/tools/kblib.py
index 45f3612..5bbdd92 100644
--- a/tools/kblib.py
+++ b/tools/kblib.py
@@ -30,11 +30,20 @@ def sha256_b64(text: str) -> str:
     return hashlib.sha256(text.encode()).hexdigest()
 
 
+def _load_extension(conn: ladybug.Connection, name: str) -> None:
+    """Install (download once) and load a ladybug extension."""
+    try:
+        conn.execute(f"INSTALL {name}")
+    except Exception:
+        pass  # already installed / offline-ok when present
+    conn.execute(f"LOAD EXTENSION {name}")
+
+
 def connect(path: Path | str | None = None, read_only: bool = True) -> tuple[ladybug.Database, ladybug.Connection]:
     db = ladybug.Database(str(path or DB_PATH), read_only=read_only)
     conn = ladybug.Connection(db)
-    conn.execute("LOAD EXTENSION FTS")
-    conn.execute("LOAD EXTENSION VECTOR")
+    _load_extension(conn, "FTS")
+    _load_extension(conn, "VECTOR")
     return db, conn