@assistant-ui/react-langchain 0.0.15 → 0.0.17

package/dist/streamingTiming.js

@@ -0,0 +1,53 @@
+"use client";
+import { getMessageType } from "./convertMessages.js";
+import { useStreamingTiming } from "@assistant-ui/core/react";
+//#region src/streamingTiming.ts
+const findAiMessage = (messages, messageId) => messages.find((m) => getMessageType(m) === "ai" && m.id === messageId);
+const reasoningTextLength = (part) => {
+	if (part.summary && part.summary.length > 0) return part.summary.map((s) => s?.text ?? "").join("\n\n\n").length;
+	return part.reasoning?.length ?? 0;
+};
+const getTextLength = (messages, messageId) => {
+	const m = findAiMessage(messages, messageId);
+	if (!m) return 0;
+	const content = m.content;
+	if (typeof content === "string") return content.length;
+	if (!Array.isArray(content)) return 0;
+	let len = 0;
+	for (const part of content) switch (part.type) {
+		case "text":
+		case "text_delta":
+			if (typeof part.text === "string") len += part.text.length;
+			break;
+		case "thinking":
+			if (typeof part.thinking === "string") len += part.thinking.length;
+			break;
+		case "reasoning":
+			len += reasoningTextLength(part);
+			break;
+	}
+	return len;
+};
+const getToolCallCount = (messages, messageId) => findAiMessage(messages, messageId)?.tool_calls?.length ?? 0;
+const getAssistantMessageId = (messages) => {
+	for (let i = messages.length - 1; i >= 0; i--) {
+		const m = messages[i];
+		if (m && getMessageType(m) === "ai" && m.id) return m.id;
+	}
+};
+const langChainStreamingTimingAccessors = {
+	getAssistantMessageId,
+	getTextLength,
+	getToolCallCount
+};
+/**
+* Tracks per-message streaming timing for LangChain messages. Delegates to
+* the shared `useStreamingTiming` primitive in `@assistant-ui/core/react`,
+* adapted to the `LangChainBaseMessage` shape (`_getType() -> "ai"`, content
+* blocks including text/thinking/reasoning, `tool_calls`).
+*/
+const useLangChainStreamingTiming = (messages, isRunning) => useStreamingTiming(messages, isRunning, langChainStreamingTimingAccessors);
+//#endregion
+export { langChainStreamingTimingAccessors, useLangChainStreamingTiming };
+
+//# sourceMappingURL=streamingTiming.js.map

