DROP INDEX IF EXISTS idx_generation_tasks_queued_stale;
DROP INDEX IF EXISTS idx_generation_tasks_running_lease;

ALTER TABLE generation_tasks
    DROP COLUMN IF EXISTS last_heartbeat_at,
    DROP COLUMN IF EXISTS attempt_count,
    DROP COLUMN IF EXISTS lease_expires_at,
    DROP COLUMN IF EXISTS lease_owner,
    DROP COLUMN IF EXISTS lease_token;