[add] migration script

This commit is contained in:
Mark
2026-04-28 13:41:46 +08:00
parent a5670bfff6
commit d85a1cb131
2 changed files with 76 additions and 0 deletions

View File

@@ -0,0 +1,42 @@
"""202604271530
Revision ID: 1f85dce125e5
Revises: 4e89970f9e7c
Create Date: 2026-04-27 15:30:35.614679
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql
# revision identifiers, used by Alembic.
revision: str = '1f85dce125e5'
down_revision: Union[str, None] = '4e89970f9e7c'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.add_column('files', sa.Column('file_key', sa.String(length=512), nullable=True, comment='storage file key for FileStorageService'))
op.create_index(op.f('ix_files_file_key'), 'files', ['file_key'], unique=False)
op.alter_column('model_configs', 'capability',
existing_type=postgresql.ARRAY(sa.VARCHAR()),
comment="模型能力列表(如['vision', 'audio', 'video', 'thinking']",
existing_comment="模型能力列表(如['vision', 'audio', 'video']",
existing_nullable=False)
# ### end Alembic commands ###
def downgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.alter_column('model_configs', 'capability',
existing_type=postgresql.ARRAY(sa.VARCHAR()),
comment="模型能力列表(如['vision', 'audio', 'video']",
existing_comment="模型能力列表(如['vision', 'audio', 'video', 'thinking']",
existing_nullable=False)
op.drop_index(op.f('ix_files_file_key'), table_name='files')
op.drop_column('files', 'file_key')
# ### end Alembic commands ###

View File

@@ -0,0 +1,34 @@
"""202604281230
Revision ID: e2d60c6d1a1a
Revises: 1f85dce125e5
Create Date: 2026-04-28 12:32:01.643954
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql
# revision identifiers, used by Alembic.
revision: str = 'e2d60c6d1a1a'
down_revision: Union[str, None] = '1f85dce125e5'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.drop_column('tenants', 'api_ops_rate_limit')
op.drop_column('tenants', 'plan')
op.drop_column('tenants', 'plan_expired_at')
# ### end Alembic commands ###
def downgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.add_column('tenants', sa.Column('plan_expired_at', postgresql.TIMESTAMP(), autoincrement=False, nullable=True))
op.add_column('tenants', sa.Column('plan', sa.VARCHAR(length=50), autoincrement=False, nullable=True))
op.add_column('tenants', sa.Column('api_ops_rate_limit', sa.VARCHAR(length=100), autoincrement=False, nullable=True))
# ### end Alembic commands ###