package/dist/streamingTiming.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streamingTiming.js","names":[],"sources":["../src/streamingTiming.ts"],"sourcesContent":["\"use client\";\n\nimport type { MessageTiming } from \"@assistant-ui/core\";\nimport {\n  useStreamingTiming,\n  type StreamingTimingAccessors,\n} from \"@assistant-ui/core/react\";\nimport type { LangChainBaseMessage, LangChainContentBlock } from \"./types\";\nimport { getMessageType } from \"./convertMessages\";\n\nconst findAiMessage = (\n  messages: readonly LangChainBaseMessage[],\n  messageId: string,\n): LangChainBaseMessage | undefined =>\n  messages.find((m) => getMessageType(m) === \"ai\" && m.id === messageId);\n\nconst reasoningTextLength = (part: {\n  readonly summary?: ReadonlyArray<{ readonly text?: string }>;\n  readonly reasoning?: string;\n}): number => {\n  if (part.summary && part.summary.length > 0)\n    return part.summary.map((s) => s?.text ?? \"\").join(\"\\n\\n\\n\").length;\n  return part.reasoning?.length ?? 0;\n};\n\nconst getTextLength = (\n  messages: readonly LangChainBaseMessage[],\n  messageId: string,\n): number => {\n  const m = findAiMessage(messages, messageId);\n  if (!m) return 0;\n  const content = m.content;\n  if (typeof content === \"string\") return content.length;\n  if (!Array.isArray(content)) return 0;\n  let len = 0;\n  for (const part of content as readonly LangChainContentBlock[]) {\n    switch (part.type) {\n      case \"text\":\n      case \"text_delta\":\n        if (typeof part.text === \"string\") len += part.text.length;\n        break;\n      case \"thinking\":\n        if (typeof part.thinking === \"string\") len += part.thinking.length;\n        break;\n      case \"reasoning\":\n        len += reasoningTextLength(part);\n        break;\n    }\n  }\n  return len;\n};\n\nconst getToolCallCount = (\n  messages: readonly LangChainBaseMessage[],\n  messageId: string,\n): number => findAiMessage(messages, messageId)?.tool_calls?.length ?? 0;\n\nconst getAssistantMessageId = (\n  messages: readonly LangChainBaseMessage[],\n): string | undefined => {\n  for (let i = messages.length - 1; i >= 0; i--) {\n    const m = messages[i];\n    if (m && getMessageType(m) === \"ai\" && m.id) return m.id;\n  }\n  return undefined;\n};\n\nexport const langChainStreamingTimingAccessors: StreamingTimingAccessors<LangChainBaseMessage> =\n  {\n    getAssistantMessageId,\n    getTextLength,\n    getToolCallCount,\n  };\n\n/**\n * Tracks per-message streaming timing for LangChain messages. Delegates to\n * the shared `useStreamingTiming` primitive in `@assistant-ui/core/react`,\n * adapted to the `LangChainBaseMessage` shape (`_getType() -> \"ai\"`, content\n * blocks including text/thinking/reasoning, `tool_calls`).\n */\nexport const useLangChainStreamingTiming = (\n  messages: readonly LangChainBaseMessage[],\n  isRunning: boolean,\n): Record<string, MessageTiming> =>\n  useStreamingTiming(messages, isRunning, langChainStreamingTimingAccessors);\n"],"mappings":";;;;AAUA,MAAM,iBACJ,UACA,cAEA,SAAS,MAAM,MAAM,eAAe,CAAC,MAAM,QAAQ,EAAE,OAAO,SAAS;AAEvE,MAAM,uBAAuB,SAGf;CACZ,IAAI,KAAK,WAAW,KAAK,QAAQ,SAAS,GACxC,OAAO,KAAK,QAAQ,KAAK,MAAM,GAAG,QAAQ,EAAE,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC;CAC/D,OAAO,KAAK,WAAW,UAAU;AACnC;AAEA,MAAM,iBACJ,UACA,cACW;CACX,MAAM,IAAI,cAAc,UAAU,SAAS;CAC3C,IAAI,CAAC,GAAG,OAAO;CACf,MAAM,UAAU,EAAE;CAClB,IAAI,OAAO,YAAY,UAAU,OAAO,QAAQ;CAChD,IAAI,CAAC,MAAM,QAAQ,OAAO,GAAG,OAAO;CACpC,IAAI,MAAM;CACV,KAAK,MAAM,QAAQ,SACjB,QAAQ,KAAK,MAAb;EACE,KAAK;EACL,KAAK;GACH,IAAI,OAAO,KAAK,SAAS,UAAU,OAAO,KAAK,KAAK;GACpD;EACF,KAAK;GACH,IAAI,OAAO,KAAK,aAAa,UAAU,OAAO,KAAK,SAAS;GAC5D;EACF,KAAK;GACH,OAAO,oBAAoB,IAAI;GAC/B;CACJ;CAEF,OAAO;AACT;AAEA,MAAM,oBACJ,UACA,cACW,cAAc,UAAU,SAAS,CAAC,EAAE,YAAY,UAAU;AAEvE,MAAM,yBACJ,aACuB;CACvB,KAAK,IAAI,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,KAAK;EAC7C,MAAM,IAAI,SAAS;EACnB,IAAI,KAAK,eAAe,CAAC,MAAM,QAAQ,EAAE,IAAI,OAAO,EAAE;CACxD;AAEF;AAEA,MAAa,oCACX;CACE;CACA;CACA;AACF;;;;;;;AAQF,MAAa,+BACX,UACA,cAEA,mBAAmB,UAAU,WAAW,iCAAiC"}

package/dist/types.d.ts

@@ -1,4 +1,6 @@
-import { MessageStatus } from "@assistant-ui/core";
+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
 /** Known content block types from @langchain/core messages. */
