@trigger.dev/sdk 0.0.0-python-preview-20250226140121 → 0.0.0-rc-4-8-0-20261021151928

package/dist/commonjs/v3/steeringContext.js

@@ -0,0 +1,118 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.retainStepMessages = retainStepMessages;
+exports.steeringMarkers = steeringMarkers;
+exports.convertSteeredMessages = convertSteeredMessages;
+const ai_shared_js_1 = require("./ai-shared.js");
+/**
+ * AI SDK 5/6 rebuild input from the original prompt on every step. Retain an
+ * override and append only the newly completed response messages. The same
+ * logic works when the installed AI SDK already retains overrides: its input
+ * is not concatenated again. State belongs to one generation, not one turn.
+ */
+function retainStepMessages(prepare) {
+    let retained;
+    let responseCount = 0;
+    let previousSteps = -1;
+    return async (input) => {
+        if (input.steps.length <= previousSteps) {
+            retained = undefined;
+            responseCount = 0;
+        }
+        const responses = input.steps.at(-1)?.response.messages ?? [];
+        const messages = retained ? [...retained, ...responses.slice(responseCount)] : input.messages;
+        const result = await prepare({ ...input, messages });
+        retained = result?.messages ?? (retained ? messages : undefined);
+        responseCount = responses.length;
+        previousSteps = input.steps.length;
+        return retained ? { ...result, messages: retained } : result;
+    };
+}
+function steeringMarkers(messages) {
+    return messages.flatMap((message) => message.parts.flatMap((part) => {
+        if (part.type !== ai_shared_js_1.PENDING_MESSAGE_INJECTED_TYPE)
+            return [];
+        const p = part;
+        return typeof p.id === "string" && Array.isArray(p.data?.messageIds)
+            ? [{ id: p.id, messageIds: p.data.messageIds, messages: p.data.messages ?? [] }]
+            : [];
+    }));
+}
+/**
+ * Project an inline UI event into a real model-message boundary. The public
+ * assistant keeps its identity; the model sees complete preceding tool steps,
+ * then the exact prepared input, then the subsequent assistant steps.
+ */
+async function convertSteeredMessages(messages, convert, injections, context = messages, requireInputPresence = false) {
+    const markers = steeringMarkers([...context, ...messages]);
+    if (markers.length === 0 && injections.size === 0)
+        return convert(messages);
+    const inlineIds = new Set(markers.flatMap((marker) => marker.messageIds));
+    const uiById = new Map([...context, ...messages].map((message) => [message.id, message]));
+    const out = [];
+    const standalone = new Map([...injections.values()].flatMap((entry) => entry.messageIds.map((id) => [id, entry])));
+    const emitted = new Set();
+    let batch = [];
+    const flush = async () => {
+        if (batch.length)
+            out.push(...(await convert(batch)));
+        batch = [];
+    };
+    for (const message of messages) {
+        if (inlineIds.has(message.id))
+            continue;
+        const preparedInput = standalone.get(message.id);
+        if (preparedInput) {
+            await flush();
+            if (!emitted.has(preparedInput.id))
+                out.push(...preparedInput.messages);
+            emitted.add(preparedInput.id);
+            continue;
+        }
+        if (message.role !== "assistant") {
+            batch.push(message);
+            continue;
+        }
+        let parts = [];
+        for (const part of message.parts) {
+            const marker = steeringMarkers([{ ...message, parts: [part] }])[0];
+            if (!marker) {
+                parts.push(part);
+                continue;
+            }
+            if (parts.length)
+                batch.push({ ...message, parts });
+            parts = [];
+            await flush();
+            // An explicit history edit removing input is authoritative. The UI
+            // marker can still describe what happened, without resurrecting it in
+            // future model context from its fallback text or private prepared form.
+            const prepared = injections.get(marker.id);
+            if (requireInputPresence &&
+                !marker.messageIds.every((id) => uiById.has(id) || prepared?.trimmedMessageIds?.includes(id)))
+                continue;
+            if (prepared)
+                out.push(...prepared.messages);
+            else {
+                // Old/custom stores may have only the UI transcript. Recover ordinary
+                // user input, never expose the private model forms in the marker.
+                const users = marker.messageIds.flatMap((id) => {
+                    const ui = uiById.get(id);
+                    if (ui)
+                        return [ui];
+                    const text = marker.messages.find((m) => m.id === id)?.text;
+                    return text === undefined
+                        ? []
+                        : [{ id, role: "user", parts: [{ type: "text", text }] }];
+                });
+                if (users.length)
+                    out.push(...(await convert(users)));
+            }
+        }
+        if (parts.length)
+            batch.push({ ...message, parts });
+    }
+    await flush();
+    return out;
+}
+//# sourceMappingURL=steeringContext.js.map

