48 lines
1.4 KiB
Python
48 lines
1.4 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.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 ###
|