"""Add prompt_source provenance to agent_operations.

Revision ID: 174
Revises: 173
"""
import sqlalchemy as sa
from alembic import op

revision = "174"
down_revision = "173"
branch_labels = None
depends_on = None


def upgrade():
    op.add_column(
        "agent_operations",
        sa.Column("prompt_source", sa.String(20), nullable=True),
    )


def downgrade():
    op.drop_column("agent_operations", "prompt_source")