@assistant-ui/react-langchain 0.0.31 → 0.0.33

package/dist/types.d.ts

@@ -1,12 +1,11 @@
-import { LangChainContentBlock } from "./converter.js";
-import { AttachmentAdapter, DictationAdapter, ExternalStoreSharedOptions, FeedbackAdapter, MessageStatus, RealtimeVoiceAdapter, RemoteThreadListAdapter, SpeechSynthesisAdapter } from "@assistant-ui/core";
-import { AnyStream, AssembledToolCall, SubagentDiscoverySnapshot, SubgraphDiscoverySnapshot, UseStreamOptions } from "@langchain/react";
-import { AssistantCloud } from "assistant-cloud";
-//#region src/types.d.ts
-type LangChainToolCall = {
-  id: string;
-  name: string;
-  args: Record<string, unknown>;
+import type { AttachmentAdapter, DictationAdapter, ExternalStoreSharedOptions, FeedbackAdapter, MessageStatus, RealtimeVoiceAdapter, RemoteThreadListAdapter, SpeechSynthesisAdapter } from "@assistant-ui/core";
+import type { AssistantCloud } from "assistant-cloud";
+import type { UseStreamOptions, AssembledToolCall, AnyStream, SubagentDiscoverySnapshot, SubgraphDiscoverySnapshot } from "@langchain/react";
+export type { LangChainContentBlock } from "./converter.js";
+export type LangChainToolCall = {
+    id: string;
+    name: string;
+    args: Record<string, unknown>;
 };
 /**
  * A generative UI component the graph accumulates in its state. Read from
@@ -15,28 +14,28 @@ type LangChainToolCall = {
  * The parent assistant message is carried in `metadata.message_id` by the
  * Python SDK and in `metadata.id` by the JS SDK; the runtime reads either.
  */
-type UIMessage<TName extends string = string, TProps extends Record<string, unknown> = Record<string, unknown>> = {
-  type: "ui";
-  id: string;
-  name: TName;
-  props: TProps;
-  metadata?: {
-    merge?: boolean;
-    run_id?: string;
-    name?: string;
-    tags?: string[];
-    message_id?: string;
-    id?: string;
-    [key: string]: unknown;
-  };
+export type UIMessage<TName extends string = string, TProps extends Record<string, unknown> = Record<string, unknown>> = {
+    type: "ui";
+    id: string;
+    name: TName;
+    props: TProps;
+    metadata?: {
+        merge?: boolean;
+        run_id?: string;
+        name?: string;
+        tags?: string[];
+        message_id?: string;
+        id?: string;
+        [key: string]: unknown;
+    };
 };
 /**
  * Emitted on the live `custom` channel by the graph's `remove_ui_message`
  * helper to drop a pushed `UIMessage` by id.
  */
-type RemoveUIMessage = {
-  type: "remove-ui";
-  id: string;
+export type RemoveUIMessage = {
+    type: "remove-ui";
+    id: string;
 };
 /**
  * Minimal duck-typed interface for BaseMessage class instances returned by
@@ -45,90 +44,88 @@ type RemoveUIMessage = {
  * Uses `unknown` for `content` to remain compatible with
  * `@langchain/core`'s `MessageContent` union.
  */
-type LangChainBaseMessage = {
-  _getType: () => string;
-  content: unknown;
-  id?: string | undefined;
-  name?: string | undefined;
-  additional_kwargs?: Record<string, unknown> | undefined;
-  /** Present on AIMessage */
-  tool_calls?: readonly LangChainToolCall[] | undefined;
-  /** Present on ToolMessage */
-  tool_call_id?: string | undefined;
-  /** Completion status (AIMessage) or outcome status (ToolMessage) */
-  status?: MessageStatus | "success" | "error" | undefined;
-  /** Present on ToolMessage */
-  artifact?: unknown;
+export type LangChainBaseMessage = {
+    _getType: () => string;
+    content: unknown;
+    id?: string | undefined;
+    name?: string | undefined;
+    additional_kwargs?: Record<string, unknown> | undefined;
+    /** Present on AIMessage */
+    tool_calls?: readonly LangChainToolCall[] | undefined;
+    /** Present on ToolMessage */
+    tool_call_id?: string | undefined;
+    /** Completion status (AIMessage) or outcome status (ToolMessage) */
+    status?: MessageStatus | "success" | "error" | undefined;
+    /** Present on ToolMessage */
+    artifact?: unknown;
 };
-type LangChainRuntimeExtras = {
-  interrupt: {
-    value?: unknown;
-  } | undefined;
-  interrupts: readonly {
-    id?: string;
-    value?: unknown;
-  }[];
-  toolCalls: readonly AssembledToolCall[];
-  subagents: ReadonlyMap<string, SubagentDiscoverySnapshot>;
-  subgraphs: ReadonlyMap<string, SubgraphDiscoverySnapshot>;
-  stream: AnyStream;
-  error: unknown;
-  submit: (values: Record<string, unknown> | null | undefined, options?: Record<string, unknown>) => Promise<void>;
-  respond: (response: unknown, options?: Record<string, unknown>) => Promise<void>;
-  respondAll: (responsesById: Record<string, unknown>, options?: Record<string, unknown>) => Promise<void>;
-  values: Record<string, unknown>;
-  messagesKey: string;
+export type LangChainRuntimeExtras = {
+    interrupt: {
+        value?: unknown;
+    } | undefined;
+    interrupts: readonly {
+        id?: string;
+        value?: unknown;
+    }[];
+    toolCalls: readonly AssembledToolCall[];
+    subagents: ReadonlyMap<string, SubagentDiscoverySnapshot>;
+    subgraphs: ReadonlyMap<string, SubgraphDiscoverySnapshot>;
+    stream: AnyStream;
+    error: unknown;
+    submit: (values: Record<string, unknown> | null | undefined, options?: Record<string, unknown>) => Promise<void>;
+    respond: (response: unknown, options?: Record<string, unknown>) => Promise<void>;
+    respondAll: (responsesById: Record<string, unknown>, options?: Record<string, unknown>) => Promise<void>;
+    values: Record<string, unknown>;
+    messagesKey: string;
 };
-type LangChainRuntimeExtraOptions = ExternalStoreSharedOptions & {
-  /**
-   * Called whenever the active thread's canonical (remote) ID changes, so the
-   * value can be treated as a managed/controlled variable (e.g. synced to a URL
-   * query param). Only the settled remote ID is emitted: while a freshly created
-   * thread is still optimistic the value is `undefined`, and the real ID is
-   * emitted once the thread is initialized; the transient local ID is never
-   * surfaced.
-   */
-  onThreadIdChange?: ((threadId: string | undefined) => void) | undefined;
-  cloud?: AssistantCloud | undefined;
-  adapters?: {
-    attachments?: AttachmentAdapter | undefined;
-    speech?: SpeechSynthesisAdapter | undefined;
-    dictation?: DictationAdapter | undefined;
-    voice?: RealtimeVoiceAdapter | undefined;
-    feedback?: FeedbackAdapter | undefined;
-  } | undefined;
-  /**
-   * When the user sends a new message while previous tool calls are
-   * still pending, automatically submit `tool` messages that cancel
-   * them so the agent's tool-call accounting stays consistent.
-   * Defaults to `true`.
-   */
-  autoCancelPendingToolCalls?: boolean | undefined;
-  /**
-   * Routes the Cancel button's click to `useStream().stop()`. On by
-   * default. Pass `false` to disable the Cancel button.
-   */
-  unstable_allowCancellation?: boolean | undefined;
-  /**
-   * Custom `RemoteThreadListAdapter`. When provided, replaces the
-   * cloud-backed thread list adapter.
-   */
-  unstable_threadListAdapter?: RemoteThreadListAdapter | undefined;
-  /** Custom thread-creation hook, forwarded to the cloud adapter. */
-  create?: (() => Promise<{
-    externalId: string | undefined;
-  }>) | undefined;
-  /** Custom thread-deletion hook, forwarded to the cloud adapter. */
-  delete?: ((threadId: string) => Promise<void>) | undefined;
-  /**
-   * State key the graph accumulates generative `UIMessage`s under. Each UI
-   * is attached to the assistant message identified by its `metadata.message_id`
-   * (Python SDK) or `metadata.id` (JS SDK) and emitted as a `data` part.
-   * Defaults to `"ui"`.
-   */
-  uiStateKey?: string | undefined;
+export type LangChainRuntimeExtraOptions = ExternalStoreSharedOptions & {
+    /**
+     * Called whenever the active thread's canonical (remote) ID changes, so the
+     * value can be treated as a managed/controlled variable (e.g. synced to a URL
+     * query param). Only the settled remote ID is emitted: while a freshly created
+     * thread is still optimistic the value is `undefined`, and the real ID is
+     * emitted once the thread is initialized; the transient local ID is never
+     * surfaced.
+     */
+    onThreadIdChange?: ((threadId: string | undefined) => void) | undefined;
+    cloud?: AssistantCloud | undefined;
+    adapters?: {
+        attachments?: AttachmentAdapter | undefined;
+        speech?: SpeechSynthesisAdapter | undefined;
+        dictation?: DictationAdapter | undefined;
+        voice?: RealtimeVoiceAdapter | undefined;
+        feedback?: FeedbackAdapter | undefined;
+    } | undefined;
+    /**
+     * When the user sends a new message while previous tool calls are
+     * still pending, automatically submit `tool` messages that cancel
+     * them so the agent's tool-call accounting stays consistent.
+     * Defaults to `true`.
+     */
+    autoCancelPendingToolCalls?: boolean | undefined;
+    /**
+     * Routes the Cancel button's click to `useStream().stop()`. On by
+     * default. Pass `false` to disable the Cancel button.
+     */
+    unstable_allowCancellation?: boolean | undefined;
+    /**
+     * Custom `RemoteThreadListAdapter`. When provided, replaces the
+     * cloud-backed thread list adapter.
+     */
+    unstable_threadListAdapter?: RemoteThreadListAdapter | undefined;
+    /** Custom thread-creation hook, forwarded to the cloud adapter. */
+    create?: (() => Promise<{
+        externalId: string | undefined;
+    }>) | undefined;
+    /** Custom thread-deletion hook, forwarded to the cloud adapter. */
+    delete?: ((threadId: string) => Promise<void>) | undefined;
+    /**
+     * State key the graph accumulates generative `UIMessage`s under. Each UI
+     * is attached to the assistant message identified by its `metadata.message_id`
+     * (Python SDK) or `metadata.id` (JS SDK) and emitted as a `data` part.
+     * Defaults to `"ui"`.
+     */
+    uiStateKey?: string | undefined;
 };
-type UseStreamRuntimeOptions = UseStreamOptions extends (infer O) ? O extends UseStreamOptions ? O & LangChainRuntimeExtraOptions : never : never;
-//#endregion
-export { LangChainBaseMessage, type LangChainContentBlock, LangChainRuntimeExtraOptions, LangChainRuntimeExtras, LangChainToolCall, RemoveUIMessage, UIMessage, UseStreamRuntimeOptions };
+export type UseStreamRuntimeOptions = UseStreamOptions extends infer O ? O extends UseStreamOptions ? O & LangChainRuntimeExtraOptions : never : never;
 //# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","names":[],"sources":["../src/types.ts"],"mappings":";;;;;KAqBY;EACV;EACA;EACA,MAAM;;;;;;;;;KAUI,UACV,+BACA,eAAe,0BAA0B;EAEzC;EACA;EACA,MAAM;EACN,OAAO;EACP;IACE;IACA;IACA;IACA;IACA;IACA;KACC;;;;;;;KAQO;EACV;EACA;;;;;;;;;KAUU;EACV;EACA;EACA;EACA;EACA,oBAAoB;;EAEpB,sBAAsB;;EAEtB;;EAEA,SAAS;;EAET;;KAGU;EACV;IAAa;;EACb;IAAuB;IAAa;;EACpC,oBAAoB;EACpB,WAAW,oBAAoB;EAC/B,WAAW,oBAAoB;EAC/B,QAAQ;EACR;EACA,SACE,QAAQ,4CACR,UAAU,4BACP;EACL,UACE,mBACA,UAAU,4BACP;EACL,aACE,eAAe,yBACf,UAAU,4BACP;EACL,QAAQ;EACR;;KAGU,+BAA+B;;;;;;;;;EASzC,qBAAqB;EACrB,QAAQ;EACR;IAEM,cAAc;IACd,SAAS;IACT,YAAY;IACZ,QAAQ;IACR,WAAW;;;;;;;;EASjB;;;;;EAKA;;;;;EAKA,6BAA6B;;EAE7B,gBAAgB;IAAU;;;EAE1B,WAAW,qBAAqB;;;;;;;EAOhC;;KAOU,0BAA0B,gCAA+B,KACjE,UAAU,mBACR,IAAI"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,iBAAiB,EACjB,gBAAgB,EAChB,0BAA0B,EAC1B,eAAe,EACf,aAAa,EACb,oBAAoB,EACpB,uBAAuB,EACvB,sBAAsB,EACvB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,KAAK,EACV,gBAAgB,EAChB,iBAAiB,EACjB,SAAS,EACT,yBAAyB,EACzB,yBAAyB,EAC1B,MAAM,kBAAkB,CAAC;AAE1B,YAAY,EAAE,qBAAqB,EAAE,uBAAoB;AAEzD,MAAM,MAAM,iBAAiB,GAAG;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC/B,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,MAAM,SAAS,CACnB,KAAK,SAAS,MAAM,GAAG,MAAM,EAC7B,MAAM,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,IAC9D;IACF,IAAI,EAAE,IAAI,CAAC;IACX,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,KAAK,CAAC;IACZ,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE;QACT,KAAK,CAAC,EAAE,OAAO,CAAC;QAChB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;QAChB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,EAAE,CAAC,EAAE,MAAM,CAAC;QACZ,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;KACxB,CAAC;CACH,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,eAAe,GAAG;IAC5B,IAAI,EAAE,WAAW,CAAC;IAClB,EAAE,EAAE,MAAM,CAAC;CACZ,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,MAAM,oBAAoB,GAAG;IACjC,QAAQ,EAAE,MAAM,MAAM,CAAC;IACvB,OAAO,EAAE,OAAO,CAAC;IACjB,EAAE,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACxB,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,iBAAiB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAC;IACxD,2BAA2B;IAC3B,UAAU,CAAC,EAAE,SAAS,iBAAiB,EAAE,GAAG,SAAS,CAAC;IACtD,6BAA6B;IAC7B,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAClC,oEAAoE;IACpE,MAAM,CAAC,EAAE,aAAa,GAAG,SAAS,GAAG,OAAO,GAAG,SAAS,CAAC;IACzD,6BAA6B;IAC7B,QAAQ,CAAC,EAAE,OAAO,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG;IACnC,SAAS,EAAE;QAAE,KAAK,CAAC,EAAE,OAAO,CAAA;KAAE,GAAG,SAAS,CAAC;IAC3C,UAAU,EAAE,SAAS;QAAE,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,OAAO,CAAA;KAAE,EAAE,CAAC;IACxD,SAAS,EAAE,SAAS,iBAAiB,EAAE,CAAC;IACxC,SAAS,EAAE,WAAW,CAAC,MAAM,EAAE,yBAAyB,CAAC,CAAC;IAC1D,SAAS,EAAE,WAAW,CAAC,MAAM,EAAE,yBAAyB,CAAC,CAAC;IAC1D,MAAM,EAAE,SAAS,CAAC;IAClB,KAAK,EAAE,OAAO,CAAC;IACf,MAAM,EAAE,CACN,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,GAAG,SAAS,EAClD,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC9B,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,OAAO,EAAE,CACP,QAAQ,EAAE,OAAO,EACjB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC9B,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,UAAU,EAAE,CACV,aAAa,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACtC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC9B,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAChC,WAAW,EAAE,MAAM,CAAC;CACrB,CAAC;AAEF,MAAM,MAAM,4BAA4B,GAAG,0BAA0B,GAAG;IACtE;;;;;;;OAOG;IACH,gBAAgB,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,MAAM,GAAG,SAAS,KAAK,IAAI,CAAC,GAAG,SAAS,CAAC;IACxE,KAAK,CAAC,EAAE,cAAc,GAAG,SAAS,CAAC;IACnC,QAAQ,CAAC,EACL;QACE,WAAW,CAAC,EAAE,iBAAiB,GAAG,SAAS,CAAC;QAC5C,MAAM,CAAC,EAAE,sBAAsB,GAAG,SAAS,CAAC;QAC5C,SAAS,CAAC,EAAE,gBAAgB,GAAG,SAAS,CAAC;QACzC,KAAK,CAAC,EAAE,oBAAoB,GAAG,SAAS,CAAC;QACzC,QAAQ,CAAC,EAAE,eAAe,GAAG,SAAS,CAAC;KACxC,GACD,SAAS,CAAC;IACd;;;;;OAKG;IACH,0BAA0B,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACjD;;;OAGG;IACH,0BAA0B,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACjD;;;OAGG;IACH,0BAA0B,CAAC,EAAE,uBAAuB,GAAG,SAAS,CAAC;IACjE,mEAAmE;IACnE,MAAM,CAAC,EAAE,CAAC,MAAM,OAAO,CAAC;QAAE,UAAU,EAAE,MAAM,GAAG,SAAS,CAAA;KAAE,CAAC,CAAC,GAAG,SAAS,CAAC;IACzE,mEAAmE;IACnE,MAAM,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,SAAS,CAAC;IAC3D;;;;;OAKG;IACH,UAAU,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACjC,CAAC;AAMF,MAAM,MAAM,uBAAuB,GAAG,gBAAgB,SAAS,MAAM,CAAC,GAClE,CAAC,SAAS,gBAAgB,GACxB,CAAC,GAAG,4BAA4B,GAChC,KAAK,GACP,KAAK,CAAC"}

package/dist/uiMessages.d.ts

@@ -1,15 +1,44 @@
-import { RemoveUIMessage, UIMessage } from "./types.js";
-//#region src/uiMessages.d.ts
-type UIUpdate = UIMessage | RemoveUIMessage;
-declare const isUIUpdate: (value: unknown) => value is UIUpdate | readonly UIUpdate[];
-declare const applyUIUpdate: (list: readonly UIMessage[], update: UIUpdate | readonly UIUpdate[]) => UIMessage[];
+import type { Channel } from "@langchain/react";
+import type { RemoveUIMessage, UIMessage } from "./types.js";
+/** Channels the generative-UI fold reads, at the root and per subagent. */
+export declare const UI_CUSTOM_CHANNELS: readonly Channel[];
+export type UIUpdate = UIMessage | RemoveUIMessage;
+export declare const isUIUpdate: (value: unknown) => value is UIUpdate | readonly UIUpdate[];
+export declare const applyUIUpdate: (list: readonly UIMessage[], update: UIUpdate | readonly UIUpdate[]) => UIMessage[];
 /**
  * Pulls a UI update out of a raw `custom`-channel event. The graph writes the
  * `UIMessage` straight to the channel, so it lands at `params.data`; some
  * transports wrap it one level deeper at `params.data.payload`.
  */
-declare const extractUIUpdate: (event: unknown) => UIUpdate | readonly UIUpdate[] | undefined;
-declare const foldUIUpdates: (events: readonly unknown[]) => UIMessage[];
+export declare const extractUIUpdate: (event: unknown) => UIUpdate | readonly UIUpdate[] | undefined;
+export type UIFoldMemo = {
+    events: readonly unknown[];
+    messages: UIMessage[];
+};
+export declare const createUIFoldMemo: () => UIFoldMemo;
+/**
+ * Folds `custom`-channel events into UI messages, continuing the fold held in
+ * `memo`. The channel buffer holds each event object once, appends new events
+ * and drops its oldest ones once full, so only events after the previously
+ * folded last event are applied, and entries no later event touches keep
+ * their identity. A buffer that no longer contains that event was replaced
+ * and is folded from scratch.
+ */
+export declare const foldUIUpdates: (events: readonly unknown[], memo?: UIFoldMemo) => UIMessage[];
+export type UISnapshotMemo = {
+    snapshot: unknown;
+    entries: readonly UIMessage[];
+};
+export declare const createUISnapshotMemo: () => UISnapshotMemo;
+/**
+ * Recovers entry identity across `values` events. The SDK rebuilds the
+ * `values` object from every snapshot and reconciles only the messages slot by
+ * id, so an unchanged UI list arrives as a new array of new objects on every
+ * superstep. An entry structurally equal to the previous entry with its id is
+ * replaced by that previous object, and a list whose entries all survive in
+ * place is replaced by the previous list.
+ */
+export declare const reconcileUISnapshot: (snapshot: unknown, memo: UISnapshotMemo) => readonly UIMessage[];
 /**
  * Merges live-streamed UI with the state snapshot. The snapshot is
  * authoritative by id: once a UI lands in graph state it supersedes its live
@@ -17,7 +46,5 @@ declare const foldUIUpdates: (events: readonly unknown[]) => UIMessage[];
  * snapshot still contains that id; the removal only takes visible effect once
  * the snapshot catches up.
  */
-declare const mergeUIMessages: (live: readonly UIMessage[], snapshot: unknown) => UIMessage[];
-//#endregion
-export { UIUpdate, applyUIUpdate, extractUIUpdate, foldUIUpdates, isUIUpdate, mergeUIMessages };
+export declare const mergeUIMessages: (live: readonly UIMessage[], snapshot: unknown) => UIMessage[];
 //# sourceMappingURL=uiMessages.d.ts.map

package/dist/uiMessages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"uiMessages.d.ts","names":[],"sources":["../src/uiMessages.ts"],"mappings":";;KAEY,WAAW,YAAY;cAEtB,aACX,mBACC,SAAS,oBAAoB;cAmBnB,gBACX,eAAe,aACf,QAAQ,oBAAoB,eAC3B;;;;;;cA0BU,kBACX,mBACC,oBAAoB;cAQV,gBAAiB,+BAA6B;;;;;;;;cAgB9C,kBACX,eAAe,aACf,sBACC"}
+{"version":3,"file":"uiMessages.d.ts","sourceRoot":"","sources":["../src/uiMessages.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAEhD,OAAO,KAAK,EAAE,eAAe,EAAE,SAAS,EAAE,mBAAgB;AAE1D,2EAA2E;AAC3E,eAAO,MAAM,kBAAkB,EAAE,SAAS,OAAO,EAAe,CAAC;AAEjE,MAAM,MAAM,QAAQ,GAAG,SAAS,GAAG,eAAe,CAAC;AAEnD,eAAO,MAAM,UAAU,GACrB,OAAO,OAAO,KACb,KAAK,IAAI,QAAQ,GAAG,SAAS,QAAQ,EAiBvC,CAAC;AAEF,eAAO,MAAM,aAAa,GACxB,MAAM,SAAS,SAAS,EAAE,EAC1B,QAAQ,QAAQ,GAAG,SAAS,QAAQ,EAAE,KACrC,SAAS,EAmBX,CAAC;AAEF;;;;GAIG;AACH,eAAO,MAAM,eAAe,GAC1B,OAAO,OAAO,KACb,QAAQ,GAAG,SAAS,QAAQ,EAAE,GAAG,SAMnC,CAAC;AAEF,MAAM,MAAM,UAAU,GAAG;IACvB,MAAM,EAAE,SAAS,OAAO,EAAE,CAAC;IAC3B,QAAQ,EAAE,SAAS,EAAE,CAAC;CACvB,CAAC;AAEF,eAAO,MAAM,gBAAgB,QAAO,UAGlC,CAAC;AAEH;;;;;;;GAOG;AACH,eAAO,MAAM,aAAa,GACxB,QAAQ,SAAS,OAAO,EAAE,EAC1B,OAAM,UAA+B,KACpC,SAAS,EAcX,CAAC;AAEF,MAAM,MAAM,cAAc,GAAG;IAC3B,QAAQ,EAAE,OAAO,CAAC;IAClB,OAAO,EAAE,SAAS,SAAS,EAAE,CAAC;CAC/B,CAAC;AAEF,eAAO,MAAM,oBAAoB,QAAO,cAGtC,CAAC;AAEH;;;;;;;GAOG;AACH,eAAO,MAAM,mBAAmB,GAC9B,UAAU,OAAO,EACjB,MAAM,cAAc,KACnB,SAAS,SAAS,EAmBpB,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,eAAe,GAC1B,MAAM,SAAS,SAAS,EAAE,EAC1B,UAAU,OAAO,KAChB,SAAS,EAOX,CAAC"}

package/dist/uiMessages.js

@@ -1,4 +1,7 @@
+import { isJSONValueEqual } from "@assistant-ui/core/internal";
 //#region src/uiMessages.ts
+/** Channels the generative-UI fold reads, at the root and per subagent. */
+const UI_CUSTOM_CHANNELS = ["custom"];
 const isUIUpdate = (value) => {
 	if (Array.isArray(value)) return value.length > 0 && value.every(isUIUpdate);
 	if (value == null || typeof value !== "object") return false;
@@ -41,14 +44,61 @@ const extractUIUpdate = (event) => {
 	const payload = data?.payload;
 	if (isUIUpdate(payload)) return payload;
 };
-const foldUIUpdates = (events) => {
-	let acc = [];
-	for (const event of events) {
-		const update = extractUIUpdate(event);
+const createUIFoldMemo = () => ({
+	events: [],
+	messages: []
+});
+/**
+* Folds `custom`-channel events into UI messages, continuing the fold held in
+* `memo`. The channel buffer holds each event object once, appends new events
+* and drops its oldest ones once full, so only events after the previously
+* folded last event are applied, and entries no later event touches keep
+* their identity. A buffer that no longer contains that event was replaced
+* and is folded from scratch.
+*/
+const foldUIUpdates = (events, memo = createUIFoldMemo()) => {
+	const previous = memo.events;
+	const resumeAt = previous.length === 0 ? 0 : events.lastIndexOf(previous.at(-1), previous.length - 1) + 1;
+	let acc = previous.length > 0 && resumeAt === 0 ? [] : memo.messages;
+	for (let i = resumeAt; i < events.length; i++) {
+		const update = extractUIUpdate(events[i]);
 		if (update) acc = applyUIUpdate(acc, update);
 	}
+	memo.events = events;
+	memo.messages = acc;
 	return acc;
 };
+const createUISnapshotMemo = () => ({
+	snapshot: void 0,
+	entries: []
+});
+/**
+* Recovers entry identity across `values` events. The SDK rebuilds the
+* `values` object from every snapshot and reconciles only the messages slot by
+* id, so an unchanged UI list arrives as a new array of new objects on every
+* superstep. An entry structurally equal to the previous entry with its id is
+* replaced by that previous object, and a list whose entries all survive in
+* place is replaced by the previous list.
+*/
+const reconcileUISnapshot = (snapshot, memo) => {
+	if (snapshot === memo.snapshot) return memo.entries;
+	memo.snapshot = snapshot;
+	if (!Array.isArray(snapshot)) {
+		if (memo.entries.length > 0) memo.entries = [];
+		return memo.entries;
+	}
+	const previous = memo.entries;
+	const previousById = new Map(previous.map((ui) => [ui.id, ui]));
+	let same = snapshot.length === previous.length;
+	const entries = snapshot.map((ui, index) => {
+		const before = previousById.get(ui.id);
+		const entry = before !== void 0 && isJSONValueEqual(before, ui) ? before : ui;
+		if (entry !== previous[index]) same = false;
+		return entry;
+	});
+	if (!same) memo.entries = entries;
+	return memo.entries;
+};
 /**
 * Merges live-streamed UI with the state snapshot. The snapshot is
 * authoritative by id: once a UI lands in graph state it supersedes its live
@@ -63,6 +113,6 @@ const mergeUIMessages = (live, snapshot) => {
 	return [...byId.values()];
 };
 //#endregion
-export { applyUIUpdate, extractUIUpdate, foldUIUpdates, isUIUpdate, mergeUIMessages };
+export { UI_CUSTOM_CHANNELS, applyUIUpdate, createUIFoldMemo, createUISnapshotMemo, extractUIUpdate, foldUIUpdates, isUIUpdate, mergeUIMessages, reconcileUISnapshot };
 
 //# sourceMappingURL=uiMessages.js.map

package/dist/uiMessages.js.map

@@ -1 +1 @@
-{"version":3,"file":"uiMessages.js","names":[],"sources":["../src/uiMessages.ts"],"sourcesContent":["import type { RemoveUIMessage, UIMessage } from \"./types\";\n\nexport type UIUpdate = UIMessage | RemoveUIMessage;\n\nexport const isUIUpdate = (\n  value: unknown,\n): value is UIUpdate | readonly UIUpdate[] => {\n  if (Array.isArray(value)) return value.length > 0 && value.every(isUIUpdate);\n  if (value == null || typeof value !== \"object\") return false;\n  const v = value as {\n    type?: unknown;\n    id?: unknown;\n    name?: unknown;\n    props?: unknown;\n  };\n  if (typeof v.id !== \"string\") return false;\n  if (v.type === \"remove-ui\") return true;\n  return (\n    v.type === \"ui\" &&\n    typeof v.name === \"string\" &&\n    typeof v.props === \"object\" &&\n    v.props !== null\n  );\n};\n\nexport const applyUIUpdate = (\n  list: readonly UIMessage[],\n  update: UIUpdate | readonly UIUpdate[],\n): UIMessage[] => {\n  const events = Array.isArray(update) ? update : [update as UIUpdate];\n  let next = list.slice();\n  for (const event of events) {\n    if (event.type === \"remove-ui\") {\n      next = next.filter((ui) => ui.id !== event.id);\n      continue;\n    }\n    const index = next.findIndex((ui) => ui.id === event.id);\n    if (index === -1) {\n      next.push(event);\n      continue;\n    }\n    next[index] =\n      event.metadata?.merge === true\n        ? { ...event, props: { ...next[index]!.props, ...event.props } }\n        : event;\n  }\n  return next;\n};\n\n/**\n * Pulls a UI update out of a raw `custom`-channel event. The graph writes the\n * `UIMessage` straight to the channel, so it lands at `params.data`; some\n * transports wrap it one level deeper at `params.data.payload`.\n */\nexport const extractUIUpdate = (\n  event: unknown,\n): UIUpdate | readonly UIUpdate[] | undefined => {\n  const data = (event as { params?: { data?: unknown } } | null)?.params?.data;\n  if (isUIUpdate(data)) return data;\n  const payload = (data as { payload?: unknown } | undefined)?.payload;\n  if (isUIUpdate(payload)) return payload;\n  return undefined;\n};\n\nexport const foldUIUpdates = (events: readonly unknown[]): UIMessage[] => {\n  let acc: UIMessage[] = [];\n  for (const event of events) {\n    const update = extractUIUpdate(event);\n    if (update) acc = applyUIUpdate(acc, update);\n  }\n  return acc;\n};\n\n/**\n * Merges live-streamed UI with the state snapshot. The snapshot is\n * authoritative by id: once a UI lands in graph state it supersedes its live\n * copy. A consequence is that a live `remove-ui` is overridden while the\n * snapshot still contains that id; the removal only takes visible effect once\n * the snapshot catches up.\n */\nexport const mergeUIMessages = (\n  live: readonly UIMessage[],\n  snapshot: unknown,\n): UIMessage[] => {\n  const byId = new Map<string, UIMessage>();\n  for (const ui of live) byId.set(ui.id, ui);\n  if (Array.isArray(snapshot)) {\n    for (const ui of snapshot as UIMessage[]) byId.set(ui.id, ui);\n  }\n  return [...byId.values()];\n};\n"],"mappings":";AAIA,MAAa,cACX,UAC4C;CAC5C,IAAI,MAAM,QAAQ,KAAK,GAAG,OAAO,MAAM,SAAS,KAAK,MAAM,MAAM,UAAU;CAC3E,IAAI,SAAS,QAAQ,OAAO,UAAU,UAAU,OAAO;CACvD,MAAM,IAAI;CAMV,IAAI,OAAO,EAAE,OAAO,UAAU,OAAO;CACrC,IAAI,EAAE,SAAS,aAAa,OAAO;CACnC,OACE,EAAE,SAAS,QACX,OAAO,EAAE,SAAS,YAClB,OAAO,EAAE,UAAU,YACnB,EAAE,UAAU;AAEhB;AAEA,MAAa,iBACX,MACA,WACgB;CAChB,MAAM,SAAS,MAAM,QAAQ,MAAM,IAAI,SAAS,CAAC,MAAkB;CACnE,IAAI,OAAO,KAAK,MAAM;CACtB,KAAK,MAAM,SAAS,QAAQ;EAC1B,IAAI,MAAM,SAAS,aAAa;GAC9B,OAAO,KAAK,QAAQ,OAAO,GAAG,OAAO,MAAM,EAAE;GAC7C;EACF;EACA,MAAM,QAAQ,KAAK,WAAW,OAAO,GAAG,OAAO,MAAM,EAAE;EACvD,IAAI,UAAU,IAAI;GAChB,KAAK,KAAK,KAAK;GACf;EACF;EACA,KAAK,SACH,MAAM,UAAU,UAAU,OACtB;GAAE,GAAG;GAAO,OAAO;IAAE,GAAG,KAAK,MAAM,CAAE;IAAO,GAAG,MAAM;GAAM;EAAE,IAC7D;CACR;CACA,OAAO;AACT;;;;;;AAOA,MAAa,mBACX,UAC+C;CAC/C,MAAM,OAAQ,OAAkD,QAAQ;CACxE,IAAI,WAAW,IAAI,GAAG,OAAO;CAC7B,MAAM,UAAW,MAA4C;CAC7D,IAAI,WAAW,OAAO,GAAG,OAAO;AAElC;AAEA,MAAa,iBAAiB,WAA4C;CACxE,IAAI,MAAmB,CAAC;CACxB,KAAK,MAAM,SAAS,QAAQ;EAC1B,MAAM,SAAS,gBAAgB,KAAK;EACpC,IAAI,QAAQ,MAAM,cAAc,KAAK,MAAM;CAC7C;CACA,OAAO;AACT;;;;;;;;AASA,MAAa,mBACX,MACA,aACgB;CAChB,MAAM,uBAAO,IAAI,IAAuB;CACxC,KAAK,MAAM,MAAM,MAAM,KAAK,IAAI,GAAG,IAAI,EAAE;CACzC,IAAI,MAAM,QAAQ,QAAQ,GACxB,KAAK,MAAM,MAAM,UAAyB,KAAK,IAAI,GAAG,IAAI,EAAE;CAE9D,OAAO,CAAC,GAAG,KAAK,OAAO,CAAC;AAC1B"}
+{"version":3,"file":"uiMessages.js","names":[],"sources":["../src/uiMessages.ts"],"sourcesContent":["import type { Channel } from \"@langchain/react\";\nimport { isJSONValueEqual } from \"@assistant-ui/core/internal\";\nimport type { RemoveUIMessage, UIMessage } from \"./types\";\n\n/** Channels the generative-UI fold reads, at the root and per subagent. */\nexport const UI_CUSTOM_CHANNELS: readonly Channel[] = [\"custom\"];\n\nexport type UIUpdate = UIMessage | RemoveUIMessage;\n\nexport const isUIUpdate = (\n  value: unknown,\n): value is UIUpdate | readonly UIUpdate[] => {\n  if (Array.isArray(value)) return value.length > 0 && value.every(isUIUpdate);\n  if (value == null || typeof value !== \"object\") return false;\n  const v = value as {\n    type?: unknown;\n    id?: unknown;\n    name?: unknown;\n    props?: unknown;\n  };\n  if (typeof v.id !== \"string\") return false;\n  if (v.type === \"remove-ui\") return true;\n  return (\n    v.type === \"ui\" &&\n    typeof v.name === \"string\" &&\n    typeof v.props === \"object\" &&\n    v.props !== null\n  );\n};\n\nexport const applyUIUpdate = (\n  list: readonly UIMessage[],\n  update: UIUpdate | readonly UIUpdate[],\n): UIMessage[] => {\n  const events = Array.isArray(update) ? update : [update as UIUpdate];\n  let next = list.slice();\n  for (const event of events) {\n    if (event.type === \"remove-ui\") {\n      next = next.filter((ui) => ui.id !== event.id);\n      continue;\n    }\n    const index = next.findIndex((ui) => ui.id === event.id);\n    if (index === -1) {\n      next.push(event);\n      continue;\n    }\n    next[index] =\n      event.metadata?.merge === true\n        ? { ...event, props: { ...next[index]!.props, ...event.props } }\n        : event;\n  }\n  return next;\n};\n\n/**\n * Pulls a UI update out of a raw `custom`-channel event. The graph writes the\n * `UIMessage` straight to the channel, so it lands at `params.data`; some\n * transports wrap it one level deeper at `params.data.payload`.\n */\nexport const extractUIUpdate = (\n  event: unknown,\n): UIUpdate | readonly UIUpdate[] | undefined => {\n  const data = (event as { params?: { data?: unknown } } | null)?.params?.data;\n  if (isUIUpdate(data)) return data;\n  const payload = (data as { payload?: unknown } | undefined)?.payload;\n  if (isUIUpdate(payload)) return payload;\n  return undefined;\n};\n\nexport type UIFoldMemo = {\n  events: readonly unknown[];\n  messages: UIMessage[];\n};\n\nexport const createUIFoldMemo = (): UIFoldMemo => ({\n  events: [],\n  messages: [],\n});\n\n/**\n * Folds `custom`-channel events into UI messages, continuing the fold held in\n * `memo`. The channel buffer holds each event object once, appends new events\n * and drops its oldest ones once full, so only events after the previously\n * folded last event are applied, and entries no later event touches keep\n * their identity. A buffer that no longer contains that event was replaced\n * and is folded from scratch.\n */\nexport const foldUIUpdates = (\n  events: readonly unknown[],\n  memo: UIFoldMemo = createUIFoldMemo(),\n): UIMessage[] => {\n  const previous = memo.events;\n  const resumeAt =\n    previous.length === 0\n      ? 0\n      : events.lastIndexOf(previous.at(-1), previous.length - 1) + 1;\n  let acc = previous.length > 0 && resumeAt === 0 ? [] : memo.messages;\n  for (let i = resumeAt; i < events.length; i++) {\n    const update = extractUIUpdate(events[i]);\n    if (update) acc = applyUIUpdate(acc, update);\n  }\n  memo.events = events;\n  memo.messages = acc;\n  return acc;\n};\n\nexport type UISnapshotMemo = {\n  snapshot: unknown;\n  entries: readonly UIMessage[];\n};\n\nexport const createUISnapshotMemo = (): UISnapshotMemo => ({\n  snapshot: undefined,\n  entries: [],\n});\n\n/**\n * Recovers entry identity across `values` events. The SDK rebuilds the\n * `values` object from every snapshot and reconciles only the messages slot by\n * id, so an unchanged UI list arrives as a new array of new objects on every\n * superstep. An entry structurally equal to the previous entry with its id is\n * replaced by that previous object, and a list whose entries all survive in\n * place is replaced by the previous list.\n */\nexport const reconcileUISnapshot = (\n  snapshot: unknown,\n  memo: UISnapshotMemo,\n): readonly UIMessage[] => {\n  if (snapshot === memo.snapshot) return memo.entries;\n  memo.snapshot = snapshot;\n  if (!Array.isArray(snapshot)) {\n    if (memo.entries.length > 0) memo.entries = [];\n    return memo.entries;\n  }\n  const previous = memo.entries;\n  const previousById = new Map(previous.map((ui) => [ui.id, ui]));\n  let same = snapshot.length === previous.length;\n  const entries = (snapshot as UIMessage[]).map((ui, index) => {\n    const before = previousById.get(ui.id);\n    const entry =\n      before !== undefined && isJSONValueEqual(before, ui) ? before : ui;\n    if (entry !== previous[index]) same = false;\n    return entry;\n  });\n  if (!same) memo.entries = entries;\n  return memo.entries;\n};\n\n/**\n * Merges live-streamed UI with the state snapshot. The snapshot is\n * authoritative by id: once a UI lands in graph state it supersedes its live\n * copy. A consequence is that a live `remove-ui` is overridden while the\n * snapshot still contains that id; the removal only takes visible effect once\n * the snapshot catches up.\n */\nexport const mergeUIMessages = (\n  live: readonly UIMessage[],\n  snapshot: unknown,\n): UIMessage[] => {\n  const byId = new Map<string, UIMessage>();\n  for (const ui of live) byId.set(ui.id, ui);\n  if (Array.isArray(snapshot)) {\n    for (const ui of snapshot as UIMessage[]) byId.set(ui.id, ui);\n  }\n  return [...byId.values()];\n};\n"],"mappings":";;;AAKA,MAAa,qBAAyC,CAAC,QAAQ;AAI/D,MAAa,cACX,UAC4C;CAC5C,IAAI,MAAM,QAAQ,KAAK,GAAG,OAAO,MAAM,SAAS,KAAK,MAAM,MAAM,UAAU;CAC3E,IAAI,SAAS,QAAQ,OAAO,UAAU,UAAU,OAAO;CACvD,MAAM,IAAI;CAMV,IAAI,OAAO,EAAE,OAAO,UAAU,OAAO;CACrC,IAAI,EAAE,SAAS,aAAa,OAAO;CACnC,OACE,EAAE,SAAS,QACX,OAAO,EAAE,SAAS,YAClB,OAAO,EAAE,UAAU,YACnB,EAAE,UAAU;AAEhB;AAEA,MAAa,iBACX,MACA,WACgB;CAChB,MAAM,SAAS,MAAM,QAAQ,MAAM,IAAI,SAAS,CAAC,MAAkB;CACnE,IAAI,OAAO,KAAK,MAAM;CACtB,KAAK,MAAM,SAAS,QAAQ;EAC1B,IAAI,MAAM,SAAS,aAAa;GAC9B,OAAO,KAAK,QAAQ,OAAO,GAAG,OAAO,MAAM,EAAE;GAC7C;EACF;EACA,MAAM,QAAQ,KAAK,WAAW,OAAO,GAAG,OAAO,MAAM,EAAE;EACvD,IAAI,UAAU,IAAI;GAChB,KAAK,KAAK,KAAK;GACf;EACF;EACA,KAAK,SACH,MAAM,UAAU,UAAU,OACtB;GAAE,GAAG;GAAO,OAAO;IAAE,GAAG,KAAK,MAAM,CAAE;IAAO,GAAG,MAAM;GAAM;EAAE,IAC7D;CACR;CACA,OAAO;AACT;;;;;;AAOA,MAAa,mBACX,UAC+C;CAC/C,MAAM,OAAQ,OAAkD,QAAQ;CACxE,IAAI,WAAW,IAAI,GAAG,OAAO;CAC7B,MAAM,UAAW,MAA4C;CAC7D,IAAI,WAAW,OAAO,GAAG,OAAO;AAElC;AAOA,MAAa,0BAAsC;CACjD,QAAQ,CAAC;CACT,UAAU,CAAC;AACb;;;;;;;;;AAUA,MAAa,iBACX,QACA,OAAmB,iBAAiB,MACpB;CAChB,MAAM,WAAW,KAAK;CACtB,MAAM,WACJ,SAAS,WAAW,IAChB,IACA,OAAO,YAAY,SAAS,GAAG,EAAE,GAAG,SAAS,SAAS,CAAC,IAAI;CACjE,IAAI,MAAM,SAAS,SAAS,KAAK,aAAa,IAAI,CAAC,IAAI,KAAK;CAC5D,KAAK,IAAI,IAAI,UAAU,IAAI,OAAO,QAAQ,KAAK;EAC7C,MAAM,SAAS,gBAAgB,OAAO,EAAE;EACxC,IAAI,QAAQ,MAAM,cAAc,KAAK,MAAM;CAC7C;CACA,KAAK,SAAS;CACd,KAAK,WAAW;CAChB,OAAO;AACT;AAOA,MAAa,8BAA8C;CACzD,UAAU,KAAA;CACV,SAAS,CAAC;AACZ;;;;;;;;;AAUA,MAAa,uBACX,UACA,SACyB;CACzB,IAAI,aAAa,KAAK,UAAU,OAAO,KAAK;CAC5C,KAAK,WAAW;CAChB,IAAI,CAAC,MAAM,QAAQ,QAAQ,GAAG;EAC5B,IAAI,KAAK,QAAQ,SAAS,GAAG,KAAK,UAAU,CAAC;EAC7C,OAAO,KAAK;CACd;CACA,MAAM,WAAW,KAAK;CACtB,MAAM,eAAe,IAAI,IAAI,SAAS,KAAK,OAAO,CAAC,GAAG,IAAI,EAAE,CAAC,CAAC;CAC9D,IAAI,OAAO,SAAS,WAAW,SAAS;CACxC,MAAM,UAAW,SAAyB,KAAK,IAAI,UAAU;EAC3D,MAAM,SAAS,aAAa,IAAI,GAAG,EAAE;EACrC,MAAM,QACJ,WAAW,KAAA,KAAa,iBAAiB,QAAQ,EAAE,IAAI,SAAS;EAClE,IAAI,UAAU,SAAS,QAAQ,OAAO;EACtC,OAAO;CACT,CAAC;CACD,IAAI,CAAC,MAAM,KAAK,UAAU;CAC1B,OAAO,KAAK;AACd;;;;;;;;AASA,MAAa,mBACX,MACA,aACgB;CAChB,MAAM,uBAAO,IAAI,IAAuB;CACxC,KAAK,MAAM,MAAM,MAAM,KAAK,IAAI,GAAG,IAAI,EAAE;CACzC,IAAI,MAAM,QAAQ,QAAQ,GACxB,KAAK,MAAM,MAAM,UAAyB,KAAK,IAAI,GAAG,IAAI,EAAE;CAE9D,OAAO,CAAC,GAAG,KAAK,OAAO,CAAC;AAC1B"}

package/dist/useStreamRuntime.d.ts

@@ -1,12 +1,11 @@
-/// <reference types="@assistant-ui/core/store" />
-import { groupUIMessagesByParent } from "./converter.js";
-import { UseStreamRuntimeOptions } from "./types.js";
-import { AppendMessage } from "@assistant-ui/core";
-//#region src/useStreamRuntime.d.ts
-declare const runConfigToSubmitOptions: (runConfig: AppendMessage["runConfig"]) => {
-  config: {
-    configurable: Record<string, unknown>;
-  };
+/// <reference types="@assistant-ui/core/store" preserve="true" />
+import type { AppendMessage } from "@assistant-ui/core";
+import type { UseStreamRuntimeOptions } from "./types.js";
+export { groupUIMessagesByParent } from "./converter.js";
+export declare const runConfigToSubmitOptions: (runConfig: AppendMessage["runConfig"]) => {
+    config: {
+        configurable: Record<string, unknown>;
+    };
 } | undefined;
 /**
  * Creates an assistant-ui runtime backed by LangChain's `useStream` hook.
@@ -32,7 +31,5 @@ declare const runConfigToSubmitOptions: (runConfig: AppendMessage["runConfig"])
  * }
  * ```
  */
-declare const useStreamRuntime: (rawOptions: UseStreamRuntimeOptions) => import("@assistant-ui/core").AssistantRuntime;
-//#endregion
-export { groupUIMessagesByParent, runConfigToSubmitOptions, useStreamRuntime };
+export declare const useStreamRuntime: (rawOptions: UseStreamRuntimeOptions) => import("@assistant-ui/core").AssistantRuntime;
 //# sourceMappingURL=useStreamRuntime.d.ts.map

package/dist/useStreamRuntime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useStreamRuntime.d.ts","names":[],"sources":["../src/useStreamRuntime.ts"],"mappings":";;;;cA0Da,2BACX,WAAW;;kBAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;cAskBA,mBAAoB,YAAY,yDAAuB"}
+{"version":3,"file":"useStreamRuntime.d.ts","sourceRoot":"","sources":["../src/useStreamRuntime.ts"],"names":[],"mappings":";AAWA,OAAO,KAAK,EAAE,aAAa,EAAuB,MAAM,oBAAoB,CAAC;AAqB7E,OAAO,KAAK,EAIV,uBAAuB,EACxB,mBAAgB;AAEjB,OAAO,EAAE,uBAAuB,EAAE,uBAAoB;AAwBtD,eAAO,MAAM,wBAAwB,GACnC,WAAW,aAAa,CAAC,WAAW,CAAC;;;;aAIxB,CAAC;AA6rBhB;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,eAAO,MAAM,gBAAgB,GAAI,YAAY,uBAAuB,kDA8BnE,CAAC"}