diff --git a/internal-packages/run-engine/src/engine/systems/runAttemptSystem.ts b/internal-packages/run-engine/src/engine/systems/runAttemptSystem.ts index ce0f8abe4d..f3cd1e075b 100644 --- a/internal-packages/run-engine/src/engine/systems/runAttemptSystem.ts +++ b/internal-packages/run-engine/src/engine/systems/runAttemptSystem.ts @@ -188,6 +188,7 @@ export class RunAttemptSystem { runTags: true, isTest: true, idempotencyKey: true, + concurrencyKey: true, startedAt: true, maxAttempts: true, taskVersion: true, @@ -255,6 +256,7 @@ export class RunAttemptSystem { createdAt: run.createdAt, startedAt: run.startedAt ?? run.createdAt, idempotencyKey: run.idempotencyKey ?? undefined, + concurrencyKey: run.concurrencyKey ?? undefined, maxAttempts: run.maxAttempts ?? undefined, version: run.taskVersion ?? "unknown", maxDuration: run.maxDurationInSeconds ?? undefined, diff --git a/packages/core/src/v3/schemas/common.ts b/packages/core/src/v3/schemas/common.ts index 2928995606..68dd0c2b07 100644 --- a/packages/core/src/v3/schemas/common.ts +++ b/packages/core/src/v3/schemas/common.ts @@ -215,6 +215,7 @@ export const TaskRun = z.object({ createdAt: z.coerce.date(), startedAt: z.coerce.date().default(() => new Date()), idempotencyKey: z.string().optional(), + concurrencyKey: z.string().optional(), maxAttempts: z.number().optional(), version: z.string().optional(), metadata: z.record(DeserializedJsonSchema).optional(),