elderberry

Author: swyxio

internal-packages/run-engine/src/engine/systems/runAttemptSystem.ts

@@ -181,6 +181,7 @@ export class RunAttemptSystem {
         lockedById: true,
         lockedQueueId: true,
         queue: true,
+        concurrencyKey: true,
         attemptNumber: true,
         status: true,
         ttl: true,
@@ -248,6 +249,7 @@ export class RunAttemptSystem {
     ]);
 
     return {
+      concurrencyKey: run.concurrencyKey ?? undefined,
       run: {
         id: run.friendlyId,
         tags: run.runTags,
@@ -396,6 +398,7 @@ export class RunAttemptSystem {
                   lockedById: true,
                   lockedQueueId: true,
                   queue: true,
+                  concurrencyKey: true,
                   attemptNumber: true,
                   status: true,
                   ttl: true,
@@ -533,6 +536,7 @@ export class RunAttemptSystem {
             ]);
 
           const execution: BackwardsCompatibleTaskRunExecution = {
+            concurrencyKey: updatedRun.concurrencyKey ?? undefined,
             attempt: {
               number: nextAttemptNumber,
               startedAt: latestSnapshot.updatedAt,

packages/core/src/v3/schemas/common.ts

@@ -334,6 +334,8 @@ export const TaskRunExecution = z.object({
       traceContext: z.record(z.unknown()).optional(),
     })
   ),
+  // The concurrencyKey creates a copy of the queue per unique value
+  concurrencyKey: z.string().optional(),
   ...StaticTaskRunExecutionShape,
 });
 
@@ -406,6 +408,8 @@ export const TaskRunContext = z.object({
     durationMs: true,
     costInCents: true,
   }),
+  // The concurrencyKey used for this run, if any
+  concurrencyKey: z.string().optional(),
   ...StaticTaskRunExecutionShape,
 });