kubiya-control-plane-api 0.1.0__py3-none-any.whl → 0.3.4__py3-none-any.whl

control_plane_api/alembic/versions/1f54bc2a37e3_add_analytics_tables.py

@@ -0,0 +1,162 @@
+"""add comprehensive analytics tables for execution tracking
+
+Revision ID: 1f54bc2a37e3
+Revises: add_llm_models_table
+Create Date: 2025-01-08 14:00:00.000000
+
+This migration adds production-grade analytics tables to track:
+- Per-turn LLM metrics (tokens, duration, cost)
+- Tool execution details (success/failure, timing)
+- Task completion tracking
+"""
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision = '1f54bc2a37e3'
+down_revision = 'add_llm_models_table'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # Create execution_turns table
+    op.create_table(
+        'execution_turns',
+        sa.Column('id', postgresql.UUID(as_uuid=True), nullable=False),
+        sa.Column('organization_id', sa.String(), nullable=False),
+        sa.Column('execution_id', postgresql.UUID(as_uuid=True), nullable=False),
+        sa.Column('turn_number', sa.Integer(), nullable=False),
+        sa.Column('turn_id', sa.String(), nullable=True),
+        sa.Column('model', sa.String(), nullable=False),
+        sa.Column('model_provider', sa.String(), nullable=True),
+        sa.Column('started_at', sa.DateTime(timezone=True), nullable=False, server_default=sa.text('now()')),
+        sa.Column('completed_at', sa.DateTime(timezone=True), nullable=True),
+        sa.Column('duration_ms', sa.Integer(), nullable=True),
+        sa.Column('input_tokens', sa.Integer(), nullable=True, server_default='0'),
+        sa.Column('output_tokens', sa.Integer(), nullable=True, server_default='0'),
+        sa.Column('cache_read_tokens', sa.Integer(), nullable=True, server_default='0'),
+        sa.Column('cache_creation_tokens', sa.Integer(), nullable=True, server_default='0'),
+        sa.Column('total_tokens', sa.Integer(), nullable=True, server_default='0'),
+        sa.Column('input_cost', sa.Float(), nullable=True, server_default='0.0'),
+        sa.Column('output_cost', sa.Float(), nullable=True, server_default='0.0'),
+        sa.Column('cache_read_cost', sa.Float(), nullable=True, server_default='0.0'),
+        sa.Column('cache_creation_cost', sa.Float(), nullable=True, server_default='0.0'),
+        sa.Column('total_cost', sa.Float(), nullable=True, server_default='0.0'),
+        sa.Column('finish_reason', sa.String(), nullable=True),
+        sa.Column('response_preview', sa.Text(), nullable=True),
+        sa.Column('tools_called_count', sa.Integer(), nullable=False, server_default='0'),
+        sa.Column('tools_called_names', postgresql.JSON(astext_type=sa.Text()), nullable=False, server_default='[]'),
+        sa.Column('error_message', sa.Text(), nullable=True),
+        sa.Column('metrics', postgresql.JSON(astext_type=sa.Text()), nullable=False, server_default='{}'),
+        sa.Column('runtime_minutes', sa.Float(), nullable=True, server_default='0.0'),
+        sa.Column('model_weight', sa.Float(), nullable=True, server_default='1.0'),
+        sa.Column('tool_calls_weight', sa.Float(), nullable=True, server_default='1.0'),
+        sa.Column('aem_value', sa.Float(), nullable=True, server_default='0.0'),
+        sa.Column('aem_cost', sa.Float(), nullable=True, server_default='0.0'),
+        sa.Column('created_at', sa.DateTime(timezone=True), nullable=False, server_default=sa.text('now()')),
+        sa.PrimaryKeyConstraint('id'),
+        sa.ForeignKeyConstraint(['execution_id'], ['executions.id'], ondelete='CASCADE')
+    )
+
+    # Create indexes for execution_turns
+    op.create_index('ix_execution_turns_organization_id', 'execution_turns', ['organization_id'])
+    op.create_index('ix_execution_turns_execution_id', 'execution_turns', ['execution_id'])
+    op.create_index('ix_execution_turns_org_execution', 'execution_turns', ['organization_id', 'execution_id'])
+    op.create_index('ix_execution_turns_org_model', 'execution_turns', ['organization_id', 'model'])
+    op.create_index('ix_execution_turns_org_created', 'execution_turns', ['organization_id', 'created_at'])
+    op.create_index('ix_execution_turns_org_cost', 'execution_turns', ['organization_id', 'total_cost'])
+
+    # Create execution_tool_calls table
+    op.create_table(
+        'execution_tool_calls',
+        sa.Column('id', postgresql.UUID(as_uuid=True), nullable=False),
+        sa.Column('organization_id', sa.String(), nullable=False),
+        sa.Column('execution_id', postgresql.UUID(as_uuid=True), nullable=False),
+        sa.Column('turn_id', postgresql.UUID(as_uuid=True), nullable=True),
+        sa.Column('tool_name', sa.String(), nullable=False),
+        sa.Column('tool_use_id', sa.String(), nullable=True),
+        sa.Column('started_at', sa.DateTime(timezone=True), nullable=False, server_default=sa.text('now()')),
+        sa.Column('completed_at', sa.DateTime(timezone=True), nullable=True),
+        sa.Column('duration_ms', sa.Integer(), nullable=True),
+        sa.Column('tool_input', postgresql.JSON(astext_type=sa.Text()), nullable=True),
+        sa.Column('tool_output', sa.Text(), nullable=True),
+        sa.Column('tool_output_size', sa.Integer(), nullable=True),
+        sa.Column('success', sa.Boolean(), nullable=False, server_default='true'),
+        sa.Column('error_message', sa.Text(), nullable=True),
+        sa.Column('error_type', sa.String(), nullable=True),
+        sa.Column('metadata', postgresql.JSON(astext_type=sa.Text()), nullable=False, server_default='{}'),
+        sa.Column('created_at', sa.DateTime(timezone=True), nullable=False, server_default=sa.text('now()')),
+        sa.PrimaryKeyConstraint('id'),
+        sa.ForeignKeyConstraint(['execution_id'], ['executions.id'], ondelete='CASCADE'),
+        sa.ForeignKeyConstraint(['turn_id'], ['execution_turns.id'], ondelete='CASCADE')
+    )
+
+    # Create indexes for execution_tool_calls
+    op.create_index('ix_execution_tool_calls_organization_id', 'execution_tool_calls', ['organization_id'])
+    op.create_index('ix_execution_tool_calls_execution_id', 'execution_tool_calls', ['execution_id'])
+    op.create_index('ix_execution_tool_calls_turn_id', 'execution_tool_calls', ['turn_id'])
+    op.create_index('ix_execution_tool_calls_tool_name', 'execution_tool_calls', ['tool_name'])
+    op.create_index('ix_execution_tool_calls_org_execution', 'execution_tool_calls', ['organization_id', 'execution_id'])
+    op.create_index('ix_execution_tool_calls_org_tool', 'execution_tool_calls', ['organization_id', 'tool_name'])
+    op.create_index('ix_execution_tool_calls_org_success', 'execution_tool_calls', ['organization_id', 'success'])
+    op.create_index('ix_execution_tool_calls_org_created', 'execution_tool_calls', ['organization_id', 'created_at'])
+
+    # Create execution_tasks table
+    op.create_table(
+        'execution_tasks',
+        sa.Column('id', postgresql.UUID(as_uuid=True), nullable=False),
+        sa.Column('organization_id', sa.String(), nullable=False),
+        sa.Column('execution_id', postgresql.UUID(as_uuid=True), nullable=False),
+        sa.Column('task_number', sa.Integer(), nullable=True),
+        sa.Column('task_id', sa.String(), nullable=True),
+        sa.Column('task_description', sa.Text(), nullable=False),
+        sa.Column('task_type', sa.String(), nullable=True),
+        sa.Column('status', sa.String(), nullable=False, server_default='pending'),
+        sa.Column('started_at', sa.DateTime(timezone=True), nullable=True),
+        sa.Column('completed_at', sa.DateTime(timezone=True), nullable=True),
+        sa.Column('duration_ms', sa.Integer(), nullable=True),
+        sa.Column('result', sa.Text(), nullable=True),
+        sa.Column('error_message', sa.Text(), nullable=True),
+        sa.Column('metadata', postgresql.JSON(astext_type=sa.Text()), nullable=False, server_default='{}'),
+        sa.Column('created_at', sa.DateTime(timezone=True), nullable=False, server_default=sa.text('now()')),
+        sa.Column('updated_at', sa.DateTime(timezone=True), nullable=False, server_default=sa.text('now()')),
+        sa.PrimaryKeyConstraint('id'),
+        sa.ForeignKeyConstraint(['execution_id'], ['executions.id'], ondelete='CASCADE')
+    )
+
+    # Create indexes for execution_tasks
+    op.create_index('ix_execution_tasks_organization_id', 'execution_tasks', ['organization_id'])
+    op.create_index('ix_execution_tasks_execution_id', 'execution_tasks', ['execution_id'])
+    op.create_index('ix_execution_tasks_org_execution', 'execution_tasks', ['organization_id', 'execution_id'])
+    op.create_index('ix_execution_tasks_org_status', 'execution_tasks', ['organization_id', 'status'])
+
+
+def downgrade():
+    # Drop execution_tasks table and indexes
+    op.drop_index('ix_execution_tasks_org_status', table_name='execution_tasks')
+    op.drop_index('ix_execution_tasks_org_execution', table_name='execution_tasks')
+    op.drop_index('ix_execution_tasks_execution_id', table_name='execution_tasks')
+    op.drop_index('ix_execution_tasks_organization_id', table_name='execution_tasks')
+    op.drop_table('execution_tasks')
+
+    # Drop execution_tool_calls table and indexes
+    op.drop_index('ix_execution_tool_calls_org_created', table_name='execution_tool_calls')
+    op.drop_index('ix_execution_tool_calls_org_success', table_name='execution_tool_calls')
+    op.drop_index('ix_execution_tool_calls_org_tool', table_name='execution_tool_calls')
+    op.drop_index('ix_execution_tool_calls_org_execution', table_name='execution_tool_calls')
+    op.drop_index('ix_execution_tool_calls_tool_name', table_name='execution_tool_calls')
+    op.drop_index('ix_execution_tool_calls_turn_id', table_name='execution_tool_calls')
+    op.drop_index('ix_execution_tool_calls_execution_id', table_name='execution_tool_calls')
+    op.drop_index('ix_execution_tool_calls_organization_id', table_name='execution_tool_calls')
+    op.drop_table('execution_tool_calls')
+
+    # Drop execution_turns table and indexes
+    op.drop_index('ix_execution_turns_org_cost', table_name='execution_turns')
+    op.drop_index('ix_execution_turns_org_created', table_name='execution_turns')
+    op.drop_index('ix_execution_turns_org_model', table_name='execution_turns')
+    op.drop_index('ix_execution_turns_org_execution', table_name='execution_turns')
+    op.drop_index('ix_execution_turns_execution_id', table_name='execution_turns')
+    op.drop_index('ix_execution_turns_organization_id', table_name='execution_turns')
+    op.drop_table('execution_turns')

