"""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')