RomanNum3ral
/
honcho
mirror of https://github.com/plastic-labs/honcho.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: down migration

This commit is contained in:
Rajat Ahuja 2026-01-12 12:02:39 -05:00
parent ed431e56d8
commit 42d4c01d47
1 changed files with 15 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
migrations/versions/f1a2b3c4d5e6_support_external_embeddings.py
View File

@ -182,13 +182,6 @@ def downgrade() -> None:
"""Remove deleted_at columns and revert embedding columns."""
inspector = sa.inspect(op.get_bind())
# Remove sync state columns from message_embeddings
if column_exists("message_embeddings", "sync_attempts", inspector):
op.drop_column("message_embeddings", "sync_attempts", schema=schema)
if column_exists("message_embeddings", "last_sync_at", inspector):
op.drop_column("message_embeddings", "last_sync_at", schema=schema)
if column_exists("message_embeddings", "sync_state", inspector):
# Drop composite index first
op.drop_index(
@ -201,14 +194,15 @@ def downgrade() -> None:
table_name="message_embeddings",
schema=schema,
)
if column_exists("message_embeddings", "sync_state", inspector):
op.drop_column("message_embeddings", "sync_state", schema=schema)
# Remove sync state columns from documents
if column_exists("documents", "sync_attempts", inspector):
op.drop_column("documents", "sync_attempts", schema=schema)
if column_exists("message_embeddings", "sync_attempts", inspector):
op.drop_column("message_embeddings", "sync_attempts", schema=schema)
if column_exists("documents", "last_sync_at", inspector):
op.drop_column("documents", "last_sync_at", schema=schema)
if column_exists("message_embeddings", "last_sync_at", inspector):
op.drop_column("message_embeddings", "last_sync_at", schema=schema)
if column_exists("documents", "sync_state", inspector):
# Drop composite index first
@ -218,8 +212,17 @@ def downgrade() -> None:
schema=schema,
)
op.drop_index("ix_documents_sync_state", table_name="documents", schema=schema)
if column_exists("documents", "sync_state", inspector):
op.drop_column("documents", "sync_state", schema=schema)
# Remove sync state columns from documents
if column_exists("documents", "sync_attempts", inspector):
op.drop_column("documents", "sync_attempts", schema=schema)
if column_exists("documents", "last_sync_at", inspector):
op.drop_column("documents", "last_sync_at", schema=schema)
# Remove deleted_at column and index from documents
if column_exists("documents", "deleted_at", inspector):
op.drop_index("ix_documents_deleted_at", table_name="documents", schema=schema)