42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
"""Add metered flag to investigation_runs
|
|
|
|
Revision ID: add_tkt_metered_001
|
|
Revises: add_tkt_row_scope_001
|
|
Create Date: 2026-07-21
|
|
|
|
Marks whether an investigation run's LLM usage counts against the org's message
|
|
budget (hosted model only). Existing rows default to false — nothing is
|
|
retroactively metered on upgrade.
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
revision = "add_tkt_metered_001"
|
|
down_revision = "add_tkt_row_scope_001"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.add_column(
|
|
"investigation_runs",
|
|
sa.Column(
|
|
"metered",
|
|
sa.Boolean(),
|
|
nullable=False,
|
|
server_default=sa.text("false"),
|
|
),
|
|
)
|
|
# The enterprise message-limit check sums llm_calls of metered runs per
|
|
# billing window — an index on (org, metered, created_at) keeps that cheap.
|
|
op.create_index(
|
|
"ix_investigation_runs_metered_period",
|
|
"investigation_runs",
|
|
["organization_id", "metered", "created_at"],
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_index("ix_investigation_runs_metered_period", table_name="investigation_runs")
|
|
op.drop_column("investigation_runs", "metered")
|