control_plane_api/alembic/versions/2e4cb136dc10_rename_toolset_ids_to_skill_ids_in_teams.py

@@ -0,0 +1,30 @@
+"""rename_toolset_ids_to_skill_ids_in_teams
+
+Revision ID: 2e4cb136dc10
+Revises: d4eaf16e3f8d
+Create Date: 2025-11-06 21:31:20.981166
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = '2e4cb136dc10'
+down_revision: Union[str, Sequence[str], None] = 'd4eaf16e3f8d'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Rename toolset_ids to skill_ids in teams table."""
+    # Rename toolset_ids column to skill_ids
+    op.alter_column('teams', 'toolset_ids', new_column_name='skill_ids')
+
+
+def downgrade() -> None:
+    """Rename skill_ids back to toolset_ids in teams table."""
+    # Reverse the column rename
+    op.alter_column('teams', 'skill_ids', new_column_name='toolset_ids')

control_plane_api/alembic/versions/31cd69a644ce_add_skill_templates_table.py

@@ -0,0 +1,28 @@
+"""add_skill_templates_table
+
+Revision ID: 31cd69a644ce
+Revises: f973b431d1ce
+Create Date: 2025-11-08 10:35:32.991694
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = '31cd69a644ce'
+down_revision: Union[str, Sequence[str], None] = 'f973b431d1ce'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    pass
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    pass

control_plane_api/alembic/versions/89e127caa47d_add_jobs_and_job_executions_tables.py

@@ -0,0 +1,161 @@
+"""add_jobs_and_job_executions_tables
+
+Revision ID: 89e127caa47d
+Revises: b0e10697f212
+Create Date: 2025-11-06 20:11:04.561446
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = '89e127caa47d'
+down_revision: Union[str, Sequence[str], None] = 'b0e10697f212'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # Create jobs table
+    op.create_table(
+        'jobs',
+        sa.Column('id', sa.String(255), primary_key=True),
+        sa.Column('organization_id', sa.String(255), nullable=False),
+        sa.Column('name', sa.String(255), nullable=False),
+        sa.Column('description', sa.Text(), nullable=True),
+        sa.Column('enabled', sa.Boolean(), nullable=False, server_default='true'),
+        sa.Column('status', sa.String(50), nullable=False, server_default='active'),
+        sa.Column('trigger_type', sa.String(50), nullable=False),
+        sa.Column('cron_schedule', sa.String(255), nullable=True),
+        sa.Column('cron_timezone', sa.String(100), server_default='UTC'),
+        sa.Column('webhook_url_path', sa.String(500), nullable=True),
+        sa.Column('webhook_secret', sa.String(500), nullable=True),
+        sa.Column('temporal_schedule_id', sa.String(255), nullable=True),
+        sa.Column('planning_mode', sa.String(50), nullable=False, server_default='predefined_agent'),
+        sa.Column('entity_type', sa.String(50), nullable=True),
+        sa.Column('entity_id', sa.String(255), nullable=True),
+        sa.Column('entity_name', sa.String(255), nullable=True),
+        sa.Column('prompt_template', sa.Text(), nullable=False),
+        sa.Column('system_prompt', sa.Text(), nullable=True),
+        sa.Column('executor_type', sa.String(50), nullable=False, server_default='auto'),
+        sa.Column('worker_queue_name', sa.String(255), nullable=True),
+        sa.Column('environment_name', sa.String(255), nullable=True),
+        sa.Column('config', sa.JSON(), server_default='{}'),
+        sa.Column('execution_environment', sa.JSON(), server_default='{}'),
+        sa.Column('last_execution_id', sa.String(255), nullable=True),
+        sa.Column('last_execution_at', sa.DateTime(timezone=True), nullable=True),
+        sa.Column('next_execution_at', sa.DateTime(timezone=True), nullable=True),
+        sa.Column('total_executions', sa.Integer(), server_default='0'),
+        sa.Column('successful_executions', sa.Integer(), server_default='0'),
+        sa.Column('failed_executions', sa.Integer(), server_default='0'),
+        sa.Column('execution_history', sa.JSON(), server_default='[]'),
+        sa.Column('created_by', sa.String(255), nullable=True),
+        sa.Column('updated_by', sa.String(255), nullable=True),
+        sa.Column('created_at', sa.DateTime(timezone=True), nullable=False, server_default=sa.text('now()')),
+        sa.Column('updated_at', sa.DateTime(timezone=True), nullable=False, server_default=sa.text('now()')),
+        sa.Column('last_triggered_at', sa.DateTime(timezone=True), nullable=True),
+    )
+
+    # Note: Skipping foreign key constraint for last_execution_id due to type mismatch
+    # The executions.id is stored as UUID in the database but defined as String in models
+    # We can add this constraint later after resolving the type discrepancy
+
+    # Add unique constraints
+    op.create_unique_constraint('uq_jobs_webhook_url_path', 'jobs', ['webhook_url_path'])
+    op.create_unique_constraint('uq_jobs_temporal_schedule_id', 'jobs', ['temporal_schedule_id'])
+
+    # Create indexes for jobs table
+    op.create_index('idx_jobs_organization_id', 'jobs', ['organization_id'])
+    op.create_index('idx_jobs_name', 'jobs', ['organization_id', 'name'])
+    op.create_index('idx_jobs_enabled', 'jobs', ['enabled'])
+    op.create_index('idx_jobs_status', 'jobs', ['status'])
+    op.create_index('idx_jobs_trigger_type', 'jobs', ['trigger_type'])
+    op.create_index('idx_jobs_webhook_url_path', 'jobs', ['webhook_url_path'])
+    op.create_index('idx_jobs_temporal_schedule_id', 'jobs', ['temporal_schedule_id'])
+    op.create_index('idx_jobs_created_at', 'jobs', ['created_at'])
+    op.create_index('idx_jobs_next_execution_at', 'jobs', ['next_execution_at'])
+
+    # Create job_executions table
+    op.create_table(
+        'job_executions',
+        sa.Column('id', sa.String(255), primary_key=True),
+        sa.Column('job_id', sa.String(255), nullable=False),
+        sa.Column('execution_id', sa.String(255), nullable=False),
+        sa.Column('organization_id', sa.String(255), nullable=False),
+        sa.Column('trigger_type', sa.String(50), nullable=False),
+        sa.Column('trigger_metadata', sa.JSON(), server_default='{}'),
+        sa.Column('execution_status', sa.String(50), nullable=True),
+        sa.Column('execution_duration_ms', sa.Integer(), nullable=True),
+        sa.Column('created_at', sa.DateTime(timezone=True), nullable=False, server_default=sa.text('now()')),
+    )
+
+    # Add foreign key constraints for job_executions
+    op.create_foreign_key(
+        'fk_job_executions_job',
+        'job_executions', 'jobs',
+        ['job_id'], ['id'],
+        ondelete='CASCADE'
+    )
+    # Note: Skipping foreign key for execution_id due to type mismatch with executions.id
+
+    # Create indexes for job_executions table
+    op.create_index('idx_job_executions_job_id', 'job_executions', ['job_id'])
+    op.create_index('idx_job_executions_execution_id', 'job_executions', ['execution_id'])
+    op.create_index('idx_job_executions_organization_id', 'job_executions', ['organization_id'])
+    op.create_index('idx_job_executions_created_at', 'job_executions', ['created_at'])
+    op.create_index('idx_job_executions_trigger_type', 'job_executions', ['trigger_type'])
+    op.create_index('idx_job_executions_execution_status', 'job_executions', ['execution_status'])
+
+    # Create trigger function for updated_at
+    op.execute("""
+        CREATE OR REPLACE FUNCTION update_updated_at_column()
+        RETURNS TRIGGER AS $$
+        BEGIN
+            NEW.updated_at = NOW();
+            RETURN NEW;
+        END;
+        $$ LANGUAGE plpgsql;
+    """)
+
+    # Create trigger for jobs table
+    op.execute("""
+        DROP TRIGGER IF EXISTS update_jobs_updated_at ON jobs;
+        CREATE TRIGGER update_jobs_updated_at
+            BEFORE UPDATE ON jobs
+            FOR EACH ROW
+            EXECUTE FUNCTION update_updated_at_column();
+    """)
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # Drop triggers
+    op.execute("DROP TRIGGER IF EXISTS update_jobs_updated_at ON jobs;")
+    op.execute("DROP FUNCTION IF EXISTS update_updated_at_column CASCADE;")
+
+    # Drop indexes for job_executions
+    op.drop_index('idx_job_executions_execution_status', 'job_executions')
+    op.drop_index('idx_job_executions_trigger_type', 'job_executions')
+    op.drop_index('idx_job_executions_created_at', 'job_executions')
+    op.drop_index('idx_job_executions_organization_id', 'job_executions')
+    op.drop_index('idx_job_executions_execution_id', 'job_executions')
+    op.drop_index('idx_job_executions_job_id', 'job_executions')
+
+    # Drop indexes for jobs
+    op.drop_index('idx_jobs_next_execution_at', 'jobs')
+    op.drop_index('idx_jobs_created_at', 'jobs')
+    op.drop_index('idx_jobs_temporal_schedule_id', 'jobs')
+    op.drop_index('idx_jobs_webhook_url_path', 'jobs')
+    op.drop_index('idx_jobs_trigger_type', 'jobs')
+    op.drop_index('idx_jobs_status', 'jobs')
+    op.drop_index('idx_jobs_enabled', 'jobs')
+    op.drop_index('idx_jobs_name', 'jobs')
+    op.drop_index('idx_jobs_organization_id', 'jobs')
+
+    # Drop tables
+    op.drop_table('job_executions')
+    op.drop_table('jobs')

