Cover consumeTracking path when cleanup set and delete both fail

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

Author: cursoragent

packages/ai/src/chatTransport.test.ts

@@ -2232,6 +2232,70 @@ describe("TriggerChatTransport", function () {
     expect(errors[0]?.error.message).toBe("tracking failed root cause");
   });
 
+  it(
+    "attempts consumeTracking cleanup set and delete when both cleanup steps throw",
+    async function () {
+      const errors: TriggerChatTransportError[] = [];
+      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_both_failure",
+          });
+          res.end(JSON.stringify({ id: "run_tracking_cleanup_both_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: function onError(error) {
+          errors.push(error);
+        },
+      });
+
+      (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-both-failure",
+        messageId: undefined,
+        messages: [],
+        abortSignal: undefined,
+      });
+
+      await expect(readChunks(stream)).rejects.toThrowError("tracking failed root cause");
+
+      await waitForCondition(function () {
+        return errors.length === 1;
+      });
+
+      expect(errors[0]).toMatchObject({
+        phase: "consumeTrackingStream",
+        chatId: "chat-tracking-cleanup-both-failure",
+        runId: "run_tracking_cleanup_both_failure",
+      });
+      expect(errors[0]?.error.message).toBe("tracking failed root cause");
+      expect(runStore.setCalls).toContain("chat-tracking-cleanup-both-failure");
+      expect(runStore.deleteCalls).toContain("chat-tracking-cleanup-both-failure");
+    }
+  );
+
   it("reports reconnect failures through onError", async function () {
     const errors: TriggerChatTransportError[] = [];
     const runStore = new InMemoryTriggerChatRunStore();