PMFlow/backend/alembic/versions/d3c63f669b0a_resize_vector.py

40 lines
1.2 KiB
Python

"""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
# revision identifiers, used by Alembic.
revision: str = 'd3c63f669b0a'
down_revision: Union[str, Sequence[str], None] = '380f4e61b41c'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
"""Upgrade schema."""
# ### commands auto generated by Alembic - please adjust! ###
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."""
# ### commands auto generated by Alembic - please adjust! ###
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 ###