@trigger.dev/sdk 0.0.0-prerelease-20260302145933 → 0.0.0-prerelease-20260305142821

package/dist/commonjs/v3/chat.test.js

@@ -0,0 +1,1557 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const vitest_1 = require("vitest");
+const chat_js_1 = require("./chat.js");
+// Helper: encode text as SSE format
+function sseEncode(chunks) {
+    return chunks.map((chunk, i) => `id: ${i}\ndata: ${JSON.stringify(chunk)}\n\n`).join("");
+}
+// Helper: create a ReadableStream from SSE text
+function createSSEStream(sseText) {
+    const encoder = new TextEncoder();
+    return new ReadableStream({
+        start(controller) {
+            controller.enqueue(encoder.encode(sseText));
+            controller.close();
+        },
+    });
+}
+// Helper: create test UIMessages with unique IDs
+let messageIdCounter = 0;
+function createUserMessage(text) {
+    return {
+        id: `msg-user-${++messageIdCounter}`,
+        role: "user",
+        parts: [{ type: "text", text }],
+    };
+}
+function createAssistantMessage(text) {
+    return {
+        id: `msg-assistant-${++messageIdCounter}`,
+        role: "assistant",
+        parts: [{ type: "text", text }],
+    };
+}
+// Sample UIMessageChunks as the AI SDK would produce
+const sampleChunks = [
+    { type: "text-start", id: "part-1" },
+    { type: "text-delta", id: "part-1", delta: "Hello" },
+    { type: "text-delta", id: "part-1", delta: " world" },
+    { type: "text-delta", id: "part-1", delta: "!" },
+    { type: "text-end", id: "part-1" },
+];
+(0, vitest_1.describe)("TriggerChatTransport", () => {
+    let originalFetch;
+    (0, vitest_1.beforeEach)(() => {
+        originalFetch = global.fetch;
+    });
+    (0, vitest_1.afterEach)(() => {
+        global.fetch = originalFetch;
+        vitest_1.vi.restoreAllMocks();
+    });
+    (0, vitest_1.describe)("constructor", () => {
+        (0, vitest_1.it)("should create transport with required options", () => {
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-chat-task",
+                accessToken: "test-token",
+            });
+            (0, vitest_1.expect)(transport).toBeInstanceOf(chat_js_1.TriggerChatTransport);
+        });
+        (0, vitest_1.it)("should accept optional configuration", () => {
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-chat-task",
+                accessToken: "test-token",
+                baseURL: "https://custom.trigger.dev",
+                streamKey: "custom-stream",
+                headers: { "X-Custom": "value" },
+            });
+            (0, vitest_1.expect)(transport).toBeInstanceOf(chat_js_1.TriggerChatTransport);
+        });
+        (0, vitest_1.it)("should accept a function for accessToken", () => {
+            let tokenCallCount = 0;
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-chat-task",
+                accessToken: () => {
+                    tokenCallCount++;
+                    return `dynamic-token-${tokenCallCount}`;
+                },
+            });
+            (0, vitest_1.expect)(transport).toBeInstanceOf(chat_js_1.TriggerChatTransport);
+        });
+    });
+    (0, vitest_1.describe)("sendMessages", () => {
+        (0, vitest_1.it)("should trigger the task and return a ReadableStream of UIMessageChunks", async () => {
+            const triggerRunId = "run_abc123";
+            const publicToken = "pub_token_xyz";
+            // Mock fetch to handle both the trigger request and the SSE stream request
+            global.fetch = vitest_1.vi.fn().mockImplementation(async (url, init) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                // Handle the task trigger request
+                if (urlStr.includes("/api/v1/tasks/") && urlStr.includes("/trigger")) {
+                    return new Response(JSON.stringify({ id: triggerRunId }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": publicToken,
+                        },
+                    });
+                }
+                // Handle the SSE stream request
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    const sseText = sseEncode(sampleChunks);
+                    return new Response(createSSEStream(sseText), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-chat-task",
+                accessToken: "test-token",
+                baseURL: "https://api.test.trigger.dev",
+            });
+            const messages = [createUserMessage("Hello!")];
+            const stream = await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-1",
+                messageId: undefined,
+                messages,
+                abortSignal: undefined,
+            });
+            (0, vitest_1.expect)(stream).toBeInstanceOf(ReadableStream);
+            // Read all chunks from the stream
+            const reader = stream.getReader();
+            const receivedChunks = [];
+            while (true) {
+                const { done, value } = await reader.read();
+                if (done)
+                    break;
+                receivedChunks.push(value);
+            }
+            (0, vitest_1.expect)(receivedChunks).toHaveLength(sampleChunks.length);
+            (0, vitest_1.expect)(receivedChunks[0]).toEqual({ type: "text-start", id: "part-1" });
+            (0, vitest_1.expect)(receivedChunks[1]).toEqual({ type: "text-delta", id: "part-1", delta: "Hello" });
+            (0, vitest_1.expect)(receivedChunks[4]).toEqual({ type: "text-end", id: "part-1" });
+        });
+        (0, vitest_1.it)("should send the correct payload to the trigger API", async () => {
+            const fetchSpy = vitest_1.vi.fn().mockImplementation(async (url, init) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/api/v1/tasks/") && urlStr.includes("/trigger")) {
+                    return new Response(JSON.stringify({ id: "run_test" }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": "pub_token",
+                        },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    return new Response(createSSEStream(""), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            global.fetch = fetchSpy;
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-chat-task",
+                accessToken: "test-token",
+                baseURL: "https://api.test.trigger.dev",
+            });
+            const messages = [createUserMessage("Hello!")];
+            await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-123",
+                messageId: undefined,
+                messages,
+                abortSignal: undefined,
+                metadata: { custom: "data" },
+            });
+            // Verify the trigger fetch call
+            const triggerCall = fetchSpy.mock.calls.find((call) => (typeof call[0] === "string" ? call[0] : call[0].toString()).includes("/trigger"));
+            (0, vitest_1.expect)(triggerCall).toBeDefined();
+            const triggerUrl = typeof triggerCall[0] === "string" ? triggerCall[0] : triggerCall[0].toString();
+            (0, vitest_1.expect)(triggerUrl).toContain("/api/v1/tasks/my-chat-task/trigger");
+            const triggerBody = JSON.parse(triggerCall[1]?.body);
+            const payload = triggerBody.payload;
+            (0, vitest_1.expect)(payload.messages).toEqual(messages);
+            (0, vitest_1.expect)(payload.chatId).toBe("chat-123");
+            (0, vitest_1.expect)(payload.trigger).toBe("submit-message");
+            (0, vitest_1.expect)(payload.metadata).toEqual({ custom: "data" });
+        });
+        (0, vitest_1.it)("should use the correct stream URL with custom streamKey", async () => {
+            const fetchSpy = vitest_1.vi.fn().mockImplementation(async (url) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/trigger")) {
+                    return new Response(JSON.stringify({ id: "run_custom" }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": "token",
+                        },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    return new Response(createSSEStream(""), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            global.fetch = fetchSpy;
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: "token",
+                baseURL: "https://api.test.trigger.dev",
+                streamKey: "my-custom-stream",
+            });
+            await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-1",
+                messageId: undefined,
+                messages: [createUserMessage("test")],
+                abortSignal: undefined,
+            });
+            // Verify the stream URL uses the custom stream key
+            const streamCall = fetchSpy.mock.calls.find((call) => (typeof call[0] === "string" ? call[0] : call[0].toString()).includes("/realtime/v1/streams/"));
+            (0, vitest_1.expect)(streamCall).toBeDefined();
+            const streamUrl = typeof streamCall[0] === "string" ? streamCall[0] : streamCall[0].toString();
+            (0, vitest_1.expect)(streamUrl).toContain("/realtime/v1/streams/run_custom/my-custom-stream");
+        });
+        (0, vitest_1.it)("should include extra headers in stream requests", async () => {
+            const fetchSpy = vitest_1.vi.fn().mockImplementation(async (url) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/trigger")) {
+                    return new Response(JSON.stringify({ id: "run_hdrs" }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": "token",
+                        },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    return new Response(createSSEStream(""), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            global.fetch = fetchSpy;
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: "token",
+                baseURL: "https://api.test.trigger.dev",
+                headers: { "X-Custom-Header": "custom-value" },
+            });
+            await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-1",
+                messageId: undefined,
+                messages: [createUserMessage("test")],
+                abortSignal: undefined,
+            });
+            // Verify the stream request includes custom headers
+            const streamCall = fetchSpy.mock.calls.find((call) => (typeof call[0] === "string" ? call[0] : call[0].toString()).includes("/realtime/v1/streams/"));
+            (0, vitest_1.expect)(streamCall).toBeDefined();
+            const requestHeaders = streamCall[1]?.headers;
+            (0, vitest_1.expect)(requestHeaders["X-Custom-Header"]).toBe("custom-value");
+        });
+    });
+    (0, vitest_1.describe)("reconnectToStream", () => {
+        (0, vitest_1.it)("should return null when no session exists for chatId", async () => {
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: "token",
+            });
+            const result = await transport.reconnectToStream({
+                chatId: "nonexistent-chat",
+            });
+            (0, vitest_1.expect)(result).toBeNull();
+        });
+        (0, vitest_1.it)("should reconnect to an existing session", async () => {
+            const triggerRunId = "run_reconnect";
+            const publicToken = "pub_reconnect_token";
+            global.fetch = vitest_1.vi.fn().mockImplementation(async (url) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/trigger")) {
+                    return new Response(JSON.stringify({ id: triggerRunId }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": publicToken,
+                        },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    const chunks = [
+                        { type: "text-start", id: "part-1" },
+                        { type: "text-delta", id: "part-1", delta: "Reconnected!" },
+                        { type: "text-end", id: "part-1" },
+                    ];
+                    return new Response(createSSEStream(sseEncode(chunks)), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: "token",
+                baseURL: "https://api.test.trigger.dev",
+            });
+            // First, send messages to establish a session
+            await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-reconnect",
+                messageId: undefined,
+                messages: [createUserMessage("Hello")],
+                abortSignal: undefined,
+            });
+            // Now reconnect
+            const stream = await transport.reconnectToStream({
+                chatId: "chat-reconnect",
+            });
+            (0, vitest_1.expect)(stream).toBeInstanceOf(ReadableStream);
+            // Read the stream
+            const reader = stream.getReader();
+            const receivedChunks = [];
+            while (true) {
+                const { done, value } = await reader.read();
+                if (done)
+                    break;
+                receivedChunks.push(value);
+            }
+            (0, vitest_1.expect)(receivedChunks.length).toBeGreaterThan(0);
+        });
+    });
+    (0, vitest_1.describe)("createChatTransport", () => {
+        (0, vitest_1.it)("should create a TriggerChatTransport instance", () => {
+            const transport = (0, chat_js_1.createChatTransport)({
+                task: "my-task",
+                accessToken: "token",
+            });
+            (0, vitest_1.expect)(transport).toBeInstanceOf(chat_js_1.TriggerChatTransport);
+        });
+        (0, vitest_1.it)("should pass options through to the transport", () => {
+            const transport = (0, chat_js_1.createChatTransport)({
+                task: "custom-task",
+                accessToken: "custom-token",
+                baseURL: "https://custom.example.com",
+                streamKey: "custom-key",
+                headers: { "X-Test": "value" },
+            });
+            (0, vitest_1.expect)(transport).toBeInstanceOf(chat_js_1.TriggerChatTransport);
+        });
+    });
+    (0, vitest_1.describe)("publicAccessToken from trigger response", () => {
+        (0, vitest_1.it)("should use x-trigger-jwt from trigger response as the stream auth token", async () => {
+            const fetchSpy = vitest_1.vi.fn().mockImplementation(async (url, init) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/trigger")) {
+                    // Return with x-trigger-jwt header — this public token should be
+                    // used for the subsequent stream subscription request.
+                    return new Response(JSON.stringify({ id: "run_pat" }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": "server-generated-public-token",
+                        },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    // Verify the Authorization header uses the server-generated token
+                    const authHeader = init?.headers?.["Authorization"];
+                    (0, vitest_1.expect)(authHeader).toBe("Bearer server-generated-public-token");
+                    const chunks = [
+                        { type: "text-start", id: "p1" },
+                        { type: "text-end", id: "p1" },
+                    ];
+                    return new Response(createSSEStream(sseEncode(chunks)), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            global.fetch = fetchSpy;
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: "caller-token",
+                baseURL: "https://api.test.trigger.dev",
+            });
+            const stream = await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-pat",
+                messageId: undefined,
+                messages: [createUserMessage("test")],
+                abortSignal: undefined,
+            });
+            // Consume the stream
+            const reader = stream.getReader();
+            while (true) {
+                const { done } = await reader.read();
+                if (done)
+                    break;
+            }
+            // Verify the stream subscription used the public token, not the caller token
+            const streamCall = fetchSpy.mock.calls.find((call) => (typeof call[0] === "string" ? call[0] : call[0].toString()).includes("/realtime/v1/streams/"));
+            (0, vitest_1.expect)(streamCall).toBeDefined();
+            const streamHeaders = streamCall[1]?.headers;
+            (0, vitest_1.expect)(streamHeaders["Authorization"]).toBe("Bearer server-generated-public-token");
+        });
+    });
+    (0, vitest_1.describe)("error handling", () => {
+        (0, vitest_1.it)("should propagate trigger API errors", async () => {
+            global.fetch = vitest_1.vi.fn().mockImplementation(async (url) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/trigger")) {
+                    return new Response(JSON.stringify({ error: "Task not found" }), {
+                        status: 404,
+                        headers: { "content-type": "application/json" },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "nonexistent-task",
+                accessToken: "token",
+                baseURL: "https://api.test.trigger.dev",
+            });
+            await (0, vitest_1.expect)(transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-error",
+                messageId: undefined,
+                messages: [createUserMessage("test")],
+                abortSignal: undefined,
+            })).rejects.toThrow();
+        });
+    });
+    (0, vitest_1.describe)("abort signal", () => {
+        (0, vitest_1.it)("should close the stream gracefully when aborted", async () => {
+            let streamResolve;
+            const streamWait = new Promise((resolve) => {
+                streamResolve = resolve;
+            });
+            global.fetch = vitest_1.vi.fn().mockImplementation(async (url) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/trigger")) {
+                    return new Response(JSON.stringify({ id: "run_abort" }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": "token",
+                        },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    // Create a slow stream that waits before sending data
+                    const stream = new ReadableStream({
+                        async start(controller) {
+                            const encoder = new TextEncoder();
+                            controller.enqueue(encoder.encode(`id: 0\ndata: ${JSON.stringify({ type: "text-start", id: "p1" })}\n\n`));
+                            // Wait for the test to signal it's done
+                            await streamWait;
+                            controller.close();
+                        },
+                    });
+                    return new Response(stream, {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            const abortController = new AbortController();
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: "token",
+                baseURL: "https://api.test.trigger.dev",
+            });
+            const stream = await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-abort",
+                messageId: undefined,
+                messages: [createUserMessage("test")],
+                abortSignal: abortController.signal,
+            });
+            // Read the first chunk
+            const reader = stream.getReader();
+            const first = await reader.read();
+            (0, vitest_1.expect)(first.done).toBe(false);
+            // Abort and clean up
+            abortController.abort();
+            streamResolve?.();
+            // The stream should close — reading should return done
+            const next = await reader.read();
+            (0, vitest_1.expect)(next.done).toBe(true);
+        });
+    });
+    (0, vitest_1.describe)("multiple sessions", () => {
+        (0, vitest_1.it)("should track multiple chat sessions independently", async () => {
+            let callCount = 0;
+            const turnCompleteChunk = { type: "__trigger_turn_complete" };
+            global.fetch = vitest_1.vi.fn().mockImplementation(async (url) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/trigger")) {
+                    callCount++;
+                    return new Response(JSON.stringify({ id: `run_multi_${callCount}` }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": `token_${callCount}`,
+                        },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    // Include turn-complete chunk so the session is preserved
+                    const chunks = [...sampleChunks, turnCompleteChunk];
+                    return new Response(createSSEStream(sseEncode(chunks)), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: "token",
+                baseURL: "https://api.test.trigger.dev",
+            });
+            // Start two independent chat sessions and consume the streams
+            const s1 = await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "session-a",
+                messageId: undefined,
+                messages: [createUserMessage("Hello A")],
+                abortSignal: undefined,
+            });
+            const r1 = s1.getReader();
+            while (!(await r1.read()).done) { }
+            const s2 = await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "session-b",
+                messageId: undefined,
+                messages: [createUserMessage("Hello B")],
+                abortSignal: undefined,
+            });
+            const r2 = s2.getReader();
+            while (!(await r2.read()).done) { }
+            // Both sessions should be independently reconnectable
+            const streamA = await transport.reconnectToStream({ chatId: "session-a" });
+            const streamB = await transport.reconnectToStream({ chatId: "session-b" });
+            const streamC = await transport.reconnectToStream({ chatId: "nonexistent" });
+            (0, vitest_1.expect)(streamA).toBeInstanceOf(ReadableStream);
+            (0, vitest_1.expect)(streamB).toBeInstanceOf(ReadableStream);
+            (0, vitest_1.expect)(streamC).toBeNull();
+        });
+    });
+    (0, vitest_1.describe)("dynamic accessToken", () => {
+        (0, vitest_1.it)("should call the accessToken function for each sendMessages call", async () => {
+            let tokenCallCount = 0;
+            global.fetch = vitest_1.vi.fn().mockImplementation(async (url) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/trigger")) {
+                    return new Response(JSON.stringify({ id: `run_dyn_${tokenCallCount}` }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": "stream-token",
+                        },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    const chunks = [
+                        { type: "text-start", id: "p1" },
+                        { type: "text-end", id: "p1" },
+                    ];
+                    return new Response(createSSEStream(sseEncode(chunks)), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: () => {
+                    tokenCallCount++;
+                    return `dynamic-token-${tokenCallCount}`;
+                },
+                baseURL: "https://api.test.trigger.dev",
+            });
+            // First call — the token function should be invoked
+            await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-dyn-1",
+                messageId: undefined,
+                messages: [createUserMessage("first")],
+                abortSignal: undefined,
+            });
+            const firstCount = tokenCallCount;
+            (0, vitest_1.expect)(firstCount).toBeGreaterThanOrEqual(1);
+            // Second call — the token function should be invoked again
+            await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-dyn-2",
+                messageId: undefined,
+                messages: [createUserMessage("second")],
+                abortSignal: undefined,
+            });
+            // Token function was called at least once more
+            (0, vitest_1.expect)(tokenCallCount).toBeGreaterThan(firstCount);
+        });
+    });
+    (0, vitest_1.describe)("body merging", () => {
+        (0, vitest_1.it)("should merge ChatRequestOptions.body into the task payload", async () => {
+            const fetchSpy = vitest_1.vi.fn().mockImplementation(async (url) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/trigger")) {
+                    return new Response(JSON.stringify({ id: "run_body" }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": "token",
+                        },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    return new Response(createSSEStream(""), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            global.fetch = fetchSpy;
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: "token",
+                baseURL: "https://api.test.trigger.dev",
+            });
+            await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-body",
+                messageId: undefined,
+                messages: [createUserMessage("test")],
+                abortSignal: undefined,
+                body: { systemPrompt: "You are helpful", temperature: 0.7 },
+            });
+            const triggerCall = fetchSpy.mock.calls.find((call) => (typeof call[0] === "string" ? call[0] : call[0].toString()).includes("/trigger"));
+            const triggerBody = JSON.parse(triggerCall[1]?.body);
+            const payload = triggerBody.payload;
+            // body properties should be merged into the payload
+            (0, vitest_1.expect)(payload.systemPrompt).toBe("You are helpful");
+            (0, vitest_1.expect)(payload.temperature).toBe(0.7);
+            // Standard fields should still be present
+            (0, vitest_1.expect)(payload.chatId).toBe("chat-body");
+            (0, vitest_1.expect)(payload.trigger).toBe("submit-message");
+        });
+    });
+    (0, vitest_1.describe)("message types", () => {
+        (0, vitest_1.it)("should handle regenerate-message trigger", async () => {
+            const fetchSpy = vitest_1.vi.fn().mockImplementation(async (url) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/trigger")) {
+                    return new Response(JSON.stringify({ id: "run_regen" }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": "token",
+                        },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    return new Response(createSSEStream(""), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            global.fetch = fetchSpy;
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: "token",
+                baseURL: "https://api.test.trigger.dev",
+            });
+            const messages = [
+                createUserMessage("Hello!"),
+                createAssistantMessage("Hi there!"),
+            ];
+            await transport.sendMessages({
+                trigger: "regenerate-message",
+                chatId: "chat-regen",
+                messageId: "msg-to-regen",
+                messages,
+                abortSignal: undefined,
+            });
+            // Verify the payload includes the regenerate trigger type and messageId
+            const triggerCall = fetchSpy.mock.calls.find((call) => (typeof call[0] === "string" ? call[0] : call[0].toString()).includes("/trigger"));
+            const triggerBody = JSON.parse(triggerCall[1]?.body);
+            const payload = triggerBody.payload;
+            (0, vitest_1.expect)(payload.trigger).toBe("regenerate-message");
+            (0, vitest_1.expect)(payload.messageId).toBe("msg-to-regen");
+        });
+    });
+    (0, vitest_1.describe)("lastEventId tracking", () => {
+        (0, vitest_1.it)("should pass lastEventId to SSE subscription on subsequent turns", async () => {
+            const turnCompleteChunk = { type: "__trigger_turn_complete" };
+            let triggerCallCount = 0;
+            const streamFetchCalls = [];
+            global.fetch = vitest_1.vi.fn().mockImplementation(async (url, init) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/api/v1/tasks/") && urlStr.includes("/trigger")) {
+                    triggerCallCount++;
+                    return new Response(JSON.stringify({ id: "run_eid" }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": "pub_token_eid",
+                        },
+                    });
+                }
+                // Handle input stream sends (for second message)
+                if (urlStr.includes("/realtime/v1/streams/") && urlStr.includes("/input/")) {
+                    return new Response(JSON.stringify({ ok: true }), {
+                        status: 200,
+                        headers: { "content-type": "application/json" },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    streamFetchCalls.push({
+                        url: urlStr,
+                        headers: init?.headers ?? {},
+                    });
+                    const chunks = [
+                        ...sampleChunks,
+                        { type: "finish", id: "part-1" },
+                        turnCompleteChunk,
+                    ];
+                    return new Response(createSSEStream(sseEncode(chunks)), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: "token",
+                baseURL: "https://api.test.trigger.dev",
+            });
+            // First message — triggers a new run
+            const stream1 = await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-eid",
+                messageId: undefined,
+                messages: [createUserMessage("Hello")],
+                abortSignal: undefined,
+            });
+            const reader1 = stream1.getReader();
+            while (true) {
+                const { done } = await reader1.read();
+                if (done)
+                    break;
+            }
+            // Second message — sends via input stream
+            const stream2 = await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-eid",
+                messageId: undefined,
+                messages: [createUserMessage("Hello"), createAssistantMessage("Hi!"), createUserMessage("What's up?")],
+                abortSignal: undefined,
+            });
+            const reader2 = stream2.getReader();
+            while (true) {
+                const { done } = await reader2.read();
+                if (done)
+                    break;
+            }
+            // The second stream subscription should include a Last-Event-ID header
+            (0, vitest_1.expect)(streamFetchCalls.length).toBe(2);
+            const secondStreamHeaders = streamFetchCalls[1].headers;
+            // SSEStreamSubscription passes lastEventId as the Last-Event-ID header
+            (0, vitest_1.expect)(secondStreamHeaders["Last-Event-ID"]).toBeDefined();
+        });
+    });
+    (0, vitest_1.describe)("minimal wire payloads", () => {
+        (0, vitest_1.it)("should send only new messages via input stream on turn 2+", async () => {
+            const turnCompleteChunk = { type: "__trigger_turn_complete" };
+            const inputStreamPayloads = [];
+            global.fetch = vitest_1.vi.fn().mockImplementation(async (url, init) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/api/v1/tasks/") && urlStr.includes("/trigger")) {
+                    return new Response(JSON.stringify({ id: "run_minimal" }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": "pub_token_minimal",
+                        },
+                    });
+                }
+                // Capture input stream payloads (ApiClient wraps in { data: ... })
+                if (urlStr.includes("/realtime/v1/streams/") && urlStr.includes("/input/")) {
+                    const body = JSON.parse(init?.body);
+                    inputStreamPayloads.push(body.data);
+                    return new Response(JSON.stringify({ ok: true }), {
+                        status: 200,
+                        headers: { "content-type": "application/json" },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    const chunks = [
+                        ...sampleChunks,
+                        turnCompleteChunk,
+                    ];
+                    return new Response(createSSEStream(sseEncode(chunks)), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: "token",
+                baseURL: "https://api.test.trigger.dev",
+            });
+            const userMsg1 = createUserMessage("Hello");
+            const assistantMsg = createAssistantMessage("Hi there!");
+            const userMsg2 = createUserMessage("What's up?");
+            // Turn 1 — triggers a new run with full history
+            const stream1 = await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-minimal",
+                messageId: undefined,
+                messages: [userMsg1],
+                abortSignal: undefined,
+            });
+            const r1 = stream1.getReader();
+            while (!(await r1.read()).done) { }
+            // Turn 2 — sends via input stream, should only include NEW messages
+            const stream2 = await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-minimal",
+                messageId: undefined,
+                messages: [userMsg1, assistantMsg, userMsg2],
+                abortSignal: undefined,
+            });
+            const r2 = stream2.getReader();
+            while (!(await r2.read()).done) { }
+            // Verify: the input stream payload should only contain the new user message
+            (0, vitest_1.expect)(inputStreamPayloads).toHaveLength(1);
+            const sentPayload = inputStreamPayloads[0];
+            // Only the new user message should be sent (backend already has the assistant response)
+            (0, vitest_1.expect)(sentPayload.messages).toHaveLength(1);
+            (0, vitest_1.expect)(sentPayload.messages[0]).toEqual(userMsg2);
+        });
+        (0, vitest_1.it)("should send full history on first message (trigger)", async () => {
+            let triggerPayload;
+            global.fetch = vitest_1.vi.fn().mockImplementation(async (url, init) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/api/v1/tasks/") && urlStr.includes("/trigger")) {
+                    triggerPayload = JSON.parse(init?.body);
+                    return new Response(JSON.stringify({ id: "run_full" }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": "pub_token_full",
+                        },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    return new Response(createSSEStream(sseEncode(sampleChunks)), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: "token",
+                baseURL: "https://api.test.trigger.dev",
+            });
+            const messages = [createUserMessage("Hello"), createAssistantMessage("Hi!"), createUserMessage("More")];
+            await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-full",
+                messageId: undefined,
+                messages,
+                abortSignal: undefined,
+            });
+            // First message always sends full history via trigger
+            (0, vitest_1.expect)(triggerPayload.payload.messages).toHaveLength(3);
+        });
+    });
+    (0, vitest_1.describe)("AbortController cleanup", () => {
+        (0, vitest_1.it)("should terminate SSE connection after intercepting control chunk", async () => {
+            const controlChunk = { type: "__trigger_turn_complete" };
+            let streamAborted = false;
+            global.fetch = vitest_1.vi.fn().mockImplementation(async (url, init) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/trigger")) {
+                    return new Response(JSON.stringify({ id: "run_abort_cleanup" }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": "pub_token",
+                        },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    // Track abort signal
+                    const signal = init?.signal;
+                    if (signal) {
+                        signal.addEventListener("abort", () => {
+                            streamAborted = true;
+                        });
+                    }
+                    const chunks = [
+                        ...sampleChunks,
+                        { type: "finish", id: "part-1" },
+                        controlChunk,
+                    ];
+                    return new Response(createSSEStream(sseEncode(chunks)), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: "token",
+                baseURL: "https://api.test.trigger.dev",
+            });
+            const stream = await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-abort-cleanup",
+                messageId: undefined,
+                messages: [createUserMessage("Hello")],
+                abortSignal: undefined,
+            });
+            // Consume all chunks
+            const reader = stream.getReader();
+            while (true) {
+                const { done } = await reader.read();
+                if (done)
+                    break;
+            }
+            // The internal AbortController should have aborted the fetch
+            (0, vitest_1.expect)(streamAborted).toBe(true);
+        });
+    });
+    (0, vitest_1.describe)("async accessToken", () => {
+        (0, vitest_1.it)("should accept an async function for accessToken", async () => {
+            let tokenCallCount = 0;
+            global.fetch = vitest_1.vi.fn().mockImplementation(async (url) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/trigger")) {
+                    return new Response(JSON.stringify({ id: `run_async_${tokenCallCount}` }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": "stream-token",
+                        },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    const chunks = [
+                        { type: "text-start", id: "p1" },
+                        { type: "text-end", id: "p1" },
+                    ];
+                    return new Response(createSSEStream(sseEncode(chunks)), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: async () => {
+                    tokenCallCount++;
+                    // Simulate async work (e.g. server action)
+                    await new Promise((r) => setTimeout(r, 1));
+                    return `async-token-${tokenCallCount}`;
+                },
+                baseURL: "https://api.test.trigger.dev",
+            });
+            await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-async",
+                messageId: undefined,
+                messages: [createUserMessage("Hello")],
+                abortSignal: undefined,
+            });
+            (0, vitest_1.expect)(tokenCallCount).toBe(1);
+        });
+        (0, vitest_1.it)("should not resolve async token for input stream send flow", async () => {
+            const turnCompleteChunk = { type: "__trigger_turn_complete" };
+            let tokenCallCount = 0;
+            let inputStreamSendCalled = false;
+            global.fetch = vitest_1.vi.fn().mockImplementation(async (url) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/api/v1/tasks/") && urlStr.includes("/trigger")) {
+                    return new Response(JSON.stringify({ id: "run_async_wp" }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": "stream-token",
+                        },
+                    });
+                }
+                // Handle input stream sends
+                if (urlStr.includes("/realtime/v1/streams/") && urlStr.includes("/input/")) {
+                    inputStreamSendCalled = true;
+                    return new Response(JSON.stringify({ ok: true }), {
+                        status: 200,
+                        headers: { "content-type": "application/json" },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    const chunks = [
+                        ...sampleChunks,
+                        { type: "finish", id: "part-1" },
+                        turnCompleteChunk,
+                    ];
+                    return new Response(createSSEStream(sseEncode(chunks)), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: async () => {
+                    tokenCallCount++;
+                    await new Promise((r) => setTimeout(r, 1));
+                    return `async-wp-token-${tokenCallCount}`;
+                },
+                baseURL: "https://api.test.trigger.dev",
+            });
+            // First message — triggers a new run (calls async token)
+            const stream1 = await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-async-wp",
+                messageId: undefined,
+                messages: [createUserMessage("Hello")],
+                abortSignal: undefined,
+            });
+            const reader1 = stream1.getReader();
+            while (true) {
+                const { done } = await reader1.read();
+                if (done)
+                    break;
+            }
+            const firstTokenCount = tokenCallCount;
+            // Second message — should send via input stream (does NOT call async token)
+            const stream2 = await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-async-wp",
+                messageId: undefined,
+                messages: [createUserMessage("Hello"), createAssistantMessage("Hi!"), createUserMessage("More")],
+                abortSignal: undefined,
+            });
+            const reader2 = stream2.getReader();
+            while (true) {
+                const { done } = await reader2.read();
+                if (done)
+                    break;
+            }
+            // Token function should NOT have been called again for the input stream path
+            (0, vitest_1.expect)(tokenCallCount).toBe(firstTokenCount);
+            (0, vitest_1.expect)(inputStreamSendCalled).toBe(true);
+        });
+    });
+    (0, vitest_1.describe)("single-run mode (input stream loop)", () => {
+        (0, vitest_1.it)("should not forward turn-complete control chunk to consumer", async () => {
+            const turnCompleteChunk = { type: "__trigger_turn_complete" };
+            global.fetch = vitest_1.vi.fn().mockImplementation(async (url) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/trigger")) {
+                    return new Response(JSON.stringify({ id: "run_single" }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": "pub_token",
+                        },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    const chunks = [
+                        ...sampleChunks,
+                        { type: "finish", id: "part-1" },
+                        turnCompleteChunk,
+                    ];
+                    return new Response(createSSEStream(sseEncode(chunks)), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: "token",
+                baseURL: "https://api.test.trigger.dev",
+            });
+            const stream = await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-single",
+                messageId: undefined,
+                messages: [createUserMessage("Hello")],
+                abortSignal: undefined,
+            });
+            // Read all chunks — the control chunk should NOT appear
+            const reader = stream.getReader();
+            const receivedChunks = [];
+            while (true) {
+                const { done, value } = await reader.read();
+                if (done)
+                    break;
+                receivedChunks.push(value);
+            }
+            // All AI SDK chunks should be forwarded
+            (0, vitest_1.expect)(receivedChunks.length).toBe(sampleChunks.length + 1); // +1 for the finish chunk
+            // Control chunk should not be in the output
+            (0, vitest_1.expect)(receivedChunks.every((c) => c.type !== "__trigger_turn_complete")).toBe(true);
+        });
+        (0, vitest_1.it)("should send via input stream on second message instead of triggering a new run", async () => {
+            const turnCompleteChunk = { type: "__trigger_turn_complete" };
+            let triggerCallCount = 0;
+            let inputStreamSendCalled = false;
+            global.fetch = vitest_1.vi.fn().mockImplementation(async (url, init) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/api/v1/tasks/") && urlStr.includes("/trigger")) {
+                    triggerCallCount++;
+                    return new Response(JSON.stringify({ id: "run_resume" }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": "pub_token",
+                        },
+                    });
+                }
+                // Handle input stream sends
+                if (urlStr.includes("/realtime/v1/streams/") && urlStr.includes("/input/")) {
+                    inputStreamSendCalled = true;
+                    return new Response(JSON.stringify({ ok: true }), {
+                        status: 200,
+                        headers: { "content-type": "application/json" },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    const chunks = [
+                        ...sampleChunks,
+                        { type: "finish", id: "part-1" },
+                        turnCompleteChunk,
+                    ];
+                    return new Response(createSSEStream(sseEncode(chunks)), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: "token",
+                baseURL: "https://api.test.trigger.dev",
+            });
+            // First message — triggers a new run
+            const stream1 = await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-resume",
+                messageId: undefined,
+                messages: [createUserMessage("Hello")],
+                abortSignal: undefined,
+            });
+            // Consume stream
+            const reader1 = stream1.getReader();
+            while (true) {
+                const { done } = await reader1.read();
+                if (done)
+                    break;
+            }
+            (0, vitest_1.expect)(triggerCallCount).toBe(1);
+            // Second message — should send via input stream instead of triggering
+            const stream2 = await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-resume",
+                messageId: undefined,
+                messages: [createUserMessage("Hello"), createAssistantMessage("Hi!"), createUserMessage("How are you?")],
+                abortSignal: undefined,
+            });
+            // Consume second stream
+            const reader2 = stream2.getReader();
+            while (true) {
+                const { done } = await reader2.read();
+                if (done)
+                    break;
+            }
+            // Should NOT have triggered a second run
+            (0, vitest_1.expect)(triggerCallCount).toBe(1);
+            // Should have sent via input stream
+            (0, vitest_1.expect)(inputStreamSendCalled).toBe(true);
+        });
+        (0, vitest_1.it)("should fall back to triggering a new run if stream closes without control chunk", async () => {
+            let triggerCallCount = 0;
+            global.fetch = vitest_1.vi.fn().mockImplementation(async (url) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/api/v1/tasks/") && urlStr.includes("/trigger")) {
+                    triggerCallCount++;
+                    return new Response(JSON.stringify({ id: `run_fallback_${triggerCallCount}` }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": "pub_token",
+                        },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    // No control chunk — stream just ends after the finish
+                    const chunks = [
+                        { type: "text-start", id: "p1" },
+                        { type: "text-delta", id: "p1", delta: "Hello" },
+                        { type: "text-end", id: "p1" },
+                    ];
+                    return new Response(createSSEStream(sseEncode(chunks)), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: "token",
+                baseURL: "https://api.test.trigger.dev",
+            });
+            // First message
+            const stream1 = await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-fallback",
+                messageId: undefined,
+                messages: [createUserMessage("Hello")],
+                abortSignal: undefined,
+            });
+            const reader1 = stream1.getReader();
+            while (true) {
+                const { done } = await reader1.read();
+                if (done)
+                    break;
+            }
+            (0, vitest_1.expect)(triggerCallCount).toBe(1);
+            // Second message — no waitpoint token stored, should trigger a new run
+            await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-fallback",
+                messageId: undefined,
+                messages: [createUserMessage("Hello"), createAssistantMessage("Hi!"), createUserMessage("Again")],
+                abortSignal: undefined,
+            });
+            // Should have triggered a second run
+            (0, vitest_1.expect)(triggerCallCount).toBe(2);
+        });
+        (0, vitest_1.it)("should fall back to new run when sendInputStream fails", async () => {
+            const turnCompleteChunk = { type: "__trigger_turn_complete" };
+            let triggerCallCount = 0;
+            global.fetch = vitest_1.vi.fn().mockImplementation(async (url) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/api/v1/tasks/") && urlStr.includes("/trigger")) {
+                    triggerCallCount++;
+                    return new Response(JSON.stringify({ id: `run_fail_${triggerCallCount}` }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": "pub_token",
+                        },
+                    });
+                }
+                // Input stream send fails
+                if (urlStr.includes("/realtime/v1/streams/") && urlStr.includes("/input/")) {
+                    return new Response(JSON.stringify({ error: "Run not found" }), {
+                        status: 404,
+                        headers: { "content-type": "application/json" },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    const chunks = [
+                        ...sampleChunks,
+                        { type: "finish", id: "part-1" },
+                        turnCompleteChunk,
+                    ];
+                    return new Response(createSSEStream(sseEncode(chunks)), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: "token",
+                baseURL: "https://api.test.trigger.dev",
+            });
+            // First message
+            const stream1 = await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-fail",
+                messageId: undefined,
+                messages: [createUserMessage("Hello")],
+                abortSignal: undefined,
+            });
+            const reader1 = stream1.getReader();
+            while (true) {
+                const { done } = await reader1.read();
+                if (done)
+                    break;
+            }
+            (0, vitest_1.expect)(triggerCallCount).toBe(1);
+            // Second message — sendInputStream will fail, should fall back to new run
+            const stream2 = await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-fail",
+                messageId: undefined,
+                messages: [createUserMessage("Hello"), createAssistantMessage("Hi!"), createUserMessage("Again")],
+                abortSignal: undefined,
+            });
+            const reader2 = stream2.getReader();
+            while (true) {
+                const { done } = await reader2.read();
+                if (done)
+                    break;
+            }
+            // Should have triggered a second run as fallback
+            (0, vitest_1.expect)(triggerCallCount).toBe(2);
+        });
+    });
+    (0, vitest_1.describe)("onSessionChange", () => {
+        (0, vitest_1.it)("should fire when a new session is created", async () => {
+            const onSessionChange = vitest_1.vi.fn();
+            const triggerRunId = "run_session_new";
+            const publicToken = "pub_session_new";
+            global.fetch = vitest_1.vi.fn().mockImplementation(async (url) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/trigger")) {
+                    return new Response(JSON.stringify({ id: triggerRunId }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": publicToken,
+                        },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    const chunks = [
+                        ...sampleChunks,
+                        { type: "__trigger_turn_complete" },
+                    ];
+                    return new Response(createSSEStream(sseEncode(chunks)), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: "token",
+                baseURL: "https://api.test.trigger.dev",
+                onSessionChange,
+            });
+            const stream = await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-1",
+                messageId: undefined,
+                messages: [createUserMessage("Hello")],
+                abortSignal: undefined,
+            });
+            // Session created notification should have fired
+            (0, vitest_1.expect)(onSessionChange).toHaveBeenCalledWith("chat-1", {
+                runId: triggerRunId,
+                publicAccessToken: publicToken,
+                lastEventId: undefined,
+            });
+            // Consume stream
+            const reader = stream.getReader();
+            while (!(await reader.read()).done) { }
+            // Should also fire with updated lastEventId on turn complete
+            const lastCall = onSessionChange.mock.calls[onSessionChange.mock.calls.length - 1];
+            (0, vitest_1.expect)(lastCall[0]).toBe("chat-1");
+            (0, vitest_1.expect)(lastCall[1]).not.toBeNull();
+            (0, vitest_1.expect)(lastCall[1].lastEventId).toBeDefined();
+        });
+        (0, vitest_1.it)("should fire with null when session is deleted (stream ends naturally)", async () => {
+            const onSessionChange = vitest_1.vi.fn();
+            global.fetch = vitest_1.vi.fn().mockImplementation(async (url) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/trigger")) {
+                    return new Response(JSON.stringify({ id: "run_end" }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": "pub_end",
+                        },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    // No turn-complete chunk — stream ends naturally (run completed)
+                    return new Response(createSSEStream(sseEncode(sampleChunks)), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: "token",
+                baseURL: "https://api.test.trigger.dev",
+                onSessionChange,
+            });
+            const stream = await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-end",
+                messageId: undefined,
+                messages: [createUserMessage("Hello")],
+                abortSignal: undefined,
+            });
+            // Consume the stream fully
+            const reader = stream.getReader();
+            while (!(await reader.read()).done) { }
+            // Session should have been created then deleted
+            (0, vitest_1.expect)(onSessionChange).toHaveBeenCalledWith("chat-end", vitest_1.expect.objectContaining({
+                runId: "run_end",
+            }));
+            (0, vitest_1.expect)(onSessionChange).toHaveBeenCalledWith("chat-end", null);
+        });
+        (0, vitest_1.it)("should be updatable via setOnSessionChange", async () => {
+            const onSessionChange1 = vitest_1.vi.fn();
+            const onSessionChange2 = vitest_1.vi.fn();
+            global.fetch = vitest_1.vi.fn().mockImplementation(async (url) => {
+                const urlStr = typeof url === "string" ? url : url.toString();
+                if (urlStr.includes("/trigger")) {
+                    return new Response(JSON.stringify({ id: "run_update" }), {
+                        status: 200,
+                        headers: {
+                            "content-type": "application/json",
+                            "x-trigger-jwt": "pub_update",
+                        },
+                    });
+                }
+                if (urlStr.includes("/realtime/v1/streams/")) {
+                    const chunks = [
+                        ...sampleChunks,
+                        { type: "__trigger_turn_complete" },
+                    ];
+                    return new Response(createSSEStream(sseEncode(chunks)), {
+                        status: 200,
+                        headers: {
+                            "content-type": "text/event-stream",
+                            "X-Stream-Version": "v1",
+                        },
+                    });
+                }
+                throw new Error(`Unexpected fetch URL: ${urlStr}`);
+            });
+            const transport = new chat_js_1.TriggerChatTransport({
+                task: "my-task",
+                accessToken: "token",
+                baseURL: "https://api.test.trigger.dev",
+                onSessionChange: onSessionChange1,
+            });
+            // Update the callback before sending
+            transport.setOnSessionChange(onSessionChange2);
+            await transport.sendMessages({
+                trigger: "submit-message",
+                chatId: "chat-update",
+                messageId: undefined,
+                messages: [createUserMessage("Hello")],
+                abortSignal: undefined,
+            });
+            // Only onSessionChange2 should have been called
+            (0, vitest_1.expect)(onSessionChange1).not.toHaveBeenCalled();
+            (0, vitest_1.expect)(onSessionChange2).toHaveBeenCalled();
+        });
+    });
+});
+//# sourceMappingURL=chat.test.js.map