veryfront 0.1.1239 → 0.1.1241

package/esm/src/agent/ag-ui/chat-ui-chunk-encoder.d.ts

@@ -0,0 +1,35 @@
+import type { ChatMessageMetadata, ChatUiMessageChunk } from "../../chat/protocol.js";
+import type { AgUiChunkEncoder } from "./chunk-encoder.js";
+import { type AgUiEncoderStateOptions, type AgUiRunFinishedMetadata, type AgUiRuntimeStreamEvent } from "./encoder.js";
+import { type CreateAgUiTrackedResponseInput } from "./tracked-response.js";
+/** Public API contract for AG-UI chat UI chunk encoder. */
+export type AgUiChatUiChunkEncoder = Pick<AgUiChunkEncoder<ChatUiMessageChunk<ChatMessageMetadata>>, "encode" | "finalize" | "timingState">;
+/** Options accepted by create AG-UI chat UI chunk encoder. */
+export interface CreateAgUiChatUiChunkEncoderOptions {
+    /**
+     * Timing clocks forwarded verbatim to the encoder state. A single object so
+     * that adding a clock is one edit in `AgUiEncoderStateOptions`, not a
+     * sweep through every wrapper that happens to sit in between.
+     */
+    timing?: AgUiEncoderStateOptions;
+    modelId?: string;
+    resolveProvider?: (modelId: string) => string | undefined;
+}
+/** Input payload for create AG-UI chat UI tracked response. */
+export interface CreateAgUiChatUiTrackedResponseInput extends Omit<CreateAgUiTrackedResponseInput<ChatUiMessageChunk<ChatMessageMetadata>>, "chunkEncoder" | "finalizeTracker"> {
+    modelId: string;
+    resolveProvider?: CreateAgUiChatUiChunkEncoderOptions["resolveProvider"];
+}
+/** Return AG-UI chat UI message metadata from chunk. */
+export declare function getAgUiChatUiMessageMetadataFromChunk(chunk: ChatUiMessageChunk<ChatMessageMetadata>): ChatMessageMetadata | undefined;
+/** Return AG-UI chat UI message usage metadata. */
+export declare function getAgUiChatUiMessageUsageMetadata(messageMetadata: ChatMessageMetadata | undefined): Pick<AgUiRunFinishedMetadata, "inputTokens" | "outputTokens" | "totalTokens" | "cachedInputTokens" | "cacheCreationInputTokens" | "cacheReadInputTokens" | "reasoningTokens">;
+/** Return AG-UI chat UI message chunk metadata. */
+export declare function getAgUiChatUiMessageChunkMetadata(chunk: ChatUiMessageChunk<ChatMessageMetadata>, options?: Pick<CreateAgUiChatUiChunkEncoderOptions, "resolveProvider">): Partial<AgUiRunFinishedMetadata> | null;
+/** Event emitted for normalize chat UI message chunk to AG-UI runtime. */
+export declare function normalizeChatUiMessageChunkToAgUiRuntimeEvent(chunk: ChatUiMessageChunk<ChatMessageMetadata>): AgUiRuntimeStreamEvent;
+/** Create AG-UI chat UI chunk encoder. */
+export declare function createAgUiChatUiChunkEncoder(options?: CreateAgUiChatUiChunkEncoderOptions): AgUiChatUiChunkEncoder;
+/** Response payload for create AG-UI chat UI tracked response. */
+export declare function createAgUiChatUiTrackedResponse(input: CreateAgUiChatUiTrackedResponseInput): Response;
+//# sourceMappingURL=chat-ui-chunk-encoder.d.ts.map

package/esm/src/agent/ag-ui/chat-ui-chunk-encoder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-ui-chunk-encoder.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/ag-ui/chat-ui-chunk-encoder.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,mBAAmB,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AACtF,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAE3D,OAAO,EACL,KAAK,uBAAuB,EAC5B,KAAK,uBAAuB,EAC5B,KAAK,sBAAsB,EAC5B,MAAM,cAAc,CAAC;AAEtB,OAAO,EAEL,KAAK,8BAA8B,EACpC,MAAM,uBAAuB,CAAC;AAE/B,2DAA2D;AAC3D,MAAM,MAAM,sBAAsB,GAAG,IAAI,CACvC,gBAAgB,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAC,EACzD,QAAQ,GAAG,UAAU,GAAG,aAAa,CACtC,CAAC;AAEF,8DAA8D;AAC9D,MAAM,WAAW,mCAAmC;IAClD;;;;OAIG;IACH,MAAM,CAAC,EAAE,uBAAuB,CAAC;IACjC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,eAAe,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,MAAM,GAAG,SAAS,CAAC;CAC3D;AAED,+DAA+D;AAC/D,MAAM,WAAW,oCAAqC,SACpD,IAAI,CACF,8BAA8B,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAC,EACvE,cAAc,GAAG,iBAAiB,CACnC;IACD,OAAO,EAAE,MAAM,CAAC;IAChB,eAAe,CAAC,EAAE,mCAAmC,CAAC,iBAAiB,CAAC,CAAC;CAC1E;AAED,wDAAwD;AACxD,wBAAgB,qCAAqC,CACnD,KAAK,EAAE,kBAAkB,CAAC,mBAAmB,CAAC,GAC7C,mBAAmB,GAAG,SAAS,CAUjC;AAED,mDAAmD;AACnD,wBAAgB,iCAAiC,CAC/C,eAAe,EAAE,mBAAmB,GAAG,SAAS,GAC/C,IAAI,CACL,uBAAuB,EACrB,aAAa,GACb,cAAc,GACd,aAAa,GACb,mBAAmB,GACnB,0BAA0B,GAC1B,sBAAsB,GACtB,iBAAiB,CACpB,CAmBA;AA6CD,mDAAmD;AACnD,wBAAgB,iCAAiC,CAC/C,KAAK,EAAE,kBAAkB,CAAC,mBAAmB,CAAC,EAC9C,OAAO,GAAE,IAAI,CAAC,mCAAmC,EAAE,iBAAiB,CAAM,GACzE,OAAO,CAAC,uBAAuB,CAAC,GAAG,IAAI,CAkFzC;AAED,0EAA0E;AAC1E,wBAAgB,6CAA6C,CAC3D,KAAK,EAAE,kBAAkB,CAAC,mBAAmB,CAAC,GAC7C,sBAAsB,CA+BxB;AAED,0CAA0C;AAC1C,wBAAgB,4BAA4B,CAC1C,OAAO,GAAE,mCAAwC,GAChD,sBAAsB,CAcxB;AAED,kEAAkE;AAClE,wBAAgB,+BAA+B,CAC7C,KAAK,EAAE,oCAAoC,GAC1C,QAAQ,CAkBV"}

