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

fix: migration order

This commit is contained in:
Rajat Ahuja 2026-01-13 11:13:53 -05:00
parent 774edc0b6c
commit b38a87088b
3 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
migrations/versions/f1a2b3c4d5e6_support_external_embeddings.py → migrations/versions/119a52b73c60_support_external_embeddings.py
View File

@ -8,8 +8,8 @@ This migration:
3. Adds sync_state, last_sync_at, and sync_attempts columns to documents and
message_embeddings tables for tracking vector store synchronization status.
Revision ID: f1a2b3c4d5e6
Revises: 110bdf470272
Revision ID: 119a52b73c60
Revises: 7c0d9a4e3b1f
Create Date: 2025-11-24 12:00:00.000000
"""
@ -23,8 +23,8 @@ from pgvector.sqlalchemy import Vector
from migrations.utils import column_exists, get_schema, index_exists
# revision identifiers, used by Alembic.
revision: str = "f1a2b3c4d5e6"
down_revision: str | None = "110bdf470272"
revision: str = "119a52b73c60"
down_revision: str | None = "7c0d9a4e3b1f"
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None

4
tests/alembic/revisions/__init__.py
View File

@ -10,6 +10,7 @@ from . import (
test_76ffba56fe8c_add_error_field_to_queueitem,
test_88b0fb10906f_add_webhooks_table,
test_110bdf470272_rename_deriver_disabled_to_deriver_,
test_119a52b73c60_support_external_embeddings,
test_556a16564f50_add_user_id_and_app_id_to_tables,
test_564ba40505c5_add_session_name_column_to_documents,
test_917195d9b5e9_add_messageembedding_table,
@ -22,7 +23,6 @@ from . import (
test_d429de0e5338_adopt_peer_paradigm,
test_e9b705f9adf9_add_server_defaults_to_timestamp_,
test_ec8f94139b02_codify_workspace_name_and_message_id_in_,
test_f1a2b3c4d5e6_support_external_embeddings,
)
__all__ = [
@ -47,5 +47,5 @@ __all__ = [
"test_d429de0e5338_adopt_peer_paradigm",
"test_e9b705f9adf9_add_server_defaults_to_timestamp_",
"test_ec8f94139b02_codify_workspace_name_and_message_id_in_",
"test_f1a2b3c4d5e6_support_external_embeddings",
"test_119a52b73c60_support_external_embeddings",
]

10
tests/alembic/revisions/test_f1a2b3c4d5e6_support_external_embeddings.py → tests/alembic/revisions/test_119a52b73c60_support_external_embeddings.py
View File

@ -1,4 +1,4 @@
"""Hooks for revision f1a2b3c4d5e6 (support_external_embeddings)."""
"""Hooks for revision 119a52b73c60 (support_external_embeddings)."""
from __future__ import annotations
@ -15,11 +15,11 @@ INDEXES = (
)
@register_before_upgrade("f1a2b3c4d5e6")
@register_before_upgrade("119a52b73c60")
def prepare_support_external_embeddings(
verifier: MigrationVerifier,
) -> None:
"""Seed state and assertions before upgrading to f1a2b3c4d5e6."""
"""Seed state and assertions before upgrading to 119a52b73c60."""
# Embedding columns should be NOT NULL before migration
verifier.assert_column_exists("message_embeddings", "embedding", nullable=False)
verifier.assert_column_exists("documents", "embedding", nullable=False)
@ -41,11 +41,11 @@ def prepare_support_external_embeddings(
verifier.assert_indexes_not_exist(INDEXES)
@register_after_upgrade("f1a2b3c4d5e6")
@register_after_upgrade("119a52b73c60")
def verify_support_external_embeddings(
verifier: MigrationVerifier,
) -> None:
"""Add assertions validating the effects of f1a2b3c4d5e6."""
"""Add assertions validating the effects of 119a52b73c60."""
# Embedding columns should now be nullable
verifier.assert_column_exists("message_embeddings", "embedding", nullable=True)
verifier.assert_column_exists("documents", "embedding", nullable=True)