afm-bridge 0.1.0

package/dist/rpc.js

@@ -0,0 +1,172 @@
+import { AbortError, AfmError, errorFromRpc, ProtocolError, RequestTimeoutError, } from './errors.js';
+const DEFAULT_CANCEL_GRACE_MS = 5000;
+/** JSON-RPC 2.0 peer over one connection. Both sides can send requests. */
+export class RpcPeer {
+    connection;
+    handlers;
+    nextId = 1;
+    pending = new Map();
+    incoming = new Map();
+    closedError;
+    constructor(connection, handlers) {
+        this.connection = connection;
+        this.handlers = handlers;
+    }
+    get pendingCount() {
+        return this.pending.size;
+    }
+    request(method, params, options = {}) {
+        if (this.closedError)
+            return Promise.reject(this.closedError);
+        const { signal, timeoutMs } = options;
+        if (signal?.aborted)
+            return Promise.reject(new AbortError());
+        const id = this.nextId++;
+        return new Promise((resolve, reject) => {
+            let timer;
+            let graceTimer;
+            let cancelError;
+            const cancel = (error) => {
+                if (cancelError || !this.pending.has(id))
+                    return;
+                cancelError = error;
+                this.notify('cancel', { id });
+                // Wait for the sidecar's answer so the session is free again when we reject,
+                // but never hang if the sidecar does not answer.
+                const grace = options.cancelGraceMs ?? DEFAULT_CANCEL_GRACE_MS;
+                if (grace <= 0)
+                    this.settle(id, { error });
+                else
+                    graceTimer = setTimeout(() => this.settle(id, { error }), grace);
+            };
+            const onAbort = () => cancel(new AbortError());
+            this.pending.set(id, {
+                resolve: (value) => resolve(value),
+                // A cancelled request is answered with `cancelled`; report the reason we cancelled for.
+                reject: (error) => reject(cancelError && error instanceof AfmError && error.type === 'cancelled' ? cancelError : error),
+                cleanup: () => {
+                    if (timer)
+                        clearTimeout(timer);
+                    if (graceTimer)
+                        clearTimeout(graceTimer);
+                    signal?.removeEventListener('abort', onAbort);
+                },
+            });
+            this.handlers.onPendingChange?.(this.pending.size);
+            signal?.addEventListener('abort', onAbort, { once: true });
+            if (timeoutMs !== undefined) {
+                timer = setTimeout(() => cancel(new RequestTimeoutError(`Request '${method}' timed out after ${timeoutMs} ms`)), timeoutMs);
+            }
+            options.onStart?.(id);
+            this.write({ jsonrpc: '2.0', id, method, params });
+        });
+    }
+    notify(method, params) {
+        if (this.closedError)
+            return;
+        this.write({ jsonrpc: '2.0', method, params });
+    }
+    /** Feeds one line received from the connection. */
+    handleLine(line) {
+        let message;
+        try {
+            message = JSON.parse(line);
+        }
+        catch {
+            this.handlers.onProtocolError?.(new ProtocolError(`Sidecar sent invalid JSON: ${line.slice(0, 200)}`));
+            return;
+        }
+        if (typeof message !== 'object' || message === null || message.jsonrpc !== '2.0') {
+            this.handlers.onProtocolError?.(new ProtocolError(`Sidecar sent a non JSON-RPC message: ${line.slice(0, 200)}`));
+            return;
+        }
+        const { id, method } = message;
+        if (typeof method === 'string') {
+            if (id === undefined)
+                this.handleNotification(method, message.params);
+            else
+                this.handleIncomingRequest(id, method, message.params);
+            return;
+        }
+        if (typeof id === 'number') {
+            if ('error' in message)
+                this.settle(id, { error: errorFromRpc(message.error) });
+            else
+                this.settle(id, { result: message.result });
+            return;
+        }
+        // Error without id: the sidecar could not parse something we sent.
+        if ('error' in message)
+            this.handlers.onProtocolError?.(errorFromRpc(message.error));
+    }
+    /** Rejects everything in flight; later calls reject with the same error. */
+    close(error) {
+        if (this.closedError)
+            return;
+        this.closedError = error;
+        for (const id of [...this.pending.keys()])
+            this.settle(id, { error });
+        for (const controller of this.incoming.values())
+            controller.abort();
+        this.incoming.clear();
+    }
+    handleNotification(method, params) {
+        if (method === 'cancel') {
+            const id = params?.id;
+            if (id !== undefined)
+                this.incoming.get(id)?.abort();
+            return;
+        }
+        this.handlers.onNotification(method, params);
+    }
+    handleIncomingRequest(id, method, params) {
+        const handler = this.handlers.onRequest;
+        if (!handler) {
+            this.write({
+                jsonrpc: '2.0',
+                id,
+                error: { code: -32601, message: `Method not found: ${method}`, data: { type: 'methodNotFound' } },
+            });
+            return;
+        }
+        const controller = new AbortController();
+        this.incoming.set(id, controller);
+        handler(method, params, controller.signal).then((result) => {
+            this.incoming.delete(id);
+            this.write({ jsonrpc: '2.0', id, result });
+        }, (error) => {
+            this.incoming.delete(id);
+            const payload = error instanceof AfmError && error.code !== undefined
+                ? { code: error.code, message: error.message, data: { ...error.data, type: error.type } }
+                : {
+                    code: -32000,
+                    message: error instanceof Error ? error.message : String(error),
+                    data: { type: 'handlerError' },
+                };
+            this.write({ jsonrpc: '2.0', id, error: payload });
+        });
+    }
+    settle(id, outcome) {
+        const entry = this.pending.get(id);
+        if (!entry)
+            return;
+        this.pending.delete(id);
+        entry.cleanup();
+        this.handlers.onPendingChange?.(this.pending.size);
+        if ('error' in outcome && outcome.error !== undefined)
+            entry.reject(outcome.error);
+        else
+            entry.resolve(outcome.result);
+    }
+    write(message) {
+        if (this.closedError)
+            return;
+        try {
+            this.connection.send(JSON.stringify(message));
+        }
+        catch (error) {
+            this.handlers.onProtocolError?.(error instanceof Error ? error : new Error(String(error)));
+        }
+    }
+}
+//# sourceMappingURL=rpc.js.map

