第一次提交
This commit is contained in:
217
backend/alembic/versions/001_initial_migration.py
Normal file
217
backend/alembic/versions/001_initial_migration.py
Normal file
@@ -0,0 +1,217 @@
|
||||
"""initial migration
|
||||
|
||||
Revision ID: 001
|
||||
Revises:
|
||||
Create Date: 2024-01-01 00:00:00.000000
|
||||
|
||||
"""
|
||||
from alembic import op
|
||||
import sqlalchemy as sa
|
||||
from sqlalchemy.dialects import mysql
|
||||
|
||||
# revision identifiers, used by Alembic.
|
||||
revision = '001'
|
||||
down_revision = None
|
||||
branch_labels = None
|
||||
depends_on = None
|
||||
|
||||
|
||||
def upgrade() -> None:
|
||||
# 创建用户表
|
||||
op.create_table(
|
||||
'users',
|
||||
sa.Column('id', mysql.CHAR(length=36), nullable=False, comment='用户ID'),
|
||||
sa.Column('username', sa.String(length=50), nullable=False, comment='用户名'),
|
||||
sa.Column('email', sa.String(length=100), nullable=False, comment='邮箱'),
|
||||
sa.Column('password_hash', sa.String(length=255), nullable=False, comment='密码哈希'),
|
||||
sa.Column('role', sa.String(length=20), server_default='user', nullable=True, comment='角色: admin/user'),
|
||||
sa.Column('created_at', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=True, comment='创建时间'),
|
||||
sa.Column('updated_at', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP'), nullable=True, comment='更新时间'),
|
||||
sa.PrimaryKeyConstraint('id'),
|
||||
sa.UniqueConstraint('email'),
|
||||
sa.UniqueConstraint('username'),
|
||||
mysql_charset='utf8mb4',
|
||||
mysql_collate='utf8mb4_unicode_ci'
|
||||
)
|
||||
|
||||
# 创建工作流表
|
||||
op.create_table(
|
||||
'workflows',
|
||||
sa.Column('id', mysql.CHAR(length=36), nullable=False, comment='工作流ID'),
|
||||
sa.Column('name', sa.String(length=100), nullable=False, comment='工作流名称'),
|
||||
sa.Column('description', sa.Text(), nullable=True, comment='描述'),
|
||||
sa.Column('nodes', sa.JSON(), nullable=False, comment='节点配置'),
|
||||
sa.Column('edges', sa.JSON(), nullable=False, comment='边配置'),
|
||||
sa.Column('version', sa.Integer(), server_default='1', nullable=True, comment='版本号'),
|
||||
sa.Column('status', sa.String(length=20), server_default='draft', nullable=True, comment='状态: draft/published/running/stopped'),
|
||||
sa.Column('user_id', mysql.CHAR(length=36), nullable=True, comment='创建者ID'),
|
||||
sa.Column('created_at', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=True, comment='创建时间'),
|
||||
sa.Column('updated_at', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP'), nullable=True, comment='更新时间'),
|
||||
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
|
||||
sa.PrimaryKeyConstraint('id'),
|
||||
mysql_charset='utf8mb4',
|
||||
mysql_collate='utf8mb4_unicode_ci'
|
||||
)
|
||||
op.create_index('ix_workflows_user_id', 'workflows', ['user_id'], unique=False)
|
||||
|
||||
# 创建工作流版本表
|
||||
op.create_table(
|
||||
'workflow_versions',
|
||||
sa.Column('id', mysql.CHAR(length=36), nullable=False, comment='版本ID'),
|
||||
sa.Column('workflow_id', mysql.CHAR(length=36), nullable=False, comment='工作流ID'),
|
||||
sa.Column('version', sa.Integer(), nullable=False, comment='版本号'),
|
||||
sa.Column('name', sa.String(length=100), nullable=False, comment='工作流名称'),
|
||||
sa.Column('description', sa.Text(), nullable=True, comment='描述'),
|
||||
sa.Column('nodes', sa.JSON(), nullable=False, comment='节点配置'),
|
||||
sa.Column('edges', sa.JSON(), nullable=False, comment='边配置'),
|
||||
sa.Column('status', sa.String(length=20), server_default='draft', nullable=True, comment='状态: draft/published/running/stopped'),
|
||||
sa.Column('created_by', mysql.CHAR(length=36), nullable=True, comment='创建者ID'),
|
||||
sa.Column('created_at', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=True, comment='创建时间'),
|
||||
sa.Column('comment', sa.Text(), nullable=True, comment='版本备注'),
|
||||
sa.ForeignKeyConstraint(['created_by'], ['users.id'], ),
|
||||
sa.ForeignKeyConstraint(['workflow_id'], ['workflows.id'], ondelete='CASCADE'),
|
||||
sa.PrimaryKeyConstraint('id'),
|
||||
mysql_charset='utf8mb4',
|
||||
mysql_collate='utf8mb4_unicode_ci'
|
||||
)
|
||||
op.create_index('ix_workflow_versions_workflow_id', 'workflow_versions', ['workflow_id'], unique=False)
|
||||
|
||||
# 创建智能体表
|
||||
op.create_table(
|
||||
'agents',
|
||||
sa.Column('id', mysql.CHAR(length=36), nullable=False, comment='智能体ID'),
|
||||
sa.Column('name', sa.String(length=100), nullable=False, comment='智能体名称'),
|
||||
sa.Column('description', sa.Text(), nullable=True, comment='描述'),
|
||||
sa.Column('workflow_config', sa.JSON(), nullable=False, comment='工作流配置'),
|
||||
sa.Column('version', sa.Integer(), server_default='1', nullable=True, comment='版本号'),
|
||||
sa.Column('status', sa.String(length=20), server_default='draft', nullable=True, comment='状态: draft/published/running/stopped'),
|
||||
sa.Column('user_id', mysql.CHAR(length=36), nullable=True, comment='创建者ID'),
|
||||
sa.Column('created_at', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=True, comment='创建时间'),
|
||||
sa.Column('updated_at', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP'), nullable=True, comment='更新时间'),
|
||||
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
|
||||
sa.PrimaryKeyConstraint('id'),
|
||||
mysql_charset='utf8mb4',
|
||||
mysql_collate='utf8mb4_unicode_ci'
|
||||
)
|
||||
op.create_index('ix_agents_user_id', 'agents', ['user_id'], unique=False)
|
||||
|
||||
# 创建执行记录表
|
||||
op.create_table(
|
||||
'executions',
|
||||
sa.Column('id', mysql.CHAR(length=36), nullable=False, comment='执行ID'),
|
||||
sa.Column('workflow_id', mysql.CHAR(length=36), nullable=True, comment='工作流ID'),
|
||||
sa.Column('agent_id', mysql.CHAR(length=36), nullable=True, comment='智能体ID'),
|
||||
sa.Column('input_data', sa.JSON(), nullable=True, comment='输入数据'),
|
||||
sa.Column('output_data', sa.JSON(), nullable=True, comment='输出数据'),
|
||||
sa.Column('status', sa.String(length=20), server_default='pending', nullable=True, comment='状态: pending/running/completed/failed'),
|
||||
sa.Column('task_id', sa.String(length=255), nullable=True, comment='Celery任务ID'),
|
||||
sa.Column('error_message', sa.Text(), nullable=True, comment='错误信息'),
|
||||
sa.Column('execution_time', sa.Integer(), nullable=True, comment='执行时间（毫秒）'),
|
||||
sa.Column('created_at', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=True, comment='创建时间'),
|
||||
sa.Column('updated_at', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP'), nullable=True, comment='更新时间'),
|
||||
sa.ForeignKeyConstraint(['agent_id'], ['agents.id'], ),
|
||||
sa.ForeignKeyConstraint(['workflow_id'], ['workflows.id'], ),
|
||||
sa.PrimaryKeyConstraint('id'),
|
||||
mysql_charset='utf8mb4',
|
||||
mysql_collate='utf8mb4_unicode_ci'
|
||||
)
|
||||
op.create_index('ix_executions_workflow_id', 'executions', ['workflow_id'], unique=False)
|
||||
op.create_index('ix_executions_agent_id', 'executions', ['agent_id'], unique=False)
|
||||
op.create_index('ix_executions_status', 'executions', ['status'], unique=False)
|
||||
|
||||
# 创建执行日志表
|
||||
op.create_table(
|
||||
'execution_logs',
|
||||
sa.Column('id', mysql.CHAR(length=36), nullable=False, comment='日志ID'),
|
||||
sa.Column('execution_id', mysql.CHAR(length=36), nullable=False, comment='执行ID'),
|
||||
sa.Column('node_id', sa.String(length=100), nullable=True, comment='节点ID'),
|
||||
sa.Column('node_type', sa.String(length=50), nullable=True, comment='节点类型'),
|
||||
sa.Column('level', sa.String(length=20), nullable=False, comment='日志级别: info/warn/error/debug'),
|
||||
sa.Column('message', sa.Text(), nullable=False, comment='日志消息'),
|
||||
sa.Column('timestamp', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=True, comment='时间戳'),
|
||||
sa.Column('duration_ms', sa.Integer(), nullable=True, comment='执行时长（毫秒）'),
|
||||
sa.Column('input_data', sa.JSON(), nullable=True, comment='输入数据'),
|
||||
sa.Column('output_data', sa.JSON(), nullable=True, comment='输出数据'),
|
||||
sa.Column('error_message', sa.Text(), nullable=True, comment='错误信息'),
|
||||
sa.Column('additional_data', sa.JSON(), nullable=True, comment='附加数据'),
|
||||
sa.ForeignKeyConstraint(['execution_id'], ['executions.id'], ondelete='CASCADE'),
|
||||
sa.PrimaryKeyConstraint('id'),
|
||||
mysql_charset='utf8mb4',
|
||||
mysql_collate='utf8mb4_unicode_ci'
|
||||
)
|
||||
op.create_index('ix_execution_logs_execution_id', 'execution_logs', ['execution_id'], unique=False)
|
||||
op.create_index('ix_execution_logs_node_id', 'execution_logs', ['node_id'], unique=False)
|
||||
op.create_index('ix_execution_logs_level', 'execution_logs', ['level'], unique=False)
|
||||
|
||||
# 创建模型配置表
|
||||
op.create_table(
|
||||
'model_configs',
|
||||
sa.Column('id', mysql.CHAR(length=36), nullable=False, comment='配置ID'),
|
||||
sa.Column('name', sa.String(length=100), nullable=False, comment='配置名称'),
|
||||
sa.Column('provider', sa.String(length=50), nullable=False, comment='提供商: openai/claude/local'),
|
||||
sa.Column('model_name', sa.String(length=100), nullable=False, comment='模型名称'),
|
||||
sa.Column('api_key', sa.String(length=500), nullable=False, comment='API密钥（加密存储）'),
|
||||
sa.Column('base_url', sa.String(length=255), nullable=True, comment='API地址'),
|
||||
sa.Column('user_id', mysql.CHAR(length=36), nullable=True, comment='所属用户ID'),
|
||||
sa.Column('created_at', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=True, comment='创建时间'),
|
||||
sa.Column('updated_at', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP'), nullable=True, comment='更新时间'),
|
||||
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
|
||||
sa.PrimaryKeyConstraint('id'),
|
||||
mysql_charset='utf8mb4',
|
||||
mysql_collate='utf8mb4_unicode_ci'
|
||||
)
|
||||
op.create_index('ix_model_configs_user_id', 'model_configs', ['user_id'], unique=False)
|
||||
|
||||
# 创建数据源表
|
||||
op.create_table(
|
||||
'data_sources',
|
||||
sa.Column('id', mysql.CHAR(length=36), nullable=False, comment='数据源ID'),
|
||||
sa.Column('name', sa.String(length=100), nullable=False, comment='数据源名称'),
|
||||
sa.Column('type', sa.String(length=50), nullable=False, comment='数据源类型: mysql/postgresql/mongodb/redis/csv/json/api/s3'),
|
||||
sa.Column('description', sa.Text(), nullable=True, comment='描述'),
|
||||
sa.Column('config', sa.JSON(), nullable=False, comment='连接配置（加密存储敏感信息）'),
|
||||
sa.Column('status', sa.String(length=20), server_default='active', nullable=True, comment='状态: active/inactive/error'),
|
||||
sa.Column('user_id', mysql.CHAR(length=36), nullable=False, comment='创建者ID'),
|
||||
sa.Column('last_connected_at', sa.DateTime(), nullable=True, comment='最后连接时间'),
|
||||
sa.Column('created_at', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=True, comment='创建时间'),
|
||||
sa.Column('updated_at', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP'), nullable=True, comment='更新时间'),
|
||||
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ondelete='CASCADE'),
|
||||
sa.PrimaryKeyConstraint('id'),
|
||||
mysql_charset='utf8mb4',
|
||||
mysql_collate='utf8mb4_unicode_ci'
|
||||
)
|
||||
op.create_index('ix_data_sources_user_id', 'data_sources', ['user_id'], unique=False)
|
||||
op.create_index('ix_data_sources_type', 'data_sources', ['type'], unique=False)
|
||||
op.create_index('ix_data_sources_status', 'data_sources', ['status'], unique=False)
|
||||
|
||||
|
||||
def downgrade() -> None:
|
||||
# 删除所有表（按依赖关系逆序）
|
||||
op.drop_index('ix_data_sources_status', table_name='data_sources')
|
||||
op.drop_index('ix_data_sources_type', table_name='data_sources')
|
||||
op.drop_index('ix_data_sources_user_id', table_name='data_sources')
|
||||
op.drop_table('data_sources')
|
||||
|
||||
op.drop_index('ix_model_configs_user_id', table_name='model_configs')
|
||||
op.drop_table('model_configs')
|
||||
|
||||
op.drop_index('ix_execution_logs_level', table_name='execution_logs')
|
||||
op.drop_index('ix_execution_logs_node_id', table_name='execution_logs')
|
||||
op.drop_index('ix_execution_logs_execution_id', table_name='execution_logs')
|
||||
op.drop_table('execution_logs')
|
||||
|
||||
op.drop_index('ix_executions_status', table_name='executions')
|
||||
op.drop_index('ix_executions_agent_id', table_name='executions')
|
||||
op.drop_index('ix_executions_workflow_id', table_name='executions')
|
||||
op.drop_table('executions')
|
||||
|
||||
op.drop_index('ix_agents_user_id', table_name='agents')
|
||||
op.drop_table('agents')
|
||||
|
||||
op.drop_index('ix_workflow_versions_workflow_id', table_name='workflow_versions')
|
||||
op.drop_table('workflow_versions')
|
||||
|
||||
op.drop_index('ix_workflows_user_id', table_name='workflows')
|
||||
op.drop_table('workflows')
|
||||
|
||||
op.drop_table('users')
|
||||
Reference in New Issue
Block a user