fix(db): add migration for skills marketplace tables

This commit is contained in:
null 2026-05-20 22:42:43 -05:00
parent 133b0ffb4c
commit d9bdc4cb3d
2 changed files with 86 additions and 0 deletions

View File

@ -0,0 +1,86 @@
"""Add skills marketplace tables.
Revision ID: f7d8e9a0b1c2
Revises: a2f6c9d4b7e8
Create Date: 2026-05-20 22:45:00.000000
"""
from __future__ import annotations
import sqlalchemy as sa
import sqlmodel
from alembic import op
# revision identifiers, used by Alembic.
revision = "f7d8e9a0b1c2"
down_revision = "a2f6c9d4b7e8"
branch_labels = None
depends_on = None
def upgrade() -> None:
"""Create marketplace_skills, skill_packs, and gateway_installed_skills tables."""
op.create_table(
"marketplace_skills",
sa.Column("id", sa.Uuid(), nullable=False),
sa.Column("organization_id", sa.Uuid(), nullable=False),
sa.Column("name", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
sa.Column("description", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
sa.Column("category", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
sa.Column("risk", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
sa.Column("source", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
sa.Column("source_url", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
sa.Column("metadata", sa.JSON(), nullable=False),
sa.Column("created_at", sa.DateTime(), nullable=False),
sa.Column("updated_at", sa.DateTime(), nullable=False),
sa.ForeignKeyConstraint(["organization_id"], ["organizations.id"]),
sa.PrimaryKeyConstraint("id"),
sa.UniqueConstraint("organization_id", "source_url", name="uq_marketplace_skills_org_source_url"),
)
op.create_index(op.f("ix_marketplace_skills_organization_id"), "marketplace_skills", ["organization_id"], unique=False)
op.create_table(
"skill_packs",
sa.Column("id", sa.Uuid(), nullable=False),
sa.Column("organization_id", sa.Uuid(), nullable=False),
sa.Column("name", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
sa.Column("description", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
sa.Column("source_url", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
sa.Column("branch", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
sa.Column("metadata", sa.JSON(), nullable=False),
sa.Column("created_at", sa.DateTime(), nullable=False),
sa.Column("updated_at", sa.DateTime(), nullable=False),
sa.ForeignKeyConstraint(["organization_id"], ["organizations.id"]),
sa.PrimaryKeyConstraint("id"),
sa.UniqueConstraint("organization_id", "source_url", name="uq_skill_packs_org_source_url"),
)
op.create_index(op.f("ix_skill_packs_organization_id"), "skill_packs", ["organization_id"], unique=False)
op.create_table(
"gateway_installed_skills",
sa.Column("id", sa.Uuid(), nullable=False),
sa.Column("gateway_id", sa.Uuid(), nullable=False),
sa.Column("skill_id", sa.Uuid(), nullable=False),
sa.Column("created_at", sa.DateTime(), nullable=False),
sa.Column("updated_at", sa.DateTime(), nullable=False),
sa.ForeignKeyConstraint(["gateway_id"], ["gateways.id"]),
sa.ForeignKeyConstraint(["skill_id"], ["marketplace_skills.id"]),
sa.PrimaryKeyConstraint("id"),
sa.UniqueConstraint("gateway_id", "skill_id", name="uq_gateway_installed_skills_gateway_id_skill_id"),
)
op.create_index(op.f("ix_gateway_installed_skills_gateway_id"), "gateway_installed_skills", ["gateway_id"], unique=False)
op.create_index(op.f("ix_gateway_installed_skills_skill_id"), "gateway_installed_skills", ["skill_id"], unique=False)
def downgrade() -> None:
"""Drop skills marketplace tables."""
op.drop_index(op.f("ix_gateway_installed_skills_skill_id"), table_name="gateway_installed_skills")
op.drop_index(op.f("ix_gateway_installed_skills_gateway_id"), table_name="gateway_installed_skills")
op.drop_table("gateway_installed_skills")
op.drop_index(op.f("ix_skill_packs_organization_id"), table_name="skill_packs")
op.drop_table("skill_packs")
op.drop_index(op.f("ix_marketplace_skills_organization_id"), table_name="marketplace_skills")
op.drop_table("marketplace_skills")