package/dist/rpc.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rpc.js","sourceRoot":"","sources":["../src/rpc.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,UAAU,EACV,QAAQ,EACR,YAAY,EACZ,aAAa,EACb,mBAAmB,GAEpB,MAAM,aAAa,CAAA;AAoBpB,MAAM,uBAAuB,GAAG,IAAI,CAAA;AAEpC,2EAA2E;AAC3E,MAAM,OAAO,OAAO;IAOC,UAAU;IACV,QAAQ;IAPnB,MAAM,GAAG,CAAC,CAAA;IACD,OAAO,GAAG,IAAI,GAAG,EAAmB,CAAA;IACpC,QAAQ,GAAG,IAAI,GAAG,EAAoC,CAAA;IAC/D,WAAW,CAAmB;IAEtC,YACmB,UAAsB,EACtB,QAKhB;0BANgB,UAAU;wBACV,QAAQ;IAMxB,CAAC;IAEJ,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAA;IAC1B,CAAC;IAED,OAAO,CAAI,MAAc,EAAE,MAAe,EAAE,OAAO,GAAmB,EAAE;QACtE,IAAI,IAAI,CAAC,WAAW;YAAE,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QAC7D,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,OAAO,CAAA;QACrC,IAAI,MAAM,EAAE,OAAO;YAAE,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,UAAU,EAAE,CAAC,CAAA;QAE5D,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAA;QACxB,OAAO,IAAI,OAAO,CAAI,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACxC,IAAI,KAAgD,CAAA;YACpD,IAAI,UAAqD,CAAA;YACzD,IAAI,WAAiC,CAAA;YAErC,MAAM,MAAM,GAAG,CAAC,KAAe,EAAE,EAAE;gBACjC,IAAI,WAAW,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;oBAAE,OAAM;gBAChD,WAAW,GAAG,KAAK,CAAA;gBACnB,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,EAAE,EAAE,CAAC,CAAA;gBAC7B,6EAA6E;gBAC7E,iDAAiD;gBACjD,MAAM,KAAK,GAAG,OAAO,CAAC,aAAa,IAAI,uBAAuB,CAAA;gBAC9D,IAAI,KAAK,IAAI,CAAC;oBAAE,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAA;;oBACrC,UAAU,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,CAAC,CAAA;YACvE,CAAC,CAAA;YACD,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,UAAU,EAAE,CAAC,CAAA;YAE9C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE;gBACnB,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,CAAC,KAAU,CAAC;gBACvC,wFAAwF;gBACxF,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE,CAChB,MAAM,CACJ,WAAW,IAAI,KAAK,YAAY,QAAQ,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,KAAK,CAC7F;gBACH,OAAO,EAAE,GAAG,EAAE;oBACZ,IAAI,KAAK;wBAAE,YAAY,CAAC,KAAK,CAAC,CAAA;oBAC9B,IAAI,UAAU;wBAAE,YAAY,CAAC,UAAU,CAAC,CAAA;oBACxC,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAA;gBAC/C,CAAC;aACF,CAAC,CAAA;YACF,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;YAElD,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YAC1D,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;gBAC5B,KAAK,GAAG,UAAU,CAChB,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,mBAAmB,CAAC,YAAY,MAAM,qBAAqB,SAAS,KAAK,CAAC,CAAC,EAC5F,SAAS,CACV,CAAA;YACH,CAAC;YACD,OAAO,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAA;YACrB,IAAI,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAA;QACpD,CAAC,CAAC,CAAA;IACJ,CAAC;IAED,MAAM,CAAC,MAAc,EAAE,MAAe;QACpC,IAAI,IAAI,CAAC,WAAW;YAAE,OAAM;QAC5B,IAAI,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAA;IAChD,CAAC;IAED,mDAAmD;IACnD,UAAU,CAAC,IAAY;QACrB,IAAI,OAAgC,CAAA;QACpC,IAAI,CAAC;YACH,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QAC5B,CAAC;QAAC,MAAM,CAAC;YACP,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,CAAC,IAAI,aAAa,CAAC,8BAA8B,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CAAA;YACtG,OAAM;QACR,CAAC;QACD,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,KAAK,IAAI,IAAI,OAAO,CAAC,OAAO,KAAK,KAAK,EAAE,CAAC;YACjF,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,CAC7B,IAAI,aAAa,CAAC,wCAAwC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAChF,CAAA;YACD,OAAM;QACR,CAAC;QACD,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,OAAO,CAAA;QAC9B,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC/B,IAAI,EAAE,KAAK,SAAS;gBAAE,IAAI,CAAC,kBAAkB,CAAC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,CAAA;;gBAChE,IAAI,CAAC,qBAAqB,CAAC,EAAqB,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,CAAA;YAC9E,OAAM;QACR,CAAC;QACD,IAAI,OAAO,EAAE,KAAK,QAAQ,EAAE,CAAC;YAC3B,IAAI,OAAO,IAAI,OAAO;gBAAE,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,YAAY,CAAC,OAAO,CAAC,KAAwB,CAAC,EAAE,CAAC,CAAA;;gBAC7F,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAA;YAChD,OAAM;QACR,CAAC;QACD,mEAAmE;QACnE,IAAI,OAAO,IAAI,OAAO;YAAE,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,CAAC,YAAY,CAAC,OAAO,CAAC,KAAwB,CAAC,CAAC,CAAA;IACzG,CAAC;IAED,4EAA4E;IAC5E,KAAK,CAAC,KAAY;QAChB,IAAI,IAAI,CAAC,WAAW;YAAE,OAAM;QAC5B,IAAI,CAAC,WAAW,GAAG,KAAK,CAAA;QACxB,KAAK,MAAM,EAAE,IAAI,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;YAAE,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAA;QACrE,KAAK,MAAM,UAAU,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE;YAAE,UAAU,CAAC,KAAK,EAAE,CAAA;QACnE,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAA;IACvB,CAAC;IAEO,kBAAkB,CAAC,MAAc,EAAE,MAAe;QACxD,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;YACxB,MAAM,EAAE,GAAI,MAA+C,EAAE,EAAE,CAAA;YAC/D,IAAI,EAAE,KAAK,SAAS;gBAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,CAAA;YACpD,OAAM;QACR,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAC9C,CAAC;IAEO,qBAAqB,CAAC,EAAmB,EAAE,MAAc,EAAE,MAAe;QAChF,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAA;QACvC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,IAAI,CAAC,KAAK,CAAC;gBACT,OAAO,EAAE,KAAK;gBACd,EAAE;gBACF,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,qBAAqB,MAAM,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,EAAE;aAClG,CAAC,CAAA;YACF,OAAM;QACR,CAAC;QACD,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAA;QACxC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,EAAE,UAAU,CAAC,CAAA;QACjC,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,CAC7C,CAAC,MAAM,EAAE,EAAE;YACT,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;YACxB,IAAI,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAA;QAC5C,CAAC,EACD,CAAC,KAAc,EAAE,EAAE;YACjB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;YACxB,MAAM,OAAO,GACX,KAAK,YAAY,QAAQ,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS;gBACnD,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE,GAAG,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,EAAE;gBACzF,CAAC,CAAC;oBACE,IAAI,EAAE,CAAC,KAAK;oBACZ,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;oBAC/D,IAAI,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE;iBAC/B,CAAA;YACP,IAAI,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAA;QACpD,CAAC,CACF,CAAA;IACH,CAAC;IAEO,MAAM,CAAC,EAAU,EAAE,OAA8C;QACvE,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;QAClC,IAAI,CAAC,KAAK;YAAE,OAAM;QAClB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;QACvB,KAAK,CAAC,OAAO,EAAE,CAAA;QACf,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA;QAClD,IAAI,OAAO,IAAI,OAAO,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS;YAAE,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;;YAC7E,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;IACpC,CAAC;IAEO,KAAK,CAAC,OAAgB;QAC5B,IAAI,IAAI,CAAC,WAAW;YAAE,OAAM;QAC5B,IAAI,CAAC;YACH,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAA;QAC/C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAA;QAC5F,CAAC;IACH,CAAC;CACF"}

