"""add_ssh_key_ids_to_tool_instances

Revision ID: 2026_05_29_add_ssh_key_ids_to_tool_instances
Revises: 2026_05_29_drop_ssh_key_id_from_config_profiles
Create Date: 2026-05-29 12:46:00.000000
"""

from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = "2026_05_29_add_ssh_key_ids_to_tool_instances"
down_revision = "2026_05_29_drop_ssh_key_id_from_config_profiles"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.add_column(
        "tool_instances",
        sa.Column("ssh_key_ids", sa.JSON(), nullable=True),
    )


def downgrade() -> None:
    op.drop_column("tool_instances", "ssh_key_ids")
