@adonis-agora/agent 0.55.0 → 0.57.0

package/dist/src/ag-ui/encoder.d.ts

@@ -1,107 +1,22 @@
+import { type AgUiEncoderOptions, type AgUiEvent } from '@dudousxd/nestjs-agent-core/ag-ui';
 import type { StreamFrame } from '../spi/token-stream-sink.js';
-import { type AgUiEvent } from './types.js';
-export interface AgUiEncoderOptions {
-    /** The ids the consumer sent on `RunAgentInput`: every boundary event echoes them. */
-    threadId: string;
-    runId: string;
-    /** The library run whose stream is being projected (it mints interrupt ids and the `agora.run` event). */
-    streamRunId: string;
-    /** The library's thread id, when it differs from what the consumer calls the thread. */
-    streamThreadId?: string;
-    /**
-     * Frames of the library run's stream that an earlier AG-UI run already delivered (a resume). They
-     * are fed to the encoder like any other — it has to know what they opened — but produce no events.
-     */
-    skip?: number;
-    /**
-     * Tool calls a resuming request just answered. The run has not necessarily said so yet — a parked
-     * run takes a moment to wake — so without this the encoder would read them as still waiting and
-     * report the interrupt again before the answer had any effect.
-     */
-    answered?: readonly string[];
-    /** Model id for the usage entry when the step frames carry none. */
-    model?: string;
-}
+export type { AgUiEncoderOptions };
 /**
- * Projects one library run's stream ({@link StreamFrame}s, in buffer order) onto AG-UI 1.0 events.
- *
- * A pure state machine: the same frames always give the same events, which is what lets a resuming
- * request — served by any replica — replay the run's buffer, skip what the interrupted AG-UI run
- * already delivered, and continue. It writes nothing itself; {@link agUiEvents} drives it and
- * decides when the run has stopped to ask.
- *
- * What maps where:
- *  - `text` → `TEXT_MESSAGE_*` (one message per stretch of text);
- *  - `reasoning` → a reasoning span with one reasoning message;
- *  - `step-start`/`step-finish` → `STEP_STARTED`/`STEP_FINISHED`, the step's usage folded into the
- *    run's `usage`;
- *  - `tool-input-*` → `TOOL_CALL_START`/`ARGS`/`END`; `tool-output*` → `TOOL_CALL_RESULT`;
- *  - an approval or a question set the run parks on → an `Interrupt` on the closing `RUN_FINISHED`;
- *  - `cancelled` → the cancelled outcome; an error frame → `RUN_ERROR`;
- *  - generative UI, title, queue and the native approval/elicitation frames → `CUSTOM` events named
- *    in {@link AG_UI_CUSTOM}, for what AG-UI has no event for.
- *
- * Not mapped: sub-agent attribution (`subagentRunId`) — a delegated agent's frames are projected
- * flat, as the protocol allows of a producer that does not attribute.
+ * The shared AG-UI encoder (`@dudousxd/nestjs-agent-core/ag-ui`, one implementation for both
+ * servers), reading this package's {@link StreamFrame}s. A pure state machine: the same frames
+ * always give the same events.
  */
 export declare class AgUiEncoder {
-    private readonly options;
-    private position;
-    private started;
-    private closed;
-    private cancelled;
-    private crossed;
-    private messageSeq;
-    private stepSeq;
-    private openText;
-    private openReasoning;
-    /** The last text message of the current step: a tool call attaches to it. */
-    private stepMessage;
-    private readonly openSteps;
-    private readonly calls;
-    private readonly pending;
-    private readonly usage;
+    private readonly shared;
     constructor(options: AgUiEncoderOptions);
     /** How many frames of the library stream this encoder has consumed. */
     get consumed(): number;
     /** The run is waiting on at least one approval or question set nobody has settled yet. */
     get waiting(): boolean;
-    /**
-     * Is anything still in flight that is NOT waiting on a person — a tool call announced and neither
-     * answered nor parked? While there is, more frames are coming without anyone's help.
-     */
+    /** A tool call is announced and neither answered nor parked: more frames are coming. */
     get busy(): boolean;
-    /** `RUN_STARTED`, and the library's own ids for the run. Call once, before the first frame. */
     start(): AgUiEvent[];
-    /** The AG-UI events the next frame of the library stream stands for (possibly none). */
     encode(frame: StreamFrame): AgUiEvent[];
-    /**
-     * The replayed frames are behind us: from here on events are written. The AG-UI run that delivered
-     * them closed everything it had open before it finished, so this run starts with nothing open, and
-     * what the resuming request answered is no longer waiting.
-     */
-    private crossBoundary;
-    private get replaying();
-    private project;
-    private event;
-    /**
-     * The events that end the AG-UI run: whatever is still open is closed first, then `RUN_FINISHED`
-     * with how it ended — interrupted when the run is waiting on someone, cancelled when it was
-     * stopped, success otherwise. Nothing after a `RUN_ERROR`. `ended` says the library run's stream
-     * itself ended, rather than this reader having stopped following it.
-     */
     finish(ended?: boolean): AgUiEvent[];
-    private interrupts;
-    private custom;
-    private nextMessageId;
-    private textDelta;
-    private reasoningDelta;
-    private closeText;
-    private closeReasoning;
-    private closeStreams;
-    private callStart;
-    private result;
-    private addUsage;
-    private usageEntries;
 }
 //# sourceMappingURL=encoder.d.ts.map