package/dist/schema.d.ts

@@ -0,0 +1,53 @@
+/** The JSON Schema subset afm-bridge supports (PROTOCOL.md section 8). */
+export interface JSONSchema {
+    type?: 'object' | 'array' | 'string' | 'integer' | 'number' | 'boolean' | 'null' | Array<string>;
+    description?: string;
+    properties?: Record<string, JSONSchema>;
+    required?: string[];
+    additionalProperties?: false;
+    items?: JSONSchema;
+    minItems?: number;
+    maxItems?: number;
+    minimum?: number;
+    maximum?: number;
+    enum?: Array<string | null>;
+    const?: string;
+    pattern?: string;
+    anyOf?: JSONSchema[];
+    oneOf?: JSONSchema[];
+    $ref?: string;
+    $defs?: Record<string, JSONSchema>;
+    definitions?: Record<string, JSONSchema>;
+    [annotation: string]: unknown;
+}
+declare const structured: unique symbol;
+/**
+ * A schema plus a parser that turns the generated JSON into `T`. Create one with
+ * `jsonSchema()` or `zodSchema()` from `afm-bridge/zod`.
+ */
+export interface StructuredSchema<T> {
+    readonly [structured]: true;
+    readonly jsonSchema: JSONSchema;
+    /** Validates and converts a generated value. Throws `DecodingError` when it does not match. */
+    parse(value: unknown): T;
+}
+/** A plain JSON Schema object or a `StructuredSchema`. */
+export type SchemaInput<T = unknown> = JSONSchema | StructuredSchema<T>;
+/**
+ * Wraps a JSON Schema. Without `parse` the result is checked by the built-in validator
+ * for the supported subset and typed as `T`.
+ */
+export declare function jsonSchema<T = unknown>(schema: JSONSchema, parse?: (value: unknown) => T): StructuredSchema<T>;
+export declare function isStructuredSchema(value: unknown): value is StructuredSchema<unknown>;
+export declare function toStructuredSchema<T>(input: SchemaInput<T>): StructuredSchema<T>;
+/** Recursively optional, for partial values while a structured response streams in. */
+export type DeepPartial<T> = T extends Array<infer U> ? Array<DeepPartial<U>> : T extends object ? {
+    [K in keyof T]?: DeepPartial<T[K]>;
+} : T;
+interface Issue {
+    path: string;
+    message: string;
+}
+/** Returns the first mismatch between `value` and `schema`, or undefined. `format` is not checked. */
+export declare function validate(schema: JSONSchema, value: unknown): Issue | undefined;
+export {};