control_plane_api/alembic/versions/add_llm_models_table.py

@@ -0,0 +1,51 @@
+"""add llm_models table
+
+Revision ID: add_llm_models_table
+Revises: f973b431d1ce
+Create Date: 2025-01-08 13:25:00.000000
+
+"""
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision = 'add_llm_models_table'
+down_revision = 'f973b431d1ce'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # Create llm_models table
+    op.create_table(
+        'llm_models',
+        sa.Column('id', sa.String(), nullable=False),
+        sa.Column('value', sa.String(), nullable=False),
+        sa.Column('label', sa.String(), nullable=False),
+        sa.Column('provider', sa.String(), nullable=False),
+        sa.Column('logo', sa.String(), nullable=True),
+        sa.Column('description', sa.Text(), nullable=True),
+        sa.Column('enabled', sa.Boolean(), nullable=False, server_default='true'),
+        sa.Column('recommended', sa.Boolean(), nullable=False, server_default='false'),
+        sa.Column('compatible_runtimes', postgresql.JSON(astext_type=sa.Text()), nullable=False, server_default='[]'),
+        sa.Column('capabilities', postgresql.JSON(astext_type=sa.Text()), nullable=False, server_default='{}'),
+        sa.Column('pricing', postgresql.JSON(astext_type=sa.Text()), nullable=True),
+        sa.Column('display_order', sa.Integer(), nullable=False, server_default='1000'),
+        sa.Column('created_at', sa.DateTime(), nullable=False, server_default=sa.text('now()')),
+        sa.Column('updated_at', sa.DateTime(), nullable=False, server_default=sa.text('now()')),
+        sa.Column('created_by', sa.String(), nullable=True),
+        sa.PrimaryKeyConstraint('id')
+    )
+
+    # Create indexes
+    op.create_index(op.f('ix_llm_models_value'), 'llm_models', ['value'], unique=True)
+    op.create_index(op.f('ix_llm_models_provider'), 'llm_models', ['provider'], unique=False)
+    op.create_index(op.f('ix_llm_models_enabled'), 'llm_models', ['enabled'], unique=False)
+
+
+def downgrade():
+    op.drop_index(op.f('ix_llm_models_enabled'), table_name='llm_models')
+    op.drop_index(op.f('ix_llm_models_provider'), table_name='llm_models')
+    op.drop_index(op.f('ix_llm_models_value'), table_name='llm_models')
+    op.drop_table('llm_models')

