danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Check table_names() before dropping chunks in recreate_embeddings_table

Browse source
This commit is contained in:
Yiorgis Gozadinos 2026-04-24 15:32:06 +03:00
parent 729899299b
commit b23bad32c3
No known key found for this signature in database
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
haiku_rag_slim/haiku/rag/store/engine.py
View file

@ -586,11 +586,12 @@ class Store:
ReadOnlyError: If the store is in read-only mode. ReadOnlyError: If the store is in read-only mode.
""" """
self._assert_writable() self._assert_writable()
# Drop and recreate chunks table # Drop and recreate chunks table. Check existence first rather than
try: # catching-and-swallowing drop_table's errors — a catch-all would
# hide real failures (permissions, storage-backend errors) and then
# the subsequent create_table would fail confusingly.
if "chunks" in await self.db.table_names():
await self.db.drop_table("chunks") await self.db.drop_table("chunks")
except Exception:
pass
# Update the ChunkRecord model with new vector dimension # Update the ChunkRecord model with new vector dimension
self.ChunkRecord = create_chunk_model(self.embedder._vector_dim) self.ChunkRecord = create_chunk_model(self.embedder._vector_dim)