40 lines
1.0 KiB
Python
40 lines
1.0 KiB
Python
"""add ai agent process tracking
|
|
|
|
Revision ID: de11b016b35a
|
|
Revises: 1a2cda4df181
|
|
Create Date: 2025-09-03 00:02:24.263636
|
|
|
|
"""
|
|
|
|
from collections.abc import Sequence
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = "de11b016b35a"
|
|
down_revision: str | Sequence[str] | None = "1a2cda4df181"
|
|
branch_labels: str | Sequence[str] | None = None
|
|
depends_on: str | Sequence[str] | None = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
"""Upgrade schema."""
|
|
# Add AI agent process tracking columns
|
|
op.add_column(
|
|
"interview_sessions", sa.Column("ai_agent_pid", sa.Integer(), nullable=True)
|
|
)
|
|
op.add_column(
|
|
"interview_sessions",
|
|
sa.Column(
|
|
"ai_agent_status", sa.String(), server_default="not_started", nullable=False
|
|
),
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
"""Downgrade schema."""
|
|
# Drop AI agent process tracking columns
|
|
op.drop_column("interview_sessions", "ai_agent_status")
|
|
op.drop_column("interview_sessions", "ai_agent_pid")
|