diff --git a/haiku_rag_slim/haiku/rag/store/upgrades/v0_50_0.py b/haiku_rag_slim/haiku/rag/store/upgrades/v0_50_0.py index 36096e3f..f87a2174 100644 --- a/haiku_rag_slim/haiku/rag/store/upgrades/v0_50_0.py +++ b/haiku_rag_slim/haiku/rag/store/upgrades/v0_50_0.py @@ -1,5 +1,6 @@ import json import logging +from datetime import timedelta from haiku.rag.store.engine import Store from haiku.rag.store.upgrades import Upgrade @@ -95,6 +96,11 @@ async def _apply_canonical_metadata_keys(store: Store) -> None: total, ) + if rewritten: + # Compact the documents table to reclaim space from the rewritten rows. + logger.info("Compacting documents table to reclaim %d row rewrites", rewritten) + await store.documents_table.optimize(cleanup_older_than=timedelta(seconds=0)) + upgrade_canonical_metadata_keys = Upgrade( version="0.50.0",