@puku-ai/sdk 4.0.0 → 4.0.2

package/dist/vendor/core/streaming.js

@@ -0,0 +1,342 @@
+import { PukuError } from './error.js';
+import { makeReadableStream } from '../internal/shims.js';
+import { findDoubleNewlineIndex, LineDecoder } from '../internal/decoders/line.js';
+import { ReadableStreamToAsyncIterable } from '../internal/shims.js';
+import { isAbortError } from '../internal/errors.js';
+import { safeJSON } from '../internal/utils/values.js';
+import { encodeUTF8 } from '../internal/utils/bytes.js';
+import { loggerFor } from '../internal/utils/log.js';
+import { APIError } from './error.js';
+import { releaseRequestSignal } from '../internal/request-signal.js';
+export class Stream {
+    iterator;
+    controller;
+    #client;
+    constructor(iterator, controller, client) {
+        this.iterator = iterator;
+        this.controller = controller;
+        this.#client = client;
+    }
+    /**
+     * Iterate the raw Server-Sent Events from `response` — `{event, data, raw}`
+     * objects, before any JSON parsing or event-name filtering.
+     *
+     * This reads `response.body` directly (not a clone), so the response is
+     * consumed. Use this in middleware that fully replaces the stream body; for
+     * read-only observation of parsed events, use `ctx.parse()` instead.
+     */
+    static rawEvents(response, controller = new AbortController()) {
+        return _iterSSEMessages(response, controller);
+    }
+    static fromSSEResponse(response, controller, client) {
+        let consumed = false;
+        const logger = client ? loggerFor(client) : console;
+        async function* iterator() {
+            if (consumed) {
+                throw new PukuError('Cannot iterate over a consumed stream, use `.tee()` to split the stream.');
+            }
+            consumed = true;
+            let done = false;
+            try {
+                for await (const sse of _iterSSEMessages(response, controller)) {
+                    if (sse.event === 'completion') {
+                        try {
+                            yield JSON.parse(sse.data);
+                        }
+                        catch (e) {
+                            logger.error(`Could not parse message into JSON:`, sse.data);
+                            logger.error(`From chunk:`, sse.raw);
+                            throw e;
+                        }
+                    }
+                    if (sse.event === 'message_start' ||
+                        sse.event === 'message_delta' ||
+                        sse.event === 'message_stop' ||
+                        sse.event === 'content_block_start' ||
+                        sse.event === 'content_block_delta' ||
+                        sse.event === 'content_block_stop' ||
+                        sse.event === 'message' ||
+                        sse.event === 'user.message' ||
+                        sse.event === 'user.interrupt' ||
+                        sse.event === 'user.tool_confirmation' ||
+                        sse.event === 'user.custom_tool_result' ||
+                        sse.event === 'user.tool_result' ||
+                        sse.event === 'agent.message' ||
+                        sse.event === 'agent.thinking' ||
+                        sse.event === 'agent.tool_use' ||
+                        sse.event === 'agent.tool_result' ||
+                        sse.event === 'agent.mcp_tool_use' ||
+                        sse.event === 'agent.mcp_tool_result' ||
+                        sse.event === 'agent.custom_tool_use' ||
+                        sse.event === 'agent.thread_context_compacted' ||
+                        sse.event === 'session.status_running' ||
+                        sse.event === 'session.status_idle' ||
+                        sse.event === 'session.status_rescheduled' ||
+                        sse.event === 'session.status_terminated' ||
+                        sse.event === 'session.error' ||
+                        sse.event === 'session.deleted' ||
+                        sse.event === 'session.updated' ||
+                        sse.event === 'span.model_request_start' ||
+                        sse.event === 'span.model_request_end' ||
+                        sse.event === 'span.outcome_evaluation_start' ||
+                        sse.event === 'span.outcome_evaluation_ongoing' ||
+                        sse.event === 'span.outcome_evaluation_end' ||
+                        sse.event === 'user.define_outcome' ||
+                        sse.event === 'agent.thread_message_received' ||
+                        sse.event === 'agent.thread_message_sent' ||
+                        sse.event === 'agent.session_thread_message_received' ||
+                        sse.event === 'agent.session_thread_message_sent' ||
+                        sse.event === 'session.thread_created' ||
+                        sse.event === 'session.thread_status_created' ||
+                        sse.event === 'session.thread_status_running' ||
+                        sse.event === 'session.thread_status_idle' ||
+                        sse.event === 'session.thread_status_rescheduled' ||
+                        sse.event === 'session.thread_status_terminated' ||
+                        sse.event === 'event_start' ||
+                        sse.event === 'event_delta' ||
+                        sse.event === 'system.message') {
+                        try {
+                            yield JSON.parse(sse.data);
+                        }
+                        catch (e) {
+                            logger.error(`Could not parse message into JSON:`, sse.data);
+                            logger.error(`From chunk:`, sse.raw);
+                            throw e;
+                        }
+                    }
+                    if (sse.event === 'ping') {
+                        continue;
+                    }
+                    if (sse.event === 'error') {
+                        const body = safeJSON(sse.data) ?? sse.data;
+                        const type = body?.error?.type;
+                        throw new APIError(undefined, body, undefined, response.headers, type);
+                    }
+                }
+                done = true;
+            }
+            catch (e) {
+                // If the user calls `stream.controller.abort()`, we should exit without throwing.
+                if (isAbortError(e))
+                    return;
+                throw e;
+            }
+            finally {
+                // If the user `break`s, abort the ongoing request.
+                if (!done)
+                    controller.abort();
+                releaseRequestSignal(controller);
+            }
+        }
+        return new Stream(iterator, controller, client);
+    }
+    /**
+     * Generates a Stream from a newline-separated ReadableStream
+     * where each item is a JSON value.
+     */
+    static fromReadableStream(readableStream, controller, client) {
+        let consumed = false;
+        async function* iterLines() {
+            const lineDecoder = new LineDecoder();
+            const iter = ReadableStreamToAsyncIterable(readableStream);
+            for await (const chunk of iter) {
+                for (const line of lineDecoder.decode(chunk)) {
+                    yield line;
+                }
+            }
+            for (const line of lineDecoder.flush()) {
+                yield line;
+            }
+        }
+        async function* iterator() {
+            if (consumed) {
+                throw new PukuError('Cannot iterate over a consumed stream, use `.tee()` to split the stream.');
+            }
+            consumed = true;
+            let done = false;
+            try {
+                for await (const line of iterLines()) {
+                    if (done)
+                        continue;
+                    if (line)
+                        yield JSON.parse(line);
+                }
+                done = true;
+            }
+            catch (e) {
+                // If the user calls `stream.controller.abort()`, we should exit without throwing.
+                if (isAbortError(e))
+                    return;
+                throw e;
+            }
+            finally {
+                // If the user `break`s, abort the ongoing request.
+                if (!done)
+                    controller.abort();
+                releaseRequestSignal(controller);
+            }
+        }
+        return new Stream(iterator, controller, client);
+    }
+    [Symbol.asyncIterator]() {
+        return this.iterator();
+    }
+    /**
+     * Splits the stream into two streams which can be
+     * independently read from at different speeds.
+     */
+    tee() {
+        const left = [];
+        const right = [];
+        const iterator = this.iterator();
+        const teeIterator = (queue) => {
+            return {
+                next: () => {
+                    if (queue.length === 0) {
+                        const result = iterator.next();
+                        left.push(result);
+                        right.push(result);
+                    }
+                    return queue.shift();
+                },
+            };
+        };
+        return [
+            new Stream(() => teeIterator(left), this.controller, this.#client),
+            new Stream(() => teeIterator(right), this.controller, this.#client),
+        ];
+    }
+    /**
+     * Converts this stream to a newline-separated ReadableStream of
+     * JSON stringified values in the stream
+     * which can be turned back into a Stream with `Stream.fromReadableStream()`.
+     */
+    toReadableStream() {
+        const self = this;
+        let iter;
+        return makeReadableStream({
+            async start() {
+                iter = self[Symbol.asyncIterator]();
+            },
+            async pull(ctrl) {
+                try {
+                    const { value, done } = await iter.next();
+                    if (done)
+                        return ctrl.close();
+                    const bytes = encodeUTF8(JSON.stringify(value) + '\n');
+                    ctrl.enqueue(bytes);
+                }
+                catch (err) {
+                    ctrl.error(err);
+                }
+            },
+            async cancel() {
+                await iter.return?.();
+            },
+        });
+    }
+}
+export async function* _iterSSEMessages(response, controller) {
+    if (!response.body) {
+        controller.abort();
+        if (typeof globalThis.navigator !== 'undefined' &&
+            globalThis.navigator.product === 'ReactNative') {
+            throw new PukuError(`The default react-native fetch implementation does not support streaming. Please use expo/fetch: https://docs.expo.dev/versions/latest/sdk/expo/#expofetch-api`);
+        }
+        throw new PukuError(`Attempted to iterate over a response with no body`);
+    }
+    const sseDecoder = new SSEDecoder();
+    const lineDecoder = new LineDecoder();
+    const iter = ReadableStreamToAsyncIterable(response.body);
+    for await (const sseChunk of iterSSEChunks(iter)) {
+        for (const line of lineDecoder.decode(sseChunk)) {
+            const sse = sseDecoder.decode(line);
+            if (sse)
+                yield sse;
+        }
+    }
+    for (const line of lineDecoder.flush()) {
+        const sse = sseDecoder.decode(line);
+        if (sse)
+            yield sse;
+    }
+}
+/**
+ * Given an async iterable iterator, iterates over it and yields full
+ * SSE chunks, i.e. yields when a double new-line is encountered.
+ */
+async function* iterSSEChunks(iterator) {
+    let data = new Uint8Array();
+    for await (const chunk of iterator) {
+        if (chunk == null) {
+            continue;
+        }
+        const binaryChunk = chunk instanceof ArrayBuffer ? new Uint8Array(chunk)
+            : typeof chunk === 'string' ? encodeUTF8(chunk)
+                : chunk;
+        let newData = new Uint8Array(data.length + binaryChunk.length);
+        newData.set(data);
+        newData.set(binaryChunk, data.length);
+        data = newData;
+        let patternIndex;
+        while ((patternIndex = findDoubleNewlineIndex(data)) !== -1) {
+            yield data.slice(0, patternIndex);
+            data = data.slice(patternIndex);
+        }
+    }
+    if (data.length > 0) {
+        yield data;
+    }
+}
+class SSEDecoder {
+    data;
+    event;
+    chunks;
+    constructor() {
+        this.event = null;
+        this.data = [];
+        this.chunks = [];
+    }
+    decode(line) {
+        if (line.endsWith('\r')) {
+            line = line.substring(0, line.length - 1);
+        }
+        if (!line) {
+            // empty line and we didn't previously encounter any messages
+            if (!this.event && !this.data.length)
+                return null;
+            const sse = {
+                event: this.event,
+                data: this.data.join('\n'),
+                raw: this.chunks,
+            };
+            this.event = null;
+            this.data = [];
+            this.chunks = [];
+            return sse;
+        }
+        this.chunks.push(line);
+        if (line.startsWith(':')) {
+            return null;
+        }
+        let [fieldname, _, value] = partition(line, ':');
+        if (value.startsWith(' ')) {
+            value = value.substring(1);
+        }
+        if (fieldname === 'event') {
+            this.event = value;
+        }
+        else if (fieldname === 'data') {
+            this.data.push(value);
+        }
+        return null;
+    }
+}
+function partition(str, delimiter) {
+    const index = str.indexOf(delimiter);
+    if (index !== -1) {
+        return [str.substring(0, index), delimiter, str.substring(index + delimiter.length)];
+    }
+    return [str, '', ''];
+}
+//# sourceMappingURL=streaming.js.map

package/dist/vendor/core/streaming.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streaming.js","sourceRoot":"","sources":["../../../src/vendor/core/streaming.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAEpC,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AACvD,OAAO,EAAE,sBAAsB,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AAChF,OAAO,EAAE,6BAA6B,EAAE,MAAM,mBAAmB,CAAC;AAClE,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,QAAQ,EAAE,MAAM,0BAA0B,CAAC;AACpD,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AACrD,OAAO,EAAE,SAAS,EAAE,MAAM,uBAAuB,CAAC;AAGlD,OAAO,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAC;AAEnC,OAAO,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;AAUlE,MAAM,OAAO,MAAM;IAKP;IAJV,UAAU,CAAkB;IAC5B,OAAO,CAAuB;IAE9B,YACU,QAAmC,EAC3C,UAA2B,EAC3B,MAAiB;QAFT,aAAQ,GAAR,QAAQ,CAA2B;QAI3C,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAED;;;;;;;OAOG;IACH,MAAM,CAAC,SAAS,CACd,QAAkB,EAClB,aAA8B,IAAI,eAAe,EAAE;QAEnD,OAAO,gBAAgB,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;IAChD,CAAC;IAED,MAAM,CAAC,eAAe,CACpB,QAAkB,EAClB,UAA2B,EAC3B,MAAiB;QAEjB,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;QAEpD,KAAK,SAAS,CAAC,CAAC,QAAQ;YACtB,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,IAAI,SAAS,CAAC,0EAA0E,CAAC,CAAC;YAClG,CAAC;YACD,QAAQ,GAAG,IAAI,CAAC;YAChB,IAAI,IAAI,GAAG,KAAK,CAAC;YACjB,IAAI,CAAC;gBACH,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,gBAAgB,CAAC,QAAQ,EAAE,UAAU,CAAC,EAAE,CAAC;oBAC/D,IAAI,GAAG,CAAC,KAAK,KAAK,YAAY,EAAE,CAAC;wBAC/B,IAAI,CAAC;4BACH,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAS,CAAC;wBACrC,CAAC;wBAAC,OAAO,CAAC,EAAE,CAAC;4BACX,MAAM,CAAC,KAAK,CAAC,oCAAoC,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;4BAC7D,MAAM,CAAC,KAAK,CAAC,aAAa,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC;4BACrC,MAAM,CAAC,CAAC;wBACV,CAAC;oBACH,CAAC;oBAED,IACE,GAAG,CAAC,KAAK,KAAK,eAAe;wBAC7B,GAAG,CAAC,KAAK,KAAK,eAAe;wBAC7B,GAAG,CAAC,KAAK,KAAK,cAAc;wBAC5B,GAAG,CAAC,KAAK,KAAK,qBAAqB;wBACnC,GAAG,CAAC,KAAK,KAAK,qBAAqB;wBACnC,GAAG,CAAC,KAAK,KAAK,oBAAoB;wBAClC,GAAG,CAAC,KAAK,KAAK,SAAS;wBACvB,GAAG,CAAC,KAAK,KAAK,cAAc;wBAC5B,GAAG,CAAC,KAAK,KAAK,gBAAgB;wBAC9B,GAAG,CAAC,KAAK,KAAK,wBAAwB;wBACtC,GAAG,CAAC,KAAK,KAAK,yBAAyB;wBACvC,GAAG,CAAC,KAAK,KAAK,kBAAkB;wBAChC,GAAG,CAAC,KAAK,KAAK,eAAe;wBAC7B,GAAG,CAAC,KAAK,KAAK,gBAAgB;wBAC9B,GAAG,CAAC,KAAK,KAAK,gBAAgB;wBAC9B,GAAG,CAAC,KAAK,KAAK,mBAAmB;wBACjC,GAAG,CAAC,KAAK,KAAK,oBAAoB;wBAClC,GAAG,CAAC,KAAK,KAAK,uBAAuB;wBACrC,GAAG,CAAC,KAAK,KAAK,uBAAuB;wBACrC,GAAG,CAAC,KAAK,KAAK,gCAAgC;wBAC9C,GAAG,CAAC,KAAK,KAAK,wBAAwB;wBACtC,GAAG,CAAC,KAAK,KAAK,qBAAqB;wBACnC,GAAG,CAAC,KAAK,KAAK,4BAA4B;wBAC1C,GAAG,CAAC,KAAK,KAAK,2BAA2B;wBACzC,GAAG,CAAC,KAAK,KAAK,eAAe;wBAC7B,GAAG,CAAC,KAAK,KAAK,iBAAiB;wBAC/B,GAAG,CAAC,KAAK,KAAK,iBAAiB;wBAC/B,GAAG,CAAC,KAAK,KAAK,0BAA0B;wBACxC,GAAG,CAAC,KAAK,KAAK,wBAAwB;wBACtC,GAAG,CAAC,KAAK,KAAK,+BAA+B;wBAC7C,GAAG,CAAC,KAAK,KAAK,iCAAiC;wBAC/C,GAAG,CAAC,KAAK,KAAK,6BAA6B;wBAC3C,GAAG,CAAC,KAAK,KAAK,qBAAqB;wBACnC,GAAG,CAAC,KAAK,KAAK,+BAA+B;wBAC7C,GAAG,CAAC,KAAK,KAAK,2BAA2B;wBACzC,GAAG,CAAC,KAAK,KAAK,uCAAuC;wBACrD,GAAG,CAAC,KAAK,KAAK,mCAAmC;wBACjD,GAAG,CAAC,KAAK,KAAK,wBAAwB;wBACtC,GAAG,CAAC,KAAK,KAAK,+BAA+B;wBAC7C,GAAG,CAAC,KAAK,KAAK,+BAA+B;wBAC7C,GAAG,CAAC,KAAK,KAAK,4BAA4B;wBAC1C,GAAG,CAAC,KAAK,KAAK,mCAAmC;wBACjD,GAAG,CAAC,KAAK,KAAK,kCAAkC;wBAChD,GAAG,CAAC,KAAK,KAAK,aAAa;wBAC3B,GAAG,CAAC,KAAK,KAAK,aAAa;wBAC3B,GAAG,CAAC,KAAK,KAAK,gBAAgB,EAC9B,CAAC;wBACD,IAAI,CAAC;4BACH,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAS,CAAC;wBACrC,CAAC;wBAAC,OAAO,CAAC,EAAE,CAAC;4BACX,MAAM,CAAC,KAAK,CAAC,oCAAoC,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;4BAC7D,MAAM,CAAC,KAAK,CAAC,aAAa,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC;4BACrC,MAAM,CAAC,CAAC;wBACV,CAAC;oBACH,CAAC;oBAED,IAAI,GAAG,CAAC,KAAK,KAAK,MAAM,EAAE,CAAC;wBACzB,SAAS;oBACX,CAAC;oBAED,IAAI,GAAG,CAAC,KAAK,KAAK,OAAO,EAAE,CAAC;wBAC1B,MAAM,IAAI,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC;wBAC5C,MAAM,IAAI,GAAG,IAAI,EAAE,KAAK,EAAE,IAA6B,CAAC;wBACxD,MAAM,IAAI,QAAQ,CAAC,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,QAAQ,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;oBACzE,CAAC;gBACH,CAAC;gBACD,IAAI,GAAG,IAAI,CAAC;YACd,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,kFAAkF;gBAClF,IAAI,YAAY,CAAC,CAAC,CAAC;oBAAE,OAAO;gBAC5B,MAAM,CAAC,CAAC;YACV,CAAC;oBAAS,CAAC;gBACT,mDAAmD;gBACnD,IAAI,CAAC,IAAI;oBAAE,UAAU,CAAC,KAAK,EAAE,CAAC;gBAC9B,oBAAoB,CAAC,UAAU,CAAC,CAAC;YACnC,CAAC;QACH,CAAC;QAED,OAAO,IAAI,MAAM,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;IAClD,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,kBAAkB,CACvB,cAA8B,EAC9B,UAA2B,EAC3B,MAAiB;QAEjB,IAAI,QAAQ,GAAG,KAAK,CAAC;QAErB,KAAK,SAAS,CAAC,CAAC,SAAS;YACvB,MAAM,WAAW,GAAG,IAAI,WAAW,EAAE,CAAC;YAEtC,MAAM,IAAI,GAAG,6BAA6B,CAAQ,cAAc,CAAC,CAAC;YAClE,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,EAAE,CAAC;gBAC/B,KAAK,MAAM,IAAI,IAAI,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;oBAC7C,MAAM,IAAI,CAAC;gBACb,CAAC;YACH,CAAC;YAED,KAAK,MAAM,IAAI,IAAI,WAAW,CAAC,KAAK,EAAE,EAAE,CAAC;gBACvC,MAAM,IAAI,CAAC;YACb,CAAC;QACH,CAAC;QAED,KAAK,SAAS,CAAC,CAAC,QAAQ;YACtB,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,IAAI,SAAS,CAAC,0EAA0E,CAAC,CAAC;YAClG,CAAC;YACD,QAAQ,GAAG,IAAI,CAAC;YAChB,IAAI,IAAI,GAAG,KAAK,CAAC;YACjB,IAAI,CAAC;gBACH,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,SAAS,EAAE,EAAE,CAAC;oBACrC,IAAI,IAAI;wBAAE,SAAS;oBACnB,IAAI,IAAI;wBAAE,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAS,CAAC;gBAC3C,CAAC;gBACD,IAAI,GAAG,IAAI,CAAC;YACd,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,kFAAkF;gBAClF,IAAI,YAAY,CAAC,CAAC,CAAC;oBAAE,OAAO;gBAC5B,MAAM,CAAC,CAAC;YACV,CAAC;oBAAS,CAAC;gBACT,mDAAmD;gBACnD,IAAI,CAAC,IAAI;oBAAE,UAAU,CAAC,KAAK,EAAE,CAAC;gBAC9B,oBAAoB,CAAC,UAAU,CAAC,CAAC;YACnC,CAAC;QACH,CAAC;QAED,OAAO,IAAI,MAAM,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;IAClD,CAAC;IAED,CAAC,MAAM,CAAC,aAAa,CAAC;QACpB,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC;IACzB,CAAC;IAED;;;OAGG;IACH,GAAG;QACD,MAAM,IAAI,GAAyC,EAAE,CAAC;QACtD,MAAM,KAAK,GAAyC,EAAE,CAAC;QACvD,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;QAEjC,MAAM,WAAW,GAAG,CAAC,KAA2C,EAAuB,EAAE;YACvF,OAAO;gBACL,IAAI,EAAE,GAAG,EAAE;oBACT,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;wBACvB,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;wBAC/B,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;wBAClB,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;oBACrB,CAAC;oBACD,OAAO,KAAK,CAAC,KAAK,EAAG,CAAC;gBACxB,CAAC;aACF,CAAC;QACJ,CAAC,CAAC;QAEF,OAAO;YACL,IAAI,MAAM,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC;YAClE,IAAI,MAAM,CAAC,GAAG,EAAE,CAAC,WAAW,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC;SACpE,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACH,gBAAgB;QACd,MAAM,IAAI,GAAG,IAAI,CAAC;QAClB,IAAI,IAAyB,CAAC;QAE9B,OAAO,kBAAkB,CAAC;YACxB,KAAK,CAAC,KAAK;gBACT,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;YACtC,CAAC;YACD,KAAK,CAAC,IAAI,CAAC,IAAS;gBAClB,IAAI,CAAC;oBACH,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;oBAC1C,IAAI,IAAI;wBAAE,OAAO,IAAI,CAAC,KAAK,EAAE,CAAC;oBAE9B,MAAM,KAAK,GAAG,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,CAAC;oBAEvD,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBACtB,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;gBAClB,CAAC;YACH,CAAC;YACD,KAAK,CAAC,MAAM;gBACV,MAAM,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC;YACxB,CAAC;SACF,CAAC,CAAC;IACL,CAAC;CACF;AAED,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC,gBAAgB,CACrC,QAAkB,EAClB,UAA2B;IAE3B,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnB,UAAU,CAAC,KAAK,EAAE,CAAC;QACnB,IACE,OAAQ,UAAkB,CAAC,SAAS,KAAK,WAAW;YACnD,UAAkB,CAAC,SAAS,CAAC,OAAO,KAAK,aAAa,EACvD,CAAC;YACD,MAAM,IAAI,SAAS,CACjB,gKAAgK,CACjK,CAAC;QACJ,CAAC;QACD,MAAM,IAAI,SAAS,CAAC,mDAAmD,CAAC,CAAC;IAC3E,CAAC;IAED,MAAM,UAAU,GAAG,IAAI,UAAU,EAAE,CAAC;IACpC,MAAM,WAAW,GAAG,IAAI,WAAW,EAAE,CAAC;IAEtC,MAAM,IAAI,GAAG,6BAA6B,CAAQ,QAAQ,CAAC,IAAI,CAAC,CAAC;IACjE,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC;QACjD,KAAK,MAAM,IAAI,IAAI,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;YAChD,MAAM,GAAG,GAAG,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACpC,IAAI,GAAG;gBAAE,MAAM,GAAG,CAAC;QACrB,CAAC;IACH,CAAC;IAED,KAAK,MAAM,IAAI,IAAI,WAAW,CAAC,KAAK,EAAE,EAAE,CAAC;QACvC,MAAM,GAAG,GAAG,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACpC,IAAI,GAAG;YAAE,MAAM,GAAG,CAAC;IACrB,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,KAAK,SAAS,CAAC,CAAC,aAAa,CAAC,QAAsC;IAClE,IAAI,IAAI,GAAG,IAAI,UAAU,EAAE,CAAC;IAE5B,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;QACnC,IAAI,KAAK,IAAI,IAAI,EAAE,CAAC;YAClB,SAAS;QACX,CAAC;QAED,MAAM,WAAW,GACf,KAAK,YAAY,WAAW,CAAC,CAAC,CAAC,IAAI,UAAU,CAAC,KAAK,CAAC;YACpD,CAAC,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC;gBAC/C,CAAC,CAAC,KAAK,CAAC;QAEV,IAAI,OAAO,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;QAC/D,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAClB,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QACtC,IAAI,GAAG,OAAO,CAAC;QAEf,IAAI,YAAY,CAAC;QACjB,OAAO,CAAC,YAAY,GAAG,sBAAsB,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;YAC5D,MAAM,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC;YAClC,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAED,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACpB,MAAM,IAAI,CAAC;IACb,CAAC;AACH,CAAC;AAED,MAAM,UAAU;IACN,IAAI,CAAW;IACf,KAAK,CAAgB;IACrB,MAAM,CAAW;IAEzB;QACE,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAClB,IAAI,CAAC,IAAI,GAAG,EAAE,CAAC;QACf,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;IACnB,CAAC;IAED,MAAM,CAAC,IAAY;QACjB,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YACxB,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC5C,CAAC;QAED,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,6DAA6D;YAC7D,IAAI,CAAC,IAAI,CAAC,KAAK,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM;gBAAE,OAAO,IAAI,CAAC;YAElD,MAAM,GAAG,GAAoB;gBAC3B,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;gBAC1B,GAAG,EAAE,IAAI,CAAC,MAAM;aACjB,CAAC;YAEF,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;YAClB,IAAI,CAAC,IAAI,GAAG,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;YAEjB,OAAO,GAAG,CAAC;QACb,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEvB,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAI,CAAC,SAAS,EAAE,CAAC,EAAE,KAAK,CAAC,GAAG,SAAS,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;QAEjD,IAAI,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YAC1B,KAAK,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QAC7B,CAAC;QAED,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;YAC1B,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACrB,CAAC;aAAM,IAAI,SAAS,KAAK,MAAM,EAAE,CAAC;YAChC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxB,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AAED,SAAS,SAAS,CAAC,GAAW,EAAE,SAAiB;IAC/C,MAAM,KAAK,GAAG,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACrC,IAAI,KAAK,KAAK,CAAC,CAAC,EAAE,CAAC;QACjB,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,SAAS,EAAE,GAAG,CAAC,SAAS,CAAC,KAAK,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;IACvF,CAAC;IAED,OAAO,CAAC,GAAG,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;AACvB,CAAC"}

package/dist/vendor/core/uploads.d.ts

@@ -0,0 +1,2 @@
+export { type Uploadable } from '../internal/uploads';
+export { toFile, type ToFileInput } from '../internal/to-file';

package/dist/vendor/core/uploads.js

@@ -0,0 +1,2 @@
+export { toFile } from '../internal/to-file.js';
+//# sourceMappingURL=uploads.js.map

package/dist/vendor/core/uploads.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"uploads.js","sourceRoot":"","sources":["../../../src/vendor/core/uploads.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAoB,MAAM,qBAAqB,CAAC"}

package/dist/vendor/error.d.ts

@@ -0,0 +1,2 @@
+/** @deprecated Import from ./core/error instead */
+export * from './core/error';

package/dist/vendor/error.js

@@ -0,0 +1,3 @@
+/** @deprecated Import from ./core/error instead */
+export * from './core/error.js';
+//# sourceMappingURL=error.js.map

package/dist/vendor/error.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"error.js","sourceRoot":"","sources":["../../src/vendor/error.ts"],"names":[],"mappings":"AAAA,mDAAmD;AACnD,cAAc,cAAc,CAAC"}

package/dist/vendor/helpers/beta/environments.d.ts

@@ -0,0 +1,22 @@
+/**
+ * Self-hosted environment runner helpers.
+ *
+ * - {@link WorkPoller} (`client.beta.environments.work.poller`) — control-plane
+ *   only: claims work items and yields each one.
+ * - {@link SessionToolRunner} (`client.beta.sessions.events.toolRunner`) — the
+ *   sessions-side counterpart to `client.beta.messages.toolRunner`: dispatches
+ *   local tools against a session's `agent.tool_use` events.
+ * - {@link EnvironmentWorker} (`client.beta.environments.work.worker`) — the full
+ *   composition: poll → set up the workdir + skills → run a
+ *   {@link SessionToolRunner} while heartbeating the work-item lease →
+ *   force-stop on exit (unless the lease was lost to another holder) → loop.
+ *   Use `handleItem()` for the per-item flow when you already hold a claimed
+ *   work item — with no arguments it reads the `PUKU_*` env vars that
+ *   `ant worker poll --on-work` sets. The class is also exported here if you
+ *   prefer `new EnvironmentWorker({ client, ... })`.
+ *
+ * The tool implementations themselves (`betaAgentToolset` and the
+ * per-tool factories) live in their own Node-only module — import them directly
+ * from `@puku-ai/sdk/tools/agent-toolset/node`.
+ */
+export { WorkPoller, EnvironmentWorker, SessionToolRunner, MANAGED_AGENTS_BETA, DEFAULT_MAX_IDLE_MS, type DispatchedToolCall, type WorkPollerOptions, type EnvironmentWorkerOptions, type EnvironmentWorkerTools, type HandleItemOptions, type SessionToolRunnerOptions, } from '../../lib/environments/index';

package/dist/vendor/helpers/beta/environments.js

@@ -0,0 +1,23 @@
+/**
+ * Self-hosted environment runner helpers.
+ *
+ * - {@link WorkPoller} (`client.beta.environments.work.poller`) — control-plane
+ *   only: claims work items and yields each one.
+ * - {@link SessionToolRunner} (`client.beta.sessions.events.toolRunner`) — the
+ *   sessions-side counterpart to `client.beta.messages.toolRunner`: dispatches
+ *   local tools against a session's `agent.tool_use` events.
+ * - {@link EnvironmentWorker} (`client.beta.environments.work.worker`) — the full
+ *   composition: poll → set up the workdir + skills → run a
+ *   {@link SessionToolRunner} while heartbeating the work-item lease →
+ *   force-stop on exit (unless the lease was lost to another holder) → loop.
+ *   Use `handleItem()` for the per-item flow when you already hold a claimed
+ *   work item — with no arguments it reads the `PUKU_*` env vars that
+ *   `ant worker poll --on-work` sets. The class is also exported here if you
+ *   prefer `new EnvironmentWorker({ client, ... })`.
+ *
+ * The tool implementations themselves (`betaAgentToolset` and the
+ * per-tool factories) live in their own Node-only module — import them directly
+ * from `@puku-ai/sdk/tools/agent-toolset/node`.
+ */
+export { WorkPoller, EnvironmentWorker, SessionToolRunner, MANAGED_AGENTS_BETA, DEFAULT_MAX_IDLE_MS, } from '../../lib/environments/index.js';
+//# sourceMappingURL=environments.js.map

package/dist/vendor/helpers/beta/environments.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"environments.js","sourceRoot":"","sources":["../../../../src/vendor/helpers/beta/environments.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,OAAO,EACL,UAAU,EACV,iBAAiB,EACjB,iBAAiB,EACjB,mBAAmB,EACnB,mBAAmB,GAOpB,MAAM,8BAA8B,CAAC"}

package/dist/vendor/helpers/beta/json-schema.d.ts

@@ -0,0 +1,36 @@
+import { FromSchema, JSONSchema } from 'json-schema-to-ts';
+import { Promisable, BetaRunnableTool, BetaToolRunContext } from '../../lib/tools/BetaRunnableTool';
+import { BetaToolResultContentBlockParam } from '../../resources/beta';
+import { AutoParseableBetaOutputFormat } from '../../lib/beta-parser';
+type NoInfer<T> = T extends infer R ? R : never;
+/**
+ * Creates a Tool with a provided JSON schema that can be passed
+ * to the `.toolRunner()` method. The schema is used to automatically validate
+ * the input arguments for the tool.
+ */
+export declare function betaTool<const Schema extends Exclude<JSONSchema, boolean> & {
+    type: 'object';
+}>(options: {
+    name: string;
+    inputSchema: Schema;
+    description: string;
+    run: (args: NoInfer<FromSchema<Schema>>, context?: BetaToolRunContext) => Promisable<string | Array<BetaToolResultContentBlockParam>>;
+    /**
+     * Optional cleanup hook for tools that hold process-level resources (e.g. a
+     * persistent shell). `client.beta.sessions.events.toolRunner` calls it once
+     * when iteration ends.
+     */
+    close?: () => void | Promise<void>;
+}): BetaRunnableTool<NoInfer<FromSchema<Schema>>>;
+/**
+ * Creates a JSON schema output format object from the given JSON schema.
+ * If this is passed to the `.parse()` method then the response message will contain a
+ * `.parsed_output` property that is the result of parsing the content with the given JSON schema.
+ *
+ */
+export declare function betaJSONSchemaOutputFormat<const Schema extends Exclude<JSONSchema, boolean> & {
+    type: 'object';
+}>(jsonSchema: Schema, options?: {
+    transform?: boolean;
+}): AutoParseableBetaOutputFormat<NoInfer<FromSchema<Schema>>>;
+export {};

package/dist/vendor/helpers/beta/json-schema.js

@@ -0,0 +1,53 @@
+import { PukuError } from '../../index.js';
+import { transformJSONSchema } from '../../lib/transform-json-schema.js';
+/**
+ * Creates a Tool with a provided JSON schema that can be passed
+ * to the `.toolRunner()` method. The schema is used to automatically validate
+ * the input arguments for the tool.
+ */
+export function betaTool(options) {
+    if (options.inputSchema.type !== 'object') {
+        throw new Error(`JSON schema for tool "${options.name}" must be an object, but got ${options.inputSchema.type}`);
+    }
+    return {
+        type: 'custom',
+        name: options.name,
+        input_schema: options.inputSchema,
+        description: options.description,
+        run: options.run,
+        parse: (content) => content,
+        ...(options.close ? { close: options.close } : {}),
+    };
+}
+/**
+ * Creates a JSON schema output format object from the given JSON schema.
+ * If this is passed to the `.parse()` method then the response message will contain a
+ * `.parsed_output` property that is the result of parsing the content with the given JSON schema.
+ *
+ */
+export function betaJSONSchemaOutputFormat(jsonSchema, options) {
+    if (jsonSchema.type !== 'object') {
+        throw new Error(`JSON schema for tool must be an object, but got ${jsonSchema.type}`);
+    }
+    const transform = options?.transform ?? true;
+    if (transform) {
+        // todo: doing this is arguably necessary, but it does change the schema the user passed in
+        // so I'm not sure how we should handle that
+        jsonSchema = transformJSONSchema(jsonSchema);
+    }
+    return {
+        type: 'json_schema',
+        schema: {
+            ...jsonSchema,
+        },
+        parse: (content) => {
+            try {
+                return JSON.parse(content);
+            }
+            catch (error) {
+                throw new PukuError(`Failed to parse structured output: ${error}`);
+            }
+        },
+    };
+}
+//# sourceMappingURL=json-schema.js.map

package/dist/vendor/helpers/beta/json-schema.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"json-schema.js","sourceRoot":"","sources":["../../../../src/vendor/helpers/beta/json-schema.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,SAAS,EAAE,MAAM,OAAO,CAAC;AAClC,OAAO,EAAE,mBAAmB,EAAE,MAAM,iCAAiC,CAAC;AAItE;;;;GAIG;AACH,MAAM,UAAU,QAAQ,CAAyE,OAchG;IACC,IAAI,OAAO,CAAC,WAAW,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC1C,MAAM,IAAI,KAAK,CACb,yBAAyB,OAAO,CAAC,IAAI,gCAAgC,OAAO,CAAC,WAAW,CAAC,IAAI,EAAE,CAChG,CAAC;IACJ,CAAC;IAED,OAAO;QACL,IAAI,EAAE,QAAQ;QACd,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,YAAY,EAAE,OAAO,CAAC,WAAW;QACjC,WAAW,EAAE,OAAO,CAAC,WAAW;QAChC,GAAG,EAAE,OAAO,CAAC,GAAG;QAChB,KAAK,EAAE,CAAC,OAAgB,EAAE,EAAE,CAAC,OAA6B;QAC1D,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC5C,CAAC;AACX,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,0BAA0B,CAGxC,UAAkB,EAClB,OAEC;IAED,IAAI,UAAU,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QACjC,MAAM,IAAI,KAAK,CAAC,mDAAmD,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC;IACxF,CAAC;IAED,MAAM,SAAS,GAAG,OAAO,EAAE,SAAS,IAAI,IAAI,CAAC;IAC7C,IAAI,SAAS,EAAE,CAAC;QACd,2FAA2F;QAC3F,4CAA4C;QAC5C,UAAU,GAAG,mBAAmB,CAAC,UAAU,CAAW,CAAC;IACzD,CAAC;IAED,OAAO;QACL,IAAI,EAAE,aAAa;QACnB,MAAM,EAAE;YACN,GAAG,UAAU;SACd;QACD,KAAK,EAAE,CAAC,OAAO,EAAE,EAAE;YACjB,IAAI,CAAC;gBACH,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC7B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,IAAI,SAAS,CAAC,sCAAsC,KAAK,EAAE,CAAC,CAAC;YACrE,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC"}