SKY-6729: Add llm_cost, proxy_cost, captcha_cost columns to task_runs (#6219)

Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
Shuchang Zheng 2026-05-28 06:46:19 -07:00 committed by GitHub
parent ad0c589bb1
commit 6a2705d82e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
41 changed files with 5020 additions and 928 deletions

View file

@ -0,0 +1,32 @@
"""add_cost_columns_to_task_runs
Revision ID: c636e511753d
Revises: 0d28973280f5
Create Date: 2026-05-28T01:06:49.862871+00:00
"""
from typing import Sequence, Union
import sqlalchemy as sa
from alembic import op
# revision identifiers, used by Alembic.
revision: str = "c636e511753d"
down_revision: Union[str, None] = "0d28973280f5"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
op.execute("SET LOCAL lock_timeout = '5s'")
op.add_column("task_runs", sa.Column("llm_cost", sa.Numeric(), nullable=True))
op.add_column("task_runs", sa.Column("proxy_cost", sa.Numeric(), nullable=True))
op.add_column("task_runs", sa.Column("captcha_cost", sa.Numeric(), nullable=True))
def downgrade() -> None:
op.drop_column("task_runs", "captcha_cost")
op.drop_column("task_runs", "proxy_cost")
op.drop_column("task_runs", "llm_cost")