Add some additional columns to raw_run_events_v1

Author: ericallam

internal-packages/clickhouse/schema/003_create_raw_run_events_v1.sql

@@ -18,6 +18,7 @@ CREATE TABLE trigger_dev.raw_run_events_v1
   /*  ─── ids & hierarchy ─────────────────────────────────────── */
   environment_id            String,
   run_id                    String,
+  friendly_id               String,
   attempt                   UInt8     DEFAULT 1,
 
   /*  ─── enums / status ──────────────────────────────────────── */
@@ -38,6 +39,16 @@ CREATE TABLE trigger_dev.raw_run_events_v1
   schedule_id               Nullable(String),
   batch_id                  Nullable(String),
 
+  /*  ─── related runs ─────────────────────────────────────────────── */
+  root_run_id               Nullable(String),
+  parent_run_id             Nullable(String),
+  depth                     UInt8 DEFAULT 0,
+
+  /*  ─── telemetry ─────────────────────────────────────────────── */
+  span_id                   Nullable(String),
+  trace_id                  Nullable(String),
+  idempotency_key           Nullable(String),
+
   /*  ─── timing ─────────────────────────────────────────────── */
   event_time         DateTime64(3),          -- when this row created
   created_at         DateTime64(3),
@@ -50,6 +61,7 @@ CREATE TABLE trigger_dev.raw_run_events_v1
   queued_at          Nullable(DateTime64(3)),
   expired_at         Nullable(DateTime64(3)),
   duration_ms        Nullable(UInt32),
+  expiration_ttl     Nullable(String),
 
   /*  ─── cost / usage ───────────────────────────────────────── */
   usage_duration_ms  UInt32  DEFAULT 0,
@@ -66,6 +78,8 @@ CREATE TABLE trigger_dev.raw_run_events_v1
   sdk_version        Nullable(String) CODEC(LZ4),
   cli_version        Nullable(String) CODEC(LZ4),
   machine_preset     LowCardinality(Nullable(String)) CODEC(LZ4),
+
+  is_test            Nullable(UInt8) DEFAULT 0,
 )
 ENGINE = MergeTree
 PARTITION BY toYYYYMM(event_time)

internal-packages/clickhouse/src/runEvents.test.ts

@@ -18,6 +18,7 @@ describe("Run Events", () => {
       {
         environment_id: "env_1234",
         run_id: "run_1234",
+        friendly_id: "friendly_1234",
         attempt: 1,
         engine: "V2",
         status: "PENDING",
@@ -49,6 +50,14 @@ describe("Run Events", () => {
         sdk_version: "1.0.0",
         cli_version: "1.0.0",
         machine_preset: "small-1x",
+        is_test: true,
+        span_id: "span_1234",
+        trace_id: "trace_1234",
+        idempotency_key: "idempotency_key_1234",
+        expiration_ttl: "1h",
+        root_run_id: "root_run_1234",
+        parent_run_id: "parent_run_1234",
+        depth: 1,
       },
     ]);
 

internal-packages/clickhouse/src/runEvents.ts

@@ -6,6 +6,7 @@ import { TaskRunError } from "@trigger.dev/core/v3/schemas";
 export const RawRunEventV1 = z.object({
   environment_id: z.string(),
   run_id: z.string(),
+  friendly_id: z.string(),
   attempt: z.number().int().default(1),
   engine: z.enum(["V1", "V2"]),
   status: z.enum([
@@ -29,19 +30,19 @@ export const RawRunEventV1 = z.object({
   queue: z.string(),
   schedule_id: z.string().optional(),
   batch_id: z.string().optional(),
-  event_time: z.coerce.number().int(),
-  created_at: z.coerce.number().int(),
-  updated_at: z.coerce.number().int(),
-  completed_at: z.coerce.number().int().optional(),
-  started_at: z.coerce.number().int().optional(),
-  executed_at: z.coerce.number().int().optional(),
-  finished_at: z.coerce.number().int().optional(),
-  delay_until: z.coerce.number().int().optional(),
-  queued_at: z.coerce.number().int().optional(),
-  expired_at: z.coerce.number().int().optional(),
-  duration_ms: z.coerce.number().int().optional(),
-  usage_duration_ms: z.coerce.number().int().optional(),
-  cost_in_cents: z.coerce.number().int().optional(),
+  event_time: z.number().int(),
+  created_at: z.number().int(),
+  updated_at: z.number().int(),
+  completed_at: z.number().int().optional(),
+  started_at: z.number().int().optional(),
+  executed_at: z.number().int().optional(),
+  finished_at: z.number().int().optional(),
+  delay_until: z.number().int().optional(),
+  queued_at: z.number().int().optional(),
+  expired_at: z.number().int().optional(),
+  duration_ms: z.number().int().optional(),
+  usage_duration_ms: z.number().int().optional(),
+  cost_in_cents: z.number().int().optional(),
   payload: z.unknown().optional(),
   output: z.unknown().optional(),
   error: TaskRunError.optional(),
@@ -50,6 +51,14 @@ export const RawRunEventV1 = z.object({
   sdk_version: z.string().optional(),
   cli_version: z.string().optional(),
   machine_preset: z.string().optional(),
+  root_run_id: z.string().optional(),
+  parent_run_id: z.string().optional(),
+  depth: z.number().int().default(0),
+  span_id: z.string().optional(),
+  trace_id: z.string().optional(),
+  idempotency_key: z.string().optional(),
+  expiration_ttl: z.string().optional(),
+  is_test: z.boolean().default(false),
 });
 
 export type RawRunEventV1 = z.infer<typeof RawRunEventV1>;