agentenkit 0.3.0 → 0.4.0

package/dist/core/snapshot.js

@@ -0,0 +1,87 @@
+import { currentRunId } from './keys.js';
+import { ACTIVE_STATES } from './publish.js';
+/** How long after its end a stream still rides on the snapshot. */
+export const RECENT_END_MS = 60_000;
+/** Events that are part of a step's content, and so of its saved messages. */
+const CONTENT = new Set([
+    'TEXT_MESSAGE_START', 'TEXT_MESSAGE_CONTENT', 'TEXT_MESSAGE_END',
+    'REASONING_START', 'REASONING_CONTENT', 'REASONING_END',
+    'TOOL_CALL_START', 'TOOL_CALL_ARGS', 'TOOL_CALL_END', 'TOOL_CALL_RESULT', 'SOURCE',
+]);
+/** Whose step an item belongs to ('' the main agent), and whether it is
+ *  content. */
+function ownerOf(item) {
+    if (item.type === 'SUBAGENT_EVENT')
+        return { agent: item.subagentId, content: CONTENT.has(item.event.type) };
+    return { agent: '', content: CONTENT.has(item.type) };
+}
+/** Drop each agent's content up to its last finished step (see
+ *  SnapshotStream.items). */
+export function cutCommitted(items) {
+    const lastStep = new Map();
+    items.forEach((item, index) => {
+        if (item.type === 'STEP_FINISHED')
+            lastStep.set(item.agentId ?? '', index);
+    });
+    return items.filter((item, index) => {
+        const { agent, content } = ownerOf(item);
+        return !content || index > (lastStep.get(agent) ?? -1);
+    });
+}
+/** The thread's current run stream, from the thread record: the latest
+ *  segment that started, while it is open or only just ended. Null when
+ *  there is none, or it is gone. */
+export async function snapshotStream(deps, threadId) {
+    if (!deps.streams)
+        return null;
+    const started = await deps.storage.events.latest(threadId, 'RUN_STARTED');
+    if (!started)
+        return null;
+    const { streamId, runId } = started.payload;
+    const ended = await deps.storage.events.latest(threadId, 'RUN_ENDED');
+    if (ended && ended.seq > started.seq &&
+        ended.payload.streamId === streamId &&
+        Date.now() - new Date(ended.createdAt).getTime() > RECENT_END_MS) {
+        return null;
+    }
+    const snap = await deps.streams.snapshot(streamId);
+    if (!snap)
+        return null;
+    return {
+        streamId,
+        runId,
+        items: cutCommitted(snap.items),
+        end: snap.end,
+        offset: snap.items.at(-1)?.offset ?? null,
+    };
+}
+/** One read for a UI: the thread, its messages, its runs, the unfinished
+ *  run's record entries and its run stream. Null when the thread is gone.
+ *  `afterMessageId` keeps only the messages after that one: what a tab that
+ *  already has it is missing (a SNAPSHOT frame). An id the thread does not
+ *  have keeps them all. */
+export async function threadSnapshot(deps, threadId, opts = {}) {
+    const thread = await deps.storage.threads.get(threadId);
+    if (!thread)
+        return null;
+    let messages = await deps.storage.messages.list(threadId, undefined);
+    if (opts.afterMessageId) {
+        const at = messages.findIndex((m) => m.id === opts.afterMessageId);
+        if (at >= 0)
+            messages = messages.slice(at + 1);
+    }
+    const runs = await deps.admin.runs.listByThread(threadId);
+    // The record is small (parks, refusals, each segment's start and end),
+    // and the run's live events come from its stream, not from here.
+    const events = await deps.storage.events.listSince(threadId, -1);
+    const lastEventSeq = events.at(-1)?.seq ?? -1;
+    // The unfinished run's record entries: its open park, a refusal.
+    let activeEvents = [];
+    if (ACTIVE_STATES.includes(thread.state)) {
+        const runId = await currentRunId(deps, threadId);
+        activeEvents = runId ? events.filter((e) => e.runId === runId) : [];
+    }
+    const stream = await snapshotStream(deps, threadId);
+    return { thread, messages, runs, lastEventSeq, activeEvents, stream };
+}
+//# sourceMappingURL=snapshot.js.map

package/dist/core/snapshot.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"snapshot.js","sourceRoot":"","sources":["../../src/core/snapshot.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAqB7C,mEAAmE;AACnE,MAAM,CAAC,MAAM,aAAa,GAAG,MAAM,CAAC;AAEpC,8EAA8E;AAC9E,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC;IACtB,oBAAoB,EAAE,sBAAsB,EAAE,kBAAkB;IAChE,iBAAiB,EAAE,mBAAmB,EAAE,eAAe;IACvD,iBAAiB,EAAE,gBAAgB,EAAE,eAAe,EAAE,kBAAkB,EAAE,QAAQ;CACnF,CAAC,CAAC;AAEH;eACe;AACf,SAAS,OAAO,CAAC,IAAgB;IAC/B,IAAI,IAAI,CAAC,IAAI,KAAK,gBAAgB;QAAE,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,UAAU,EAAE,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;IAC7G,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;AACxD,CAAC;AAED;6BAC6B;AAC7B,MAAM,UAAU,YAAY,CAAC,KAAmB;IAC9C,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC3C,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;QAC5B,IAAI,IAAI,CAAC,IAAI,KAAK,eAAe;YAAE,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,EAAE,KAAK,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;IACH,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;QAClC,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;QACzC,OAAO,CAAC,OAAO,IAAI,KAAK,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;oCAEoC;AACpC,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,IAAkB,EAAE,QAAgB;IACvE,IAAI,CAAC,IAAI,CAAC,OAAO;QAAE,OAAO,IAAI,CAAC;IAC/B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;IAC1E,IAAI,CAAC,OAAO;QAAE,OAAO,IAAI,CAAC;IAC1B,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC,OAA8C,CAAC;IACnF,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IACtE,IACE,KAAK,IAAI,KAAK,CAAC,GAAG,GAAG,OAAO,CAAC,GAAG;QAC/B,KAAK,CAAC,OAAiC,CAAC,QAAQ,KAAK,QAAQ;QAC9D,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,GAAG,aAAa,EAChE,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IACD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACnD,IAAI,CAAC,IAAI;QAAE,OAAO,IAAI,CAAC;IACvB,OAAO;QACL,QAAQ;QACR,KAAK;QACL,KAAK,EAAE,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC;QAC/B,GAAG,EAAE,IAAI,CAAC,GAAG;QACb,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,IAAI,IAAI;KAC1C,CAAC;AACJ,CAAC;AAED;;;;2BAI2B;AAC3B,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,IAAkB,EAClB,QAAgB,EAChB,OAAoC,EAAE;IAEtC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACxD,IAAI,CAAC,MAAM;QAAE,OAAO,IAAI,CAAC;IAEzB,IAAI,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IACrE,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;QACxB,MAAM,EAAE,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,cAAc,CAAC,CAAC;QACnE,IAAI,EAAE,IAAI,CAAC;YAAE,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;IACjD,CAAC;IACD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;IAC1D,uEAAuE;IACvE,iEAAiE;IACjE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC;IACjE,MAAM,YAAY,GAAG,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC;IAC9C,iEAAiE;IACjE,IAAI,YAAY,GAAiB,EAAE,CAAC;IACpC,IAAI,aAAa,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QACjD,YAAY,GAAG,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACtE,CAAC;IACD,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IACpD,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC;AACxE,CAAC"}

