Better types for the traceContext

Author: ericallam

apps/webapp/app/runEngine/services/triggerTask.server.ts

@@ -356,7 +356,7 @@ export class RunEngineTriggerTaskService {
     traceContext: Record<string, unknown>,
     parentRunTraceContext: unknown,
     parentSpanId: string | undefined
-  ): Record<string, unknown> {
+  ): TriggerTraceContext {
     if (!parentRunTraceContext) {
       return traceContext;
     }

internal-packages/run-engine/src/engine/index.ts

@@ -385,7 +385,7 @@ export class RunEngine {
               payload,
               payloadType,
               context,
-              traceContext: traceContext as any,
+              traceContext,
               traceId,
               spanId,
               parentSpanId,

internal-packages/run-engine/src/engine/types.ts

@@ -6,6 +6,7 @@ import {
   MachinePresetName,
   RetryOptions,
   RunChainState,
+  TriggerTraceContext,
 } from "@trigger.dev/core/v3";
 import { PrismaClient, PrismaReplicaClient } from "@trigger.dev/database";
 import { Worker, type WorkerConcurrencyOptions } from "@trigger.dev/redis-worker";
@@ -89,7 +90,7 @@ export type TriggerParams = {
   payload: string;
   payloadType: string;
   context: any;
-  traceContext: Record<string, unknown>;
+  traceContext: TriggerTraceContext;
   traceId: string;
   spanId: string;
   parentSpanId?: string;