@@ -11,17 +13,18 @@ type LangChainContentBlock = {
 } | {
   type: "image_url";
   image_url: string | {
-    url: string;
+    url?: string;
   };
 } | {
   type: "thinking";
   thinking: string;
 } | {
   type: "reasoning";
-  summary: Array<{
+  summary?: Array<{
     type: "summary_text";
-    text: string;
+    text?: string;
   }>;
+  reasoning?: string;
 } | {
   type: "file";
   data: string;
@@ -38,6 +41,36 @@ type LangChainToolCall = {
   name: string;
   args: Record<string, unknown>;
 };
+/**
+ * A generative UI component the graph accumulates in its state. Read from
+ * `stream.values[uiStateKey]` and rendered via `makeAssistantDataUI`.
+ *
+ * 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;
+  };
+};
+/**
+ * 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;
+};
 /**
  * Minimal duck-typed interface for BaseMessage class instances returned by
  * `useStream`. Used internally by the message converter.
@@ -56,6 +89,73 @@ type LangChainBaseMessage = {
   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;
+};
+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, LangChainContentBlock, LangChainToolCall };
+export { LangChainBaseMessage, LangChainContentBlock, LangChainRuntimeExtraOptions, LangChainRuntimeExtras, LangChainToolCall, RemoveUIMessage, UIMessage, UseStreamRuntimeOptions };
 //# 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":";;;;KAGY,qBAAA;EACN,IAAA;EAAc,IAAA;AAAA;EACd,IAAA;EAAoB,IAAA;AAAA;EACpB,IAAA;EAAmB,SAAA;IAAsB,GAAA;EAAA;AAAA;EACzC,IAAA;EAAkB,QAAA;AAAA;EAElB,IAAA;EACA,OAAA,EAAS,KAAK;IAAG,IAAA;IAAsB,IAAA;EAAA;AAAA;EAGvC,IAAA;EACA,IAAA;EACA,SAAA;EACA,WAAA;EACA,QAAA;IAAa,QAAA;EAAA;AAAA;EAEb,IAAA;AAAA;AAAA,KAEM,iBAAA;EACV,EAAA;EACA,IAAA;EACA,IAAA,EAAM,MAAM;AAAA;;AAAA;AAUd;;;;;KAAY,oBAAA;EACV,QAAA;EACA,OAAA;EACA,EAAA;EACA,IAAA;EACA,iBAAA,GAAoB,MAAA,+BADpB;EAGA,UAAA,YAAsB,iBAAA,gBAFF;EAIpB,YAAA,uBAFsB;EAItB,MAAA,GAAS,aAAA,oCAAT;EAEA,QAAA;AAAA"}
+{"version":3,"file":"types.d.ts","names":[],"sources":["../src/types.ts"],"mappings":";;;;;;KAoBY,qBAAA;EACN,IAAA;EAAc,IAAA;AAAA;EACd,IAAA;EAAoB,IAAA;AAAA;EACpB,IAAA;EAAmB,SAAA;IAAsB,GAAA;EAAA;AAAA;EACzC,IAAA;EAAkB,QAAA;AAAA;EAElB,IAAA;EACA,OAAA,GAAU,KAAK;IAAG,IAAA;IAAsB,IAAA;EAAA;EACxC,SAAA;AAAA;EAGA,IAAA;EACA,IAAA;EACA,SAAA;EACA,WAAA;EACA,QAAA;IAAa,QAAA;EAAA;AAAA;EAEb,IAAA;AAAA;AAAA,KAEM,iBAAA;EACV,EAAA;EACA,IAAA;EACA,IAAA,EAAM,MAAM;AAAA;;AAAA;AAUd;;;;;KAAY,SAAA,+CAEK,MAAA,oBAA0B,MAAA;EAEzC,IAAA;EACA,EAAA;EACA,IAAA,EAAM,KAAA;EACN,KAAA,EAAO,MAAA;EACP,QAAA;IACE,KAAA;IACA,MAAA;IACA,IAAA;IACA,IAAA;IACA,UAAA;IACA,EAAA;IAAA,CACC,GAAA;EAAA;AAAA;;;;;KAQO,eAAA;EACV,IAAA;EACA,EAAE;AAAA;AAVY;AAQhB;;;;AAEI;AAUJ;AApBgB,KAoBJ,oBAAA;EACV,QAAA;EACA,OAAA;EACA,EAAA;EACA,IAAA;EACA,iBAAA,GAAoB,MAAA,+BAME;EAJtB,UAAA,YAAsB,iBAAA,gBALtB;EAOA,YAAA,uBALA;EAOA,MAAA,GAAS,aAAA,oCANW;EAQpB,QAAA;AAAA;AAAA,KAGU,sBAAA;EACV,SAAA;IAAa,KAAA;EAAA;EACb,UAAA;IAAuB,EAAA;IAAa,KAAA;EAAA;EACpC,SAAA,WAAoB,iBAAA;EACpB,SAAA,EAAW,WAAA,SAAoB,yBAAA;EAC/B,SAAA,EAAW,WAAA,SAAoB,yBAAA;EAC/B,MAAA,EAAQ,SAAA;EACR,KAAA;EACA,MAAA,GACE,MAAA,EAAQ,MAAA,sCACR,OAAA,GAAU,MAAA,sBACP,OAAA;EACL,OAAA,GACE,QAAA,WACA,OAAA,GAAU,MAAA,sBACP,OAAA;EACL,UAAA,GACE,aAAA,EAAe,MAAA,mBACf,OAAA,GAAU,MAAA,sBACP,OAAA;EACL,MAAA,EAAQ,MAAA;EACR,WAAA;AAAA;AAAA,KAGU,4BAAA,GAA+B,0BAAA;EAPxB;;;;;;;;EAgBjB,gBAAA,KAAqB,QAAA;EACrB,KAAA,GAAQ,cAAA;EACR,QAAA;IAEM,WAAA,GAAc,iBAAA;IACd,MAAA,GAAS,sBAAA;IACT,SAAA,GAAY,gBAAA;IACZ,KAAA,GAAQ,oBAAA;IACR,QAAA,GAAW,eAAA;EAAA;EApCN;;;;;;EA6CX,0BAAA;EAzCE;;;;EA8CF,0BAAA;EA1CE;;;;EA+CF,0BAAA,GAA6B,uBAAA,cA3CZ;EA6CjB,MAAA,UAAgB,OAAA;IAAU,UAAA;EAAA,iBA3CrB;EA6CL,MAAA,KAAW,QAAA,aAAqB,OAAA;EA5CxB;;;AACG;AAGb;;EA+CE,UAAA;AAAA;AAAA,KAOU,uBAAA,GAA0B,gBAAA,mBAClC,CAAA,SAAU,gBAAA,GACR,CAAA,GAAI,4BAAA"}

package/dist/uiMessages.d.ts

@@ -0,0 +1,24 @@
+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[];
+/**
+ * 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[];
+/**
+ * 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
+ * copy. A consequence is that a live `remove-ui` is overridden while the
+ * 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 };
+//# sourceMappingURL=uiMessages.d.ts.map

package/dist/uiMessages.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"uiMessages.d.ts","names":[],"sources":["../src/uiMessages.ts"],"mappings":";;;KAEY,QAAA,GAAW,SAAA,GAAY,eAAe;AAAA,cAErC,UAAA,GACX,KAAA,cACC,KAAA,IAAS,QAAA,YAAoB,QAAQ;AAAA,cAmB3B,aAAA,GACX,IAAA,WAAe,SAAA,IACf,MAAA,EAAQ,QAAA,YAAoB,QAAA,OAC3B,SAAA;;;;AA1B+C;AAElD;cAkDa,eAAA,GACX,KAAA,cACC,QAAA,YAAoB,QAAQ;AAAA,cAQlB,aAAA,GAAiB,MAAA,yBAA6B,SAAS;;;;;;;AA1D5B;cA0E3B,eAAA,GACX,IAAA,WAAe,SAAA,IACf,QAAA,cACC,SAAS"}

package/dist/uiMessages.js

@@ -0,0 +1,68 @@
+//#region src/uiMessages.ts
+const isUIUpdate = (value) => {
+	if (Array.isArray(value)) return value.length > 0 && value.every(isUIUpdate);
+	if (value == null || typeof value !== "object") return false;
+	const v = value;
+	if (typeof v.id !== "string") return false;
+	if (v.type === "remove-ui") return true;
+	return v.type === "ui" && typeof v.name === "string" && typeof v.props === "object" && v.props !== null;
+};
+const applyUIUpdate = (list, update) => {
+	const events = Array.isArray(update) ? update : [update];
+	let next = list.slice();
+	for (const event of events) {
+		if (event.type === "remove-ui") {
+			next = next.filter((ui) => ui.id !== event.id);
+			continue;
+		}
+		const index = next.findIndex((ui) => ui.id === event.id);
+		if (index === -1) {
+			next.push(event);
+			continue;
+		}
+		next[index] = event.metadata?.merge === true ? {
+			...event,
+			props: {
+				...next[index].props,
+				...event.props
+			}
+		} : event;
+	}
+	return next;
+};
+/**
+* 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`.
+*/
+const extractUIUpdate = (event) => {
+	const data = event?.params?.data;
+	if (isUIUpdate(data)) return data;
+	const payload = data?.payload;
+	if (isUIUpdate(payload)) return payload;
+};
+const foldUIUpdates = (events) => {
+	let acc = [];
+	for (const event of events) {
+		const update = extractUIUpdate(event);
+		if (update) acc = applyUIUpdate(acc, update);
+	}
+	return acc;
+};
+/**
+* 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
+* copy. A consequence is that a live `remove-ui` is overridden while the
+* snapshot still contains that id; the removal only takes visible effect once
+* the snapshot catches up.
+*/
+const mergeUIMessages = (live, snapshot) => {
+	const byId = /* @__PURE__ */ new Map();
+	for (const ui of live) byId.set(ui.id, ui);
+	if (Array.isArray(snapshot)) for (const ui of snapshot) byId.set(ui.id, ui);
+	return [...byId.values()];
+};
+//#endregion
+export { applyUIUpdate, extractUIUpdate, foldUIUpdates, isUIUpdate, mergeUIMessages };
+
+//# sourceMappingURL=uiMessages.js.map

package/dist/uiMessages.js.map

@@ -0,0 +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"}

package/dist/useStreamRuntime.d.ts

@@ -1,47 +1,20 @@
 /// <reference types="@assistant-ui/core/store" />
-import { LangChainBaseMessage } from "./types.js";
-import { AppendMessage, AttachmentAdapter, DictationAdapter, ExternalStoreSharedOptions, FeedbackAdapter, RealtimeVoiceAdapter, RemoteThreadListAdapter, SpeechSynthesisAdapter } from "@assistant-ui/core";
-import { AssembledToolCall, UseStreamOptions } from "@langchain/react";
-import { AssistantCloud } from "assistant-cloud";
+import { UIMessage, UseStreamRuntimeOptions } from "./types.js";
+import { AppendMessage } from "@assistant-ui/core";
 
 //#region src/useStreamRuntime.d.ts
-type LangChainRuntimeExtraOptions = ExternalStoreSharedOptions & {
-  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;
-};
 declare const runConfigToSubmitOptions: (runConfig: AppendMessage["runConfig"]) => {
   config: {
     configurable: Record<string, unknown>;
   };
 } | undefined;
-type UseStreamRuntimeOptions = UseStreamOptions extends infer O ? O extends UseStreamOptions ? O & LangChainRuntimeExtraOptions : never : never;
+/**
+ * Group the graph's accumulated `UIMessage`s by the assistant message they
+ * belong to. Non-array state and entries without a parent link are dropped.
+ * The parent id comes from `metadata.message_id` (Python SDK) or
+ * `metadata.id` (JS SDK).
+ */
+declare const groupUIMessagesByParent: (value: unknown) => Map<string, UIMessage[]>;
 /**
  * Creates an assistant-ui runtime backed by LangChain's `useStream` hook.
  * Accepts the same options as `useStream` from `@langchain/react`, plus
@@ -67,54 +40,6 @@ type UseStreamRuntimeOptions = UseStreamOptions extends infer O ? O extends UseS
  * ```
  */
 declare const useStreamRuntime: (rawOptions: UseStreamRuntimeOptions) => import("@assistant-ui/core").AssistantRuntime;
-/**
- * Read the current LangGraph interrupt state from the runtime extras.
- */
-declare const useLangChainInterruptState: () => {
-  value?: unknown;
-} | undefined;
-/** Read the last run/hydration error from the runtime extras. */
-declare const useLangChainError: () => unknown;
-/**
- * Read the root tool calls assembled by `useStream` from the `tools`
- * channel. Defaults to an empty array, so consumers can `.map` without
- * a guard. Useful for rendering pending/streamed tool calls and
- * approval UIs.
- */
-declare const useLangChainToolCalls: () => readonly AssembledToolCall<string, unknown, unknown>[];
-/**
- * Returns a function to submit raw state updates to the LangGraph agent,
- * bypassing the normal message flow. Useful for sending interrupt resume
- * commands.
- */
-declare const useLangChainSubmit: () => (values: Record<string, unknown> | null | undefined, options?: Record<string, unknown>) => Promise<void>;
-/**
- * Submit a list of LangChain-shaped messages on the current thread.
- * Parity helper for migrating from `useLangGraphSend`. Routes to
- * `useStream().submit({ [messagesKey]: messages }, options)`.
- */
-declare const useLangChainSend: () => (messages: readonly LangChainBaseMessage[], options?: Record<string, unknown>) => Promise<void>;
-/**
- * Submit a `useStream` command (e.g. interrupt resume). Parity helper
- * for migrating from `useLangGraphSendCommand`. Note that v1's command
- * shape (`{ resume?, goto?, update? }`) differs from the legacy
- * `{ resume: string }` form — to carry a payload, use the input or
- * `stream.respond` instead.
- */
-declare const useLangChainSendCommand: () => (command: Record<string, unknown>) => Promise<void>;
-/**
- * Read a custom LangGraph state key from the current thread. Mirrors
- * `useStream().values[key]` from `@langchain/react` and updates when the
- * stream emits new state.
- *
- * @example
- * ```tsx
- * const todos = useLangChainState<Todo[]>("todos");
- * const files = useLangChainState<Record<string, string>>("files", {});
- * ```
- */
-declare function useLangChainState<T>(key: string): T | undefined;
-declare function useLangChainState<T>(key: string, defaultValue: T): T;
 //#endregion
-export { UseStreamRuntimeOptions, runConfigToSubmitOptions, useLangChainError, useLangChainInterruptState, useLangChainSend, useLangChainSendCommand, useLangChainState, useLangChainSubmit, useLangChainToolCalls, useStreamRuntime };
+export { groupUIMessagesByParent, runConfigToSubmitOptions, useStreamRuntime };
 //# 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":";;;;;;KA8DK,4BAAA,GAA+B,0BAAA;EAClC,KAAA,GAAQ,cAAA;EACR,QAAA;IAEM,WAAA,GAAc,iBAAA;IACd,MAAA,GAAS,sBAAA;IACT,SAAA,GAAY,gBAAA;IACZ,KAAA,GAAQ,oBAAA;IACR,QAAA,GAAW,eAAA;EAAA;EAFC;;;;;;EAWlB,0BAAA;EAcuC;;;;EATvC,0BAAA;EAlBM;;;;EAuBN,0BAAA,GAA6B,uBAAA,cArBX;EAuBlB,MAAA,UAAgB,OAAA;IAAU,UAAA;EAAA,iBArBT;EAuBjB,MAAA,KAAW,QAAA,aAAqB,OAAA;AAAA;AAAA,cAGrB,wBAAA,GACX,SAAA,EAAW,aAAA;;kBAAA,MAAA;EAAA;AAAA;AAAA,KAyBD,uBAAA,GAA0B,gBAAA,mBAClC,CAAA,SAAU,gBAAA,GACR,CAAA,GAAI,4BAAA;;;;AA/B+B;AAGzC;;;;;;;;;;;AA0BA;;;;;;;;;cA2Ma,gBAAA,GAAoB,UAAA,EAAY,uBAAuB,kCAAA,gBAAA;;;;cA+BvD,0BAAA;EAMZ,KAAA;AAAA;AA9OqC;AAAA,cAiPzB,iBAAA;;;;;;;cAcA,qBAAA,iBAAqB,iBAAA;AA5BjC;AAKD;;;;AALC,cAyCY,kBAAA,SAGT,MAAA,EAAQ,MAAA,sCACR,OAAA,GAAU,MAAA,sBAAuB,OAAA;AA/BrC;;;;AAMC;AAND,cA4Ca,gBAAA,SAGT,QAAA,WAAmB,oBAAA,IACnB,OAAA,GAAU,MAAA,sBAAuB,OAAA;;;;AAlCH;AAalC;;;cAqCa,uBAAA,SAEH,OAAA,EAAS,MAAA,sBAAuB,OAAA;;;;;;;;;;;AAnCL;iBAiDrB,iBAAA,IAAqB,GAAA,WAAc,CAAC;AAAA,iBACpC,iBAAA,IAAqB,GAAA,UAAa,YAAA,EAAc,CAAA,GAAI,CAAC"}
+{"version":3,"file":"useStreamRuntime.d.ts","names":[],"sources":["../src/useStreamRuntime.ts"],"mappings":";;;;cAsCa,wBAAA,GACX,SAAA,EAAW,aAAA;;kBAAA,MAAA;EAAA;AAAA;;;;;;;cAYA,uBAAA,GACX,KAAA,cACC,GAAG,SAAS,SAAA;;AAFf;;;;;;;;AAEwB;AAyZxB;;;;;;;;AA4BC;;;;;;cA5BY,gBAAA,GAAoB,UAAA,EAAY,uBAAuB,kCAAA,gBAAA"}