package/dist/src/ag-ui/encoder.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"encoder.d.ts","sourceRoot":"","sources":["../../../src/ag-ui/encoder.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,6BAA6B,CAAC;AAI/D,OAAO,EAGL,KAAK,SAAS,EAGf,MAAM,YAAY,CAAC;AAEpB,MAAM,WAAW,kBAAkB;IACjC,sFAAsF;IACtF,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,0GAA0G;IAC1G,WAAW,EAAE,MAAM,CAAC;IACpB,wFAAwF;IACxF,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;IACd;;;;OAIG;IACH,QAAQ,CAAC,EAAE,SAAS,MAAM,EAAE,CAAC;IAC7B,oEAAoE;IACpE,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AA0DD;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,qBAAa,WAAW;IAiBV,OAAO,CAAC,QAAQ,CAAC,OAAO;IAhBpC,OAAO,CAAC,QAAQ,CAAK;IACrB,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,UAAU,CAAK;IACvB,OAAO,CAAC,OAAO,CAAK;IACpB,OAAO,CAAC,QAAQ,CAAqB;IACrC,OAAO,CAAC,aAAa,CAAqB;IAC1C,6EAA6E;IAC7E,OAAO,CAAC,WAAW,CAAqB;IACxC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAgB;IAC1C,OAAO,CAAC,QAAQ,CAAC,KAAK,CAA+B;IACrD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAA8B;IACtD,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAqC;IAE3D,YAA6B,OAAO,EAAE,kBAAkB,EAAI;IAE5D,uEAAuE;IACvE,IAAI,QAAQ,IAAI,MAAM,CAErB;IAED,0FAA0F;IAC1F,IAAI,OAAO,IAAI,OAAO,CAErB;IAED;;;OAGG;IACH,IAAI,IAAI,IAAI,OAAO,CAKlB;IAED,+FAA+F;IAC/F,KAAK,IAAI,SAAS,EAAE,CAmBnB;IAED,wFAAwF;IACxF,MAAM,CAAC,KAAK,EAAE,WAAW,GAAG,SAAS,EAAE,CAgBtC;IAED;;;;OAIG;IACH,OAAO,CAAC,aAAa;IAarB,OAAO,KAAK,SAAS,GAEpB;IAED,OAAO,CAAC,OAAO;IA+Ef,OAAO,CAAC,KAAK;IAgHb;;;;;OAKG;IACH,MAAM,CAAC,KAAK,UAAQ,GAAG,SAAS,EAAE,CA+BjC;IAED,OAAO,CAAC,UAAU;IAalB,OAAO,CAAC,MAAM;IAId,OAAO,CAAC,aAAa;IAKrB,OAAO,CAAC,SAAS;IAYjB,OAAO,CAAC,cAAc;IActB,OAAO,CAAC,SAAS;IAOjB,OAAO,CAAC,cAAc;IAUtB,OAAO,CAAC,YAAY;IAIpB,OAAO,CAAC,SAAS;IASjB,OAAO,CAAC,MAAM;IA0Bd,OAAO,CAAC,QAAQ;IAsBhB,OAAO,CAAC,YAAY;CAGrB"}
+{"version":3,"file":"encoder.d.ts","sourceRoot":"","sources":["../../../src/ag-ui/encoder.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,kBAAkB,EACvB,KAAK,SAAS,EAEf,MAAM,mCAAmC,CAAC;AAC3C,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,6BAA6B,CAAC;AAG/D,YAAY,EAAE,kBAAkB,EAAE,CAAC;AAEnC;;;;GAIG;AACH,qBAAa,WAAW;IACtB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAgB;IAEvC,YAAY,OAAO,EAAE,kBAAkB,EAEtC;IAED,uEAAuE;IACvE,IAAI,QAAQ,IAAI,MAAM,CAErB;IAED,0FAA0F;IAC1F,IAAI,OAAO,IAAI,OAAO,CAErB;IAED,wFAAwF;IACxF,IAAI,IAAI,IAAI,OAAO,CAElB;IAED,KAAK,IAAI,SAAS,EAAE,CAEnB;IAED,MAAM,CAAC,KAAK,EAAE,WAAW,GAAG,SAAS,EAAE,CAEtC;IAED,MAAM,CAAC,KAAK,UAAQ,GAAG,SAAS,EAAE,CAEjC;CACF"}

package/dist/src/ag-ui/encoder.js

@@ -1,519 +1,35 @@
-import { encodeInterruptId } from './interrupt-id.js';
-import { AG_UI_CUSTOM, AG_UI_PROTOCOL_VERSION, } from './types.js';
-const APPROVAL_SCHEMA = {
-    type: 'object',
-    properties: {
-        approved: { type: 'boolean' },
-        reason: { type: 'string' },
-        remember: { type: 'boolean' },
-    },
-    required: ['approved'],
-};
-/** The answer an elicitation expects: `{ answers: { <questionId>: string[] } }`. */
-function elicitationSchema(request) {
-    const properties = {};
-    for (const question of request.questions) {
-        const options = question.options?.map((option) => option.value);
-        properties[question.id] = {
-            type: 'array',
-            title: question.prompt,
-            ...(question.description !== undefined ? { description: question.description } : {}),
-            items: options !== undefined && options.length > 0 && question.allowFreeText !== true
-                ? { type: 'string', enum: options }
-                : { type: 'string' },
-            ...(question.multiple === true ? {} : { maxItems: 1 }),
-            ...(question.defaults !== undefined ? { default: question.defaults } : {}),
-        };
-    }
-    return {
-        type: 'object',
-        properties: { answers: { type: 'object', properties } },
-        required: ['answers'],
-    };
-}
-function text(value) {
-    if (typeof value === 'string')
-        return value;
-    const json = JSON.stringify(value);
-    return json === undefined ? '' : json;
-}
+import { AgUiEncoder as SharedEncoder, } from '@dudousxd/nestjs-agent-core/ag-ui';
+import { toAgUiFrame } from './frames.js';
 /**
- * Projects one library run's stream ({@link StreamFrame}s, in buffer order) onto AG-UI 1.0 events.
- *
- * A pure state machine: the same frames always give the same events, which is what lets a resuming
- * request — served by any replica — replay the run's buffer, skip what the interrupted AG-UI run
- * already delivered, and continue. It writes nothing itself; {@link agUiEvents} drives it and
- * decides when the run has stopped to ask.
- *
- * What maps where:
- *  - `text` → `TEXT_MESSAGE_*` (one message per stretch of text);
- *  - `reasoning` → a reasoning span with one reasoning message;
- *  - `step-start`/`step-finish` → `STEP_STARTED`/`STEP_FINISHED`, the step's usage folded into the
- *    run's `usage`;
- *  - `tool-input-*` → `TOOL_CALL_START`/`ARGS`/`END`; `tool-output*` → `TOOL_CALL_RESULT`;
- *  - an approval or a question set the run parks on → an `Interrupt` on the closing `RUN_FINISHED`;
- *  - `cancelled` → the cancelled outcome; an error frame → `RUN_ERROR`;
- *  - generative UI, title, queue and the native approval/elicitation frames → `CUSTOM` events named
- *    in {@link AG_UI_CUSTOM}, for what AG-UI has no event for.
- *
- * Not mapped: sub-agent attribution (`subagentRunId`) — a delegated agent's frames are projected
- * flat, as the protocol allows of a producer that does not attribute.
+ * The shared AG-UI encoder (`@dudousxd/nestjs-agent-core/ag-ui`, one implementation for both
+ * servers), reading this package's {@link StreamFrame}s. A pure state machine: the same frames
+ * always give the same events.
  */
 export class AgUiEncoder {
-    options;
-    position = 0;
-    started = false;
-    closed = false;
-    cancelled = false;
-    crossed = false;
-    messageSeq = 0;
-    stepSeq = 0;
-    openText;
-    openReasoning;
-    /** The last text message of the current step: a tool call attaches to it. */
-    stepMessage;
-    openSteps = [];
-    calls = new Map();
-    pending = new Map();
-    usage = new Map();
+    shared;
     constructor(options) {
-        this.options = options;
+        this.shared = new SharedEncoder(options);
     }
     /** How many frames of the library stream this encoder has consumed. */
     get consumed() {
-        return this.position;
+        return this.shared.consumed;
     }
     /** The run is waiting on at least one approval or question set nobody has settled yet. */
     get waiting() {
-        return this.pending.size > 0;
+        return this.shared.waiting;
     }
-    /**
-     * Is anything still in flight that is NOT waiting on a person — a tool call announced and neither
-     * answered nor parked? While there is, more frames are coming without anyone's help.
-     */
+    /** A tool call is announced and neither answered nor parked: more frames are coming. */
     get busy() {
-        for (const [id, call] of this.calls) {
-            if (!call.answered && !this.pending.has(id))
-                return true;
-        }
-        return false;
+        return this.shared.busy;
     }
-    /** `RUN_STARTED`, and the library's own ids for the run. Call once, before the first frame. */
     start() {
-        if (this.started)
-            return [];
-        this.started = true;
-        return [
-            {
-                type: 'RUN_STARTED',
-                threadId: this.options.threadId,
-                runId: this.options.runId,
-                protocolVersion: AG_UI_PROTOCOL_VERSION,
-            },
-            {
-                type: 'CUSTOM',
-                name: AG_UI_CUSTOM.run,
-                value: {
-                    runId: this.options.streamRunId,
-                    threadId: this.options.streamThreadId ?? this.options.threadId,
-                },
-            },
-        ];
+        return this.shared.start();
     }
-    /** The AG-UI events the next frame of the library stream stands for (possibly none). */
     encode(frame) {
-        const position = this.position;
-        this.position += 1;
-        if (this.closed)
-            return [];
-        const skip = this.options.skip ?? 0;
-        if (!this.crossed && position >= skip)
-            this.crossBoundary();
-        const events = this.project(frame, position);
-        // A frame an earlier AG-UI run already delivered still moves the state (what is open, what is
-        // pending; NOT what was spent — usage is per run) but writes nothing. The last of them is the
-        // boundary: crossed at once, not when the next frame happens to arrive, because a run that was
-        // just answered may take a moment to write one.
-        if (position < skip) {
-            if (this.position >= skip)
-                this.crossBoundary();
-            return [];
-        }
-        return events;
-    }
-    /**
-     * The replayed frames are behind us: from here on events are written. The AG-UI run that delivered
-     * them closed everything it had open before it finished, so this run starts with nothing open, and
-     * what the resuming request answered is no longer waiting.
-     */
-    crossBoundary() {
-        this.crossed = true;
-        this.openText = undefined;
-        this.openReasoning = undefined;
-        this.stepMessage = undefined;
-        this.openSteps.length = 0;
-        for (const call of this.calls.values()) {
-            call.started = false;
-            call.ended = true;
-        }
-        for (const id of this.options.answered ?? [])
-            this.pending.delete(id);
-    }
-    get replaying() {
-        return this.position <= (this.options.skip ?? 0);
+        return this.shared.encode(toAgUiFrame(frame));
     }
-    project(frame, position) {
-        switch (frame.t) {
-            case 'text':
-                return frame.v.length > 0 ? this.textDelta(frame.v) : [];
-            case 'component':
-                return [
-                    this.custom(AG_UI_CUSTOM.ui, {
-                        id: frame.id ?? `ui:${position}`,
-                        component: frame.name,
-                        props: frame.data,
-                        ...(frame.version !== undefined ? { version: frame.version } : {}),
-                        ...(frame.toolCallId !== undefined ? { toolCallId: frame.toolCallId } : {}),
-                    }),
-                ];
-            case 'approval': {
-                this.pending.set(frame.id, {
-                    kind: 'approval',
-                    parked: frame.runId,
-                    toolCallId: frame.id,
-                    interrupt: {
-                        reason: 'tool_approval',
-                        message: `Approve ${frame.toolName}?`,
-                        toolCallId: frame.id,
-                        responseSchema: APPROVAL_SCHEMA,
-                        ...(frame.expiresAt !== undefined ? { expiresAt: frame.expiresAt } : {}),
-                        metadata: {
-                            'agora.toolName': frame.toolName,
-                            'agora.input': frame.input ?? null,
-                            'agora.approver': frame.approver ?? 'requester',
-                        },
-                    },
-                });
-                return [
-                    this.custom(AG_UI_CUSTOM.approvalRequested, {
-                        id: frame.id,
-                        runId: frame.runId,
-                        toolName: frame.toolName,
-                        input: frame.input ?? null,
-                        approver: frame.approver ?? 'requester',
-                        ...(frame.expiresAt !== undefined ? { expiresAt: frame.expiresAt } : {}),
-                    }),
-                ];
-            }
-            case 'elicitation': {
-                this.pending.set(frame.id, {
-                    kind: 'elicitation',
-                    parked: frame.runId,
-                    toolCallId: frame.id,
-                    interrupt: {
-                        reason: 'input_required',
-                        message: frame.request.preamble ?? frame.request.questions[0]?.prompt ?? 'Input required',
-                        // An intake's question set belongs to no tool call the stream announced.
-                        ...(this.calls.has(frame.id) ? { toolCallId: frame.id } : {}),
-                        responseSchema: elicitationSchema(frame.request),
-                        metadata: { 'agora.request': frame.request },
-                    },
-                });
-                return [
-                    this.custom(AG_UI_CUSTOM.elicitation, { runId: frame.runId, request: frame.request }),
-                ];
-            }
-            case 'error': {
-                this.closed = true;
-                const usage = this.usageEntries();
-                return [
-                    {
-                        type: 'RUN_ERROR',
-                        message: frame.message,
-                        code: frame.code,
-                        ...(usage.length > 0 ? { usage } : {}),
-                    },
-                ];
-            }
-            case 'event':
-                return this.event(frame.event);
-        }
-    }
-    event(event) {
-        switch (event.kind) {
-            case 'text':
-                return event.text.length > 0 ? this.textDelta(event.text) : [];
-            case 'reasoning':
-                return event.text.length > 0 ? this.reasoningDelta(event.text) : [];
-            case 'step-start': {
-                this.stepSeq += 1;
-                const stepName = `step-${this.stepSeq}`;
-                this.openSteps.push(stepName);
-                this.stepMessage = undefined;
-                return [...this.closeStreams(), { type: 'STEP_STARTED', stepName }];
-            }
-            case 'step-finish': {
-                const out = this.closeStreams();
-                if (!this.replaying)
-                    this.addUsage(event.usage, event.model);
-                const stepName = this.openSteps.pop();
-                if (stepName !== undefined)
-                    out.push({ type: 'STEP_FINISHED', stepName });
-                if (event.usage !== undefined || event.costUsd !== undefined) {
-                    out.push(this.custom(AG_UI_CUSTOM.stepUsage, {
-                        ...(event.usage !== undefined ? { usage: event.usage } : {}),
-                        ...(event.costUsd !== undefined ? { costUsd: event.costUsd } : {}),
-                        ...(event.reasoningMs !== undefined ? { reasoningMs: event.reasoningMs } : {}),
-                    }));
-                }
-                return out;
-            }
-            case 'tool-input-start': {
-                if (this.calls.has(event.id))
-                    return [];
-                const out = this.closeStreams();
-                this.calls.set(event.id, {
-                    name: event.name,
-                    started: true,
-                    ended: false,
-                    sawDelta: false,
-                    answered: false,
-                });
-                out.push(this.callStart(event.id, event.name));
-                return out;
-            }
-            case 'tool-input-delta': {
-                const call = this.calls.get(event.id);
-                if (call === undefined || call.ended || event.delta.length === 0)
-                    return [];
-                call.sawDelta = true;
-                return [{ type: 'TOOL_CALL_ARGS', toolCallId: event.id, delta: event.delta }];
-            }
-            case 'tool-input-available': {
-                const known = this.calls.get(event.id);
-                if (known?.ended === true)
-                    return [];
-                const out = this.closeStreams();
-                const call = known ?? {
-                    name: event.name,
-                    started: false,
-                    ended: false,
-                    sawDelta: false,
-                    answered: false,
-                };
-                if (!call.started) {
-                    call.started = true;
-                    out.push(this.callStart(event.id, event.name));
-                }
-                if (!call.sawDelta) {
-                    out.push({
-                        type: 'TOOL_CALL_ARGS',
-                        toolCallId: event.id,
-                        delta: text(event.input ?? {}),
-                    });
-                }
-                call.ended = true;
-                this.calls.set(event.id, call);
-                out.push({ type: 'TOOL_CALL_END', toolCallId: event.id });
-                return out;
-            }
-            case 'tool-output':
-                return this.result(event.id, text(event.output));
-            case 'tool-output-error':
-                return this.result(event.id, event.error, { 'agora.outcome': 'error' });
-            case 'tool-output-denied':
-                return this.result(event.id, event.reason ?? 'denied', { 'agora.outcome': 'denied' });
-            case 'approval-requested':
-                // Written by the loop as a typed `approval` frame; an event of this kind reaching the sink
-                // directly is carried as-is rather than guessed into an interrupt without its run id.
-                return [this.custom(AG_UI_CUSTOM.approvalRequested, event)];
-            case 'approval-settled':
-                this.pending.delete(event.id);
-                return [this.custom(AG_UI_CUSTOM.approvalSettled, event)];
-            case 'elicitation':
-                return [this.custom(AG_UI_CUSTOM.elicitation, { request: event.request })];
-            case 'ui': {
-                const { kind: _kind, ...component } = event;
-                return [this.custom(AG_UI_CUSTOM.ui, component)];
-            }
-            case 'title':
-                return [this.custom(AG_UI_CUSTOM.title, { title: event.title })];
-            case 'queue':
-                return [
-                    this.custom(AG_UI_CUSTOM.queue, {
-                        queue: event.queue,
-                        ...(event.started !== undefined ? { started: event.started } : {}),
-                    }),
-                ];
-            case 'cancelled':
-                this.cancelled = true;
-                return [];
-            default:
-                // A kind newer than this encoder: tolerated, as every reader of the stream must.
-                return [];
-        }
-    }
-    /**
-     * The events that end the AG-UI run: whatever is still open is closed first, then `RUN_FINISHED`
-     * with how it ended — interrupted when the run is waiting on someone, cancelled when it was
-     * stopped, success otherwise. Nothing after a `RUN_ERROR`. `ended` says the library run's stream
-     * itself ended, rather than this reader having stopped following it.
-     */
     finish(ended = false) {
-        if (this.closed)
-            return [];
-        this.closed = true;
-        if (!this.crossed)
-            this.crossBoundary();
-        // A library run that ended is waiting on no one, whatever it asked along the way.
-        if (ended)
-            this.pending.clear();
-        const out = this.start();
-        out.push(...this.closeStreams());
-        for (const [id, call] of this.calls) {
-            if (call.started && !call.ended) {
-                call.ended = true;
-                out.push({ type: 'TOOL_CALL_END', toolCallId: id });
-            }
-        }
-        while (this.openSteps.length > 0) {
-            out.push({ type: 'STEP_FINISHED', stepName: this.openSteps.pop() });
-        }
-        const usage = this.usageEntries();
-        const interrupts = this.cancelled ? [] : this.interrupts();
-        out.push({
-            type: 'RUN_FINISHED',
-            threadId: this.options.threadId,
-            runId: this.options.runId,
-            ...(this.cancelled
-                ? { outcome: { type: 'cancelled' } }
-                : interrupts.length > 0
-                    ? { outcome: { type: 'interrupt', interrupts } }
-                    : {}),
-            ...(usage.length > 0 ? { usage } : {}),
-        });
-        return out;
-    }
-    interrupts() {
-        return [...this.pending.values()].map((entry) => ({
-            id: encodeInterruptId({
-                kind: entry.kind,
-                parked: entry.parked,
-                stream: this.options.streamRunId,
-                toolCallId: entry.toolCallId,
-                position: this.position,
-            }),
-            ...entry.interrupt,
-        }));
-    }
-    custom(name, value) {
-        return { type: 'CUSTOM', name, value };
-    }
-    nextMessageId() {
-        this.messageSeq += 1;
-        return `${this.options.runId}:m${this.messageSeq}`;
-    }
-    textDelta(delta) {
-        const out = [];
-        if (this.openReasoning !== undefined)
-            out.push(...this.closeReasoning());
-        if (this.openText === undefined) {
-            this.openText = this.nextMessageId();
-            this.stepMessage = this.openText;
-            out.push({ type: 'TEXT_MESSAGE_START', messageId: this.openText, role: 'assistant' });
-        }
-        out.push({ type: 'TEXT_MESSAGE_CONTENT', messageId: this.openText, delta });
-        return out;
-    }
-    reasoningDelta(delta) {
-        const out = [];
-        if (this.openText !== undefined)
-            out.push(...this.closeText());
-        if (this.openReasoning === undefined) {
-            this.openReasoning = this.nextMessageId();
-            out.push({ type: 'REASONING_START', messageId: this.openReasoning }, { type: 'REASONING_MESSAGE_START', messageId: this.openReasoning, role: 'reasoning' });
-        }
-        out.push({ type: 'REASONING_MESSAGE_CONTENT', messageId: this.openReasoning, delta });
-        return out;
-    }
-    closeText() {
-        if (this.openText === undefined)
-            return [];
-        const messageId = this.openText;
-        this.openText = undefined;
-        return [{ type: 'TEXT_MESSAGE_END', messageId }];
-    }
-    closeReasoning() {
-        if (this.openReasoning === undefined)
-            return [];
-        const messageId = this.openReasoning;
-        this.openReasoning = undefined;
-        return [
-            { type: 'REASONING_MESSAGE_END', messageId },
-            { type: 'REASONING_END', messageId },
-        ];
-    }
-    closeStreams() {
-        return [...this.closeReasoning(), ...this.closeText()];
-    }
-    callStart(id, name) {
-        return {
-            type: 'TOOL_CALL_START',
-            toolCallId: id,
-            toolCallName: name,
-            ...(this.stepMessage !== undefined ? { parentMessageId: this.stepMessage } : {}),
-        };
-    }
-    result(id, content, metadata) {
-        const call = this.calls.get(id);
-        if (call !== undefined)
-            call.answered = true;
-        else {
-            this.calls.set(id, {
-                name: '',
-                started: false,
-                ended: true,
-                sawDelta: false,
-                answered: true,
-            });
-        }
-        this.pending.delete(id);
-        return [
-            ...this.closeStreams(),
-            {
-                type: 'TOOL_CALL_RESULT',
-                messageId: `${this.options.runId}:tool:${id}`,
-                toolCallId: id,
-                content,
-                role: 'tool',
-                ...(metadata !== undefined ? { metadata } : {}),
-            },
-        ];
-    }
-    addUsage(usage, model) {
-        if (usage === undefined)
-            return;
-        const name = typeof model === 'string' && model.length > 0 ? model : this.options.model;
-        const key = name ?? '';
-        const entry = this.usage.get(key) ?? (name !== undefined ? { model: name } : {});
-        // The library's accounting is already the protocol's: input and output are totals, cache and
-        // reasoning counts are parts of them. A count the provider did not report stays absent.
-        entry.inputTokens = (entry.inputTokens ?? 0) + usage.inputTokens;
-        entry.outputTokens = (entry.outputTokens ?? 0) + usage.outputTokens;
-        entry.totalTokens = entry.inputTokens + entry.outputTokens;
-        if (usage.cacheReadTokens !== undefined) {
-            entry.cachedInputTokens = (entry.cachedInputTokens ?? 0) + usage.cacheReadTokens;
-        }
-        if (usage.cacheWriteTokens !== undefined) {
-            entry.cacheWriteInputTokens = (entry.cacheWriteInputTokens ?? 0) + usage.cacheWriteTokens;
-        }
-        if (usage.reasoningTokens !== undefined) {
-            entry.reasoningTokens = (entry.reasoningTokens ?? 0) + usage.reasoningTokens;
-        }
-        this.usage.set(key, entry);
-    }
-    usageEntries() {
-        return [...this.usage.values()];
+        return this.shared.finish(ended);
     }
 }
 //# sourceMappingURL=encoder.js.map