package/dist/core/stop.d.ts

@@ -1,8 +1,23 @@
 import type { RuntimePorts } from '../ports/runtime.js';
 import type { StopResult } from '../ports/runtime.js';
+import type { RegisteredAgent } from './agent.js';
+export interface StopOptions {
+    /** Resolves the registered agent a run belongs to, by the name on its
+     *  record. A stop that ends a run no worker holds (queued, or parked on an
+     *  approval) runs that agent's `onSettle` itself (§5.6); with no resolver
+     *  the run is left unsettled for the sweep. */
+    agent?: (name: string) => RegisteredAgent | null;
+}
 /** The whole stop mechanism (§2.1): one button, one behavior — everything
  *  stops immediately. The engine's poller sees CANCELLED on the hot cache and
  *  fires the abort; the durable state is the recovery truth (§3.4). */
-export declare function stop(deps: RuntimePorts, threadId: string): Promise<StopResult>;
+export declare function stop(deps: RuntimePorts, threadId: string, opts?: StopOptions): Promise<StopResult>;
 export declare function recordStoppedRun(deps: RuntimePorts, runId: string, endedAt: Date): Promise<void>;
+/** Answer every open approval with a cancellation, so the history the next
+ *  run sends is well-formed (§2.5). A park persists the assistant's tool call
+ *  and defers its result to the resume; a stop means that resume never comes,
+ *  and a dangling call is a prompt no strict provider accepts. Nested parks
+ *  close their whole chain: the child's call and every spawnSubagent call
+ *  waiting on it (§2.7), whose run records end CANCELLED too. */
+export declare function closeOpenParks(deps: RuntimePorts, threadId: string): Promise<void>;
 //# sourceMappingURL=stop.d.ts.map