package/dist/schema.js

@@ -0,0 +1,149 @@
+import { DecodingError } from './errors.js';
+const structured = Symbol.for('afm-bridge.schema');
+/**
+ * Wraps a JSON Schema. Without `parse` the result is checked by the built-in validator
+ * for the supported subset and typed as `T`.
+ */
+export function jsonSchema(schema, parse) {
+    return {
+        [structured]: true,
+        jsonSchema: schema,
+        parse: parse ??
+            ((value) => {
+                const issue = validate(schema, value);
+                if (issue) {
+                    throw new DecodingError(`Generated value does not match the schema at '${issue.path}': ${issue.message}`, {
+                        type: 'decodingFailure',
+                        data: { path: issue.path, rawContent: JSON.stringify(value) },
+                    });
+                }
+                return value;
+            }),
+    };
+}
+export function isStructuredSchema(value) {
+    return typeof value === 'object' && value !== null && structured in value;
+}
+export function toStructuredSchema(input) {
+    return isStructuredSchema(input) ? input : jsonSchema(input);
+}
+/** Returns the first mismatch between `value` and `schema`, or undefined. `format` is not checked. */
+export function validate(schema, value) {
+    return check(schema, value, '#', schema, 0);
+}
+function check(schema, value, path, root, depth) {
+    if (depth > 64)
+        return undefined;
+    const fail = (message) => ({ path, message });
+    if (schema.$ref !== undefined) {
+        const target = resolveRef(schema.$ref, root);
+        if (!target)
+            return fail(`cannot resolve ${schema.$ref}`);
+        return check(target, value, path, root, depth + 1);
+    }
+    const branches = schema.anyOf ?? schema.oneOf;
+    if (branches) {
+        const ok = branches.some((b) => !check(b, value, path, root, depth + 1));
+        return ok ? undefined : fail('does not match any of the allowed schemas');
+    }
+    if (schema.enum) {
+        return schema.enum.includes(value)
+            ? undefined
+            : fail(`must be one of ${JSON.stringify(schema.enum)}`);
+    }
+    if (schema.const !== undefined) {
+        return value === schema.const ? undefined : fail(`must be ${JSON.stringify(schema.const)}`);
+    }
+    const types = schema.type === undefined ? [] : Array.isArray(schema.type) ? schema.type : [schema.type];
+    if (value === null) {
+        return types.includes('null') ? undefined : fail('must not be null');
+    }
+    const type = types.find((t) => t !== 'null');
+    switch (type) {
+        case 'string': {
+            if (typeof value !== 'string')
+                return fail('must be a string');
+            // A pattern that is valid for Swift but not for JavaScript is skipped, not failed.
+            const pattern = schema.pattern === undefined ? undefined : safeRegExp(schema.pattern);
+            if (pattern && !pattern.test(value))
+                return fail(`must match /${schema.pattern}/`);
+            return undefined;
+        }
+        case 'integer':
+        case 'number':
+            if (typeof value !== 'number' || !Number.isFinite(value))
+                return fail(`must be a ${type}`);
+            if (type === 'integer' && !Number.isInteger(value))
+                return fail('must be an integer');
+            if (schema.minimum !== undefined && value < schema.minimum)
+                return fail(`must be >= ${schema.minimum}`);
+            if (schema.maximum !== undefined && value > schema.maximum)
+                return fail(`must be <= ${schema.maximum}`);
+            return undefined;
+        case 'boolean':
+            return typeof value === 'boolean' ? undefined : fail('must be a boolean');
+        case 'array': {
+            if (!Array.isArray(value))
+                return fail('must be an array');
+            if (schema.minItems !== undefined && value.length < schema.minItems) {
+                return fail(`must have at least ${schema.minItems} items`);
+            }
+            if (schema.maxItems !== undefined && value.length > schema.maxItems) {
+                return fail(`must have at most ${schema.maxItems} items`);
+            }
+            if (!schema.items)
+                return undefined;
+            for (const [i, item] of value.entries()) {
+                const issue = check(schema.items, item, `${path}/${i}`, root, depth + 1);
+                if (issue)
+                    return issue;
+            }
+            return undefined;
+        }
+        case 'object': {
+            if (typeof value !== 'object' || Array.isArray(value))
+                return fail('must be an object');
+            const record = value;
+            const properties = schema.properties ?? {};
+            for (const key of schema.required ?? []) {
+                if (!(key in record))
+                    return fail(`is missing required property '${key}'`);
+            }
+            for (const [key, v] of Object.entries(record)) {
+                const sub = properties[key];
+                if (!sub) {
+                    if (schema.additionalProperties === false)
+                        return fail(`has unexpected property '${key}'`);
+                    continue;
+                }
+                const issue = check(sub, v, `${path}/${escapePointer(key)}`, root, depth + 1);
+                if (issue)
+                    return issue;
+            }
+            return undefined;
+        }
+        default:
+            return undefined;
+    }
+}
+function resolveRef(ref, root) {
+    if (ref === '#')
+        return root;
+    const match = /^#\/(\$defs|definitions)\/(.+)$/.exec(ref);
+    if (!match?.[1] || !match[2])
+        return undefined;
+    const defs = (match[1] === '$defs' ? root.$defs : root.definitions) ?? {};
+    return defs[match[2].replace(/~1/g, '/').replace(/~0/g, '~')];
+}
+function escapePointer(key) {
+    return key.replace(/~/g, '~0').replace(/\//g, '~1');
+}
+function safeRegExp(pattern) {
+    try {
+        return new RegExp(pattern, 'u');
+    }
+    catch {
+        return undefined;
+    }
+}
+//# sourceMappingURL=schema.js.map

package/dist/schema.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.js","sourceRoot":"","sources":["../src/schema.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAA;AAyB3C,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAA;AAgBlD;;;GAGG;AACH,MAAM,UAAU,UAAU,CACxB,MAAkB,EAClB,KAA6B;IAE7B,OAAO;QACL,CAAC,UAAU,CAAC,EAAE,IAAI;QAClB,UAAU,EAAE,MAAM;QAClB,KAAK,EACH,KAAK;YACL,CAAC,CAAC,KAAc,EAAE,EAAE;gBAClB,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,KAAK,CAAC,CAAA;gBACrC,IAAI,KAAK,EAAE,CAAC;oBACV,MAAM,IAAI,aAAa,CACrB,iDAAiD,KAAK,CAAC,IAAI,MAAM,KAAK,CAAC,OAAO,EAAE,EAChF;wBACE,IAAI,EAAE,iBAAiB;wBACvB,IAAI,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE;qBAC9D,CACF,CAAA;gBACH,CAAC;gBACD,OAAO,KAAU,CAAA;YACnB,CAAC,CAAC;KACL,CAAA;AACH,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,KAAc;IAC/C,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,UAAU,IAAI,KAAK,CAAA;AAC3E,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAI,KAAqB;IACzD,OAAO,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAE,KAA6B,CAAC,CAAC,CAAC,UAAU,CAAI,KAAmB,CAAC,CAAA;AACxG,CAAC;AAiBD,sGAAsG;AACtG,MAAM,UAAU,QAAQ,CAAC,MAAkB,EAAE,KAAc;IACzD,OAAO,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC,CAAA;AAC7C,CAAC;AAED,SAAS,KAAK,CACZ,MAAkB,EAClB,KAAc,EACd,IAAY,EACZ,IAAgB,EAChB,KAAa;IAEb,IAAI,KAAK,GAAG,EAAE;QAAE,OAAO,SAAS,CAAA;IAChC,MAAM,IAAI,GAAG,CAAC,OAAe,EAAS,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAA;IAE5D,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,CAAA;QAC5C,IAAI,CAAC,MAAM;YAAE,OAAO,IAAI,CAAC,kBAAkB,MAAM,CAAC,IAAI,EAAE,CAAC,CAAA;QACzD,OAAO,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,GAAG,CAAC,CAAC,CAAA;IACpD,CAAC;IACD,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAA;IAC7C,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC,CAAA;QACxE,OAAO,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,2CAA2C,CAAC,CAAA;IAC3E,CAAC;IACD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;QAChB,OAAO,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAsB,CAAC;YACjD,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,IAAI,CAAC,kBAAkB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;IAC3D,CAAC;IACD,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QAC/B,OAAO,KAAK,KAAK,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAA;IAC7F,CAAC;IAED,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;IACvG,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QACnB,OAAO,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAA;IACtE,CAAC;IACD,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,MAAM,CAAC,CAAA;IAC5C,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,QAAQ,EAAE,CAAC;YACd,IAAI,OAAO,KAAK,KAAK,QAAQ;gBAAE,OAAO,IAAI,CAAC,kBAAkB,CAAC,CAAA;YAC9D,mFAAmF;YACnF,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YACrF,IAAI,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC;gBAAE,OAAO,IAAI,CAAC,eAAe,MAAM,CAAC,OAAO,GAAG,CAAC,CAAA;YAClF,OAAO,SAAS,CAAA;QAClB,CAAC;QACD,KAAK,SAAS,CAAC;QACf,KAAK,QAAQ;YACX,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC;gBAAE,OAAO,IAAI,CAAC,aAAa,IAAI,EAAE,CAAC,CAAA;YAC1F,IAAI,IAAI,KAAK,SAAS,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC;gBAAE,OAAO,IAAI,CAAC,oBAAoB,CAAC,CAAA;YACrF,IAAI,MAAM,CAAC,OAAO,KAAK,SAAS,IAAI,KAAK,GAAG,MAAM,CAAC,OAAO;gBAAE,OAAO,IAAI,CAAC,cAAc,MAAM,CAAC,OAAO,EAAE,CAAC,CAAA;YACvG,IAAI,MAAM,CAAC,OAAO,KAAK,SAAS,IAAI,KAAK,GAAG,MAAM,CAAC,OAAO;gBAAE,OAAO,IAAI,CAAC,cAAc,MAAM,CAAC,OAAO,EAAE,CAAC,CAAA;YACvG,OAAO,SAAS,CAAA;QAClB,KAAK,SAAS;YACZ,OAAO,OAAO,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAA;QAC3E,KAAK,OAAO,EAAE,CAAC;YACb,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;gBAAE,OAAO,IAAI,CAAC,kBAAkB,CAAC,CAAA;YAC1D,IAAI,MAAM,CAAC,QAAQ,KAAK,SAAS,IAAI,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;gBACpE,OAAO,IAAI,CAAC,sBAAsB,MAAM,CAAC,QAAQ,QAAQ,CAAC,CAAA;YAC5D,CAAC;YACD,IAAI,MAAM,CAAC,QAAQ,KAAK,SAAS,IAAI,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;gBACpE,OAAO,IAAI,CAAC,qBAAqB,MAAM,CAAC,QAAQ,QAAQ,CAAC,CAAA;YAC3D,CAAC;YACD,IAAI,CAAC,MAAM,CAAC,KAAK;gBAAE,OAAO,SAAS,CAAA;YACnC,KAAK,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;gBACxC,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,GAAG,IAAI,IAAI,CAAC,EAAE,EAAE,IAAI,EAAE,KAAK,GAAG,CAAC,CAAC,CAAA;gBACxE,IAAI,KAAK;oBAAE,OAAO,KAAK,CAAA;YACzB,CAAC;YACD,OAAO,SAAS,CAAA;QAClB,CAAC;QACD,KAAK,QAAQ,EAAE,CAAC;YACd,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;gBAAE,OAAO,IAAI,CAAC,mBAAmB,CAAC,CAAA;YACvF,MAAM,MAAM,GAAG,KAAgC,CAAA;YAC/C,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,EAAE,CAAA;YAC1C,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,QAAQ,IAAI,EAAE,EAAE,CAAC;gBACxC,IAAI,CAAC,CAAC,GAAG,IAAI,MAAM,CAAC;oBAAE,OAAO,IAAI,CAAC,iCAAiC,GAAG,GAAG,CAAC,CAAA;YAC5E,CAAC;YACD,KAAK,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC9C,MAAM,GAAG,GAAG,UAAU,CAAC,GAAG,CAAC,CAAA;gBAC3B,IAAI,CAAC,GAAG,EAAE,CAAC;oBACT,IAAI,MAAM,CAAC,oBAAoB,KAAK,KAAK;wBAAE,OAAO,IAAI,CAAC,4BAA4B,GAAG,GAAG,CAAC,CAAA;oBAC1F,SAAQ;gBACV,CAAC;gBACD,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,GAAG,IAAI,IAAI,aAAa,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,EAAE,KAAK,GAAG,CAAC,CAAC,CAAA;gBAC7E,IAAI,KAAK;oBAAE,OAAO,KAAK,CAAA;YACzB,CAAC;YACD,OAAO,SAAS,CAAA;QAClB,CAAC;QACD;YACE,OAAO,SAAS,CAAA;IACpB,CAAC;AACH,CAAC;AAED,SAAS,UAAU,CAAC,GAAW,EAAE,IAAgB;IAC/C,IAAI,GAAG,KAAK,GAAG;QAAE,OAAO,IAAI,CAAA;IAC5B,MAAM,KAAK,GAAG,iCAAiC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IACzD,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QAAE,OAAO,SAAS,CAAA;IAC9C,MAAM,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,CAAA;IACzE,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,CAAA;AAC/D,CAAC;AAED,SAAS,aAAa,CAAC,GAAW;IAChC,OAAO,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAA;AACrD,CAAC;AAED,SAAS,UAAU,CAAC,OAAe;IACjC,IAAI,CAAC;QACH,OAAO,IAAI,MAAM,CAAC,OAAO,EAAE,GAAG,CAAC,CAAA;IACjC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAA;IAClB,CAAC;AACH,CAAC"}

package/dist/session.d.ts

@@ -0,0 +1,63 @@
+import type { RequestOptions } from './rpc.js';
+import type { GenerateResult, ModelId, Prompt, RespondOptions, SessionInfo, StreamChunk, StructuredOptions, StructuredResult, StructuredStreamChunk } from './types.js';
+/** What a session needs from the client. Internal. */
+export interface SessionHost {
+    request<T>(epoch: number, method: string, params: unknown, options?: RequestOptions): Promise<T>;
+    listen(epoch: number, requestId: number, listener: (params: DeltaParams) => void): () => void;
+    disposed(epoch: number, sessionId: string): void;
+}
+export interface DeltaParams {
+    requestId: number;
+    delta?: string;
+    reset?: boolean;
+    text?: string;
+    partial?: unknown;
+}
+/**
+ * A conversation with the model. The sidecar keeps its history, so every call sees
+ * the previous prompts and responses. Only one `respond`/`stream` may run at a time.
+ */
+export declare class Session {
+    private readonly host;
+    private readonly epoch;
+    readonly id: string;
+    /** Context window in tokens. */
+    readonly contextSize: number;
+    readonly model: ModelId;
+    private disposed;
+    /** @internal Use `AppleFoundationModels.createSession()`. */
+    constructor(host: SessionHost, epoch: number, id: string, 
+    /** Context window in tokens. */
+    contextSize: number, model: ModelId);
+    /**
+     * Generates a complete response. Returns the text, or with `schema` the parsed and
+     * validated value.
+     */
+    respond(prompt: Prompt, options?: RespondOptions & {
+        schema?: undefined;
+    }): Promise<string>;
+    respond<T>(prompt: Prompt, options: StructuredOptions<T>): Promise<T>;
+    /** Like `respond`, but also returns token usage (macOS 27+). */
+    generate(prompt: Prompt, options?: RespondOptions & {
+        schema?: undefined;
+    }): Promise<GenerateResult>;
+    generate<T>(prompt: Prompt, options: StructuredOptions<T>): Promise<StructuredResult<T>>;
+    /**
+     * Streams the response. Text: each chunk carries the new `delta` and the full `text`
+     * so far. With `schema`: each chunk carries the `partial` value so far. The last chunk
+     * has `done: true` with the complete result. Breaking out of the loop cancels the generation.
+     */
+    stream(prompt: Prompt, options?: RespondOptions & {
+        schema?: undefined;
+    }): AsyncGenerator<StreamChunk, void, undefined>;
+    stream<T>(prompt: Prompt, options: StructuredOptions<T>): AsyncGenerator<StructuredStreamChunk<T>, void, undefined>;
+    /** Loads model resources ahead of time to reduce the latency of the first response. */
+    prewarm(promptPrefix?: string): Promise<void>;
+    /** Context usage of the session. */
+    info(): Promise<SessionInfo>;
+    /** Frees the session in the sidecar and cancels a running generation. Safe to call twice. */
+    dispose(): Promise<void>;
+    private call;
+    private assertUsable;
+    private respondParams;
+}

package/dist/session.js

@@ -0,0 +1,202 @@
+import { AbortError, AfmError, ClientClosedError, SessionLostError, SessionNotFoundError } from './errors.js';
+import { toStructuredSchema } from './schema.js';
+/**
+ * A conversation with the model. The sidecar keeps its history, so every call sees
+ * the previous prompts and responses. Only one `respond`/`stream` may run at a time.
+ */
+export class Session {
+    host;
+    epoch;
+    id;
+    contextSize;
+    model;
+    disposed = false;
+    /** @internal Use `AppleFoundationModels.createSession()`. */
+    constructor(host, epoch, id, 
+    /** Context window in tokens. */
+    contextSize, model) {
+        this.host = host;
+        this.epoch = epoch;
+        this.id = id;
+        this.contextSize = contextSize;
+        this.model = model;
+    }
+    async respond(prompt, options = {}) {
+        const result = await this.generate(prompt, options);
+        return options.schema ? result.content : result.text;
+    }
+    async generate(prompt, options = {}) {
+        const { signal, timeoutMs } = options;
+        const schema = options.schema ? toStructuredSchema(options.schema) : undefined;
+        const result = await this.call('session/respond', this.respondParams(prompt, options, schema), {
+            signal,
+            timeoutMs,
+        });
+        if (!schema)
+            return withUsage({ text: result.text }, result.usage);
+        return withUsage({ content: parseContent(schema, result), text: result.text }, result.usage);
+    }
+    async *stream(prompt, options = {}) {
+        const { signal, timeoutMs } = options;
+        if (signal?.aborted)
+            throw new AbortError();
+        this.assertUsable();
+        const schema = options.schema ? toStructuredSchema(options.schema) : undefined;
+        const controller = new AbortController();
+        const onAbort = () => controller.abort();
+        signal?.addEventListener('abort', onAbort, { once: true });
+        const queue = new ChunkQueue();
+        let text = '';
+        let unsubscribe = () => { };
+        const request = this.host.request(this.epoch, 'session/respond', { ...this.respondParams(prompt, options, schema), stream: true }, {
+            signal: controller.signal,
+            timeoutMs,
+            onStart: (id) => {
+                unsubscribe = this.host.listen(this.epoch, id, (p) => {
+                    if (schema) {
+                        if (p.partial !== undefined)
+                            queue.push({ done: false, partial: p.partial });
+                    }
+                    else if (p.reset) {
+                        text = p.text ?? '';
+                        queue.push({ done: false, delta: '', text, reset: true });
+                    }
+                    else if (p.delta) {
+                        text += p.delta;
+                        queue.push({ done: false, delta: p.delta, text });
+                    }
+                });
+            },
+        });
+        request.then((result) => queue.end(result), (error) => queue.fail(error));
+        let finished = false;
+        try {
+            while (true) {
+                const item = await queue.next();
+                if (item.kind === 'chunk') {
+                    yield item.chunk;
+                    continue;
+                }
+                finished = true;
+                const { result } = item;
+                yield schema
+                    ? withUsage({ done: true, content: parseContent(schema, result), text: result.text }, result.usage)
+                    : withUsage({ done: true, delta: '', text: result.text }, result.usage);
+                return;
+            }
+        }
+        finally {
+            unsubscribe();
+            signal?.removeEventListener('abort', onAbort);
+            if (!finished) {
+                // The consumer stopped early or an error was thrown: make sure the
+                // generation is cancelled and the session is free before we return.
+                controller.abort();
+                await request.catch(() => { });
+            }
+        }
+    }
+    /** Loads model resources ahead of time to reduce the latency of the first response. */
+    async prewarm(promptPrefix) {
+        await this.call('session/prewarm', { sessionId: this.id, ...(promptPrefix ? { promptPrefix } : {}) });
+    }
+    /** Context usage of the session. */
+    async info() {
+        return this.call('session/info', { sessionId: this.id });
+    }
+    /** Frees the session in the sidecar and cancels a running generation. Safe to call twice. */
+    async dispose() {
+        if (this.disposed)
+            return;
+        this.disposed = true;
+        this.host.disposed(this.epoch, this.id);
+        try {
+            await this.host.request(this.epoch, 'session/dispose', { sessionId: this.id });
+        }
+        catch (error) {
+            // Nothing to free when the process is already gone.
+            if (error instanceof SessionLostError || error instanceof ClientClosedError)
+                return;
+            throw error;
+        }
+    }
+    call(method, params, options) {
+        try {
+            this.assertUsable();
+        }
+        catch (error) {
+            return Promise.reject(error);
+        }
+        return this.host.request(this.epoch, method, params, options);
+    }
+    assertUsable() {
+        if (this.disposed) {
+            throw new SessionNotFoundError(`Session ${this.id} was disposed`, {
+                type: 'sessionNotFound',
+                data: { sessionId: this.id },
+            });
+        }
+    }
+    respondParams(prompt, options, schema) {
+        const { temperature, maximumResponseTokens, sampling, toolCalling, reasoningLevel } = options;
+        const generation = {
+            temperature,
+            maximumResponseTokens,
+            sampling,
+            toolCalling,
+            reasoningLevel,
+        };
+        const defined = Object.fromEntries(Object.entries(generation).filter(([, v]) => v !== undefined));
+        return {
+            sessionId: this.id,
+            prompt,
+            ...(Object.keys(defined).length > 0 ? { options: defined } : {}),
+            ...(schema ? { schema: schema.jsonSchema } : {}),
+            ...(schema && options.includeSchemaInPrompt !== undefined
+                ? { includeSchemaInPrompt: options.includeSchemaInPrompt }
+                : {}),
+        };
+    }
+}
+function parseContent(schema, result) {
+    return schema.parse(result.content !== undefined ? result.content : JSON.parse(result.text));
+}
+function withUsage(value, usage) {
+    return usage ? { ...value, usage } : value;
+}
+/** Single-consumer queue that delivers buffered chunks before an error. */
+class ChunkQueue {
+    items = [];
+    error;
+    waiter;
+    push(chunk) {
+        this.items.push({ kind: 'chunk', chunk });
+        this.wake();
+    }
+    end(result) {
+        this.items.push({ kind: 'end', result });
+        this.wake();
+    }
+    fail(error) {
+        this.error = { value: error };
+        this.wake();
+    }
+    async next() {
+        while (true) {
+            const item = this.items.shift();
+            if (item)
+                return item;
+            if (this.error)
+                throw this.error.value instanceof Error ? this.error.value : new AfmError(String(this.error.value));
+            await new Promise((resolve) => {
+                this.waiter = resolve;
+            });
+        }
+    }
+    wake() {
+        const waiter = this.waiter;
+        this.waiter = undefined;
+        waiter?.();
+    }
+}
+//# sourceMappingURL=session.js.map