package/esm/src/agent/ag-ui/{chat-ui-chunk-browser-encoder.js → chat-ui-chunk-encoder.js}

@@ -1,7 +1,7 @@
 import { tryGetVeryfrontCloudProviderFromModelId } from "../../provider/veryfront-cloud/model-catalog.js";
-import { createAgUiBrowserFinalizeTracker } from "./browser-finalize-tracker.js";
-import { createAgUiBrowserChunkEncoder } from "./browser-chunk-encoder.js";
-import { createAgUiTrackedBrowserResponse, } from "./tracked-browser-response.js";
+import { createAgUiFinalizeTracker } from "./finalize-tracker.js";
+import { createAgUiChunkEncoder } from "./chunk-encoder.js";
+import { createAgUiTrackedResponse, } from "./tracked-response.js";
 /** Return AG-UI chat UI message metadata from chunk. */
 export function getAgUiChatUiMessageMetadataFromChunk(chunk) {
     if (chunk.type === "start" || chunk.type === "finish") {
@@ -161,12 +161,12 @@ export function normalizeChatUiMessageChunkToAgUiRuntimeEvent(chunk) {
             };
     }
 }
-/** Create AG-UI chat UI chunk browser encoder. */
-export function createAgUiChatUiChunkBrowserEncoder(options = {}) {
+/** Create AG-UI chat UI chunk encoder. */
+export function createAgUiChatUiChunkEncoder(options = {}) {
     const provider = options.modelId
         ? (options.resolveProvider ?? tryGetVeryfrontCloudProviderFromModelId)(options.modelId)
         : undefined;
-    return createAgUiBrowserChunkEncoder({
+    return createAgUiChunkEncoder({
         initialMetadata: {
             ...(provider ? { provider } : {}),
             ...(options.modelId ? { model: options.modelId } : {}),
@@ -176,16 +176,16 @@ export function createAgUiChatUiChunkBrowserEncoder(options = {}) {
         ...(options.timing === undefined ? {} : { timing: options.timing }),
     });
 }
-/** Response payload for create AG-UI chat UI tracked browser. */
-export function createAgUiChatUiTrackedBrowserResponse(input) {
-    const finalizeTracker = createAgUiBrowserFinalizeTracker({
+/** Response payload for create AG-UI chat UI tracked response. */
+export function createAgUiChatUiTrackedResponse(input) {
+    const finalizeTracker = createAgUiFinalizeTracker({
         getMetadataFromChunk: (chunk) => getAgUiChatUiMessageChunkMetadata(chunk, {
             resolveProvider: input.resolveProvider,
         }),
     });
-    return createAgUiTrackedBrowserResponse({
+    return createAgUiTrackedResponse({
         ...input,
-        chunkEncoder: createAgUiChatUiChunkBrowserEncoder({
+        chunkEncoder: createAgUiChatUiChunkEncoder({
             modelId: input.modelId,
             resolveProvider: input.resolveProvider,
         }),

package/esm/src/agent/ag-ui/chunk-encoder-bridge.d.ts

@@ -1,20 +1,22 @@
-import { type AgUiBrowserEncodedEvent, type AgUiBrowserEncoderState, type AgUiBrowserEncoderStateOptions, type AgUiRuntimeStreamEvent } from "./browser-encoder.js";
+import { type AgUiEncodedEvent, type AgUiEncoderState, type AgUiEncoderStateOptions, type AgUiRuntimeStreamEvent } from "./encoder.js";
 import type { AgentResponse } from "../types.js";
 /** Public API contract for AG-UI chunk encoder bridge. */
 export interface AgUiChunkEncoderBridge<TChunk> {
-    encode: (chunk: TChunk) => AgUiBrowserEncodedEvent[];
-    finalize: (response: AgentResponse | null) => AgUiBrowserEncodedEvent[];
-    state: AgUiBrowserEncoderState;
+    encode: (chunk: TChunk) => AgUiEncodedEvent[];
+    finalize: (response: AgentResponse | null) => AgUiEncodedEvent[];
+    state: AgUiEncoderState;
+    /** Timing anchor consumed by the response composition root. */
+    timingState: AgUiEncoderState;
 }
 /** Options accepted by create AG-UI chunk encoder bridge. */
 export interface CreateAgUiChunkEncoderBridgeOptions<TChunk> {
     getRuntimeEvents: (chunk: TChunk) => readonly AgUiRuntimeStreamEvent[];
     /**
      * Timing clocks forwarded verbatim to the encoder state. A single object so
-     * that adding a clock is one edit in `AgUiBrowserEncoderStateOptions`, not a
+     * that adding a clock is one edit in `AgUiEncoderStateOptions`, not a
      * sweep through every wrapper that happens to sit in between.
      */
-    timing?: AgUiBrowserEncoderStateOptions;
+    timing?: AgUiEncoderStateOptions;
 }
 /** Create AG-UI chunk encoder bridge. */
 export declare function createAgUiChunkEncoderBridge<TChunk>(options: CreateAgUiChunkEncoderBridgeOptions<TChunk>): AgUiChunkEncoderBridge<TChunk>;

package/esm/src/agent/ag-ui/chunk-encoder-bridge.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chunk-encoder-bridge.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/ag-ui/chunk-encoder-bridge.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,uBAAuB,EAC5B,KAAK,uBAAuB,EAC5B,KAAK,8BAA8B,EACnC,KAAK,sBAAsB,EAI5B,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAEjD,0DAA0D;AAC1D,MAAM,WAAW,sBAAsB,CAAC,MAAM;IAC5C,MAAM,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,uBAAuB,EAAE,CAAC;IACrD,QAAQ,EAAE,CAAC,QAAQ,EAAE,aAAa,GAAG,IAAI,KAAK,uBAAuB,EAAE,CAAC;IACxE,KAAK,EAAE,uBAAuB,CAAC;CAChC;AAED,6DAA6D;AAC7D,MAAM,WAAW,mCAAmC,CAAC,MAAM;IACzD,gBAAgB,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,SAAS,sBAAsB,EAAE,CAAC;IACvE;;;;OAIG;IACH,MAAM,CAAC,EAAE,8BAA8B,CAAC;CACzC;AAED,yCAAyC;AACzC,wBAAgB,4BAA4B,CAAC,MAAM,EACjD,OAAO,EAAE,mCAAmC,CAAC,MAAM,CAAC,GACnD,sBAAsB,CAAC,MAAM,CAAC,CAWhC"}
+{"version":3,"file":"chunk-encoder-bridge.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/ag-ui/chunk-encoder-bridge.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,gBAAgB,EACrB,KAAK,gBAAgB,EACrB,KAAK,uBAAuB,EAC5B,KAAK,sBAAsB,EAI5B,MAAM,cAAc,CAAC;AACtB,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAEjD,0DAA0D;AAC1D,MAAM,WAAW,sBAAsB,CAAC,MAAM;IAC5C,MAAM,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,gBAAgB,EAAE,CAAC;IAC9C,QAAQ,EAAE,CAAC,QAAQ,EAAE,aAAa,GAAG,IAAI,KAAK,gBAAgB,EAAE,CAAC;IACjE,KAAK,EAAE,gBAAgB,CAAC;IACxB,+DAA+D;IAC/D,WAAW,EAAE,gBAAgB,CAAC;CAC/B;AAED,6DAA6D;AAC7D,MAAM,WAAW,mCAAmC,CAAC,MAAM;IACzD,gBAAgB,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,SAAS,sBAAsB,EAAE,CAAC;IACvE;;;;OAIG;IACH,MAAM,CAAC,EAAE,uBAAuB,CAAC;CAClC;AAED,yCAAyC;AACzC,wBAAgB,4BAA4B,CAAC,MAAM,EACjD,OAAO,EAAE,mCAAmC,CAAC,MAAM,CAAC,GACnD,sBAAsB,CAAC,MAAM,CAAC,CAYhC"}

package/esm/src/agent/ag-ui/chunk-encoder-bridge.js

@@ -1,10 +1,11 @@
-import { createAgUiBrowserEncoderState, finalizeAgUiBrowserEvents, mapRuntimeStreamEventToAgUiBrowserEvents, } from "./browser-encoder.js";
+import { createAgUiEncoderState, finalizeAgUiEvents, mapRuntimeStreamEventToAgUiEvents, } from "./encoder.js";
 /** Create AG-UI chunk encoder bridge. */
 export function createAgUiChunkEncoderBridge(options) {
-    const state = createAgUiBrowserEncoderState(options.timing ?? {});
+    const state = createAgUiEncoderState(options.timing ?? {});
     return {
         state,
-        encode: (chunk) => options.getRuntimeEvents(chunk).flatMap((event) => mapRuntimeStreamEventToAgUiBrowserEvents(state, event)),
-        finalize: (response) => finalizeAgUiBrowserEvents(state, response),
+        timingState: state,
+        encode: (chunk) => options.getRuntimeEvents(chunk).flatMap((event) => mapRuntimeStreamEventToAgUiEvents(state, event)),
+        finalize: (response) => finalizeAgUiEvents(state, response),
     };
 }

package/esm/src/agent/ag-ui/chunk-encoder.d.ts

@@ -0,0 +1,26 @@
+import { type AgUiEncoderStateOptions } from "./encoder.js";
+import type { AgUiEncodedEvent, AgUiEncoderState, AgUiRunFinishedMetadata, AgUiRuntimeStreamEvent } from "./encoder.js";
+import type { AgentResponse } from "../types.js";
+/** Public API contract for AG-UI chunk encoder. */
+export interface AgUiChunkEncoder<TChunk> {
+    state: AgUiEncoderState;
+    /** Optional timing anchor consumed by the response composition root. */
+    timingState?: AgUiEncoderState;
+    encode: (chunk: TChunk) => AgUiEncodedEvent[];
+    finalize: (response: AgentResponse | null) => AgUiEncodedEvent[];
+}
+/** Options accepted by create AG-UI chunk encoder. */
+export interface CreateAgUiChunkEncoderOptions<TChunk> {
+    /**
+     * Timing clocks forwarded verbatim to the encoder state. A single object so
+     * that adding a clock is one edit in `AgUiEncoderStateOptions`, not a
+     * sweep through every wrapper that happens to sit in between.
+     */
+    timing?: AgUiEncoderStateOptions;
+    getRuntimeEvents: (chunk: TChunk) => readonly AgUiRuntimeStreamEvent[];
+    getMetadataFromChunk?: (chunk: TChunk) => Partial<AgUiRunFinishedMetadata> | null | undefined;
+    initialMetadata?: Partial<AgUiRunFinishedMetadata>;
+}
+/** Create AG-UI chunk encoder. */
+export declare function createAgUiChunkEncoder<TChunk>(options: CreateAgUiChunkEncoderOptions<TChunk>): AgUiChunkEncoder<TChunk>;
+//# sourceMappingURL=chunk-encoder.d.ts.map

package/esm/src/agent/ag-ui/chunk-encoder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chunk-encoder.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/ag-ui/chunk-encoder.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,uBAAuB,EAAE,MAAM,cAAc,CAAC;AAC5D,OAAO,KAAK,EACV,gBAAgB,EAChB,gBAAgB,EAChB,uBAAuB,EACvB,sBAAsB,EACvB,MAAM,cAAc,CAAC;AAKtB,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAEjD,mDAAmD;AACnD,MAAM,WAAW,gBAAgB,CAAC,MAAM;IACtC,KAAK,EAAE,gBAAgB,CAAC;IACxB,wEAAwE;IACxE,WAAW,CAAC,EAAE,gBAAgB,CAAC;IAC/B,MAAM,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,gBAAgB,EAAE,CAAC;IAC9C,QAAQ,EAAE,CAAC,QAAQ,EAAE,aAAa,GAAG,IAAI,KAAK,gBAAgB,EAAE,CAAC;CAClE;AAED,sDAAsD;AACtD,MAAM,WAAW,6BAA6B,CAAC,MAAM;IACnD;;;;OAIG;IACH,MAAM,CAAC,EAAE,uBAAuB,CAAC;IACjC,gBAAgB,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,SAAS,sBAAsB,EAAE,CAAC;IACvE,oBAAoB,CAAC,EAAE,CACrB,KAAK,EAAE,MAAM,KACV,OAAO,CAAC,uBAAuB,CAAC,GAAG,IAAI,GAAG,SAAS,CAAC;IACzD,eAAe,CAAC,EAAE,OAAO,CAAC,uBAAuB,CAAC,CAAC;CACpD;AAkED,kCAAkC;AAClC,wBAAgB,sBAAsB,CAAC,MAAM,EAC3C,OAAO,EAAE,6BAA6B,CAAC,MAAM,CAAC,GAC7C,gBAAgB,CAAC,MAAM,CAAC,CAe1B"}

package/esm/src/agent/ag-ui/{browser-chunk-encoder.js → chunk-encoder.js}

@@ -68,14 +68,15 @@ function mergeMetadata(target, metadata) {
     if (typeof metadata.finishReason === "string")
         target.finishReason = metadata.finishReason;
 }
-/** Create AG-UI browser chunk encoder. */
-export function createAgUiBrowserChunkEncoder(options) {
+/** Create AG-UI chunk encoder. */
+export function createAgUiChunkEncoder(options) {
     const runtimeEventEncoder = createAgUiRuntimeEventEncoder({
         initialMetadata: options.initialMetadata,
         ...(options.timing === undefined ? {} : { timing: options.timing }),
     });
     return {
         state: runtimeEventEncoder.state,
+        timingState: runtimeEventEncoder.state,
         encode: (chunk) => {
             mergeMetadata(runtimeEventEncoder.state.metadata, options.getMetadataFromChunk?.(chunk));
             return options.getRuntimeEvents(chunk).flatMap((event) => runtimeEventEncoder.encode(event));

package/esm/src/agent/ag-ui/{browser-encoder.d.ts → encoder.d.ts}

@@ -3,8 +3,8 @@ import type { AgentResponse } from "../types.js";
 export type AgUiRuntimeStreamEvent = Record<string, unknown> & {
     type: string;
 };
-/** Public API contract for AG-UI browser run finished metadata. */
-export interface AgUiBrowserRunFinishedMetadata {
+/** Public API contract for AG-UI run finished metadata. */
+export interface AgUiRunFinishedMetadata {
     provider?: string;
     model?: string;
     inputTokens?: number;
@@ -30,8 +30,8 @@ export interface AgUiBrowserRunFinishedMetadata {
     finishReason?: string;
     usageCaptureStatus?: "complete" | "partial" | "missing";
 }
-/** State for AG-UI browser encoder. */
-export interface AgUiBrowserEncoderState {
+/** State for AG-UI encoder. */
+export interface AgUiEncoderState {
     messageId: string | null;
     textOpen: boolean;
     activeTextContentId: string | null;
@@ -45,12 +45,24 @@ export interface AgUiBrowserEncoderState {
     activeStepName: string | null;
     stepCount: number;
     streamedToolInputIds: Set<string>;
+    /**
+     * Tool calls whose `ToolCallStart` has been emitted but not yet closed with
+     * a `ToolCallEnd`. Distinct from `streamedToolInputIds`, which tracks
+     * whether any args were streamed, not whether the call is still open.
+     *
+     * Optional, and populated lazily, so a state object built against the shape
+     * this type had before the tracker existed stays valid — the same reason
+     * `reasoningSpanIndex` above is optional. This type is re-exported from
+     * `veryfront/agent`, so a required field would crash existing callers on the
+     * first `tool-input-start`.
+     */
+    openToolCallIds?: Set<string>;
     sawVisibleOutput: boolean;
     sawTerminalError: boolean;
-    metadata: AgUiBrowserRunFinishedMetadata;
+    metadata: AgUiRunFinishedMetadata;
     /**
      * Clock for `elapsedMs`, and the run-relative anchor it measures from. Absent
-     * only when a caller opts out; see `createAgUiBrowserEncoderState`.
+     * only when a caller opts out; see `createAgUiEncoderState`.
      */
     nowMs?: () => number;
     startedMs?: number;
@@ -63,8 +75,8 @@ export interface AgUiBrowserEncoderState {
      */
     epochMs?: () => number;
 }
-/** Options for create AG-UI browser encoder state. */
-export interface AgUiBrowserEncoderStateOptions {
+/** Options for create AG-UI encoder state. */
+export interface AgUiEncoderStateOptions {
     /**
      * Clock used to stamp `elapsedMs`. Defaults to `performance.now`. Pass null
      * to omit the stamp, which keeps exact-payload assertions deterministic.
@@ -75,18 +87,20 @@ export interface AgUiBrowserEncoderStateOptions {
      * `Date.now`. Pass null to omit the stamp.
      */
     epochMs?: (() => number) | null;
+    startedMs?: number;
 }
-/** Event emitted for AG-UI browser encoded. */
-export interface AgUiBrowserEncodedEvent {
+/** Event emitted for AG-UI encoded. */
+export interface AgUiEncodedEvent {
     event: string;
     payload: Record<string, unknown>;
 }
-/** State for create AG-UI browser encoder. */
-export declare function createAgUiBrowserEncoderState(options?: AgUiBrowserEncoderStateOptions): AgUiBrowserEncoderState;
-/** Response payload for build AG-UI browser finalize. */
-export declare function buildAgUiBrowserFinalizeResponse(metadata: AgUiBrowserRunFinishedMetadata): AgentResponse | null;
-/** Map runtime stream event to AG-UI browser events. */
-export declare function mapRuntimeStreamEventToAgUiBrowserEvents(state: AgUiBrowserEncoderState, event: AgUiRuntimeStreamEvent): AgUiBrowserEncodedEvent[];
-/** Finalize AG-UI browser events helper. */
-export declare function finalizeAgUiBrowserEvents(state: AgUiBrowserEncoderState, response: AgentResponse | null): AgUiBrowserEncodedEvent[];
-//# sourceMappingURL=browser-encoder.d.ts.map
+/** State for create AG-UI encoder. */
+export declare function createAgUiEncoderState(options?: AgUiEncoderStateOptions): AgUiEncoderState;
+/** Response payload for build AG-UI finalize. */
+export declare function buildAgUiFinalizeResponse(metadata: AgUiRunFinishedMetadata): AgentResponse | null;
+/** Map runtime stream event to AG-UI events. */
+export declare function mapRuntimeStreamEventToAgUiEvents(state: AgUiEncoderState, event: AgUiRuntimeStreamEvent): AgUiEncodedEvent[];
+export declare function stampAgUiEventTiming(state: AgUiEncoderState, events: AgUiEncodedEvent[]): AgUiEncodedEvent[];
+/** Finalize AG-UI events helper. */
+export declare function finalizeAgUiEvents(state: AgUiEncoderState, response: AgentResponse | null): AgUiEncodedEvent[];
+//# sourceMappingURL=encoder.d.ts.map

package/esm/src/agent/ag-ui/encoder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"encoder.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/ag-ui/encoder.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAEjD,8CAA8C;AAC9C,MAAM,MAAM,sBAAsB,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG;IAAE,IAAI,EAAE,MAAM,CAAA;CAAE,CAAC;AAEhF,2DAA2D;AAC3D,MAAM,WAAW,uBAAuB;IACtC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,wBAAwB,CAAC,EAAE,MAAM,CAAC;IAClC,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,uBAAuB,CAAC,EAAE,MAAM,CAAC;IACjC,wBAAwB,CAAC,EAAE,MAAM,CAAC;IAClC,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,SAAS,GAAG,SAAS,GAAG,SAAS,CAAC;IAC/C,WAAW,CAAC,EAAE,QAAQ,GAAG,UAAU,CAAC;IACpC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,kBAAkB,CAAC,EAAE,UAAU,GAAG,SAAS,GAAG,SAAS,CAAC;CACzD;AAED,+BAA+B;AAC/B,MAAM,WAAW,gBAAgB;IAC/B,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,QAAQ,EAAE,OAAO,CAAC;IAClB,mBAAmB,EAAE,MAAM,GAAG,IAAI,CAAC;IACnC,gBAAgB,EAAE,MAAM,CAAC;IACzB,kBAAkB,EAAE,MAAM,GAAG,IAAI,CAAC;IAClC;;;OAGG;IACH,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,cAAc,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9B,SAAS,EAAE,MAAM,CAAC;IAClB,oBAAoB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAClC;;;;;;;;;;OAUG;IACH,eAAe,CAAC,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAC9B,gBAAgB,EAAE,OAAO,CAAC;IAC1B,gBAAgB,EAAE,OAAO,CAAC;IAC1B,QAAQ,EAAE,uBAAuB,CAAC;IAClC;;;OAGG;IACH,KAAK,CAAC,EAAE,MAAM,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;;;;;OAMG;IACH,OAAO,CAAC,EAAE,MAAM,MAAM,CAAC;CACxB;AAED,8CAA8C;AAC9C,MAAM,WAAW,uBAAuB;IACtC;;;OAGG;IACH,KAAK,CAAC,EAAE,CAAC,MAAM,MAAM,CAAC,GAAG,IAAI,CAAC;IAC9B;;;OAGG;IACH,OAAO,CAAC,EAAE,CAAC,MAAM,MAAM,CAAC,GAAG,IAAI,CAAC;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,uCAAuC;AACvC,MAAM,WAAW,gBAAgB;IAC/B,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAClC;AAED,sCAAsC;AACtC,wBAAgB,sBAAsB,CACpC,OAAO,GAAE,uBAA4B,GACpC,gBAAgB,CAuBlB;AA6SD,iDAAiD;AACjD,wBAAgB,yBAAyB,CACvC,QAAQ,EAAE,uBAAuB,GAChC,aAAa,GAAG,IAAI,CA+HtB;AAuJD,gDAAgD;AAChD,wBAAgB,iCAAiC,CAC/C,KAAK,EAAE,gBAAgB,EACvB,KAAK,EAAE,sBAAsB,GAC5B,gBAAgB,EAAE,CAKpB;AAED,wBAAgB,oBAAoB,CAClC,KAAK,EAAE,gBAAgB,EACvB,MAAM,EAAE,gBAAgB,EAAE,GACzB,gBAAgB,EAAE,CAyCpB;AAgQD,oCAAoC;AACpC,wBAAgB,kBAAkB,CAChC,KAAK,EAAE,gBAAgB,EACvB,QAAQ,EAAE,aAAa,GAAG,IAAI,GAC7B,gBAAgB,EAAE,CAEpB"}

package/esm/src/agent/ag-ui/{browser-encoder.js → encoder.js}

@@ -1,13 +1,13 @@
 import * as dntShim from "../../../_dnt.shims.js";
-/** State for create AG-UI browser encoder. */
-export function createAgUiBrowserEncoderState(options = {}) {
+/** State for create AG-UI encoder. */
+export function createAgUiEncoderState(options = {}) {
     // Clocked by default. This state is built at three separate composition
     // roots, so an opt-in clock only has to be forgotten once to lose elapsedMs
     // for every run -- which is exactly what happened twice before.
     const nowMs = options.nowMs === null ? undefined : options.nowMs ?? (() => performance.now());
     const epochMs = options.epochMs === null ? undefined : options.epochMs ?? (() => Date.now());
     return {
-        ...(nowMs ? { nowMs, startedMs: nowMs() } : {}),
+        ...(nowMs ? { nowMs, startedMs: options.startedMs ?? nowMs() } : {}),
         ...(epochMs ? { epochMs } : {}),
         messageId: null,
         textOpen: false,
@@ -18,6 +18,7 @@ export function createAgUiBrowserEncoderState(options = {}) {
         activeStepName: null,
         stepCount: 0,
         streamedToolInputIds: new Set(),
+        openToolCallIds: new Set(),
         sawVisibleOutput: false,
         sawTerminalError: false,
         metadata: {},
@@ -270,8 +271,8 @@ function applyResponseMetadata(state, response) {
         state.metadata.usageCaptureStatus = usageCaptureStatus;
     }
 }
-/** Response payload for build AG-UI browser finalize. */
-export function buildAgUiBrowserFinalizeResponse(metadata) {
+/** Response payload for build AG-UI finalize. */
+export function buildAgUiFinalizeResponse(metadata) {
     const responseMetadata = {};
     if (typeof metadata.finishReason === "string" && metadata.finishReason.length > 0) {
         responseMetadata.finishReason = metadata.finishReason;
@@ -396,6 +397,19 @@ export function buildAgUiBrowserFinalizeResponse(metadata) {
         ...(Object.keys(responseMetadata).length > 0 ? { metadata: responseMetadata } : {}),
     };
 }
+/**
+ * Emit the `ToolCallEnd` for a call whose input never reached a terminal
+ * input event. Returns nothing when the call was already closed, so a normal
+ * tool failure does not produce a second end.
+ */
+function closeOpenToolInput(state, toolCallId) {
+    if (typeof toolCallId !== "string" || toolCallId.length === 0)
+        return [];
+    if (state.openToolCallIds?.delete(toolCallId) !== true)
+        return [];
+    state.streamedToolInputIds.delete(toolCallId);
+    return [{ event: "ToolCallEnd", payload: { toolCallId } }];
+}
 function completeToolInput(state, event) {
     const toolCallId = typeof event.toolCallId === "string" ? event.toolCallId : "";
     const events = [];
@@ -410,6 +424,7 @@ function completeToolInput(state, event) {
     }
     if (toolCallId.length > 0) {
         state.streamedToolInputIds.delete(toolCallId);
+        state.openToolCallIds?.delete(toolCallId);
     }
     events.push({
         event: "ToolCallEnd",
@@ -485,11 +500,11 @@ function closeOpenReasoningEvent(state) {
             payload: { messageId },
         }];
 }
-/** Map runtime stream event to AG-UI browser events. */
-export function mapRuntimeStreamEventToAgUiBrowserEvents(state, event) {
-    return stampTiming(state, mapRuntimeStreamEventToAgUiBrowserEventsUnstamped(state, event));
+/** Map runtime stream event to AG-UI events. */
+export function mapRuntimeStreamEventToAgUiEvents(state, event) {
+    return stampAgUiEventTiming(state, mapRuntimeStreamEventToAgUiEventsUnstamped(state, event));
 }
-function stampTiming(state, events) {
+export function stampAgUiEventTiming(state, events) {
     if (events.length === 0) {
         return events;
     }
@@ -500,19 +515,49 @@ function stampTiming(state, events) {
     // wall-clock traces and logs, and turns ingest lag into `created_at -
     // emittedAt`. Both are stamped because wall clocks can step backwards and
     // the monotonic reading cannot.
-    const timing = {};
-    if (state.nowMs && state.startedMs !== undefined) {
-        timing.elapsedMs = Math.max(0, Math.round(state.nowMs() - state.startedMs));
+    for (const { payload } of events) {
+        if (Object.hasOwn(payload, "elapsedMs"))
+            assertValidElapsedMs(payload.elapsedMs);
+        if (Object.hasOwn(payload, "emittedAt"))
+            assertValidEmittedAt(payload.emittedAt);
+    }
+    const needsElapsedMs = events.some(({ payload }) => !Object.hasOwn(payload, "elapsedMs"));
+    const needsEmittedAt = events.some(({ payload }) => !Object.hasOwn(payload, "emittedAt"));
+    const elapsedMs = needsElapsedMs && state.nowMs && state.startedMs !== undefined
+        ? Math.max(0, Math.round(state.nowMs() - state.startedMs))
+        : undefined;
+    const emittedAt = needsEmittedAt && state.epochMs ? Math.round(state.epochMs()) : undefined;
+    if (elapsedMs !== undefined)
+        assertValidElapsedMs(elapsedMs);
+    if (emittedAt !== undefined)
+        assertValidEmittedAt(emittedAt);
+    if (elapsedMs === undefined && emittedAt === undefined) {
+        return events;
     }
-    if (state.epochMs) {
-        timing.emittedAt = Math.round(state.epochMs());
+    return events.map((entry) => ({
+        ...entry,
+        payload: {
+            ...entry.payload,
+            ...(elapsedMs !== undefined && !Object.hasOwn(entry.payload, "elapsedMs")
+                ? { elapsedMs }
+                : {}),
+            ...(emittedAt !== undefined && !Object.hasOwn(entry.payload, "emittedAt")
+                ? { emittedAt }
+                : {}),
+        },
+    }));
+}
+function assertValidElapsedMs(value) {
+    if (typeof value !== "number" || !Number.isFinite(value) || value < 0) {
+        throw new TypeError("elapsedMs must be a finite non-negative number");
     }
-    if (Object.keys(timing).length === 0) {
-        return events;
+}
+function assertValidEmittedAt(value) {
+    if (typeof value !== "number" || !Number.isInteger(value) || value < 0) {
+        throw new TypeError("emittedAt must be a non-negative integer");
     }
-    return events.map((entry) => ({ ...entry, payload: { ...entry.payload, ...timing } }));
 }
-function mapRuntimeStreamEventToAgUiBrowserEventsUnstamped(state, event) {
+function mapRuntimeStreamEventToAgUiEventsUnstamped(state, event) {
     if (event.type.startsWith("data-")) {
         const name = event.type.slice("data-".length);
         if (name.length === 0) {
@@ -596,6 +641,9 @@ function mapRuntimeStreamEventToAgUiBrowserEventsUnstamped(state, event) {
                 ...closeOpenReasoningEvent(state),
             ];
             state.sawVisibleOutput = true;
+            if (typeof event.toolCallId === "string" && event.toolCallId.length > 0) {
+                (state.openToolCallIds ??= new Set()).add(event.toolCallId);
+            }
             events.push({
                 event: "ToolCallStart",
                 payload: {
@@ -623,15 +671,11 @@ function mapRuntimeStreamEventToAgUiBrowserEventsUnstamped(state, event) {
             ];
         case "tool-input-available": {
             state.sawVisibleOutput = true;
-            const events = [
+            return [
                 ...closeOpenTextEvent(state),
                 ...closeOpenReasoningEvent(state),
                 ...completeToolInput(state, event),
             ];
-            if (event.providerExecuted === true) {
-                events.push(createToolResultEvent(event.toolCallId, null));
-            }
-            return events;
         }
         case "tool-input-error": {
             state.sawVisibleOutput = true;
@@ -653,6 +697,8 @@ function mapRuntimeStreamEventToAgUiBrowserEventsUnstamped(state, event) {
             return events;
         }
         case "tool-output-available":
+            if (event.preliminary === true)
+                return [];
             state.sawVisibleOutput = true;
             return [
                 ...closeOpenTextEvent(state),
@@ -664,6 +710,14 @@ function mapRuntimeStreamEventToAgUiBrowserEventsUnstamped(state, event) {
             return [
                 ...closeOpenTextEvent(state),
                 ...closeOpenReasoningEvent(state),
+                // A truncated local tool call terminalizes as `tool-input-start`
+                // (plus any partial deltas) and then straight to this event, so the
+                // input is still open. `tool-input-available` and `tool-input-error`
+                // close it via `completeToolInput`; this branch has to close it too,
+                // or the client is left with ToolCallStart and ToolCallResult and no
+                // ToolCallEnd. No synthetic args are emitted: the model never
+                // committed any, and inventing `{}` would claim it did.
+                ...closeOpenToolInput(state, event.toolCallId),
                 createToolResultEvent(event.toolCallId, { error: event.errorText }, true),
             ];
         case "tool-output-denied":
@@ -709,11 +763,11 @@ function mapRuntimeStreamEventToAgUiBrowserEventsUnstamped(state, event) {
             return [];
     }
 }
-/** Finalize AG-UI browser events helper. */
-export function finalizeAgUiBrowserEvents(state, response) {
-    return stampTiming(state, finalizeAgUiBrowserEventsUnstamped(state, response));
+/** Finalize AG-UI events helper. */
+export function finalizeAgUiEvents(state, response) {
+    return stampAgUiEventTiming(state, finalizeAgUiEventsUnstamped(state, response));
 }
-function finalizeAgUiBrowserEventsUnstamped(state, response) {
+function finalizeAgUiEventsUnstamped(state, response) {
     applyResponseMetadata(state, response);
     if (state.sawTerminalError) {
         return [];

package/esm/src/agent/ag-ui/finalize-tracker.d.ts

@@ -0,0 +1,15 @@
+import { type AgUiEncodedEvent, type AgUiRunFinishedMetadata } from "./encoder.js";
+import type { AgentResponse } from "../types.js";
+/** Public API contract for AG-UI finalize tracker. */
+export interface AgUiFinalizeTracker<TChunk> {
+    observeChunk: (chunk: TChunk) => void;
+    observeEncodedEvents: (events: readonly AgUiEncodedEvent[]) => void;
+    getFinalResponse: () => AgentResponse | null;
+}
+/** Options accepted by create AG-UI finalize tracker. */
+export interface CreateAgUiFinalizeTrackerOptions<TChunk> {
+    getMetadataFromChunk: (chunk: TChunk) => Partial<AgUiRunFinishedMetadata> | null | undefined;
+}
+/** Create AG-UI finalize tracker. */
+export declare function createAgUiFinalizeTracker<TChunk>(options: CreateAgUiFinalizeTrackerOptions<TChunk>): AgUiFinalizeTracker<TChunk>;
+//# sourceMappingURL=finalize-tracker.d.ts.map

package/esm/src/agent/ag-ui/finalize-tracker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"finalize-tracker.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/ag-ui/finalize-tracker.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,gBAAgB,EACrB,KAAK,uBAAuB,EAE7B,MAAM,cAAc,CAAC;AACtB,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAEjD,sDAAsD;AACtD,MAAM,WAAW,mBAAmB,CAAC,MAAM;IACzC,YAAY,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IACtC,oBAAoB,EAAE,CAAC,MAAM,EAAE,SAAS,gBAAgB,EAAE,KAAK,IAAI,CAAC;IACpE,gBAAgB,EAAE,MAAM,aAAa,GAAG,IAAI,CAAC;CAC9C;AAED,yDAAyD;AACzD,MAAM,WAAW,gCAAgC,CAAC,MAAM;IACtD,oBAAoB,EAAE,CACpB,KAAK,EAAE,MAAM,KACV,OAAO,CAAC,uBAAuB,CAAC,GAAG,IAAI,GAAG,SAAS,CAAC;CAC1D;AAED,qCAAqC;AACrC,wBAAgB,yBAAyB,CAAC,MAAM,EAC9C,OAAO,EAAE,gCAAgC,CAAC,MAAM,CAAC,GAChD,mBAAmB,CAAC,MAAM,CAAC,CAoG7B"}

package/esm/src/agent/ag-ui/{browser-finalize-tracker.js → finalize-tracker.js}

@@ -1,6 +1,6 @@
-import { buildAgUiBrowserFinalizeResponse, } from "./browser-encoder.js";
-/** Create AG-UI browser finalize tracker. */
-export function createAgUiBrowserFinalizeTracker(options) {
+import { buildAgUiFinalizeResponse, } from "./encoder.js";
+/** Create AG-UI finalize tracker. */
+export function createAgUiFinalizeTracker(options) {
     let sawRunError = false;
     const metadata = {};
     return {
@@ -94,7 +94,7 @@ export function createAgUiBrowserFinalizeTracker(options) {
             if (sawRunError) {
                 return null;
             }
-            return buildAgUiBrowserFinalizeResponse(metadata);
+            return buildAgUiFinalizeResponse(metadata);
         },
     };
 }

package/esm/src/agent/ag-ui/response-stream.d.ts

@@ -0,0 +1,36 @@
+import type { AgentResponse } from "../types.js";
+import type { AgUiSseEvent } from "./host-support.js";
+import { type AgUiEncoderState } from "./encoder.js";
+/** State for AG-UI response request. */
+export interface AgUiResponseRequestState {
+    runId?: string;
+    threadId?: string;
+    state?: unknown;
+    messages: unknown[];
+}
+/** Public API contract for AG-UI response execution. */
+export interface AgUiResponseExecution<TChunk> {
+    agentUIStream: AsyncIterable<TChunk>;
+    fail: (error: unknown) => Promise<void>;
+    waitForFinish: () => Promise<void>;
+}
+/** Public API contract for AG-UI response encoder. */
+export interface AgUiResponseEncoder<TChunk> {
+    encode: (chunk: TChunk) => AgUiSseEvent[];
+    finalize: (response: AgentResponse | null) => AgUiSseEvent[];
+    /** Shared timing anchor for bootstrap, chunk, and final events. */
+    timingState?: AgUiEncoderState;
+}
+/** Input payload for create AG-UI response stream. */
+export interface CreateAgUiResponseStreamInput<TChunk, TState> {
+    agUiInput: AgUiResponseRequestState;
+    agentId: string;
+    execution: AgUiResponseExecution<TChunk>;
+    encoder: AgUiResponseEncoder<TChunk>;
+    initialState: TState;
+    onChunk?: (state: TState, chunk: TChunk) => void;
+    getFinalResponse?: (state: TState) => AgentResponse | null;
+}
+/** Create AG-UI response stream. */
+export declare function createAgUiResponseStream<TChunk, TState>(input: CreateAgUiResponseStreamInput<TChunk, TState>): ReadableStream<Uint8Array>;
+//# sourceMappingURL=response-stream.d.ts.map

package/esm/src/agent/ag-ui/response-stream.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"response-stream.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/ag-ui/response-stream.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AACjD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACtD,OAAO,EAAE,KAAK,gBAAgB,EAAgD,MAAM,cAAc,CAAC;AAkBnG,wCAAwC;AACxC,MAAM,WAAW,wBAAwB;IACvC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,QAAQ,EAAE,OAAO,EAAE,CAAC;CACrB;AAED,wDAAwD;AACxD,MAAM,WAAW,qBAAqB,CAAC,MAAM;IAC3C,aAAa,EAAE,aAAa,CAAC,MAAM,CAAC,CAAC;IACrC,IAAI,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACxC,aAAa,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CACpC;AAED,sDAAsD;AACtD,MAAM,WAAW,mBAAmB,CAAC,MAAM;IACzC,MAAM,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,YAAY,EAAE,CAAC;IAC1C,QAAQ,EAAE,CAAC,QAAQ,EAAE,aAAa,GAAG,IAAI,KAAK,YAAY,EAAE,CAAC;IAC7D,mEAAmE;IACnE,WAAW,CAAC,EAAE,gBAAgB,CAAC;CAChC;AAED,sDAAsD;AACtD,MAAM,WAAW,6BAA6B,CAAC,MAAM,EAAE,MAAM;IAC3D,SAAS,EAAE,wBAAwB,CAAC;IACpC,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,qBAAqB,CAAC,MAAM,CAAC,CAAC;IACzC,OAAO,EAAE,mBAAmB,CAAC,MAAM,CAAC,CAAC;IACrC,YAAY,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IACjD,gBAAgB,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,aAAa,GAAG,IAAI,CAAC;CAC5D;AAUD,oCAAoC;AACpC,wBAAgB,wBAAwB,CAAC,MAAM,EAAE,MAAM,EACrD,KAAK,EAAE,6BAA6B,CAAC,MAAM,EAAE,MAAM,CAAC,GACnD,cAAc,CAAC,UAAU,CAAC,CAuH5B"}

package/esm/src/agent/ag-ui/{browser-response-stream.js → response-stream.js}

@@ -1,3 +1,4 @@
+import { createAgUiEncoderState, stampAgUiEventTiming } from "./encoder.js";
 const encoder = new TextEncoder();
 function formatAgUiSseEventWithId(event, eventId) {
     const idLine = eventId === null ? "" : `id: ${eventId}\n`;
@@ -12,18 +13,20 @@ function isRecord(value) {
 function normalizeSnapshot(snapshot) {
     return isRecord(snapshot) ? snapshot : {};
 }
-/** Create AG-UI browser response stream. */
-export function createAgUiBrowserResponseStream(input) {
+/** Create AG-UI response stream. */
+export function createAgUiResponseStream(input) {
     let streamClosed = false;
     let nextEventId = 1;
     return new ReadableStream({
         start(controller) {
+            const timingState = input.encoder.timingState ?? createAgUiEncoderState();
             const writeEvent = (event) => {
                 if (streamClosed) {
                     return false;
                 }
                 try {
-                    controller.enqueue(formatAgUiSseEventWithId(event, nextEventId));
+                    const [timed] = stampAgUiEventTiming(timingState, [event]);
+                    controller.enqueue(formatAgUiSseEventWithId(timed ?? event, nextEventId));
                     nextEventId += 1;
                     return true;
                 }

package/esm/src/agent/ag-ui/runtime-chat-stream-encoder.d.ts

@@ -1,4 +1,4 @@
-import type { AgUiRuntimeStreamEvent } from "./browser-encoder.js";
+import type { AgUiRuntimeStreamEvent } from "./encoder.js";
 import type { ChatFinishReason, ChatStreamEvent } from "../../chat/protocol.js";
 /** Usage metadata captured from an AG-UI runtime finish event. */
 export type AgUiRuntimeChatStreamUsage = {

package/esm/src/agent/ag-ui/runtime-chat-stream-encoder.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"runtime-chat-stream-encoder.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/ag-ui/runtime-chat-stream-encoder.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AACnE,OAAO,KAAK,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAGhF,kEAAkE;AAClE,MAAM,MAAM,0BAA0B,GAAG;IACvC,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;IACpB,iBAAiB,EAAE;QACjB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;KAC3B,CAAC;IACF,kBAAkB,EAAE;QAClB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,eAAe,CAAC,EAAE,MAAM,CAAC;KAC1B,CAAC;IACF,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,uBAAuB,CAAC,EAAE,MAAM,CAAC;IACjC,wBAAwB,CAAC,EAAE,MAAM,CAAC;IAClC,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,SAAS,GAAG,SAAS,GAAG,SAAS,CAAC;IAC/C,WAAW,CAAC,EAAE,QAAQ,GAAG,UAAU,CAAC;IACpC,kBAAkB,CAAC,EAAE,UAAU,GAAG,SAAS,GAAG,SAAS,CAAC;CACzD,CAAC;AAEF,mDAAmD;AACnD,MAAM,WAAW,iCAAiC;IAChD,UAAU,EAAE,OAAO,CAAC;IACpB,YAAY,EAAE,gBAAgB,CAAC;IAC/B,UAAU,EAAE,0BAA0B,GAAG,IAAI,CAAC;CAC/C;AAED,iEAAiE;AACjE,MAAM,WAAW,4BAA4B;IAC3C,KAAK,EAAE,iCAAiC,CAAC;IACzC,MAAM,EAAE,CAAC,KAAK,EAAE,sBAAsB,KAAK,eAAe,EAAE,CAAC;CAC9D;AAED,oEAAoE;AACpE,MAAM,WAAW,yCAAyC;IACxD,iBAAiB,EAAE,MAAM,CAAC;IAC1B,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,MAAM,CAAC;CACtC;AAmMD,gDAAgD;AAChD,wBAAgB,kCAAkC,CAChD,OAAO,EAAE,yCAAyC,GACjD,4BAA4B,CA+U9B"}
+{"version":3,"file":"runtime-chat-stream-encoder.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/ag-ui/runtime-chat-stream-encoder.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,cAAc,CAAC;AAC3D,OAAO,KAAK,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAGhF,kEAAkE;AAClE,MAAM,MAAM,0BAA0B,GAAG;IACvC,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;IACpB,iBAAiB,EAAE;QACjB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;KAC3B,CAAC;IACF,kBAAkB,EAAE;QAClB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,eAAe,CAAC,EAAE,MAAM,CAAC;KAC1B,CAAC;IACF,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,uBAAuB,CAAC,EAAE,MAAM,CAAC;IACjC,wBAAwB,CAAC,EAAE,MAAM,CAAC;IAClC,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,SAAS,GAAG,SAAS,GAAG,SAAS,CAAC;IAC/C,WAAW,CAAC,EAAE,QAAQ,GAAG,UAAU,CAAC;IACpC,kBAAkB,CAAC,EAAE,UAAU,GAAG,SAAS,GAAG,SAAS,CAAC;CACzD,CAAC;AAEF,mDAAmD;AACnD,MAAM,WAAW,iCAAiC;IAChD,UAAU,EAAE,OAAO,CAAC;IACpB,YAAY,EAAE,gBAAgB,CAAC;IAC/B,UAAU,EAAE,0BAA0B,GAAG,IAAI,CAAC;CAC/C;AAED,iEAAiE;AACjE,MAAM,WAAW,4BAA4B;IAC3C,KAAK,EAAE,iCAAiC,CAAC;IACzC,MAAM,EAAE,CAAC,KAAK,EAAE,sBAAsB,KAAK,eAAe,EAAE,CAAC;CAC9D;AAED,oEAAoE;AACpE,MAAM,WAAW,yCAAyC;IACxD,iBAAiB,EAAE,MAAM,CAAC;IAC1B,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,MAAM,CAAC;CACtC;AAmMD,gDAAgD;AAChD,wBAAgB,kCAAkC,CAChD,OAAO,EAAE,yCAAyC,GACjD,4BAA4B,CA+U9B"}