agent-platform/services/tool-service/alembic/versions/20260427_0002_add_tool_credentials.py

"""add tool credentials

Revision ID: 20260427_0002
Revises: 20260422_0001
Create Date: 2026-04-27 00:00:00
"""

from collections.abc import Sequence

from alembic import op
import sqlalchemy as sa


revision: str = "20260427_0002"
down_revision: str | None = "20260422_0001"
branch_labels: Sequence[str] | None = None
depends_on: Sequence[str] | None = None


def upgrade() -> None:
    op.create_table(
        "tool_credential",
        sa.Column("name", sa.String(length=128), nullable=False),
        sa.Column("credential_type", sa.String(length=64), nullable=False),
        sa.Column("encrypted_payload_text", sa.Text(), nullable=False),
        sa.Column("secret_fingerprint", sa.String(length=64), nullable=False),
        sa.Column("encryption_algorithm", sa.String(length=64), nullable=False),
        sa.Column("metadata_json", sa.JSON(), nullable=True),
        sa.Column("id", sa.String(length=36), nullable=False),
        sa.Column("tenant_id", sa.String(length=36), nullable=False),
        sa.Column("created_by", sa.String(length=36), nullable=True),
        sa.Column("updated_by", sa.String(length=36), nullable=True),
        sa.Column("created_time", sa.DateTime(), nullable=False),
        sa.Column("updated_time", sa.DateTime(), nullable=False),
        sa.Column("deleted_time", sa.DateTime(), nullable=True),
        sa.Column("version", sa.Integer(), nullable=False),
        sa.PrimaryKeyConstraint("id"),
    )
    op.create_index("ix_tool_credential_tenant_id", "tool_credential", ["tenant_id"])
    op.create_index(
        "ix_tool_credential_secret_fingerprint",
        "tool_credential",
        ["secret_fingerprint"],
    )


def downgrade() -> None:
    op.drop_index("ix_tool_credential_secret_fingerprint", table_name="tool_credential")
    op.drop_index("ix_tool_credential_tenant_id", table_name="tool_credential")
    op.drop_table("tool_credential")