haiku.rag 0.11.0__tar.gz → 0.11.1__tar.gz

{haiku_rag-0.11.0 → haiku_rag-0.11.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: haiku.rag
-Version: 0.11.0
+Version: 0.11.1
 Summary: Agentic Retrieval Augmented Generation (RAG) with LanceDB
 Author-email: Yiorgis Gozadinos <ggozadinos@gmail.com>
 License: MIT

{haiku_rag-0.11.0 → haiku_rag-0.11.1}/pyproject.toml

@@ -2,7 +2,7 @@
 
 name = "haiku.rag"
 description = "Agentic Retrieval Augmented Generation (RAG) with LanceDB"
-version = "0.11.0"
+version = "0.11.1"
 authors = [{ name = "Yiorgis Gozadinos", email = "ggozadinos@gmail.com" }]
 license = { text = "MIT" }
 readme = { file = "README.md", content-type = "text/markdown" }

{haiku_rag-0.11.0 → haiku_rag-0.11.1}/src/haiku/rag/reader.py

@@ -10,6 +10,8 @@ from haiku.rag.utils import text_to_docling_document
 class FileReader:
     # Extensions supported by docling
     docling_extensions: ClassVar[list[str]] = [
+        ".adoc",
+        ".asc",
         ".asciidoc",
         ".bmp",
         ".csv",

{haiku_rag-0.11.0 → haiku_rag-0.11.1}/src/haiku/rag/store/engine.py

@@ -157,7 +157,8 @@ class Store:
             current_version = metadata.version("haiku.rag")
             db_version = self.get_haiku_version()
 
-            run_pending_upgrades(self, db_version, current_version)
+            if db_version != "0.0.0":
+                run_pending_upgrades(self, db_version, current_version)
 
             # After upgrades complete (or if none), set stored version
             # to the greater of the installed package version and the

{haiku_rag-0.11.0 → haiku_rag-0.11.1}/tests/test_versioning.py

@@ -92,3 +92,33 @@ async def test_version_rollback_on_update_failure(temp_db_path):
     chunks_repo = ChunkRepository(store)
     original_chunks = await chunks_repo.get_by_document_id(created.id)  # type: ignore[arg-type]
     assert len(original_chunks) > 0
+
+
+def test_new_database_does_not_run_upgrades(monkeypatch, temp_db_path):
+    def fail_if_called(*_args, **_kwargs):
+        raise AssertionError("run_pending_upgrades should not be called for new DB")
+
+    monkeypatch.setattr(
+        "haiku.rag.store.upgrades.run_pending_upgrades",
+        fail_if_called,
+    )
+
+    Store(temp_db_path)
+
+
+def test_existing_database_runs_upgrades(monkeypatch, temp_db_path):
+    Store(temp_db_path)
+
+    called = {"value": False}
+
+    def mark_called(*_args, **_kwargs):
+        called["value"] = True
+
+    monkeypatch.setattr(
+        "haiku.rag.store.upgrades.run_pending_upgrades",
+        mark_called,
+    )
+
+    Store(temp_db_path)
+
+    assert called["value"]

{haiku_rag-0.11.0 → haiku_rag-0.11.1}/uv.lock

@@ -1111,7 +1111,7 @@ wheels = [
 
 [[package]]
 name = "haiku-rag"
-version = "0.11.0"
+version = "0.11.1"
 source = { editable = "." }
 dependencies = [
     { name = "docling" },