agent-platform/services/session-service/alembic/versions/20260512_0002_session_runtime_targets.py

"""add session runtime target fields

Revision ID: 20260512_0002_session
Revises: 20260429_9001_session
Create Date: 2026-05-12 00:00:00.000000
"""

from collections.abc import Sequence

import sqlalchemy as sa
from alembic import op

revision: str = "20260512_0002_session"
down_revision: str | None = "20260429_9001_session"
branch_labels: Sequence[str] | None = None
depends_on: Sequence[str] | None = None


def upgrade() -> None:
    op.add_column("session", sa.Column("runtime_target_type", sa.String(length=32), nullable=True))
    op.add_column("session", sa.Column("runtime_target_id", sa.String(length=36), nullable=True))
    op.add_column("session", sa.Column("runtime_target_config_id", sa.String(length=36), nullable=True))


def downgrade() -> None:
    op.drop_column("session", "runtime_target_config_id")
    op.drop_column("session", "runtime_target_id")
    op.drop_column("session", "runtime_target_type")