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

chore: suppress deprecation wwarning from 'lancedb'

Browse source
This commit is contained in:
Tres Seaver 2026-03-12 17:11:58 -04:00
parent 7696b3f98f
commit 53481a7d7a
No known key found for this signature in database
GPG key ID: F2A968348913D1D8
3 changed files with 9 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
haiku_rag_slim/haiku/rag/inspector/widgets/info_modal.py
View file

@ -79,7 +79,7 @@ class InfoModal(ModalScreen):
# Connect to get table info # Connect to get table info
try: try:
db = lancedb.connect(self.db_path) db = lancedb.connect(self.db_path)
table_names = set(db.table_names()) table_names = set(db.list_tables().tables)
except Exception as e: except Exception as e:
lines.append(f"[red]Failed to open database: {e}[/red]") lines.append(f"[red]Failed to open database: {e}[/red]")
self._content_widget.update("\n".join(lines)) self._content_widget.update("\n".join(lines))

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

@ -153,7 +153,7 @@ class Store:
The stored vector dimension, or None if not found. The stored vector dimension, or None if not found.
""" """
try: try:
existing_tables = self.db.table_names() existing_tables = self.db.list_tables().tables
if "settings" not in existing_tables: if "settings" not in existing_tables:
return None return None
@ -333,7 +333,7 @@ class Store:
def _init_tables(self): def _init_tables(self):
"""Initialize database tables (create if they don't exist).""" """Initialize database tables (create if they don't exist)."""
existing_tables = self.db.table_names() existing_tables = self.db.list_tables().tables
required_tables = {"documents", "chunks", "settings"} required_tables = {"documents", "chunks", "settings"}
missing_tables = required_tables - set(existing_tables) missing_tables = required_tables - set(existing_tables)

12
haiku_rag_slim/haiku/rag/store/upgrades/v0_25_0.py
View file

@ -88,7 +88,7 @@ def _apply_compress_docling_document(store: Store) -> None: # pragma: no cover
if not ids: if not ids:
# Check if there's a staging table from a failed migration to recover from # Check if there's a staging table from a failed migration to recover from
if "documents_v4_staging" in store.db.table_names(): if "documents_v4_staging" in store.db.list_tables().tables:
staging_table = store.db.open_table("documents_v4_staging") staging_table = store.db.open_table("documents_v4_staging")
staging_ids = [ staging_ids = [
row["id"] row["id"]
@ -100,7 +100,7 @@ def _apply_compress_docling_document(store: Store) -> None: # pragma: no cover
) )
# Create new documents table and copy from staging # Create new documents table and copy from staging
store.documents_table = None store.documents_table = None
if "documents" in store.db.table_names(): if "documents" in store.db.list_tables().tables:
store.db.drop_table("documents") store.db.drop_table("documents")
store.documents_table = store.db.create_table( store.documents_table = store.db.create_table(
"documents", schema=get_documents_arrow_schema_v4() "documents", schema=get_documents_arrow_schema_v4()
@ -142,7 +142,7 @@ def _apply_compress_docling_document(store: Store) -> None: # pragma: no cover
# No documents and no staging to recover, just recreate table with new schema # No documents and no staging to recover, just recreate table with new schema
store.documents_table = None store.documents_table = None
if "documents" in store.db.table_names(): if "documents" in store.db.list_tables().tables:
store.db.drop_table("documents") store.db.drop_table("documents")
store.documents_table = store.db.create_table( store.documents_table = store.db.create_table(
"documents", schema=get_documents_arrow_schema_v4() "documents", schema=get_documents_arrow_schema_v4()
@ -150,7 +150,7 @@ def _apply_compress_docling_document(store: Store) -> None: # pragma: no cover
return return
# Create staging table with new schema # Create staging table with new schema
if "documents_v4_staging" in store.db.table_names(): if "documents_v4_staging" in store.db.list_tables().tables:
store.db.drop_table("documents_v4_staging") store.db.drop_table("documents_v4_staging")
staging_table = store.db.create_table( staging_table = store.db.create_table(
"documents_v4_staging", schema=get_documents_arrow_schema_v4() "documents_v4_staging", schema=get_documents_arrow_schema_v4()
@ -185,7 +185,7 @@ def _apply_compress_docling_document(store: Store) -> None: # pragma: no cover
# Replace old table with staging table # Replace old table with staging table
store.documents_table = None store.documents_table = None
if "documents" in store.db.table_names(): if "documents" in store.db.list_tables().tables:
store.db.drop_table("documents") store.db.drop_table("documents")
store.documents_table = store.db.create_table( store.documents_table = store.db.create_table(
"documents", schema=get_documents_arrow_schema_v4() "documents", schema=get_documents_arrow_schema_v4()
@ -225,7 +225,7 @@ def _apply_compress_docling_document(store: Store) -> None: # pragma: no cover
logger.info("Copied batch %d/%d", batch_num, total_batches) logger.info("Copied batch %d/%d", batch_num, total_batches)
# Cleanup staging table # Cleanup staging table
if "documents_v4_staging" in store.db.table_names(): if "documents_v4_staging" in store.db.list_tables().tables:
store.db.drop_table("documents_v4_staging") store.db.drop_table("documents_v4_staging")
# Vacuum all tables (destructive migration, no history preserved) # Vacuum all tables (destructive migration, no history preserved)