Refactor run-state cleanup into shared transport helper

Co-authored-by: Eric Allam <eric@trigger.dev>

Author: cursoragent

packages/ai/src/chatTransport.ts

@@ -235,9 +235,7 @@ export class TriggerChatTransport<
     try {
       stream = await this.fetchRunStream(runState, options.abortSignal);
     } catch (error) {
-      runState.isActive = false;
-      await this.runStore.set(runState);
-      await this.runStore.delete(runState.chatId);
+      await this.markRunInactiveAndDelete(runState);
       await this.reportError({
         phase: "streamSubscribe",
         chatId: runState.chatId,
@@ -268,9 +266,7 @@ export class TriggerChatTransport<
     try {
       stream = await this.fetchRunStream(runState, undefined, runState.lastEventId);
     } catch (error) {
-      runState.isActive = false;
-      await this.runStore.set(runState);
-      await this.runStore.delete(options.chatId);
+      await this.markRunInactiveAndDelete(runState);
       await this.reportError({
         phase: "reconnect",
         chatId: runState.chatId,
@@ -347,16 +343,12 @@ export class TriggerChatTransport<
 
       const runState = await this.runStore.get(chatId);
       if (runState) {
-        runState.isActive = false;
-        await this.runStore.set(runState);
-        await this.runStore.delete(chatId);
+        await this.markRunInactiveAndDelete(runState);
       }
     } catch (error) {
       const runState = await this.runStore.get(chatId);
       if (runState) {
-        runState.isActive = false;
-        await this.runStore.set(runState);
-        await this.runStore.delete(chatId);
+        await this.markRunInactiveAndDelete(runState);
         await this.reportError({
           phase: "consumeTrackingStream",
           chatId: runState.chatId,
@@ -387,6 +379,12 @@ export class TriggerChatTransport<
     return `${normalizedBaseUrl}/realtime/v1/streams/${encodedRunId}/${encodedStreamKey}`;
   }
 
+  private async markRunInactiveAndDelete(runState: TriggerChatRunState) {
+    runState.isActive = false;
+    await this.runStore.set(runState);
+    await this.runStore.delete(runState.chatId);
+  }
+
   private async reportError(event: TriggerChatTransportError) {
     if (!this.onError) {
       return;