[add] migration script

This commit is contained in:
Mark
2025-12-19 18:06:49 +08:00
parent 01a5bed11c
commit 185e262db8
2 changed files with 37 additions and 1 deletions

View File

@@ -1,7 +1,7 @@
import datetime
from enum import StrEnum
import uuid
from sqlalchemy import Column, Integer, String, DateTime, ForeignKey, Boolean
from sqlalchemy import Column, String, DateTime, ForeignKey, Boolean
from sqlalchemy.dialects.postgresql import UUID
from sqlalchemy.orm import relationship
from app.db import Base

View File

@@ -0,0 +1,36 @@
"""202512191805
Revision ID: f96a53af914c
Revises: 87a6537b4074
Create Date: 2025-12-19 18:05:14.964454
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
# revision identifiers, used by Alembic.
revision: str = 'f96a53af914c'
down_revision: Union[str, None] = '87a6537b4074'
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('data_config', sa.Column('reflection_model_id', sa.String(), nullable=True, comment='反思模型ID'))
op.add_column('data_config', sa.Column('memory_verify', sa.Boolean(), nullable=True, comment='记忆验证'))
op.add_column('data_config', sa.Column('quality_assessment', sa.Boolean(), nullable=True, comment='质量评估'))
op.add_column('end_users', sa.Column('reflection_time', sa.DateTime(), nullable=True))
# ### end Alembic commands ###
def downgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.drop_column('end_users', 'reflection_time')
op.drop_column('data_config', 'quality_assessment')
op.drop_column('data_config', 'memory_verify')
op.drop_column('data_config', 'reflection_model_id')
# ### end Alembic commands ###