67 lines
3.1 KiB
Python
67 lines
3.1 KiB
Python
"""MCP-Server
|
|
|
|
Revision ID: 0006
|
|
Revises: 0005
|
|
Erstellt: 2026-10-21 23:43:44.357067
|
|
"""
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
|
|
revision = '0006'
|
|
down_revision = '0005'
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.create_table('mcp_servers',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('name', sa.String(length=16), nullable=False),
|
|
sa.Column('label', sa.String(length=200), nullable=False),
|
|
sa.Column('url', sa.String(length=500), nullable=False),
|
|
sa.Column('secret', sa.Text(), nullable=True),
|
|
sa.Column('secret_hint', sa.String(length=8), nullable=False),
|
|
sa.Column('active', sa.Boolean(), nullable=False),
|
|
sa.Column('status', sa.String(length=20), nullable=False),
|
|
sa.Column('last_error', sa.Text(), nullable=False),
|
|
sa.Column('discovered_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('server_info', sa.JSON(), nullable=False),
|
|
sa.Column('created_by', sa.Integer(), nullable=False),
|
|
sa.Column('created', sa.DateTime(timezone=True), nullable=False),
|
|
sa.ForeignKeyConstraint(['created_by'], ['users.id'], name=op.f('fk_mcp_servers_created_by_users')),
|
|
sa.PrimaryKeyConstraint('id', name=op.f('pk_mcp_servers')),
|
|
sa.UniqueConstraint('name', name=op.f('uq_mcp_servers_name'))
|
|
)
|
|
op.create_table('mcp_tools',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('server_id', sa.Integer(), nullable=False),
|
|
sa.Column('name', sa.String(length=64), nullable=False),
|
|
sa.Column('remote_name', sa.String(length=200), nullable=False),
|
|
sa.Column('description', sa.Text(), nullable=False),
|
|
sa.Column('input_schema', sa.JSON(), nullable=False),
|
|
sa.Column('annotations', sa.JSON(), nullable=False),
|
|
sa.ForeignKeyConstraint(['server_id'], ['mcp_servers.id'], name=op.f('fk_mcp_tools_server_id_mcp_servers'), ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id', name=op.f('pk_mcp_tools')),
|
|
sa.UniqueConstraint('name', name=op.f('uq_mcp_tools_name')),
|
|
sa.UniqueConstraint('server_id', 'remote_name', name=op.f('uq_mcp_tools_server_id'))
|
|
)
|
|
op.create_table('workspace_mcp_servers',
|
|
sa.Column('workspace_id', sa.Integer(), nullable=False),
|
|
sa.Column('server_id', sa.Integer(), nullable=False),
|
|
sa.Column('secret', sa.Text(), nullable=True),
|
|
sa.Column('secret_hint', sa.String(length=8), nullable=False),
|
|
sa.Column('enabled_by', sa.Integer(), nullable=False),
|
|
sa.Column('enabled', sa.DateTime(timezone=True), nullable=False),
|
|
sa.ForeignKeyConstraint(['enabled_by'], ['users.id'], name=op.f('fk_workspace_mcp_servers_enabled_by_users')),
|
|
sa.ForeignKeyConstraint(['server_id'], ['mcp_servers.id'], name=op.f('fk_workspace_mcp_servers_server_id_mcp_servers'), ondelete='CASCADE'),
|
|
sa.ForeignKeyConstraint(['workspace_id'], ['workspaces.id'], name=op.f('fk_workspace_mcp_servers_workspace_id_workspaces'), ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('workspace_id', 'server_id', name=op.f('pk_workspace_mcp_servers'))
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_table('workspace_mcp_servers')
|
|
op.drop_table('mcp_tools')
|
|
op.drop_table('mcp_servers')
|