25 lines
962 B
Python
25 lines
962 B
Python
"""Hooks for revision f1a2b3c4d5e6 (add_chunk_index_to_message_embeddings)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from tests.alembic.registry import register_after_upgrade, register_before_upgrade
|
|
from tests.alembic.verifier import MigrationVerifier
|
|
|
|
|
|
@register_before_upgrade("f1a2b3c4d5e6")
|
|
def prepare_add_chunk_index_to_message_embeddings(
|
|
verifier: MigrationVerifier,
|
|
) -> None:
|
|
"""Seed state and assertions before upgrading to f1a2b3c4d5e6."""
|
|
# Verify chunk_index column doesn't exist before migration
|
|
verifier.assert_column_exists("message_embeddings", "chunk_index", exists=False)
|
|
|
|
|
|
@register_after_upgrade("f1a2b3c4d5e6")
|
|
def verify_add_chunk_index_to_message_embeddings(
|
|
verifier: MigrationVerifier,
|
|
) -> None:
|
|
"""Add assertions validating the effects of f1a2b3c4d5e6."""
|
|
# Verify chunk_index column was added with correct properties
|
|
verifier.assert_column_exists("message_embeddings", "chunk_index", nullable=False)
|