"""resize_vector

Revision ID: d3c63f669b0a
Revises: 380f4e61b41c
Create Date: 2026-02-18 22:10:28.693146

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa
import pgvector.sqlalchemy


# revision identifiers, used by Alembic.
revision: str = 'd3c63f669b0a'
down_revision: Union[str, Sequence[str], None] = 'add_users_table'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    """Upgrade schema."""
    bind = op.get_bind()
    is_sqlite = bind.dialect.name == 'sqlite'

    # ### commands auto generated by Alembic - please adjust! ###
    if not is_sqlite:
        op.alter_column('context_nodes', 'embedding',
               existing_type=pgvector.sqlalchemy.vector.VECTOR(dim=1536),
               type_=pgvector.sqlalchemy.vector.VECTOR(dim=768),
               existing_nullable=True)
    # ### end Alembic commands ###


def downgrade() -> None:
    """Downgrade schema."""
    bind = op.get_bind()
    is_sqlite = bind.dialect.name == 'sqlite'
    
    # ### commands auto generated by Alembic - please adjust! ###
    if not is_sqlite:
        op.alter_column('context_nodes', 'embedding',
               existing_type=pgvector.sqlalchemy.vector.VECTOR(dim=768),
               type_=pgvector.sqlalchemy.vector.VECTOR(dim=1536),
               existing_nullable=True)
    # ### end Alembic commands ###