Propagate the replayedFrom run ID in the replay flow

Author: myftija

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

@@ -264,6 +264,7 @@ export class RunEngineTriggerTaskService {
                   spanId: event.spanId,
                   parentSpanId:
                     options.parentAsLinkType === "replay" ? undefined : event.traceparent?.spanId,
+                  replayedFromTaskRunFriendlyId: options.replayedFromTaskRunFriendlyId,
                   lockedToVersionId: lockedToBackgroundWorker?.id,
                   taskVersion: lockedToBackgroundWorker?.version,
                   sdkVersion: lockedToBackgroundWorker?.sdkVersion,

apps/webapp/app/v3/services/replayTaskRun.server.ts

@@ -112,6 +112,7 @@ export class ReplayTaskRunService extends BaseService {
         {
           spanParentAsLink: true,
           parentAsLinkType: "replay",
+          replayedFromTaskRunFriendlyId: existingTaskRun.friendlyId,
           traceContext: {
             traceparent: `00-${existingTaskRun.traceId}-${existingTaskRun.spanId}-01`,
           },

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

@@ -33,6 +33,7 @@ export type TriggerTaskServiceOptions = {
   scheduleInstanceId?: string;
   queueTimestamp?: Date;
   overrideCreatedAt?: Date;
+  replayedFromTaskRunFriendlyId?: string;
 };
 
 export class OutOfEntitlementError extends Error {

apps/webapp/app/v3/services/triggerTaskV1.server.ts

@@ -427,6 +427,7 @@ export class TriggerTaskServiceV1 extends BaseService {
                       parentAttempt?.taskRun.id ??
                       dependentBatchRun?.dependentTaskAttempt?.taskRun.rootTaskRunId ??
                       dependentBatchRun?.dependentTaskAttempt?.taskRun.id,
+                    replayedFromTaskRunFriendlyId: options.replayedFromTaskRunFriendlyId,
                     batchId: dependentBatchRun?.id ?? parentBatchRun?.id,
                     resumeParentOnCompletion: !!(dependentAttempt ?? dependentBatchRun),
                     depth,

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

@@ -372,6 +372,7 @@ export class RunEngine {
       tags,
       parentTaskRunId,
       rootTaskRunId,
+      replayedFromTaskRunFriendlyId,
       batch,
       resumeParentOnCompletion,
       depth,
@@ -450,6 +451,7 @@ export class RunEngine {
               oneTimeUseToken,
               parentTaskRunId,
               rootTaskRunId,
+              replayedFromTaskRunFriendlyId,
               batchId: batch?.id,
               resumeParentOnCompletion,
               depth,

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

@@ -126,6 +126,7 @@ export type TriggerParams = {
   tags: { id: string; name: string }[];
   parentTaskRunId?: string;
   rootTaskRunId?: string;
+  replayedFromTaskRunFriendlyId?: string;
   batch?: {
     id: string;
     index: number;