control_plane_api/alembic/versions/b0e10697f212_add_runtime_column_to_teams_simple.py

@@ -0,0 +1,42 @@
+"""add_runtime_column_to_teams_simple
+
+Revision ID: b0e10697f212
+Revises: 1382bec74309
+Create Date: 2025-11-06 11:48:06.515460
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = 'b0e10697f212'
+down_revision: Union[str, Sequence[str], None] = '1382bec74309'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema: Add runtime column to teams table."""
+    # Create the RuntimeType enum if it doesn't exist
+    op.execute("CREATE TYPE runtimetype AS ENUM ('default', 'claude_code')")
+
+    # Add runtime column to teams table
+    op.add_column('teams', sa.Column('runtime', sa.Enum('default', 'claude_code', name='runtimetype'), server_default='default', nullable=False))
+
+    # Create index on runtime column
+    op.create_index(op.f('ix_teams_runtime'), 'teams', ['runtime'], unique=False)
+
+
+def downgrade() -> None:
+    """Downgrade schema: Remove runtime column from teams table."""
+    # Drop index
+    op.drop_index(op.f('ix_teams_runtime'), table_name='teams')
+
+    # Drop column
+    op.drop_column('teams', 'runtime')
+
+    # Drop enum type
+    op.execute("DROP TYPE runtimetype")

