Cover consumeTracking behavior when cleanup and onError all fail

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

Author: cursoragent

packages/ai/src/chatTransport.test.ts

@@ -2474,6 +2474,58 @@ describe("TriggerChatTransport", function () {
     }
   );
 
+  it(
+    "preserves consumeTracking root failures when cleanup set/delete and onError callbacks all fail",
+    async function () {
+      const runStore = new FailingCleanupSetAndDeleteRunStore();
+
+      const server = await startServer(function (req, res) {
+        if (req.method === "POST" && req.url === "/api/v1/tasks/chat-task/trigger") {
+          res.writeHead(200, {
+            "content-type": "application/json",
+            "x-trigger-jwt": "pk_run_tracking_cleanup_all_failure",
+          });
+          res.end(JSON.stringify({ id: "run_tracking_cleanup_all_failure" }));
+          return;
+        }
+
+        res.writeHead(404);
+        res.end();
+      });
+
+      const transport = new TriggerChatTransport({
+        task: "chat-task",
+        stream: "chat-stream",
+        accessToken: "pk_trigger",
+        baseURL: server.url,
+        runStore,
+        onError: async function onError() {
+          throw new Error("onError failed");
+        },
+      });
+
+      (transport as any).fetchRunStream = async function fetchRunStream() {
+        return new ReadableStream({
+          start(controller) {
+            controller.error(new Error("tracking failed root cause"));
+          },
+        });
+      };
+
+      const stream = await transport.sendMessages({
+        trigger: "submit-message",
+        chatId: "chat-tracking-cleanup-all-failure",
+        messageId: undefined,
+        messages: [],
+        abortSignal: undefined,
+      });
+
+      await expect(readChunks(stream)).rejects.toThrowError("tracking failed root cause");
+      expect(runStore.setCalls).toContain("chat-tracking-cleanup-all-failure");
+      expect(runStore.deleteCalls).toContain("chat-tracking-cleanup-all-failure");
+    }
+  );
+
   it("reports reconnect failures through onError", async function () {
     const errors: TriggerChatTransportError[] = [];
     const runStore = new InMemoryTriggerChatRunStore();