Use immutable run-state updates in tracking and cleanup paths

cursoragent · ericallam · cursoragent · commit fe5cd3d95812 · 2026-02-15T01:26:06.000Z
Co-authored-by: Eric Allam &lt;eric@trigger.dev&gt;
diff --git a/packages/ai/src/chatTransport.ts b/packages/ai/src/chatTransport.ts
@@ -337,8 +337,10 @@ export class TriggerChatTransport<
           return;
         }
 
-        runState.lastEventId = part.id;
-        await this.runStore.set(runState);
+        await this.runStore.set({
+          ...runState,
+          lastEventId: part.id,
+        });
       }
 
       const runState = await this.runStore.get(chatId);
@@ -380,8 +382,10 @@ export class TriggerChatTransport<
   }
 
   private async markRunInactiveAndDelete(runState: TriggerChatRunState) {
-    runState.isActive = false;
-    await this.runStore.set(runState);
+    await this.runStore.set({
+      ...runState,
+      isActive: false,
+    });
     await this.runStore.delete(runState.chatId);
   }
 

Original file line number	Diff line number	Diff line change
`@@ -337,8 +337,10 @@ export class TriggerChatTransport<`
`337`	`337`	`return;`
`338`	`338`	`}`
`339`	`339`
`340`		`- runState.lastEventId = part.id;`
`341`		`- await this.runStore.set(runState);`
	`340`	`+ await this.runStore.set({`
	`341`	`+ ...runState,`
	`342`	`+ lastEventId: part.id,`
	`343`	`+ });`
`342`	`344`	`}`
`343`	`345`
`344`	`346`	`const runState = await this.runStore.get(chatId);`
`@@ -380,8 +382,10 @@ export class TriggerChatTransport<`
`380`	`382`	`}`
`381`	`383`
`382`	`384`	`private async markRunInactiveAndDelete(runState: TriggerChatRunState) {`
`383`		`- runState.isActive = false;`
`384`		`- await this.runStore.set(runState);`
	`385`	`+ await this.runStore.set({`
	`386`	`+ ...runState,`
	`387`	`+ isActive: false,`
	`388`	`+ });`
`385`	`389`	`await this.runStore.delete(runState.chatId);`
`386`	`390`	`}`
`387`	`391`