control_plane_api/alembic/versions/ce43b24b63bf_add_execution_trigger_source_and_fix_.py

@@ -0,0 +1,155 @@
+"""add_execution_trigger_source_and_fix_types
+
+Revision ID: ce43b24b63bf
+Revises: 89e127caa47d
+Create Date: 2025-11-06 20:15:09.289858
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = 'ce43b24b63bf'
+down_revision: Union[str, Sequence[str], None] = '89e127caa47d'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # 1. Add trigger_source enum type
+    op.execute("""
+        CREATE TYPE executiontriggersource AS ENUM (
+            'user', 'job_cron', 'job_webhook', 'job_manual', 'system', 'api', 'chat'
+        )
+    """)
+
+    # 2. Add trigger_source and trigger_metadata columns to executions table
+    op.add_column('executions',
+        sa.Column('trigger_source', sa.Enum(
+            'user', 'job_cron', 'job_webhook', 'job_manual', 'system', 'api', 'chat',
+            name='executiontriggersource'
+        ), nullable=False, server_default='user')
+    )
+    op.add_column('executions',
+        sa.Column('trigger_metadata', sa.JSON(), server_default='{}', nullable=True)
+    )
+
+    # 3. Create index on trigger_source
+    op.create_index('ix_executions_trigger_source', 'executions', ['trigger_source'])
+
+    # 4. Fix jobs table column types (total_executions, successful_executions, failed_executions from JSON to INTEGER)
+    # First, convert any existing data
+    op.execute("UPDATE jobs SET total_executions = COALESCE((total_executions::text)::integer, 0) WHERE total_executions IS NOT NULL")
+    op.execute("UPDATE jobs SET successful_executions = COALESCE((successful_executions::text)::integer, 0) WHERE successful_executions IS NOT NULL")
+    op.execute("UPDATE jobs SET failed_executions = COALESCE((failed_executions::text)::integer, 0) WHERE failed_executions IS NOT NULL")
+
+    # Then alter column types
+    op.alter_column('jobs', 'total_executions',
+        type_=sa.Integer(),
+        existing_type=sa.JSON(),
+        server_default='0',
+        nullable=False
+    )
+    op.alter_column('jobs', 'successful_executions',
+        type_=sa.Integer(),
+        existing_type=sa.JSON(),
+        server_default='0',
+        nullable=False
+    )
+    op.alter_column('jobs', 'failed_executions',
+        type_=sa.Integer(),
+        existing_type=sa.JSON(),
+        server_default='0',
+        nullable=False
+    )
+
+    # 5. Fix job_executions.execution_duration_ms from JSON to INTEGER
+    op.execute("UPDATE job_executions SET execution_duration_ms = COALESCE((execution_duration_ms::text)::integer, NULL) WHERE execution_duration_ms IS NOT NULL")
+    op.alter_column('job_executions', 'execution_duration_ms',
+        type_=sa.Integer(),
+        existing_type=sa.JSON(),
+        nullable=True
+    )
+
+    # 6. Update last_execution_id in jobs to be UUID type instead of VARCHAR
+    # First drop the existing column (safe since we're still setting up)
+    op.drop_column('jobs', 'last_execution_id')
+
+    # Add it back with correct UUID type
+    op.add_column('jobs',
+        sa.Column('last_execution_id', sa.dialects.postgresql.UUID(as_uuid=True), nullable=True)
+    )
+
+    # 7. Update execution_id in job_executions to be UUID type
+    op.drop_column('job_executions', 'execution_id')
+    op.add_column('job_executions',
+        sa.Column('execution_id', sa.dialects.postgresql.UUID(as_uuid=True), nullable=False)
+    )
+    op.create_index('idx_job_executions_execution_id', 'job_executions', ['execution_id'])
+
+    # 8. Now add the foreign key constraints with proper types
+    op.create_foreign_key(
+        'fk_jobs_last_execution',
+        'jobs', 'executions',
+        ['last_execution_id'], ['id'],
+        ondelete='SET NULL'
+    )
+
+    op.create_foreign_key(
+        'fk_job_executions_execution',
+        'job_executions', 'executions',
+        ['execution_id'], ['id'],
+        ondelete='CASCADE'
+    )
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # Drop foreign keys
+    op.drop_constraint('fk_job_executions_execution', 'job_executions', type_='foreignkey')
+    op.drop_constraint('fk_jobs_last_execution', 'jobs', type_='foreignkey')
+
+    # Revert execution_id in job_executions back to VARCHAR
+    op.drop_index('idx_job_executions_execution_id', 'job_executions')
+    op.drop_column('job_executions', 'execution_id')
+    op.add_column('job_executions',
+        sa.Column('execution_id', sa.String(255), nullable=False)
+    )
+
+    # Revert last_execution_id in jobs back to VARCHAR
+    op.drop_column('jobs', 'last_execution_id')
+    op.add_column('jobs',
+        sa.Column('last_execution_id', sa.String(255), nullable=True)
+    )
+
+    # Revert execution_duration_ms back to JSON
+    op.alter_column('job_executions', 'execution_duration_ms',
+        type_=sa.JSON(),
+        existing_type=sa.Integer()
+    )
+
+    # Revert job execution counters back to JSON
+    op.alter_column('jobs', 'failed_executions',
+        type_=sa.JSON(),
+        existing_type=sa.Integer()
+    )
+    op.alter_column('jobs', 'successful_executions',
+        type_=sa.JSON(),
+        existing_type=sa.Integer()
+    )
+    op.alter_column('jobs', 'total_executions',
+        type_=sa.JSON(),
+        existing_type=sa.Integer()
+    )
+
+    # Drop trigger_source index and columns
+    op.drop_index('ix_executions_trigger_source', 'executions')
+    op.drop_column('executions', 'trigger_metadata')
+    op.drop_column('executions', 'trigger_source')
+
+    # Drop enum type
+    op.execute("DROP TYPE executiontriggersource")