package/dist/commonjs/v3/steeringContext.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"steeringContext.js","sourceRoot":"","sources":["../../../src/v3/steeringContext.ts"],"names":[],"mappings":";;;;;AACA,iDAA+D;AAgB/D;;;;;GAKG;AACH,4BACE,OAA6C;IAE7C,IAAI,QAAoC,CAAC;IACzC,IAAI,aAAa,GAAG,CAAC,CAAC;IACtB,IAAI,aAAa,GAAG,CAAC,CAAC,CAAC;IACvB,OAAO,KAAK,EAAE,KAAK,EAAE,EAAE;QACrB,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,IAAI,aAAa,EAAE,CAAC;YACxC,QAAQ,GAAG,SAAS,CAAC;YACrB,aAAa,GAAG,CAAC,CAAC;QACpB,CAAC;QACD,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,QAAQ,IAAI,EAAE,CAAC;QAC9D,MAAM,QAAQ,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,GAAG,QAAQ,EAAE,GAAG,SAAS,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC;QAC9F,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;QACrD,QAAQ,GAAG,MAAM,EAAE,QAAQ,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QACjE,aAAa,GAAG,SAAS,CAAC,MAAM,CAAC;QACjC,aAAa,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC;QACnC,OAAO,QAAQ,CAAC,CAAC,CAAE,EAAE,GAAG,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAQ,CAAC,CAAC,CAAC,MAAM,CAAC;IACtE,CAAC,CAAC;AACJ,CAAC;AAED,yBAAgC,QAA8B;IAC5D,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAClC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QAC7B,IAAI,IAAI,CAAC,IAAI,KAAK,4CAA6B;YAAE,OAAO,EAAE,CAAC;QAC3D,MAAM,CAAC,GAAG,IAGT,CAAC;QACF,OAAO,OAAO,CAAC,CAAC,EAAE,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,EAAE,UAAU,CAAC;YAClE,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,EAAE,CAAC;YAChF,CAAC,CAAC,EAAE,CAAC;IACT,CAAC,CAAC,CACH,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACI,KAAK,iCACV,QAAqB,EACrB,OAA2D,EAC3D,UAAkD,EAClD,OAAO,GAAyB,QAAQ,EACxC,oBAAoB,GAAG,KAAK;IAE5B,MAAM,OAAO,GAAG,eAAe,CAAC,CAAC,GAAG,OAAO,EAAE,GAAG,QAAQ,CAAC,CAAC,CAAC;IAC3D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,IAAI,UAAU,CAAC,IAAI,KAAK,CAAC;QAAE,OAAO,OAAO,CAAC,QAAQ,CAAC,CAAC;IAC5E,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC;IAC1E,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,OAAO,EAAE,GAAG,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;IAC1F,MAAM,GAAG,GAAmB,EAAE,CAAC;IAC/B,MAAM,UAAU,GAAG,IAAI,GAAG,CACxB,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,KAAK,CAAU,CAAC,CAAC,CAChG,CAAC;IACF,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;IAClC,IAAI,KAAK,GAAgB,EAAE,CAAC;IAC5B,MAAM,KAAK,GAAG,KAAK,IAAI,EAAE;QACvB,IAAI,KAAK,CAAC,MAAM;YAAE,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACtD,KAAK,GAAG,EAAE,CAAC;IACb,CAAC,CAAC;IACF,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;YAAE,SAAS;QACxC,MAAM,aAAa,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACjD,IAAI,aAAa,EAAE,CAAC;YAClB,MAAM,KAAK,EAAE,CAAC;YACd,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC;gBAAE,GAAG,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;YACxE,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;YAC9B,SAAS;QACX,CAAC;QACD,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACjC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACpB,SAAS;QACX,CAAC;QACD,IAAI,KAAK,GAAuB,EAAE,CAAC;QACnC,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YACjC,MAAM,MAAM,GAAG,eAAe,CAAC,CAAC,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACnE,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACjB,SAAS;YACX,CAAC;YACD,IAAI,KAAK,CAAC,MAAM;gBAAE,KAAK,CAAC,IAAI,CAAC,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;YACpD,KAAK,GAAG,EAAE,CAAC;YACX,MAAM,KAAK,EAAE,CAAC;YACd,mEAAmE;YACnE,sEAAsE;YACtE,wEAAwE;YACxE,MAAM,QAAQ,GAAG,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAC3C,IACE,oBAAoB;gBACpB,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CACtB,CAAC,EAAE,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,QAAQ,EAAE,iBAAiB,EAAE,QAAQ,CAAC,EAAE,CAAC,CACpE;gBAED,SAAS;YACX,IAAI,QAAQ;gBAAE,GAAG,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;iBACxC,CAAC;gBACJ,sEAAsE;gBACtE,kEAAkE;gBAClE,MAAM,KAAK,GAAG,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE;oBAC7C,MAAM,EAAE,GAAG,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;oBAC1B,IAAI,EAAE;wBAAE,OAAO,CAAC,EAAE,CAAC,CAAC;oBACpB,MAAM,IAAI,GAAG,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,CAAC;oBAC5D,OAAO,IAAI,KAAK,SAAS;wBACvB,CAAC,CAAC,EAAE;wBACJ,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAe,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;gBAChF,CAAC,CAAC,CAAC;gBACH,IAAI,KAAK,CAAC,MAAM;oBAAE,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YACxD,CAAC;QACH,CAAC;QACD,IAAI,KAAK,CAAC,MAAM;YAAE,KAAK,CAAC,IAAI,CAAC,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;IACtD,CAAC;IACD,MAAM,KAAK,EAAE,CAAC;IACd,OAAO,GAAG,CAAC;AACb,CAAC"}