package/dist/core/stop.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"stop.d.ts","sourceRoot":"","sources":["../../src/core/stop.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAGxD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAGtD;;uEAEuE;AACvE,wBAAsB,IAAI,CAAC,IAAI,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC,CAwBpF;AAED,wBAAsB,gBAAgB,CAAC,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAWtG"}
+{"version":3,"file":"stop.d.ts","sourceRoot":"","sources":["../../src/core/stop.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAGxD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAEtD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAIlD,MAAM,WAAW,WAAW;IAC1B;;;mDAG+C;IAC/C,KAAK,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,eAAe,GAAG,IAAI,CAAC;CAClD;AAED;;uEAEuE;AACvE,wBAAsB,IAAI,CAAC,IAAI,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,GAAE,WAAgB,GAAG,OAAO,CAAC,UAAU,CAAC,CAmC5G;AAED,wBAAsB,gBAAgB,CAAC,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAWtG;AAkCD;;;;;iEAKiE;AACjE,wBAAsB,cAAc,CAAC,IAAI,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAkCxF"}

package/dist/core/stop.js

@@ -1,12 +1,14 @@
-import { publish, setThreadState } from './publish.js';
+import { ACTIVE_STATES, publish, transition } from './publish.js';
 import { loadOpenHitls } from './hitl.js';
 import { currentRunId } from './keys.js';
+import { Lease } from './lease.js';
+import { settleEndedRun } from './settle.js';
 /** The whole stop mechanism (§2.1): one button, one behavior — everything
  *  stops immediately. The engine's poller sees CANCELLED on the hot cache and
  *  fires the abort; the durable state is the recovery truth (§3.4). */
-export async function stop(deps, threadId) {
+export async function stop(deps, threadId, opts = {}) {
     const thread = await deps.storage.threads.get(threadId);
-    if (thread?.state !== 'RUNNING' && thread?.state !== 'WAITING_FOR_INPUT') {
+    if (!thread || !ACTIVE_STATES.includes(thread.state)) {
         return { accepted: false, error: `Cannot stop thread in state ${thread?.state ?? 'unknown'}` };
     }
     // Read before the write: an adapter may hand back the very object the
@@ -14,12 +16,23 @@ export async function stop(deps, threadId) {
     const wasParked = thread.state === 'WAITING_FOR_INPUT';
     const runId = await currentRunId(deps, threadId);
     const endedAt = new Date();
-    await deps.kv.set(`agent:state:${threadId}`, 'CANCELLED');
-    await setThreadState(deps, threadId, 'CANCELLED', thread.model);
+    // A compare-and-set on the state and the run (§3.4): a finish, a failure or
+    // a newer run that got there first keeps its own ending, and this stop is
+    // refused with the state it found.
+    const won = await transition(deps, threadId, {
+        from: ACTIVE_STATES, to: 'CANCELLED', runId, model: thread.model,
+    });
+    if (!won) {
+        const now = await deps.storage.threads.get(threadId);
+        return { accepted: false, error: `Cannot stop thread in state ${now?.state ?? 'unknown'}` };
+    }
     // A queued or parked run may never execute again. Close its record here,
     // without touching usage that a running worker can still be accruing.
-    if (runId)
+    if (runId) {
         await recordStoppedRun(deps, runId, endedAt);
+        if (opts.agent)
+            await settleAfterStop(deps, opts.agent, threadId, runId);
+    }
     await publish(deps, threadId, 'STATE_CHANGE', {
         state: 'CANCELLED', stopReason: 'cancelled', runId, endedAt,
     });
@@ -41,27 +54,68 @@ export async function recordStoppedRun(deps, runId, endedAt) {
         // Operational history must not prevent cancellation.
     }
 }
+/** Settle a stopped run when no worker is there to (§5.6). A queued run never
+ *  reached a worker; a parked run's worker is long gone. Their steps were
+ *  priced as they happened, so the bill is real, and the spec's `onSettle` is
+ *  where it gets charged.
+ *
+ *  It takes the run lock and settles under it, exactly as a worker would. A
+ *  held lock means a worker owns the run right now (the lock is renewed while
+ *  a worker runs, §3.4), and that worker settles it: from its own cancel
+ *  path, or when it finds the thread cancelled at its segment start. Either
+ *  way the settle claim makes the second arrival a no-op. */
+async function settleAfterStop(deps, lookup, threadId, runId) {
+    const rec = await deps.admin.runs.get(runId).catch(() => null);
+    if (!rec || rec.settledAt)
+        return;
+    const agent = lookup(rec.agent);
+    if (!agent)
+        return;
+    const lease = await Lease.acquire(deps, threadId, runId, undefined);
+    if (!lease)
+        return; // a worker holds the run; it settles the run itself
+    // Renewed while the hook runs: a slow settle must not let a queued job take
+    // the lock and settle the same run beside it.
+    lease.keep();
+    try {
+        await settleEndedRun(deps, agent, threadId, runId);
+    }
+    finally {
+        await lease.release();
+    }
+}
 /** Answer every open approval with a cancellation, so the history the next
  *  run sends is well-formed (§2.5). A park persists the assistant's tool call
  *  and defers its result to the resume; a stop means that resume never comes,
  *  and a dangling call is a prompt no strict provider accepts. Nested parks
  *  close their whole chain: the child's call and every spawnSubagent call
  *  waiting on it (§2.7), whose run records end CANCELLED too. */
-async function closeOpenParks(deps, threadId) {
+export async function closeOpenParks(deps, threadId) {
     const result = { cancelled: true, reason: 'stopped' };
     const toolResult = (toolCallId, toolName) => [
         { type: 'tool-result', toolCallId, toolName, result },
     ];
+    const closed = new Set(); // two parks in one child share its frames
     for (const pending of await loadOpenHitls(deps, threadId)) {
-        await deps.storage.messages.append(threadId, {
-            role: 'tool',
-            agentId: pending.agentId,
-            content: toolResult(pending.toolCallId, pending.toolName),
-        });
-        if (pending.nested) {
-            await recordStoppedRun(deps, pending.nested.agentId, new Date());
+        let frames = pending.frames;
+        if (pending.landed) {
+            // Its own call has its result already; only the levels above wait.
+            frames = frames.slice(pending.resumeFrame ?? 0);
+        }
+        else {
+            await deps.storage.messages.append(threadId, {
+                role: 'tool',
+                agentId: pending.agentId,
+                content: toolResult(pending.toolCallId, pending.toolName),
+            });
+            if (pending.nested) {
+                await recordStoppedRun(deps, pending.nested.agentId, new Date());
+            }
         }
-        for (const frame of pending.frames) {
+        for (const frame of frames) {
+            if (closed.has(frame.toolCallId))
+                continue;
+            closed.add(frame.toolCallId);
             await deps.storage.messages.append(threadId, {
                 role: 'tool',
                 agentId: frame.agentId,

package/dist/core/stop.js.map

@@ -1 +1 @@
-{"version":3,"file":"stop.js","sourceRoot":"","sources":["../../src/core/stop.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AACvD,OAAO,EAAE,aAAa,EAAE,MAAM,WAAW,CAAC;AAE1C,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAEzC;;uEAEuE;AACvE,MAAM,CAAC,KAAK,UAAU,IAAI,CAAC,IAAkB,EAAE,QAAgB;IAC7D,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACxD,IAAI,MAAM,EAAE,KAAK,KAAK,SAAS,IAAI,MAAM,EAAE,KAAK,KAAK,mBAAmB,EAAE,CAAC;QACzE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,+BAA+B,MAAM,EAAE,KAAK,IAAI,SAAS,EAAE,EAAE,CAAC;IACjG,CAAC;IAED,sEAAsE;IACtE,uBAAuB;IACvB,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,KAAK,mBAAmB,CAAC;IACvD,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IACjD,MAAM,OAAO,GAAG,IAAI,IAAI,EAAE,CAAC;IAE3B,MAAM,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,eAAe,QAAQ,EAAE,EAAE,WAAW,CAAC,CAAC;IAC1D,MAAM,cAAc,CAAC,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;IAChE,yEAAyE;IACzE,sEAAsE;IACtE,IAAI,KAAK;QAAE,MAAM,gBAAgB,CAAC,IAAI,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;IACxD,MAAM,OAAO,CAAC,IAAI,EAAE,QAAQ,EAAE,cAAc,EAAE;QAC5C,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,WAAW,EAAE,KAAK,EAAE,OAAO;KAC5D,CAAC,CAAC;IAEH,IAAI,SAAS;QAAE,MAAM,cAAc,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IAEpD,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;AAC5B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,IAAkB,EAAE,KAAa,EAAE,OAAa;IACrF,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC/C,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,KAAK,KAAK,WAAW;YAAE,OAAO;QAClD,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE;YACjC,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,WAAW,EAAE,OAAO;YACpD,UAAU,EAAE,OAAO,CAAC,OAAO,EAAE,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE;SACpE,CAAC,CAAC;IACL,CAAC;IAAC,MAAM,CAAC;QACP,qDAAqD;IACvD,CAAC;AACH,CAAC;AAED;;;;;iEAKiE;AACjE,KAAK,UAAU,cAAc,CAAC,IAAkB,EAAE,QAAgB;IAChE,MAAM,MAAM,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;IACtD,MAAM,UAAU,GAAG,CAAC,UAAkB,EAAE,QAAgB,EAAE,EAAE,CAAC;QAC3D,EAAE,IAAI,EAAE,aAAa,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE;KACtD,CAAC;IACF,KAAK,MAAM,OAAO,IAAI,MAAM,aAAa,CAAC,IAAI,EAAE,QAAQ,CAAC,EAAE,CAAC;QAC1D,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,EAAE;YAC3C,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,OAAO,EAAE,UAAU,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,QAAQ,CAAC;SAC1D,CAAC,CAAC;QACH,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,MAAM,gBAAgB,CAAC,IAAI,EAAE,OAAO,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC;QACnE,CAAC;QACD,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnC,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,EAAE;gBAC3C,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,OAAO,EAAE,UAAU,CAAC,KAAK,CAAC,UAAU,EAAE,eAAe,CAAC;aACvD,CAAC,CAAC;YACH,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;gBACjB,MAAM,gBAAgB,CAAC,IAAI,EAAE,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC;YACjE,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC"}
+{"version":3,"file":"stop.js","sourceRoot":"","sources":["../../src/core/stop.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,aAAa,EAAE,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAClE,OAAO,EAAE,aAAa,EAAE,MAAM,WAAW,CAAC;AAE1C,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAEzC,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAU7C;;uEAEuE;AACvE,MAAM,CAAC,KAAK,UAAU,IAAI,CAAC,IAAkB,EAAE,QAAgB,EAAE,OAAoB,EAAE;IACrF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACxD,IAAI,CAAC,MAAM,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;QACrD,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,+BAA+B,MAAM,EAAE,KAAK,IAAI,SAAS,EAAE,EAAE,CAAC;IACjG,CAAC;IAED,sEAAsE;IACtE,uBAAuB;IACvB,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,KAAK,mBAAmB,CAAC;IACvD,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IACjD,MAAM,OAAO,GAAG,IAAI,IAAI,EAAE,CAAC;IAE3B,4EAA4E;IAC5E,0EAA0E;IAC1E,mCAAmC;IACnC,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC,IAAI,EAAE,QAAQ,EAAE;QAC3C,IAAI,EAAE,aAAa,EAAE,EAAE,EAAE,WAAW,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK;KACjE,CAAC,CAAC;IACH,IAAI,CAAC,GAAG,EAAE,CAAC;QACT,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACrD,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,+BAA+B,GAAG,EAAE,KAAK,IAAI,SAAS,EAAE,EAAE,CAAC;IAC9F,CAAC;IACD,yEAAyE;IACzE,sEAAsE;IACtE,IAAI,KAAK,EAAE,CAAC;QACV,MAAM,gBAAgB,CAAC,IAAI,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QAC7C,IAAI,IAAI,CAAC,KAAK;YAAE,MAAM,eAAe,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;IAC3E,CAAC;IACD,MAAM,OAAO,CAAC,IAAI,EAAE,QAAQ,EAAE,cAAc,EAAE;QAC5C,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,WAAW,EAAE,KAAK,EAAE,OAAO;KAC5D,CAAC,CAAC;IAEH,IAAI,SAAS;QAAE,MAAM,cAAc,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IAEpD,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;AAC5B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,IAAkB,EAAE,KAAa,EAAE,OAAa;IACrF,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC/C,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,KAAK,KAAK,WAAW;YAAE,OAAO;QAClD,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE;YACjC,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,WAAW,EAAE,OAAO;YACpD,UAAU,EAAE,OAAO,CAAC,OAAO,EAAE,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE;SACpE,CAAC,CAAC;IACL,CAAC;IAAC,MAAM,CAAC;QACP,qDAAqD;IACvD,CAAC;AACH,CAAC;AAED;;;;;;;;;6DAS6D;AAC7D,KAAK,UAAU,eAAe,CAC5B,IAAkB,EAClB,MAAgD,EAChD,QAAgB,EAChB,KAAa;IAEb,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;IAC/D,IAAI,CAAC,GAAG,IAAI,GAAG,CAAC,SAAS;QAAE,OAAO;IAClC,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAChC,IAAI,CAAC,KAAK;QAAE,OAAO;IACnB,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;IACpE,IAAI,CAAC,KAAK;QAAE,OAAO,CAAC,oDAAoD;IACxE,4EAA4E;IAC5E,8CAA8C;IAC9C,KAAK,CAAC,IAAI,EAAE,CAAC;IACb,IAAI,CAAC;QACH,MAAM,cAAc,CAAC,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;IACrD,CAAC;YAAS,CAAC;QACT,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;IACxB,CAAC;AACH,CAAC;AAED;;;;;iEAKiE;AACjE,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,IAAkB,EAAE,QAAgB;IACvE,MAAM,MAAM,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;IACtD,MAAM,UAAU,GAAG,CAAC,UAAkB,EAAE,QAAgB,EAAE,EAAE,CAAC;QAC3D,EAAE,IAAI,EAAE,aAAa,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE;KACtD,CAAC;IACF,MAAM,MAAM,GAAG,IAAI,GAAG,EAAU,CAAC,CAAC,0CAA0C;IAC5E,KAAK,MAAM,OAAO,IAAI,MAAM,aAAa,CAAC,IAAI,EAAE,QAAQ,CAAC,EAAE,CAAC;QAC1D,IAAI,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC5B,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,mEAAmE;YACnE,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,WAAW,IAAI,CAAC,CAAC,CAAC;QAClD,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,EAAE;gBAC3C,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,OAAO,EAAE,UAAU,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,QAAQ,CAAC;aAC1D,CAAC,CAAC;YACH,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACnB,MAAM,gBAAgB,CAAC,IAAI,EAAE,OAAO,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC;YACnE,CAAC;QACH,CAAC;QACD,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,IAAI,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC;gBAAE,SAAS;YAC3C,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAC7B,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,EAAE;gBAC3C,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,OAAO,EAAE,UAAU,CAAC,KAAK,CAAC,UAAU,EAAE,eAAe,CAAC;aACvD,CAAC,CAAC;YACH,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;gBACjB,MAAM,gBAAgB,CAAC,IAAI,EAAE,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC;YACjE,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC"}

package/dist/core/stream-events.d.ts

@@ -0,0 +1,184 @@
+/** The events a run stream carries (see RunStreams). One typed union: a
+ *  reader checks `type` and the compiler knows the fields. The shapes follow
+ *  AG-UI, so an AG-UI encoder only renames a few fields. The Go runtime has
+ *  the same events, with the same names and the same JSON. */
+/** Tokens a step or a run used. */
+export interface StreamUsage {
+    inputTokens: number;
+    cachedInputTokens: number;
+    outputTokens: number;
+    totalTokens: number;
+}
+/** Money spent in one currency, in millionths of a unit. */
+export interface StreamCost {
+    currency: string;
+    micros: number;
+}
+export interface RunStartedEvent {
+    type: 'RUN_STARTED';
+    threadId: string;
+    runId: string;
+    streamId: string;
+    /** 1 for the first pickup, 2 for the resume after a park, and so on. */
+    segment: number;
+}
+export interface TextMessageStartEvent {
+    type: 'TEXT_MESSAGE_START';
+    messageId: string;
+    role: 'assistant';
+}
+export interface TextMessageContentEvent {
+    type: 'TEXT_MESSAGE_CONTENT';
+    messageId: string;
+    delta: string;
+}
+export interface TextMessageEndEvent {
+    type: 'TEXT_MESSAGE_END';
+    messageId: string;
+}
+export interface ReasoningStartEvent {
+    type: 'REASONING_START';
+    messageId: string;
+}
+export interface ReasoningContentEvent {
+    type: 'REASONING_CONTENT';
+    messageId: string;
+    delta: string;
+}
+export interface ReasoningEndEvent {
+    type: 'REASONING_END';
+    messageId: string;
+}
+export interface ToolCallStartEvent {
+    type: 'TOOL_CALL_START';
+    toolCallId: string;
+    toolName: string;
+}
+export interface ToolCallArgsEvent {
+    type: 'TOOL_CALL_ARGS';
+    toolCallId: string;
+    delta: string;
+}
+/** The call is complete. `args` is the whole, parsed argument object, so a
+ *  reader that missed the ARGS deltas still has it. */
+export interface ToolCallEndEvent {
+    type: 'TOOL_CALL_END';
+    toolCallId: string;
+    toolName: string;
+    args: unknown;
+}
+export interface ToolCallResultEvent {
+    type: 'TOOL_CALL_RESULT';
+    toolCallId: string;
+    toolName: string;
+    result: unknown;
+}
+/** A source the model cited (a web page, a document). Not in AG-UI: the
+ *  encoder sends it as CUSTOM. */
+export interface SourceEvent {
+    type: 'SOURCE';
+    source: unknown;
+}
+/** One model call is done and its messages are saved. */
+export interface StepFinishedEvent {
+    type: 'STEP_FINISHED';
+    /** 1-based, per agent. */
+    step: number;
+    /** null for the main agent, the nested run's id otherwise. */
+    agentId: string | null;
+    finishReason: string;
+    usage: StreamUsage;
+}
+export interface MessageAppendedEvent {
+    type: 'MESSAGE_APPENDED';
+    message: {
+        id: string;
+        role: string;
+        content: unknown;
+        agentId: string | null;
+        createdAt: string;
+    };
+    /** The sender's own name for the message, echoed back. */
+    clientMessageId?: string;
+}
+/** A tool is waiting for a person. Also kept in the thread record. */
+export interface InputRequiredEvent {
+    type: 'INPUT_REQUIRED';
+    toolCallId: string;
+    toolName: string;
+    agentId: string | null;
+    arguments: unknown;
+    inputSchema: unknown;
+    /** The unwind chain for a nested park; empty for the main agent. */
+    frames: unknown[];
+    nested?: unknown;
+    /** The dispatch ticket the answer resumes with. */
+    resume?: unknown;
+    reason: string;
+    expiresAt?: string;
+}
+export interface SubagentStartedEvent {
+    type: 'SUBAGENT_STARTED';
+    subagentId: string;
+    name: string;
+    depth: number;
+}
+/** One event from a nested run, wrapped. */
+export interface SubagentEventEvent {
+    type: 'SUBAGENT_EVENT';
+    subagentId: string;
+    event: StreamEvent;
+}
+export interface SubagentFinishedEvent {
+    type: 'SUBAGENT_FINISHED';
+    subagentId: string;
+    /** `cancelled`: a user stop reached the child. */
+    status: 'completed' | 'failed' | 'cancelled';
+    error?: string;
+}
+/** An app's own event, sent with publishEvent. */
+export interface CustomStreamEvent {
+    type: 'CUSTOM';
+    name: string;
+    value: unknown;
+}
+/** The run's segment ended normally. Always the last item of a stream. */
+export interface RunFinishedEvent {
+    type: 'RUN_FINISHED';
+    status: 'finished' | 'parked' | 'stopped';
+    usage?: StreamUsage;
+    costs?: StreamCost[];
+    finishReason?: string;
+    /** The final text of a one-shot agent. */
+    text?: string;
+}
+/** The run's segment ended badly. Always the last item of a stream. */
+export interface RunErrorEvent {
+    type: 'RUN_ERROR';
+    /** `lost`: the worker died and the sweep closed its stream. */
+    status: 'error' | 'lost';
+    error: string;
+}
+export type StreamEvent = RunStartedEvent | TextMessageStartEvent | TextMessageContentEvent | TextMessageEndEvent | ReasoningStartEvent | ReasoningContentEvent | ReasoningEndEvent | ToolCallStartEvent | ToolCallArgsEvent | ToolCallEndEvent | ToolCallResultEvent | SourceEvent | StepFinishedEvent | MessageAppendedEvent | InputRequiredEvent | SubagentStartedEvent | SubagentEventEvent | SubagentFinishedEvent | CustomStreamEvent | RunFinishedEvent | RunErrorEvent;
+export type StreamEventType = StreamEvent['type'];
+/** How a stream ends: the last item it will ever hold. */
+export type StreamEnd = RunFinishedEvent | RunErrorEvent;
+/** An event as a stream holds it: with the offset the store gave it. */
+export type StreamItem = StreamEvent & {
+    offset: string;
+};
+/** True for the event that ends a stream. */
+export declare function isStreamEnd(e: {
+    type: string;
+}): e is StreamEnd;
+/** A run segment's stream id: `<runId>:<segment>`. A park closes a
+ *  segment's stream and the resume opens the next, so a closed stream never
+ *  opens again. */
+export declare function streamIdOf(runId: string, segment: number): string;
+/** The run id and segment in a stream id, or null for one not made by
+ *  streamIdOf. */
+export declare function parseStreamId(streamId: string): {
+    runId: string;
+    segment: number;
+} | null;
+//# sourceMappingURL=stream-events.d.ts.map

package/dist/core/stream-events.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream-events.d.ts","sourceRoot":"","sources":["../../src/core/stream-events.ts"],"names":[],"mappings":"AAAA;;;8DAG8D;AAE9D,mCAAmC;AACnC,MAAM,WAAW,WAAW;IAC1B,WAAW,EAAE,MAAM,CAAC;IACpB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,4DAA4D;AAC5D,MAAM,WAAW,UAAU;IACzB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,aAAa,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC;IACjB,wEAAwE;IACxE,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,qBAAqB;IACpC,IAAI,EAAE,oBAAoB,CAAC;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,WAAW,CAAC;CACnB;AACD,MAAM,WAAW,uBAAuB;IACtC,IAAI,EAAE,sBAAsB,CAAC;IAC7B,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;CACf;AACD,MAAM,WAAW,mBAAmB;IAClC,IAAI,EAAE,kBAAkB,CAAC;IACzB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,mBAAmB;IAClC,IAAI,EAAE,iBAAiB,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;CACnB;AACD,MAAM,WAAW,qBAAqB;IACpC,IAAI,EAAE,mBAAmB,CAAC;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;CACf;AACD,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,eAAe,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,iBAAiB,CAAC;IACxB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;CAClB;AACD,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,gBAAgB,CAAC;IACvB,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;CACf;AACD;uDACuD;AACvD,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,eAAe,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,OAAO,CAAC;CACf;AACD,MAAM,WAAW,mBAAmB;IAClC,IAAI,EAAE,kBAAkB,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,OAAO,CAAC;CACjB;AAED;kCACkC;AAClC,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,QAAQ,CAAC;IACf,MAAM,EAAE,OAAO,CAAC;CACjB;AAED,yDAAyD;AACzD,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,eAAe,CAAC;IACtB,0BAA0B;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,8DAA8D;IAC9D,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,YAAY,EAAE,MAAM,CAAC;IACrB,KAAK,EAAE,WAAW,CAAC;CACpB;AAED,MAAM,WAAW,oBAAoB;IACnC,IAAI,EAAE,kBAAkB,CAAC;IACzB,OAAO,EAAE;QACP,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,OAAO,CAAC;QACjB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;QACvB,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;IACF,0DAA0D;IAC1D,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED,sEAAsE;AACtE,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,gBAAgB,CAAC;IACvB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,SAAS,EAAE,OAAO,CAAC;IACnB,WAAW,EAAE,OAAO,CAAC;IACrB,oEAAoE;IACpE,MAAM,EAAE,OAAO,EAAE,CAAC;IAClB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,mDAAmD;IACnD,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,oBAAoB;IACnC,IAAI,EAAE,kBAAkB,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;CACf;AACD,4CAA4C;AAC5C,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,gBAAgB,CAAC;IACvB,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,WAAW,CAAC;CACpB;AACD,MAAM,WAAW,qBAAqB;IACpC,IAAI,EAAE,mBAAmB,CAAC;IAC1B,UAAU,EAAE,MAAM,CAAC;IACnB,kDAAkD;IAClD,MAAM,EAAE,WAAW,GAAG,QAAQ,GAAG,WAAW,CAAC;IAC7C,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,kDAAkD;AAClD,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,QAAQ,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,OAAO,CAAC;CAChB;AAED,0EAA0E;AAC1E,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,cAAc,CAAC;IACrB,MAAM,EAAE,UAAU,GAAG,QAAQ,GAAG,SAAS,CAAC;IAC1C,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,KAAK,CAAC,EAAE,UAAU,EAAE,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,0CAA0C;IAC1C,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,uEAAuE;AACvE,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,WAAW,CAAC;IAClB,+DAA+D;IAC/D,MAAM,EAAE,OAAO,GAAG,MAAM,CAAC;IACzB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,MAAM,WAAW,GACnB,eAAe,GACf,qBAAqB,GACrB,uBAAuB,GACvB,mBAAmB,GACnB,mBAAmB,GACnB,qBAAqB,GACrB,iBAAiB,GACjB,kBAAkB,GAClB,iBAAiB,GACjB,gBAAgB,GAChB,mBAAmB,GACnB,WAAW,GACX,iBAAiB,GACjB,oBAAoB,GACpB,kBAAkB,GAClB,oBAAoB,GACpB,kBAAkB,GAClB,qBAAqB,GACrB,iBAAiB,GACjB,gBAAgB,GAChB,aAAa,CAAC;AAElB,MAAM,MAAM,eAAe,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;AAElD,0DAA0D;AAC1D,MAAM,MAAM,SAAS,GAAG,gBAAgB,GAAG,aAAa,CAAC;AAEzD,wEAAwE;AACxE,MAAM,MAAM,UAAU,GAAG,WAAW,GAAG;IAAE,MAAM,EAAE,MAAM,CAAA;CAAE,CAAC;AAE1D,6CAA6C;AAC7C,wBAAgB,WAAW,CAAC,CAAC,EAAE;IAAE,IAAI,EAAE,MAAM,CAAA;CAAE,GAAG,CAAC,IAAI,SAAS,CAE/D;AAED;;mBAEmB;AACnB,wBAAgB,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,MAAM,CAEjE;AAED;kBACkB;AAClB,wBAAgB,aAAa,CAAC,QAAQ,EAAE,MAAM,GAAG;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GAAG,IAAI,CAMzF"}

package/dist/core/stream-events.js

@@ -0,0 +1,26 @@
+/** The events a run stream carries (see RunStreams). One typed union: a
+ *  reader checks `type` and the compiler knows the fields. The shapes follow
+ *  AG-UI, so an AG-UI encoder only renames a few fields. The Go runtime has
+ *  the same events, with the same names and the same JSON. */
+/** True for the event that ends a stream. */
+export function isStreamEnd(e) {
+    return e.type === 'RUN_FINISHED' || e.type === 'RUN_ERROR';
+}
+/** A run segment's stream id: `<runId>:<segment>`. A park closes a
+ *  segment's stream and the resume opens the next, so a closed stream never
+ *  opens again. */
+export function streamIdOf(runId, segment) {
+    return `${runId}:${segment}`;
+}
+/** The run id and segment in a stream id, or null for one not made by
+ *  streamIdOf. */
+export function parseStreamId(streamId) {
+    const at = streamId.lastIndexOf(':');
+    if (at <= 0)
+        return null;
+    const segment = Number(streamId.slice(at + 1));
+    if (!Number.isInteger(segment) || segment < 1)
+        return null;
+    return { runId: streamId.slice(0, at), segment };
+}
+//# sourceMappingURL=stream-events.js.map

package/dist/core/stream-events.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream-events.js","sourceRoot":"","sources":["../../src/core/stream-events.ts"],"names":[],"mappings":"AAAA;;;8DAG8D;AA4M9D,6CAA6C;AAC7C,MAAM,UAAU,WAAW,CAAC,CAAmB;IAC7C,OAAO,CAAC,CAAC,IAAI,KAAK,cAAc,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC;AAC7D,CAAC;AAED;;mBAEmB;AACnB,MAAM,UAAU,UAAU,CAAC,KAAa,EAAE,OAAe;IACvD,OAAO,GAAG,KAAK,IAAI,OAAO,EAAE,CAAC;AAC/B,CAAC;AAED;kBACkB;AAClB,MAAM,UAAU,aAAa,CAAC,QAAgB;IAC5C,MAAM,EAAE,GAAG,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IACrC,IAAI,EAAE,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC;IACzB,MAAM,OAAO,GAAG,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;IAC/C,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,OAAO,GAAG,CAAC;QAAE,OAAO,IAAI,CAAC;IAC3D,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC;AACnD,CAAC"}

package/dist/core/stream.d.ts

@@ -9,4 +9,33 @@
  *  Draining fully first also means `onChunk` fires for every part that did
  *  arrive before the failure. */
 export declare function drainOrThrow(fullStream: AsyncIterable<unknown>): Promise<void>;
+/** Merges token deltas before they go out as events (§2.2). Every published
+ *  event costs a seq, a stored row and a bus send; one per token is thousands
+ *  a reply. Consecutive text (or reasoning) deltas are joined into one delta
+ *  of the same shape, so a client reads them exactly as before. A held delta
+ *  goes out after `windowMs`, or at once when any other kind of chunk comes,
+ *  or on `flush`: the loop flushes before a step is committed, so its text is
+ *  never published after the step that produced it. Sends go out in order. */
+export declare class ChunkBatcher {
+    private readonly send;
+    private readonly windowMs;
+    private held;
+    private timer?;
+    private chain;
+    constructor(send: (chunk: unknown) => Promise<void>, windowMs?: number);
+    /** Take one chunk. A delta is held; anything else is sent, after what is
+     *  held, and the promise resolves once it is out, so the step does not move
+     *  on (a tool starting, say) before what it streamed is stored. */
+    push(chunk: unknown): Promise<void>;
+    /** Send what is held, and resolve once everything pushed so far is out. */
+    flush(): Promise<void>;
+    private enqueue;
+}
+/** A chunk as it is published: the SDK part, except that a tool result goes
+ *  out in the shape the Go runtime sends, `{type, toolCallId, toolName,
+ *  result}`, with `{error}` as the result of a call whose tool failed (the
+ *  model got `error: <message>`, but a client marks the call failed from
+ *  this), and a parked call's placeholder is not sent at all (its verdict
+ *  arrives with the resume). Null means "do not publish". */
+export declare function chunkPayload(chunk: unknown, toolErrors?: Map<string, string>): unknown | null;
 //# sourceMappingURL=stream.d.ts.map

package/dist/core/stream.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"stream.d.ts","sourceRoot":"","sources":["../../src/core/stream.ts"],"names":[],"mappings":"AAAA;;;;;;;;;iCASiC;AACjC,wBAAsB,YAAY,CAAC,UAAU,EAAE,aAAa,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,CAQpF"}
+{"version":3,"file":"stream.d.ts","sourceRoot":"","sources":["../../src/core/stream.ts"],"names":[],"mappings":"AAAA;;;;;;;;;iCASiC;AACjC,wBAAsB,YAAY,CAAC,UAAU,EAAE,aAAa,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,CAQpF;AAED;;;;;;8EAM8E;AAC9E,qBAAa,YAAY;IAMrB,OAAO,CAAC,QAAQ,CAAC,IAAI;IACrB,OAAO,CAAC,QAAQ,CAAC,QAAQ;IAN3B,OAAO,CAAC,IAAI,CAAoD;IAChE,OAAO,CAAC,KAAK,CAAC,CAAgC;IAC9C,OAAO,CAAC,KAAK,CAAoC;gBAG9B,IAAI,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,OAAO,CAAC,IAAI,CAAC,EACvC,QAAQ,SAAK;IAGhC;;uEAEmE;IACnE,IAAI,CAAC,KAAK,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IAmBnC,2EAA2E;IAC3E,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAStB,OAAO,CAAC,OAAO;CAOhB;AAED;;;;;6DAK6D;AAC7D,wBAAgB,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE,UAAU,CAAC,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,OAAO,GAAG,IAAI,CAa7F"}

package/dist/core/stream.js

@@ -18,4 +18,83 @@ export async function drainOrThrow(fullStream) {
     if (streamError !== undefined)
         throw streamError;
 }
+/** Merges token deltas before they go out as events (§2.2). Every published
+ *  event costs a seq, a stored row and a bus send; one per token is thousands
+ *  a reply. Consecutive text (or reasoning) deltas are joined into one delta
+ *  of the same shape, so a client reads them exactly as before. A held delta
+ *  goes out after `windowMs`, or at once when any other kind of chunk comes,
+ *  or on `flush`: the loop flushes before a step is committed, so its text is
+ *  never published after the step that produced it. Sends go out in order. */
+export class ChunkBatcher {
+    send;
+    windowMs;
+    held = null;
+    timer;
+    chain = Promise.resolve();
+    constructor(send, windowMs = 50) {
+        this.send = send;
+        this.windowMs = windowMs;
+    }
+    /** Take one chunk. A delta is held; anything else is sent, after what is
+     *  held, and the promise resolves once it is out, so the step does not move
+     *  on (a tool starting, say) before what it streamed is stored. */
+    push(chunk) {
+        const c = chunk;
+        const isDelta = (c?.type === 'text-delta' || c?.type === 'reasoning') && typeof c.textDelta === 'string';
+        if (isDelta && this.held && this.held.type === c.type) {
+            this.held.textDelta += c.textDelta;
+            return Promise.resolve();
+        }
+        void this.flush();
+        if (isDelta) {
+            this.held = { ...c, type: c.type, textDelta: c.textDelta };
+            this.timer = setTimeout(() => void this.flush(), this.windowMs);
+            this.timer.unref?.();
+            return Promise.resolve();
+        }
+        this.enqueue(chunk);
+        return this.chain;
+    }
+    /** Send what is held, and resolve once everything pushed so far is out. */
+    flush() {
+        if (this.timer)
+            clearTimeout(this.timer);
+        this.timer = undefined;
+        const held = this.held;
+        this.held = null;
+        if (held)
+            this.enqueue(held);
+        return this.chain;
+    }
+    enqueue(chunk) {
+        this.chain = this.chain.then(() => this.send(chunk)).catch((err) => {
+            // A lost live chunk is not a lost step: the step's messages are stored
+            // at commit. It must not stop the chunks after it either.
+            console.error('chunk not published', err);
+        });
+    }
+}
+/** A chunk as it is published: the SDK part, except that a tool result goes
+ *  out in the shape the Go runtime sends, `{type, toolCallId, toolName,
+ *  result}`, with `{error}` as the result of a call whose tool failed (the
+ *  model got `error: <message>`, but a client marks the call failed from
+ *  this), and a parked call's placeholder is not sent at all (its verdict
+ *  arrives with the resume). Null means "do not publish". */
+export function chunkPayload(chunk, toolErrors) {
+    const c = chunk;
+    if (c?.type !== 'tool-result')
+        return chunk;
+    const r = c.result;
+    if (r && typeof r === 'object' && '__hitl_parked__' in r)
+        return null;
+    const failed = c.toolCallId ? toolErrors?.get(c.toolCallId) : undefined;
+    if (failed !== undefined)
+        toolErrors.delete(c.toolCallId);
+    return {
+        type: 'tool-result',
+        toolCallId: c.toolCallId,
+        toolName: c.toolName,
+        result: failed !== undefined ? { error: failed } : c.result,
+    };
+}
 //# sourceMappingURL=stream.js.map

package/dist/core/stream.js.map

@@ -1 +1 @@
-{"version":3,"file":"stream.js","sourceRoot":"","sources":["../../src/core/stream.ts"],"names":[],"mappings":"AAAA;;;;;;;;;iCASiC;AACjC,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,UAAkC;IACnE,IAAI,WAAoB,CAAC;IACzB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;QACpC,IAAK,IAAiC,EAAE,IAAI,KAAK,OAAO,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;YACtF,WAAW,GAAI,IAA4B,CAAC,KAAK,CAAC;QACpD,CAAC;IACH,CAAC;IACD,IAAI,WAAW,KAAK,SAAS;QAAE,MAAM,WAAW,CAAC;AACnD,CAAC"}
+{"version":3,"file":"stream.js","sourceRoot":"","sources":["../../src/core/stream.ts"],"names":[],"mappings":"AAAA;;;;;;;;;iCASiC;AACjC,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,UAAkC;IACnE,IAAI,WAAoB,CAAC;IACzB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;QACpC,IAAK,IAAiC,EAAE,IAAI,KAAK,OAAO,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;YACtF,WAAW,GAAI,IAA4B,CAAC,KAAK,CAAC;QACpD,CAAC;IACH,CAAC;IACD,IAAI,WAAW,KAAK,SAAS;QAAE,MAAM,WAAW,CAAC;AACnD,CAAC;AAED;;;;;;8EAM8E;AAC9E,MAAM,OAAO,YAAY;IAMJ;IACA;IANX,IAAI,GAA+C,IAAI,CAAC;IACxD,KAAK,CAAiC;IACtC,KAAK,GAAkB,OAAO,CAAC,OAAO,EAAE,CAAC;IAEjD,YACmB,IAAuC,EACvC,WAAW,EAAE;QADb,SAAI,GAAJ,IAAI,CAAmC;QACvC,aAAQ,GAAR,QAAQ,CAAK;IAC7B,CAAC;IAEJ;;uEAEmE;IACnE,IAAI,CAAC,KAAc;QACjB,MAAM,CAAC,GAAG,KAAuD,CAAC;QAClE,MAAM,OAAO,GACX,CAAC,CAAC,EAAE,IAAI,KAAK,YAAY,IAAI,CAAC,EAAE,IAAI,KAAK,WAAW,CAAC,IAAI,OAAO,CAAC,CAAC,SAAS,KAAK,QAAQ,CAAC;QAC3F,IAAI,OAAO,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,CAAE,CAAC,IAAI,EAAE,CAAC;YACvD,IAAI,CAAC,IAAI,CAAC,SAAS,IAAI,CAAE,CAAC,SAAmB,CAAC;YAC9C,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;QAC3B,CAAC;QACD,KAAK,IAAI,CAAC,KAAK,EAAE,CAAC;QAClB,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC,IAAI,GAAG,EAAE,GAAI,CAAY,EAAE,IAAI,EAAE,CAAE,CAAC,IAAc,EAAE,SAAS,EAAE,CAAE,CAAC,SAAmB,EAAE,CAAC;YAC7F,IAAI,CAAC,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,KAAK,IAAI,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC/D,IAAI,CAAC,KAA2C,CAAC,KAAK,EAAE,EAAE,CAAC;YAC5D,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;QAC3B,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACpB,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED,2EAA2E;IAC3E,KAAK;QACH,IAAI,IAAI,CAAC,KAAK;YAAE,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzC,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC;QACvB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;QACvB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,IAAI;YAAE,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC7B,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAEO,OAAO,CAAC,KAAc;QAC5B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;YACjE,uEAAuE;YACvE,0DAA0D;YAC1D,OAAO,CAAC,KAAK,CAAC,qBAAqB,EAAE,GAAG,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAED;;;;;6DAK6D;AAC7D,MAAM,UAAU,YAAY,CAAC,KAAc,EAAE,UAAgC;IAC3E,MAAM,CAAC,GAAG,KAA2F,CAAC;IACtG,IAAI,CAAC,EAAE,IAAI,KAAK,aAAa;QAAE,OAAO,KAAK,CAAC;IAC5C,MAAM,CAAC,GAAG,CAAC,CAAC,MAAwC,CAAC;IACrD,IAAI,CAAC,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,iBAAiB,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC;IACtE,MAAM,MAAM,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACxE,IAAI,MAAM,KAAK,SAAS;QAAE,UAAW,CAAC,MAAM,CAAC,CAAC,CAAC,UAAW,CAAC,CAAC;IAC5D,OAAO;QACL,IAAI,EAAE,aAAa;QACnB,UAAU,EAAE,CAAC,CAAC,UAAU;QACxB,QAAQ,EAAE,CAAC,CAAC,QAAQ;QACpB,MAAM,EAAE,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM;KAC5D,CAAC;AACJ,CAAC"}