package/dist/commonjs/v3/streams.d.ts

@@ -0,0 +1,298 @@
+import { type AsyncIterableStream, type WriterStreamOptions, type PipeStreamOptions, type PipeStreamResult, type ReadStreamOptions, type AppendStreamOptions, type RealtimeDefinedStream, type InferStreamType, type RealtimeDefinedInputStream, type InferInputStreamType } from "@trigger.dev/core/v3";
+/**
+ * Pipes data to a realtime stream using the default stream key (`"default"`).
+ *
+ * This is a convenience overload that allows you to pipe data without specifying a stream key.
+ * The stream will be created/accessed with the key `"default"`.
+ *
+ * @template T - The type of data chunks in the stream
+ * @param value - The stream of data to pipe from. Can be an `AsyncIterable<T>` or `ReadableStream<T>`.
+ * @param options - Optional configuration for the stream operation
+ * @returns A promise that resolves to an object containing:
+ *   - `stream`: The original stream (can be consumed in your task)
+ *   - `waitUntilComplete`: A function that returns a promise resolving when the stream is fully sent
+ *
+ * @example
+ * ```ts
+ * import { streams } from "@trigger.dev/sdk";
+ *
+ * // Stream OpenAI completion chunks to the default stream
+ * const completion = await openai.chat.completions.create({
+ *   model: "gpt-4",
+ *   messages: [{ role: "user", content: "Hello" }],
+ *   stream: true,
+ * });
+ *
+ * const { waitUntilComplete } = await streams.pipe(completion);
+ *
+ * // Process the stream locally
+ * for await (const chunk of completion) {
+ *   console.log(chunk);
+ * }
+ *
+ * // Or alternatievely wait for all chunks to be sent to the realtime stream
+ * await waitUntilComplete();
+ * ```
+ */
+declare function pipe<T>(value: AsyncIterable<T> | ReadableStream<T>, options?: PipeStreamOptions): PipeStreamResult<T>;
+/**
+ * Pipes data to a realtime stream with a specific stream key.
+ *
+ * Use this overload when you want to use a custom stream key instead of the default.
+ *
+ * @template T - The type of data chunks in the stream
+ * @param key - The unique identifier for this stream. If multiple streams use the same key,
+ *   they will be merged into a single stream. Defaults to `"default"` if not provided.
+ * @param value - The stream of data to pipe from. Can be an `AsyncIterable<T>` or `ReadableStream<T>`.
+ * @param options - Optional configuration for the stream operation
+ * @returns A promise that resolves to an object containing:
+ *   - `stream`: The original stream (can be consumed in your task)
+ *   - `waitUntilComplete`: A function that returns a promise resolving when the stream is fully sent
+ *
+ * @example
+ * ```ts
+ * import { streams } from "@trigger.dev/sdk";
+ *
+ * // Stream data to a specific stream key
+ * const myStream = createAsyncGenerator();
+ * const { waitUntilComplete } = await streams.pipe("my-custom-stream", myStream);
+ *
+ * // Process the stream locally
+ * for await (const chunk of myStream) {
+ *   console.log(chunk);
+ * }
+ *
+ * // Wait for all chunks to be sent
+ * await waitUntilComplete();
+ * ```
+ *
+ * @example
+ * ```ts
+ * // Stream to a parent run
+ * await streams.pipe("output", myStream, {
+ *   target: "parent",
+ * });
+ * ```
+ */
+declare function pipe<T>(key: string, value: AsyncIterable<T> | ReadableStream<T>, options?: PipeStreamOptions): PipeStreamResult<T>;
+/**
+ * Reads data from a realtime stream using the default stream key (`"default"`).
+ *
+ * This is a convenience overload that allows you to read from the default stream without
+ * specifying a stream key. The stream will be accessed with the key `"default"`.
+ *
+ * @template T - The type of data chunks in the stream
+ * @param runId - The unique identifier of the run to read the stream from
+ * @param options - Optional configuration for reading the stream
+ * @returns A promise that resolves to an `AsyncIterableStream<T>` that can be consumed
+ *   using `for await...of` or as a `ReadableStream`.
+ *
+ * @example
+ * ```ts
+ * import { streams } from "@trigger.dev/sdk/v3";
+ *
+ * // Read from the default stream
+ * const stream = await streams.read<string>(runId);
+ *
+ * for await (const chunk of stream) {
+ *   console.log("Received chunk:", chunk);
+ * }
+ * ```
+ *
+ * @example
+ * ```ts
+ * // Read with custom timeout and starting position
+ * const stream = await streams.read<string>(runId, {
+ *   timeoutInSeconds: 120,
+ *   startIndex: 10, // Start from the 10th chunk
+ * });
+ * ```
+ */
+declare function read<T>(runId: string, options?: ReadStreamOptions): Promise<AsyncIterableStream<T>>;
+/**
+ * Reads data from a realtime stream with a specific stream key.
+ *
+ * Use this overload when you want to read from a stream with a custom key.
+ *
+ * @template T - The type of data chunks in the stream
+ * @param runId - The unique identifier of the run to read the stream from
+ * @param key - The unique identifier of the stream to read from. Defaults to `"default"` if not provided.
+ * @param options - Optional configuration for reading the stream
+ * @returns A promise that resolves to an `AsyncIterableStream<T>` that can be consumed
+ *   using `for await...of` or as a `ReadableStream`.
+ *
+ * @example
+ * ```ts
+ * import { streams } from "@trigger.dev/sdk";
+ *
+ * // Read from a specific stream key
+ * const stream = await streams.read<string>(runId, "my-custom-stream");
+ *
+ * for await (const chunk of stream) {
+ *   console.log("Received chunk:", chunk);
+ * }
+ * ```
+ *
+ * @example
+ * ```ts
+ * // Read with signal for cancellation
+ * const controller = new AbortController();
+ * const stream = await streams.read<string>(runId, "my-stream", {
+ *   signal: controller.signal,
+ *   timeoutInSeconds: 30,
+ * });
+ *
+ * // Cancel after 5 seconds
+ * setTimeout(() => controller.abort(), 5000);
+ * ```
+ */
+declare function read<T>(runId: string, key: string, options?: ReadStreamOptions): Promise<AsyncIterableStream<T>>;
+declare function append<TPart extends BodyInit>(value: TPart, options?: AppendStreamOptions): Promise<void>;
+declare function append<TPart extends BodyInit>(key: string, value: TPart, options?: AppendStreamOptions): Promise<void>;
+/**
+ * Writes data to a realtime stream using the default stream key (`"default"`).
+ *
+ * This is a convenience overload that allows you to write to the default stream without
+ * specifying a stream key. The stream will be created/accessed with the key `"default"`.
+ *
+ * @template TPart - The type of data chunks in the stream
+ * @param options - The options for writing to the stream
+ * @returns A promise that resolves to an object containing:
+ *   - `stream`: The original stream (can be consumed in your task)
+ *   - `waitUntilComplete`: A function that returns a promise resolving when the stream is fully sent
+ *
+ * @example
+ * ```ts
+ * import { streams } from "@trigger.dev/sdk";
+ *
+ * // Write to the default stream
+ * const { waitUntilComplete } = await streams.writer({
+ *   execute: ({ write, merge }) => {
+ *     write("chunk 1");
+ *     write("chunk 2");
+ *     write("chunk 3");
+ *   },
+ * });
+ *
+ * // Wait for all chunks to be written
+ * await waitUntilComplete();
+ * ```
+ *
+ * @example
+ * ```ts
+ * // Write to a specific stream key
+ * const { waitUntilComplete } = await streams.writer("my-custom-stream", {
+ *   execute: ({ write, merge }) => {
+ *     write("chunk 1");
+ *     write("chunk 2");
+ *     write("chunk 3");
+ *   },
+ * });
+ *
+ * // Wait for all chunks to be written
+ * await waitUntilComplete();
+ * ```
+ *
+ * @example
+ * ```ts
+ * // Write to a parent run
+ * await streams.writer("output", {
+ *   execute: ({ write, merge }) => {
+ *     write("chunk 1");
+ *     write("chunk 2");
+ *     write("chunk 3");
+ *   },
+ * });
+ *
+ * // Wait for all chunks to be written
+ * await waitUntilComplete();
+ * ```
+ *
+ * @example
+ * ```ts
+ * // Write to a specific stream key
+ * await streams.writer("my-custom-stream", {
+ *   execute: ({ write, merge }) => {
+ *     write("chunk 1");
+ *     write("chunk 2");
+ *     write("chunk 3");
+ *   },
+ * });
+ *
+ * // Wait for all chunks to be written
+ * await waitUntilComplete();
+ * ```
+ */
+declare function writer<TPart>(options: WriterStreamOptions<TPart>): PipeStreamResult<TPart>;
+/**
+ * Writes data to a realtime stream with a specific stream key.
+ *
+ * @template TPart - The type of data chunks in the stream
+ * @param key - The unique identifier of the stream to write to. Defaults to `"default"` if not provided.
+ * @param options - The options for writing to the stream
+ * @returns A promise that resolves to an object containing:
+ *   - `stream`: The original stream (can be consumed in your task)
+ *   - `waitUntilComplete`: A function that returns a promise resolving when the stream is fully sent
+ *
+ * @example
+ * ```ts
+ * import { streams } from "@trigger.dev/sdk";
+ *
+ * // Write to a specific stream key
+ * const { waitUntilComplete } = await streams.writer("my-custom-stream", {
+ *   execute: ({ write, merge }) => {
+ *     write("chunk 1");
+ *     write("chunk 2");
+ *     write("chunk 3");
+ *   },
+ * });
+ *
+ * // Wait for all chunks to be written
+ * await waitUntilComplete();
+ * ```
+ */
+declare function writer<TPart>(key: string, options: WriterStreamOptions<TPart>): PipeStreamResult<TPart>;
+export type RealtimeDefineStreamOptions = {
+    id: string;
+};
+declare function define<TPart>(opts: RealtimeDefineStreamOptions): RealtimeDefinedStream<TPart>;
+export type { InferStreamType, InferInputStreamType };
+/**
+ * Define an input stream that can receive typed data from external callers.
+ *
+ * Inside a task, use `.on()`, `.once()`, or `.peek()` to receive data.
+ * Outside a task (e.g., from your backend), use `.send(runId, data)` to send data.
+ *
+ * @template TData - The type of data this input stream receives
+ * @param opts - Options including a unique `id` for this input stream
+ *
+ * @example
+ * ```ts
+ * import { streams, task } from "@trigger.dev/sdk";
+ *
+ * const approval = streams.input<{ approved: boolean; reviewer: string }>({ id: "approval" });
+ *
+ * export const myTask = task({
+ *   id: "my-task",
+ *   run: async (payload) => {
+ *     // Wait for the next approval
+ *     const data = await approval.once().unwrap();
+ *     console.log(data.approved, data.reviewer);
+ *   },
+ * });
+ *
+ * // From your backend:
+ * // await approval.send(runId, { approved: true, reviewer: "alice" });
+ * ```
+ */
+declare function input<TData>(opts: {
+    id: string;
+}): RealtimeDefinedInputStream<TData>;
+export declare const streams: {
+    pipe: typeof pipe;
+    read: typeof read;
+    append: typeof append;
+    writer: typeof writer;
+    define: typeof define;
+    input: typeof input;
+};