@langchain/react 0.3.5 → 0.4.0-0

package/dist/react-ui/server/server.d.cts

@@ -0,0 +1,54 @@
+import { RemoveUIMessage, UIMessage } from "../types.cjs";
+import { ComponentPropsWithoutRef, ElementType } from "react";
+
+//#region src/react-ui/server/server.d.ts
+interface MessageLike {
+  id?: string;
+}
+/**
+ * Helper to send and persist UI messages. Accepts a map of component names to React components
+ * as type argument to provide type safety. Will also write to the `options?.stateKey` state.
+ *
+ * @param config LangGraphRunnableConfig
+ * @param options
+ * @returns
+ */
+declare const typedUi: <Decl extends Record<string, ElementType>>(config: {
+  writer?: ((chunk: unknown) => void) | undefined;
+  runId?: string | undefined;
+  metadata?: Record<string, unknown> | undefined;
+  tags?: string[] | undefined;
+  runName?: string | undefined;
+  configurable?: {
+    [key: string]: unknown;
+    __pregel_send?: ((writes_: [string, unknown][]) => void) | undefined;
+  } | undefined;
+}, options?: {
+  /** The key to write the UI messages to. Defaults to `ui`. */stateKey?: string | undefined;
+} | undefined) => {
+  push: {
+    <K extends keyof Decl & string>(message: {
+      id?: string | undefined;
+      name: K;
+      props: { [K_1 in keyof Decl]: ComponentPropsWithoutRef<Decl[K_1]> }[K];
+      metadata?: Record<string, unknown> | undefined;
+    }, options?: {
+      message?: MessageLike | undefined;
+      merge?: boolean | undefined;
+    } | undefined): UIMessage<K, { [K_1 in keyof Decl]: ComponentPropsWithoutRef<Decl[K_1]> }[K]>;
+    <K extends keyof Decl & string>(message: {
+      id?: string | undefined;
+      name: K;
+      props: Partial<{ [K_1 in keyof Decl]: ComponentPropsWithoutRef<Decl[K_1]> }[K]>;
+      metadata?: Record<string, unknown> | undefined;
+    }, options: {
+      message?: MessageLike | undefined;
+      merge: true;
+    }): UIMessage<K, Partial<{ [K_1 in keyof Decl]: ComponentPropsWithoutRef<Decl[K_1]> }[K]>>;
+  };
+  delete: (id: string) => RemoveUIMessage;
+  items: (RemoveUIMessage | UIMessage<string, Record<string, unknown>>)[];
+};
+//#endregion
+export { typedUi };
+//# sourceMappingURL=server.d.cts.map

package/dist/react-ui/server/server.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.d.cts","names":[],"sources":["../../../src/react-ui/server/server.ts"],"mappings":";;;;UAIU,WAAA;EACR,EAAA;AAAA;;;;;AAWF;;;;cAAa,OAAA,gBAAe,MAAA,SAAA,WAAA,GAAA,MAAA;;;;;;;;;;;;;;qBAwBL,IAAA,WAAA,OAAA;;;;;;;;;qBAUA,IAAA,WAAA,OAAA"}

package/dist/react-ui/server/server.d.ts

@@ -0,0 +1,54 @@
+import { RemoveUIMessage, UIMessage } from "../types.js";
+import { ComponentPropsWithoutRef, ElementType } from "react";
+
+//#region src/react-ui/server/server.d.ts
+interface MessageLike {
+  id?: string;
+}
+/**
+ * Helper to send and persist UI messages. Accepts a map of component names to React components
+ * as type argument to provide type safety. Will also write to the `options?.stateKey` state.
+ *
+ * @param config LangGraphRunnableConfig
+ * @param options
+ * @returns
+ */
+declare const typedUi: <Decl extends Record<string, ElementType>>(config: {
+  writer?: ((chunk: unknown) => void) | undefined;
+  runId?: string | undefined;
+  metadata?: Record<string, unknown> | undefined;
+  tags?: string[] | undefined;
+  runName?: string | undefined;
+  configurable?: {
+    [key: string]: unknown;
+    __pregel_send?: ((writes_: [string, unknown][]) => void) | undefined;
+  } | undefined;
+}, options?: {
+  /** The key to write the UI messages to. Defaults to `ui`. */stateKey?: string | undefined;
+} | undefined) => {
+  push: {
+    <K extends keyof Decl & string>(message: {
+      id?: string | undefined;
+      name: K;
+      props: { [K_1 in keyof Decl]: ComponentPropsWithoutRef<Decl[K_1]> }[K];
+      metadata?: Record<string, unknown> | undefined;
+    }, options?: {
+      message?: MessageLike | undefined;
+      merge?: boolean | undefined;
+    } | undefined): UIMessage<K, { [K_1 in keyof Decl]: ComponentPropsWithoutRef<Decl[K_1]> }[K]>;
+    <K extends keyof Decl & string>(message: {
+      id?: string | undefined;
+      name: K;
+      props: Partial<{ [K_1 in keyof Decl]: ComponentPropsWithoutRef<Decl[K_1]> }[K]>;
+      metadata?: Record<string, unknown> | undefined;
+    }, options: {
+      message?: MessageLike | undefined;
+      merge: true;
+    }): UIMessage<K, Partial<{ [K_1 in keyof Decl]: ComponentPropsWithoutRef<Decl[K_1]> }[K]>>;
+  };
+  delete: (id: string) => RemoveUIMessage;
+  items: (RemoveUIMessage | UIMessage<string, Record<string, unknown>>)[];
+};
+//#endregion
+export { typedUi };
+//# sourceMappingURL=server.d.ts.map

package/dist/react-ui/server/server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.d.ts","names":[],"sources":["../../../src/react-ui/server/server.ts"],"mappings":";;;;UAIU,WAAA;EACR,EAAA;AAAA;;;;;AAWF;;;;cAAa,OAAA,gBAAe,MAAA,SAAA,WAAA,GAAA,MAAA;;;;;;;;;;;;;;qBAwBL,IAAA,WAAA,OAAA;;;;;;;;;qBAUA,IAAA,WAAA,OAAA"}

package/dist/react-ui/server/server.js

@@ -0,0 +1,55 @@
+import { v4 } from "uuid";
+//#region src/react-ui/server/server.ts
+/**
+* Helper to send and persist UI messages. Accepts a map of component names to React components
+* as type argument to provide type safety. Will also write to the `options?.stateKey` state.
+*
+* @param config LangGraphRunnableConfig
+* @param options
+* @returns
+*/
+const typedUi = (config, options) => {
+	const items = [];
+	const stateKey = options?.stateKey ?? "ui";
+	const runId = config.metadata?.run_id ?? config.runId;
+	if (!runId) throw new Error("run_id is required");
+	function handlePush(message, options) {
+		const evt = {
+			type: "ui",
+			id: message?.id ?? v4(),
+			name: message?.name,
+			props: message?.props,
+			metadata: {
+				merge: options?.merge || void 0,
+				run_id: runId,
+				tags: config.tags,
+				name: config.runName,
+				...message?.metadata,
+				...options?.message ? { message_id: options.message.id } : null
+			}
+		};
+		items.push(evt);
+		config.writer?.(evt);
+		config.configurable?.__pregel_send?.([[stateKey, evt]]);
+		return evt;
+	}
+	const handleDelete = (id) => {
+		const evt = {
+			type: "remove-ui",
+			id
+		};
+		items.push(evt);
+		config.writer?.(evt);
+		config.configurable?.__pregel_send?.([[stateKey, evt]]);
+		return evt;
+	};
+	return {
+		push: handlePush,
+		delete: handleDelete,
+		items
+	};
+};
+//#endregion
+export { typedUi };
+
+//# sourceMappingURL=server.js.map

package/dist/react-ui/server/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","names":["uuidv4"],"sources":["../../../src/react-ui/server/server.ts"],"sourcesContent":["import { v4 as uuidv4 } from \"uuid\";\nimport type { ComponentPropsWithoutRef, ElementType } from \"react\";\nimport type { RemoveUIMessage, UIMessage } from \"../types.js\";\n\ninterface MessageLike {\n  id?: string;\n}\n\n/**\n * Helper to send and persist UI messages. Accepts a map of component names to React components\n * as type argument to provide type safety. Will also write to the `options?.stateKey` state.\n *\n * @param config LangGraphRunnableConfig\n * @param options\n * @returns\n */\nexport const typedUi = <Decl extends Record<string, ElementType>>(\n  config: {\n    writer?: (chunk: unknown) => void;\n    runId?: string;\n    metadata?: Record<string, unknown>;\n    tags?: string[];\n    runName?: string;\n    configurable?: {\n      __pregel_send?: (writes_: [string, unknown][]) => void;\n      [key: string]: unknown;\n    };\n  },\n  options?: {\n    /** The key to write the UI messages to. Defaults to `ui`. */\n    stateKey?: string;\n  },\n) => {\n  type PropMap = { [K in keyof Decl]: ComponentPropsWithoutRef<Decl[K]> };\n  const items: (UIMessage | RemoveUIMessage)[] = [];\n  const stateKey = options?.stateKey ?? \"ui\";\n\n  const runId = (config.metadata?.run_id as string | undefined) ?? config.runId;\n  if (!runId) throw new Error(\"run_id is required\");\n\n  function handlePush<K extends keyof PropMap & string>(\n    message: {\n      id?: string;\n      name: K;\n      props: PropMap[K];\n      metadata?: Record<string, unknown>;\n    },\n    options?: { message?: MessageLike; merge?: boolean },\n  ): UIMessage<K, PropMap[K]>;\n\n  function handlePush<K extends keyof PropMap & string>(\n    message: {\n      id?: string;\n      name: K;\n      props: Partial<PropMap[K]>;\n      metadata?: Record<string, unknown>;\n    },\n    options: { message?: MessageLike; merge: true },\n  ): UIMessage<K, Partial<PropMap[K]>>;\n\n  function handlePush<K extends keyof PropMap & string>(\n    message: {\n      id?: string;\n      name: K;\n      props: PropMap[K] | Partial<PropMap[K]>;\n      metadata?: Record<string, unknown>;\n    },\n    options?: { message?: MessageLike; merge?: boolean },\n  ): UIMessage<K, PropMap[K] | Partial<PropMap[K]>> {\n    const evt: UIMessage<K, PropMap[K] | Partial<PropMap[K]>> = {\n      type: \"ui\" as const,\n      id: message?.id ?? uuidv4(),\n      name: message?.name,\n      props: message?.props,\n      metadata: {\n        merge: options?.merge || undefined,\n        run_id: runId,\n        tags: config.tags,\n        name: config.runName,\n        ...message?.metadata,\n        ...(options?.message ? { message_id: options.message.id } : null),\n      },\n    };\n    items.push(evt);\n    config.writer?.(evt);\n    config.configurable?.__pregel_send?.([[stateKey, evt]]);\n    return evt;\n  }\n\n  const handleDelete = (id: string): RemoveUIMessage => {\n    const evt: RemoveUIMessage = { type: \"remove-ui\", id };\n    items.push(evt);\n    config.writer?.(evt);\n    config.configurable?.__pregel_send?.([[stateKey, evt]]);\n    return evt;\n  };\n\n  return { push: handlePush, delete: handleDelete, items };\n};\n"],"mappings":";;;;;;;;;;AAgBA,MAAa,WACX,QAWA,YAIG;CAEH,MAAM,QAAyC,EAAE;CACjD,MAAM,WAAW,SAAS,YAAY;CAEtC,MAAM,QAAS,OAAO,UAAU,UAAiC,OAAO;AACxE,KAAI,CAAC,MAAO,OAAM,IAAI,MAAM,qBAAqB;CAsBjD,SAAS,WACP,SAMA,SACgD;EAChD,MAAM,MAAsD;GAC1D,MAAM;GACN,IAAI,SAAS,MAAMA,IAAQ;GAC3B,MAAM,SAAS;GACf,OAAO,SAAS;GAChB,UAAU;IACR,OAAO,SAAS,SAAS,KAAA;IACzB,QAAQ;IACR,MAAM,OAAO;IACb,MAAM,OAAO;IACb,GAAG,SAAS;IACZ,GAAI,SAAS,UAAU,EAAE,YAAY,QAAQ,QAAQ,IAAI,GAAG;IAC7D;GACF;AACD,QAAM,KAAK,IAAI;AACf,SAAO,SAAS,IAAI;AACpB,SAAO,cAAc,gBAAgB,CAAC,CAAC,UAAU,IAAI,CAAC,CAAC;AACvD,SAAO;;CAGT,MAAM,gBAAgB,OAAgC;EACpD,MAAM,MAAuB;GAAE,MAAM;GAAa;GAAI;AACtD,QAAM,KAAK,IAAI;AACf,SAAO,SAAS,IAAI;AACpB,SAAO,cAAc,gBAAgB,CAAC,CAAC,UAAU,IAAI,CAAC,CAAC;AACvD,SAAO;;AAGT,QAAO;EAAE,MAAM;EAAY,QAAQ;EAAc;EAAO"}

package/dist/react-ui/types.cjs

@@ -0,0 +1,37 @@
+//#region src/react-ui/types.ts
+function isUIMessage(message) {
+	if (typeof message !== "object" || message == null) return false;
+	if (!("type" in message)) return false;
+	return message.type === "ui";
+}
+function isRemoveUIMessage(message) {
+	if (typeof message !== "object" || message == null) return false;
+	if (!("type" in message)) return false;
+	return message.type === "remove-ui";
+}
+function uiMessageReducer(state, update) {
+	const events = Array.isArray(update) ? update : [update];
+	let newState = state.slice();
+	for (const event of events) {
+		if (event.type === "remove-ui") {
+			newState = newState.filter((ui) => ui.id !== event.id);
+			continue;
+		}
+		const index = state.findIndex((ui) => ui.id === event.id);
+		if (index !== -1) newState[index] = typeof event.metadata === "object" && event.metadata != null && event.metadata.merge ? {
+			...event,
+			props: {
+				...state[index].props,
+				...event.props
+			}
+		} : event;
+		else newState.push(event);
+	}
+	return newState;
+}
+//#endregion
+exports.isRemoveUIMessage = isRemoveUIMessage;
+exports.isUIMessage = isUIMessage;
+exports.uiMessageReducer = uiMessageReducer;
+
+//# sourceMappingURL=types.cjs.map

package/dist/react-ui/types.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.cjs","names":[],"sources":["../../src/react-ui/types.ts"],"sourcesContent":["export interface UIMessage<\n  TName extends string = string,\n  TProps extends Record<string, unknown> = Record<string, unknown>,\n> {\n  type: \"ui\";\n\n  id: string;\n  name: TName;\n  props: TProps;\n  metadata?: {\n    merge?: boolean;\n    run_id?: string;\n    name?: string;\n    tags?: string[];\n    message_id?: string;\n    [key: string]: unknown;\n  };\n}\n\nexport interface RemoveUIMessage {\n  type: \"remove-ui\";\n  id: string;\n}\n\nexport function isUIMessage(message: unknown): message is UIMessage {\n  if (typeof message !== \"object\" || message == null) return false;\n  if (!(\"type\" in message)) return false;\n  return message.type === \"ui\";\n}\n\nexport function isRemoveUIMessage(\n  message: unknown,\n): message is RemoveUIMessage {\n  if (typeof message !== \"object\" || message == null) return false;\n  if (!(\"type\" in message)) return false;\n  return message.type === \"remove-ui\";\n}\n\nexport function uiMessageReducer(\n  state: UIMessage[],\n  update: UIMessage | RemoveUIMessage | (UIMessage | RemoveUIMessage)[],\n) {\n  const events = Array.isArray(update) ? update : [update];\n  let newState = state.slice();\n\n  for (const event of events) {\n    if (event.type === \"remove-ui\") {\n      newState = newState.filter((ui) => ui.id !== event.id);\n      continue;\n    }\n\n    const index = state.findIndex((ui) => ui.id === event.id);\n    if (index !== -1) {\n      newState[index] =\n        typeof event.metadata === \"object\" &&\n        event.metadata != null &&\n        event.metadata.merge\n          ? { ...event, props: { ...state[index].props, ...event.props } }\n          : event;\n    } else {\n      newState.push(event);\n    }\n  }\n\n  return newState;\n}\n"],"mappings":";AAwBA,SAAgB,YAAY,SAAwC;AAClE,KAAI,OAAO,YAAY,YAAY,WAAW,KAAM,QAAO;AAC3D,KAAI,EAAE,UAAU,SAAU,QAAO;AACjC,QAAO,QAAQ,SAAS;;AAG1B,SAAgB,kBACd,SAC4B;AAC5B,KAAI,OAAO,YAAY,YAAY,WAAW,KAAM,QAAO;AAC3D,KAAI,EAAE,UAAU,SAAU,QAAO;AACjC,QAAO,QAAQ,SAAS;;AAG1B,SAAgB,iBACd,OACA,QACA;CACA,MAAM,SAAS,MAAM,QAAQ,OAAO,GAAG,SAAS,CAAC,OAAO;CACxD,IAAI,WAAW,MAAM,OAAO;AAE5B,MAAK,MAAM,SAAS,QAAQ;AAC1B,MAAI,MAAM,SAAS,aAAa;AAC9B,cAAW,SAAS,QAAQ,OAAO,GAAG,OAAO,MAAM,GAAG;AACtD;;EAGF,MAAM,QAAQ,MAAM,WAAW,OAAO,GAAG,OAAO,MAAM,GAAG;AACzD,MAAI,UAAU,GACZ,UAAS,SACP,OAAO,MAAM,aAAa,YAC1B,MAAM,YAAY,QAClB,MAAM,SAAS,QACX;GAAE,GAAG;GAAO,OAAO;IAAE,GAAG,MAAM,OAAO;IAAO,GAAG,MAAM;IAAO;GAAE,GAC9D;MAEN,UAAS,KAAK,MAAM;;AAIxB,QAAO"}

package/dist/react-ui/types.d.cts

@@ -0,0 +1,25 @@
+//#region src/react-ui/types.d.ts
+interface 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;
+    [key: string]: unknown;
+  };
+}
+interface RemoveUIMessage {
+  type: "remove-ui";
+  id: string;
+}
+declare function isUIMessage(message: unknown): message is UIMessage;
+declare function isRemoveUIMessage(message: unknown): message is RemoveUIMessage;
+declare function uiMessageReducer(state: UIMessage[], update: UIMessage | RemoveUIMessage | (UIMessage | RemoveUIMessage)[]): UIMessage<string, Record<string, unknown>>[];
+//#endregion
+export { RemoveUIMessage, UIMessage, isRemoveUIMessage, isUIMessage, uiMessageReducer };
+//# sourceMappingURL=types.d.cts.map

package/dist/react-ui/types.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.cts","names":[],"sources":["../../src/react-ui/types.ts"],"mappings":";UAAiB,SAAA,+CAEA,MAAA,oBAA0B,MAAA;EAEzC,IAAA;EAEA,EAAA;EACA,IAAA,EAAM,KAAA;EACN,KAAA,EAAO,MAAA;EACP,QAAA;IACE,KAAA;IACA,MAAA;IACA,IAAA;IACA,IAAA;IACA,UAAA;IAAA,CACC,GAAA;EAAA;AAAA;AAAA,UAIY,eAAA;EACf,IAAA;EACA,EAAA;AAAA;AAAA,iBAGc,WAAA,CAAY,OAAA,YAAmB,OAAA,IAAW,SAAA;AAAA,iBAM1C,iBAAA,CACd,OAAA,YACC,OAAA,IAAW,eAAA;AAAA,iBAME,gBAAA,CACd,KAAA,EAAO,SAAA,IACP,MAAA,EAAQ,SAAA,GAAY,eAAA,IAAmB,SAAA,GAAY,eAAA,MAAkB,SAAA,SAAA,MAAA"}

package/dist/react-ui/types.d.ts

@@ -0,0 +1,25 @@
+//#region src/react-ui/types.d.ts
+interface 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;
+    [key: string]: unknown;
+  };
+}
+interface RemoveUIMessage {
+  type: "remove-ui";
+  id: string;
+}
+declare function isUIMessage(message: unknown): message is UIMessage;
+declare function isRemoveUIMessage(message: unknown): message is RemoveUIMessage;
+declare function uiMessageReducer(state: UIMessage[], update: UIMessage | RemoveUIMessage | (UIMessage | RemoveUIMessage)[]): UIMessage<string, Record<string, unknown>>[];
+//#endregion
+export { RemoveUIMessage, UIMessage, isRemoveUIMessage, isUIMessage, uiMessageReducer };
+//# sourceMappingURL=types.d.ts.map

package/dist/react-ui/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","names":[],"sources":["../../src/react-ui/types.ts"],"mappings":";UAAiB,SAAA,+CAEA,MAAA,oBAA0B,MAAA;EAEzC,IAAA;EAEA,EAAA;EACA,IAAA,EAAM,KAAA;EACN,KAAA,EAAO,MAAA;EACP,QAAA;IACE,KAAA;IACA,MAAA;IACA,IAAA;IACA,IAAA;IACA,UAAA;IAAA,CACC,GAAA;EAAA;AAAA;AAAA,UAIY,eAAA;EACf,IAAA;EACA,EAAA;AAAA;AAAA,iBAGc,WAAA,CAAY,OAAA,YAAmB,OAAA,IAAW,SAAA;AAAA,iBAM1C,iBAAA,CACd,OAAA,YACC,OAAA,IAAW,eAAA;AAAA,iBAME,gBAAA,CACd,KAAA,EAAO,SAAA,IACP,MAAA,EAAQ,SAAA,GAAY,eAAA,IAAmB,SAAA,GAAY,eAAA,MAAkB,SAAA,SAAA,MAAA"}

package/dist/react-ui/types.js

@@ -0,0 +1,35 @@
+//#region src/react-ui/types.ts
+function isUIMessage(message) {
+	if (typeof message !== "object" || message == null) return false;
+	if (!("type" in message)) return false;
+	return message.type === "ui";
+}
+function isRemoveUIMessage(message) {
+	if (typeof message !== "object" || message == null) return false;
+	if (!("type" in message)) return false;
+	return message.type === "remove-ui";
+}
+function uiMessageReducer(state, update) {
+	const events = Array.isArray(update) ? update : [update];
+	let newState = state.slice();
+	for (const event of events) {
+		if (event.type === "remove-ui") {
+			newState = newState.filter((ui) => ui.id !== event.id);
+			continue;
+		}
+		const index = state.findIndex((ui) => ui.id === event.id);
+		if (index !== -1) newState[index] = typeof event.metadata === "object" && event.metadata != null && event.metadata.merge ? {
+			...event,
+			props: {
+				...state[index].props,
+				...event.props
+			}
+		} : event;
+		else newState.push(event);
+	}
+	return newState;
+}
+//#endregion
+export { isRemoveUIMessage, isUIMessage, uiMessageReducer };
+
+//# sourceMappingURL=types.js.map

package/dist/react-ui/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","names":[],"sources":["../../src/react-ui/types.ts"],"sourcesContent":["export interface UIMessage<\n  TName extends string = string,\n  TProps extends Record<string, unknown> = Record<string, unknown>,\n> {\n  type: \"ui\";\n\n  id: string;\n  name: TName;\n  props: TProps;\n  metadata?: {\n    merge?: boolean;\n    run_id?: string;\n    name?: string;\n    tags?: string[];\n    message_id?: string;\n    [key: string]: unknown;\n  };\n}\n\nexport interface RemoveUIMessage {\n  type: \"remove-ui\";\n  id: string;\n}\n\nexport function isUIMessage(message: unknown): message is UIMessage {\n  if (typeof message !== \"object\" || message == null) return false;\n  if (!(\"type\" in message)) return false;\n  return message.type === \"ui\";\n}\n\nexport function isRemoveUIMessage(\n  message: unknown,\n): message is RemoveUIMessage {\n  if (typeof message !== \"object\" || message == null) return false;\n  if (!(\"type\" in message)) return false;\n  return message.type === \"remove-ui\";\n}\n\nexport function uiMessageReducer(\n  state: UIMessage[],\n  update: UIMessage | RemoveUIMessage | (UIMessage | RemoveUIMessage)[],\n) {\n  const events = Array.isArray(update) ? update : [update];\n  let newState = state.slice();\n\n  for (const event of events) {\n    if (event.type === \"remove-ui\") {\n      newState = newState.filter((ui) => ui.id !== event.id);\n      continue;\n    }\n\n    const index = state.findIndex((ui) => ui.id === event.id);\n    if (index !== -1) {\n      newState[index] =\n        typeof event.metadata === \"object\" &&\n        event.metadata != null &&\n        event.metadata.merge\n          ? { ...event, props: { ...state[index].props, ...event.props } }\n          : event;\n    } else {\n      newState.push(event);\n    }\n  }\n\n  return newState;\n}\n"],"mappings":";AAwBA,SAAgB,YAAY,SAAwC;AAClE,KAAI,OAAO,YAAY,YAAY,WAAW,KAAM,QAAO;AAC3D,KAAI,EAAE,UAAU,SAAU,QAAO;AACjC,QAAO,QAAQ,SAAS;;AAG1B,SAAgB,kBACd,SAC4B;AAC5B,KAAI,OAAO,YAAY,YAAY,WAAW,KAAM,QAAO;AAC3D,KAAI,EAAE,UAAU,SAAU,QAAO;AACjC,QAAO,QAAQ,SAAS;;AAG1B,SAAgB,iBACd,OACA,QACA;CACA,MAAM,SAAS,MAAM,QAAQ,OAAO,GAAG,SAAS,CAAC,OAAO;CACxD,IAAI,WAAW,MAAM,OAAO;AAE5B,MAAK,MAAM,SAAS,QAAQ;AAC1B,MAAI,MAAM,SAAS,aAAa;AAC9B,cAAW,SAAS,QAAQ,OAAO,GAAG,OAAO,MAAM,GAAG;AACtD;;EAGF,MAAM,QAAQ,MAAM,WAAW,OAAO,GAAG,OAAO,MAAM,GAAG;AACzD,MAAI,UAAU,GACZ,UAAS,SACP,OAAO,MAAM,aAAa,YAC1B,MAAM,YAAY,QAClB,MAAM,SAAS,QACX;GAAE,GAAG;GAAO,OAAO;IAAE,GAAG,MAAM,OAAO;IAAO,GAAG,MAAM;IAAO;GAAE,GAC9D;MAEN,UAAS,KAAK,MAAM;;AAIxB,QAAO"}

package/dist/stream.cjs

@@ -1,3 +1,4 @@
+require("./_virtual/_rolldown/runtime.cjs");
 const require_stream_lgp = require("./stream.lgp.cjs");
 const require_stream_custom = require("./stream.custom.cjs");
 let react = require("react");
@@ -5,12 +6,10 @@ let react = require("react");
 function isCustomOptions(options) {
 	return "transport" in options;
 }
-function selectStreamImplementation(options) {
-	return isCustomOptions(options) ? require_stream_custom.useStreamCustom : require_stream_lgp.useStreamLGP;
-}
 function useStream(options) {
-	const [useSelectedStream] = (0, react.useState)(() => selectStreamImplementation(options));
-	return useSelectedStream(options);
+	const [isCustom] = (0, react.useState)(isCustomOptions(options));
+	if (isCustom) return require_stream_custom.useStreamCustom(options);
+	return require_stream_lgp.useStreamLGP(options);
 }
 //#endregion
 exports.useStream = useStream;

package/dist/stream.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"stream.cjs","names":["useStreamCustom","useStreamLGP"],"sources":["../src/stream.tsx"],"sourcesContent":["import { useState } from \"react\";\nimport type { BagTemplate } from \"@langchain/langgraph-sdk\";\nimport type {\n  UseStreamOptions,\n  ResolveStreamInterface,\n  ResolveStreamOptions,\n  InferBag,\n  InferStateType,\n  WithClassMessages,\n} from \"@langchain/langgraph-sdk/ui\";\nimport { useStreamLGP } from \"./stream.lgp.js\";\nimport { useStreamCustom } from \"./stream.custom.js\";\nimport type { UseStreamCustomOptions } from \"./types.js\";\n\nfunction isCustomOptions<\n  StateType extends Record<string, unknown> = Record<string, unknown>,\n  Bag extends BagTemplate = BagTemplate,\n>(\n  options:\n    | UseStreamOptions<StateType, Bag>\n    | UseStreamCustomOptions<StateType, Bag>\n): options is UseStreamCustomOptions<StateType, Bag> {\n  return \"transport\" in options;\n}\n\ntype UseStreamImplementation = typeof useStreamLGP | typeof useStreamCustom;\n\ntype AnyUseStreamOptions =\n  | UseStreamOptions<Record<string, unknown>, BagTemplate>\n  | UseStreamCustomOptions<Record<string, unknown>, BagTemplate>;\n\nfunction selectStreamImplementation(\n  options: AnyUseStreamOptions\n): UseStreamImplementation {\n  return isCustomOptions(options) ? useStreamCustom : useStreamLGP;\n}\n\nexport type {\n  WithClassMessages,\n  ClassSubagentStreamInterface,\n} from \"@langchain/langgraph-sdk/ui\";\n\n/**\n * A React hook that provides seamless integration with LangGraph streaming capabilities.\n *\n * The `useStream` hook handles all the complexities of streaming, state management, and branching logic,\n * letting you focus on building great chat experiences. It provides automatic state management for\n * messages, interrupts, loading states, subagent streams, and errors.\n *\n * ## Usage with ReactAgent (recommended for createAgent users)\n *\n * When using `createAgent` from `@langchain/langgraph`, you can pass `typeof agent` as the\n * type parameter to automatically infer tool call types:\n *\n * @example\n * ```typescript\n * // In your agent file (e.g., agent.ts)\n * import { createAgent, tool } from \"langchain\";\n * import { z } from \"zod\";\n *\n * const getWeather = tool(\n *   async ({ location }) => `Weather in ${location}`,\n *   { name: \"get_weather\", schema: z.object({ location: z.string() }) }\n * );\n *\n * export const agent = createAgent({\n *   model: \"openai:gpt-4o\",\n *   tools: [getWeather],\n * });\n *\n * // In your React component\n * import { agent } from \"./agent\";\n *\n * function Chat() {\n *   // Tool calls are automatically typed from the agent's tools!\n *   const stream = useStream<typeof agent>({\n *     assistantId: \"agent\",\n *     apiUrl: \"http://localhost:2024\",\n *   });\n *\n *   // stream.toolCalls[0].call.name is typed as \"get_weather\"\n *   // stream.toolCalls[0].call.args is typed as { location: string }\n * }\n * ```\n *\n * ## Usage with StateGraph (for custom LangGraph applications)\n *\n * When building custom graphs with `StateGraph`, embed your tool call types directly\n * in your state's messages property using `Message<MyToolCalls>`:\n *\n * @example\n * ```typescript\n * import { Message } from \"@langchain/langgraph-sdk\";\n *\n * // Define your tool call types as a discriminated union\n * type MyToolCalls =\n *   | { name: \"search\"; args: { query: string }; id?: string }\n *   | { name: \"calculate\"; args: { expression: string }; id?: string };\n *\n * // Embed tool call types in your state's messages\n * interface MyGraphState {\n *   messages: Message<MyToolCalls>[];\n *   context?: string;\n * }\n *\n * function Chat() {\n *   const stream = useStream<MyGraphState>({\n *     assistantId: \"my-graph\",\n *     apiUrl: \"http://localhost:2024\",\n *   });\n *\n *   // stream.values is typed as MyGraphState\n *   // stream.toolCalls[0].call.name is typed as \"search\" | \"calculate\"\n * }\n * ```\n *\n * @example\n * ```typescript\n * // With additional type configuration (interrupts, configurable)\n * interface MyGraphState {\n *   messages: Message<MyToolCalls>[];\n * }\n *\n * function Chat() {\n *   const stream = useStream<MyGraphState, {\n *     InterruptType: { question: string };\n *     ConfigurableType: { userId: string };\n *   }>({\n *     assistantId: \"my-graph\",\n *     apiUrl: \"http://localhost:2024\",\n *   });\n *\n *   // stream.interrupt is typed as { question: string } | undefined\n * }\n * ```\n *\n * ## Usage with Deep Agents (subagent streaming, experimental)\n *\n * For agents that spawn subagents (nested graphs), use `filterSubagentMessages`\n * to keep the main message stream clean while tracking subagent activity separately:\n *\n * @example\n * ```typescript\n * import { useStream, SubagentStream } from \"@langchain/langgraph-sdk/react\";\n * import type { agent } from \"./agent\";\n *\n * function DeepAgentChat() {\n *   const stream = useStream<typeof agent>({\n *     assistantId: \"deepagent\",\n *     apiUrl: \"http://localhost:2024\",\n *     // Filter subagent messages from main stream\n *     filterSubagentMessages: true,\n *   });\n *\n *   const handleSubmit = (content: string) => {\n *     stream.submit(\n *       { messages: [{ content, type: \"human\" }] },\n *       { streamSubgraphs: true } // Enable subgraph streaming\n *     );\n *   };\n *\n *   // Access subagent streams via stream.subagents (Map<string, SubagentStream>)\n *   const subagentList = [...stream.subagents.values()];\n *\n *   return (\n *     <div>\n *       {stream.messages.map((msg) => <Message key={msg.id} message={msg} />)}\n *\n *       {subagentList.map((subagent) => (\n *         <SubagentCard\n *           key={subagent.id}\n *           status={subagent.status} // \"pending\" | \"running\" | \"complete\" | \"error\"\n *           messages={subagent.messages}\n *           toolCalls={subagent.toolCalls}\n *         />\n *       ))}\n *     </div>\n *   );\n * }\n * ```\n *\n * @template T Either a ReactAgent type (with `~agentTypes`) or a state type (`Record<string, unknown>`)\n * @template Bag Type configuration bag containing:\n *   - `ConfigurableType`: Type for the `config.configurable` property\n *   - `InterruptType`: Type for interrupt values\n *   - `CustomEventType`: Type for custom events\n *   - `UpdateType`: Type for the submit function updates\n *\n * @see {@link https://docs.langchain.com/langgraph-platform/use-stream-react | LangGraph React Integration Guide}\n */\nexport function useStream<\n  T = Record<string, unknown>,\n  Bag extends BagTemplate = BagTemplate,\n>(\n  options: ResolveStreamOptions<T, InferBag<T, Bag>>\n): WithClassMessages<ResolveStreamInterface<T, InferBag<T, Bag>>>;\n\n/**\n * A React hook that provides seamless integration with LangGraph streaming capabilities.\n *\n * The `useStream` hook handles all the complexities of streaming, state management, and branching logic,\n * letting you focus on building great chat experiences. It provides automatic state management for\n * messages, interrupts, loading states, and errors.\n *\n * @template T Either a ReactAgent type (with `~agentTypes`) or a state type (`Record<string, unknown>`)\n * @template Bag Type configuration bag containing:\n *   - `ConfigurableType`: Type for the `config.configurable` property\n *   - `InterruptType`: Type for interrupt values\n *   - `CustomEventType`: Type for custom events\n *   - `UpdateType`: Type for the submit function updates\n *\n * @see {@link https://docs.langchain.com/langgraph-platform/use-stream-react | LangGraph React Integration Guide}\n */\nexport function useStream<\n  T = Record<string, unknown>,\n  Bag extends BagTemplate = BagTemplate,\n>(\n  options: UseStreamCustomOptions<InferStateType<T>, InferBag<T, Bag>>\n): WithClassMessages<ResolveStreamInterface<T, InferBag<T, Bag>>>;\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport function useStream(options: any): any {\n  // Keep implementation stable for the lifetime of this hook instance.\n  const [useSelectedStream] = useState(() =>\n    selectStreamImplementation(options)\n  );\n  return useSelectedStream(options);\n}\n"],"mappings":";;;;AAcA,SAAS,gBAIP,SAGmD;AACnD,QAAO,eAAe;;AASxB,SAAS,2BACP,SACyB;AACzB,QAAO,gBAAgB,QAAQ,GAAGA,sBAAAA,kBAAkBC,mBAAAA;;AA2LtD,SAAgB,UAAU,SAAmB;CAE3C,MAAM,CAAC,sBAAA,GAAA,MAAA,gBACL,2BAA2B,QAAQ,CACpC;AACD,QAAO,kBAAkB,QAAQ"}
+{"version":3,"file":"stream.cjs","names":["useStreamCustom","useStreamLGP"],"sources":["../src/stream.tsx"],"sourcesContent":["import { useState } from \"react\";\nimport type {\n  BaseMessage,\n  ToolMessage as CoreToolMessage,\n  AIMessage as CoreAIMessage,\n} from \"@langchain/core/messages\";\nimport type {\n  BagTemplate,\n  ToolCallWithResult,\n  DefaultToolCall,\n} from \"@langchain/langgraph-sdk\";\nimport type {\n  UseStreamOptions,\n  AcceptBaseMessages,\n  ResolveStreamInterface,\n  ResolveStreamOptions,\n  InferBag,\n  InferStateType,\n  MessageMetadata,\n  SubagentStreamInterface,\n  HistoryWithBaseMessages,\n} from \"@langchain/langgraph-sdk/ui\";\nimport { useStreamLGP } from \"./stream.lgp.js\";\nimport { useStreamCustom } from \"./stream.custom.js\";\nimport type { UseStreamCustomOptions } from \"./types.js\";\n\nfunction isCustomOptions<\n  StateType extends Record<string, unknown> = Record<string, unknown>,\n  Bag extends BagTemplate = BagTemplate,\n>(\n  options:\n    | UseStreamOptions<StateType, Bag>\n    | UseStreamCustomOptions<StateType, Bag>,\n): options is UseStreamCustomOptions<StateType, Bag> {\n  return \"transport\" in options;\n}\n\ntype ClassToolCallWithResult<T> =\n  T extends ToolCallWithResult<infer TC, unknown, unknown>\n    ? ToolCallWithResult<TC, CoreToolMessage, CoreAIMessage>\n    : T;\n\nexport type ClassSubagentStreamInterface<\n  StateType = Record<string, unknown>,\n  ToolCall = DefaultToolCall,\n  SubagentName extends string = string,\n> = Omit<\n  SubagentStreamInterface<StateType, ToolCall, SubagentName>,\n  \"messages\"\n> & {\n  messages: BaseMessage[];\n};\n\n/**\n * Maps a stream interface to use @langchain/core BaseMessage class instances\n * instead of plain Message objects for the `messages` property, and remaps\n * tool call types to use @langchain/core message classes.\n */\ntype WithClassMessages<T> = Omit<\n  T,\n  | \"messages\"\n  | \"history\"\n  | \"getMessagesMetadata\"\n  | \"toolCalls\"\n  | \"getToolCalls\"\n  | \"submit\"\n  | \"subagents\"\n  | \"activeSubagents\"\n  | \"getSubagent\"\n  | \"getSubagentsByType\"\n  | \"getSubagentsByMessage\"\n> & {\n  messages: BaseMessage[];\n  getMessagesMetadata: (\n    message: BaseMessage,\n    index?: number,\n  ) => MessageMetadata<Record<string, unknown>> | undefined;\n} & (\"history\" extends keyof T\n    ? { history: HistoryWithBaseMessages<T[\"history\"]> }\n    : unknown) &\n  (\"submit\" extends keyof T\n    ? {\n        submit: T extends {\n          submit: (values: infer V, options?: infer O) => infer Ret;\n        }\n          ? (\n              values:\n                | AcceptBaseMessages<Exclude<V, null | undefined>>\n                | null\n                | undefined,\n              options?: O,\n            ) => Ret\n          : never;\n      }\n    : unknown) &\n  (\"toolCalls\" extends keyof T\n    ? {\n        toolCalls: T extends { toolCalls: (infer TC)[] }\n          ? ClassToolCallWithResult<TC>[]\n          : never;\n      }\n    : unknown) &\n  (\"getToolCalls\" extends keyof T\n    ? {\n        getToolCalls: T extends {\n          getToolCalls: (message: infer _M) => (infer TC)[];\n        }\n          ? (message: CoreAIMessage) => ClassToolCallWithResult<TC>[]\n          : never;\n      }\n    : unknown) &\n  (\"subagents\" extends keyof T\n    ? {\n        subagents: T extends {\n          subagents: Map<\n            string,\n            SubagentStreamInterface<infer S, infer TC, infer N>\n          >;\n        }\n          ? Map<string, ClassSubagentStreamInterface<S, TC, N>>\n          : never;\n        activeSubagents: T extends {\n          activeSubagents: SubagentStreamInterface<\n            infer S,\n            infer TC,\n            infer N\n          >[];\n        }\n          ? ClassSubagentStreamInterface<S, TC, N>[]\n          : never;\n        getSubagent: T extends {\n          getSubagent: (\n            id: string,\n          ) => SubagentStreamInterface<infer S, infer TC, infer N> | undefined;\n        }\n          ? (\n              toolCallId: string,\n            ) => ClassSubagentStreamInterface<S, TC, N> | undefined\n          : never;\n        getSubagentsByType: T extends {\n          getSubagentsByType: (\n            type: string,\n          ) => SubagentStreamInterface<infer S, infer TC, infer N>[];\n        }\n          ? (type: string) => ClassSubagentStreamInterface<S, TC, N>[]\n          : never;\n        getSubagentsByMessage: T extends {\n          getSubagentsByMessage: (\n            id: string,\n          ) => SubagentStreamInterface<infer S, infer TC, infer N>[];\n        }\n          ? (messageId: string) => ClassSubagentStreamInterface<S, TC, N>[]\n          : never;\n      }\n    : unknown);\n\n/**\n * A React hook that provides seamless integration with LangGraph streaming capabilities.\n *\n * The `useStream` hook handles all the complexities of streaming, state management, and branching logic,\n * letting you focus on building great chat experiences. It provides automatic state management for\n * messages, interrupts, loading states, subagent streams, and errors.\n *\n * ## Usage with ReactAgent (recommended for createAgent users)\n *\n * When using `createAgent` from `@langchain/langgraph`, you can pass `typeof agent` as the\n * type parameter to automatically infer tool call types:\n *\n * @example\n * ```typescript\n * // In your agent file (e.g., agent.ts)\n * import { createAgent, tool } from \"langchain\";\n * import { z } from \"zod\";\n *\n * const getWeather = tool(\n *   async ({ location }) => `Weather in ${location}`,\n *   { name: \"get_weather\", schema: z.object({ location: z.string() }) }\n * );\n *\n * export const agent = createAgent({\n *   model: \"openai:gpt-4o\",\n *   tools: [getWeather],\n * });\n *\n * // In your React component\n * import { agent } from \"./agent\";\n *\n * function Chat() {\n *   // Tool calls are automatically typed from the agent's tools!\n *   const stream = useStream<typeof agent>({\n *     assistantId: \"agent\",\n *     apiUrl: \"http://localhost:2024\",\n *   });\n *\n *   // stream.toolCalls[0].call.name is typed as \"get_weather\"\n *   // stream.toolCalls[0].call.args is typed as { location: string }\n * }\n * ```\n *\n * ## Usage with StateGraph (for custom LangGraph applications)\n *\n * When building custom graphs with `StateGraph`, embed your tool call types directly\n * in your state's messages property using `Message<MyToolCalls>`:\n *\n * @example\n * ```typescript\n * import { Message } from \"@langchain/langgraph-sdk\";\n *\n * // Define your tool call types as a discriminated union\n * type MyToolCalls =\n *   | { name: \"search\"; args: { query: string }; id?: string }\n *   | { name: \"calculate\"; args: { expression: string }; id?: string };\n *\n * // Embed tool call types in your state's messages\n * interface MyGraphState {\n *   messages: Message<MyToolCalls>[];\n *   context?: string;\n * }\n *\n * function Chat() {\n *   const stream = useStream<MyGraphState>({\n *     assistantId: \"my-graph\",\n *     apiUrl: \"http://localhost:2024\",\n *   });\n *\n *   // stream.values is typed as MyGraphState\n *   // stream.toolCalls[0].call.name is typed as \"search\" | \"calculate\"\n * }\n * ```\n *\n * @example\n * ```typescript\n * // With additional type configuration (interrupts, configurable)\n * interface MyGraphState {\n *   messages: Message<MyToolCalls>[];\n * }\n *\n * function Chat() {\n *   const stream = useStream<MyGraphState, {\n *     InterruptType: { question: string };\n *     ConfigurableType: { userId: string };\n *   }>({\n *     assistantId: \"my-graph\",\n *     apiUrl: \"http://localhost:2024\",\n *   });\n *\n *   // stream.interrupt is typed as { question: string } | undefined\n * }\n * ```\n *\n * ## Usage with Deep Agents (subagent streaming, experimental)\n *\n * For agents that spawn subagents (nested graphs), use `filterSubagentMessages`\n * to keep the main message stream clean while tracking subagent activity separately:\n *\n * @example\n * ```typescript\n * import { useStream, SubagentStream } from \"@langchain/langgraph-sdk/react\";\n * import type { agent } from \"./agent\";\n *\n * function DeepAgentChat() {\n *   const stream = useStream<typeof agent>({\n *     assistantId: \"deepagent\",\n *     apiUrl: \"http://localhost:2024\",\n *     // Filter subagent messages from main stream\n *     filterSubagentMessages: true,\n *   });\n *\n *   const handleSubmit = (content: string) => {\n *     stream.submit(\n *       { messages: [{ content, type: \"human\" }] },\n *       { streamSubgraphs: true } // Enable subgraph streaming\n *     );\n *   };\n *\n *   // Access subagent streams via stream.subagents (Map<string, SubagentStream>)\n *   const subagentList = [...stream.subagents.values()];\n *\n *   return (\n *     <div>\n *       {stream.messages.map((msg) => <Message key={msg.id} message={msg} />)}\n *\n *       {subagentList.map((subagent) => (\n *         <SubagentCard\n *           key={subagent.id}\n *           status={subagent.status} // \"pending\" | \"running\" | \"complete\" | \"error\"\n *           messages={subagent.messages}\n *           toolCalls={subagent.toolCalls}\n *         />\n *       ))}\n *     </div>\n *   );\n * }\n * ```\n *\n * @template T Either a ReactAgent type (with `~agentTypes`) or a state type (`Record<string, unknown>`)\n * @template Bag Type configuration bag containing:\n *   - `ConfigurableType`: Type for the `config.configurable` property\n *   - `InterruptType`: Type for interrupt values\n *   - `CustomEventType`: Type for custom events\n *   - `UpdateType`: Type for the submit function updates\n *\n * @see {@link https://docs.langchain.com/langgraph-platform/use-stream-react | LangGraph React Integration Guide}\n */\nexport function useStream<\n  T = Record<string, unknown>,\n  Bag extends BagTemplate = BagTemplate,\n>(\n  options: ResolveStreamOptions<T, InferBag<T, Bag>>,\n): WithClassMessages<ResolveStreamInterface<T, InferBag<T, Bag>>>;\n\n/**\n * A React hook that provides seamless integration with LangGraph streaming capabilities.\n *\n * The `useStream` hook handles all the complexities of streaming, state management, and branching logic,\n * letting you focus on building great chat experiences. It provides automatic state management for\n * messages, interrupts, loading states, and errors.\n *\n * @template T Either a ReactAgent type (with `~agentTypes`) or a state type (`Record<string, unknown>`)\n * @template Bag Type configuration bag containing:\n *   - `ConfigurableType`: Type for the `config.configurable` property\n *   - `InterruptType`: Type for interrupt values\n *   - `CustomEventType`: Type for custom events\n *   - `UpdateType`: Type for the submit function updates\n *\n * @see {@link https://docs.langchain.com/langgraph-platform/use-stream-react | LangGraph React Integration Guide}\n */\nexport function useStream<\n  T = Record<string, unknown>,\n  Bag extends BagTemplate = BagTemplate,\n>(\n  options: UseStreamCustomOptions<InferStateType<T>, InferBag<T, Bag>>,\n): WithClassMessages<ResolveStreamInterface<T, InferBag<T, Bag>>>;\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport function useStream(options: any): any {\n  // Store this in useState to make sure we're not changing the implementation in re-renders\n  const [isCustom] = useState(isCustomOptions(options));\n\n  if (isCustom) {\n    // eslint-disable-next-line react-hooks/rules-of-hooks\n    return useStreamCustom(options);\n  }\n\n  // eslint-disable-next-line react-hooks/rules-of-hooks\n  return useStreamLGP(options);\n}\n"],"mappings":";;;;;AA0BA,SAAS,gBAIP,SAGmD;AACnD,QAAO,eAAe;;AA6SxB,SAAgB,UAAU,SAAmB;CAE3C,MAAM,CAAC,aAAA,GAAA,MAAA,UAAqB,gBAAgB,QAAQ,CAAC;AAErD,KAAI,SAEF,QAAOA,sBAAAA,gBAAgB,QAAQ;AAIjC,QAAOC,mBAAAA,aAAa,QAAQ"}

package/dist/stream.custom.cjs

@@ -1,26 +1,11 @@
 "use client";
+require("./_virtual/_rolldown/runtime.cjs");
 const require_thread = require("./thread.cjs");
 let react = require("react");
 let _langchain_langgraph_sdk_ui = require("@langchain/langgraph-sdk/ui");
 let _langchain_langgraph_sdk_utils = require("@langchain/langgraph-sdk/utils");
 let _langchain_langgraph_sdk = require("@langchain/langgraph-sdk");
 //#region src/stream.custom.tsx
-function createCustomTransportThreadState(values, threadId) {
-	return {
-		values,
-		next: [],
-		tasks: [],
-		metadata: void 0,
-		created_at: null,
-		checkpoint: {
-			thread_id: threadId,
-			checkpoint_id: null,
-			checkpoint_ns: "",
-			checkpoint_map: null
-		},
-		parent_checkpoint: null
-	};
-}
 function useStreamCustom(options) {
 	const [messageManager] = (0, react.useState)(() => new _langchain_langgraph_sdk_ui.MessageTupleManager());
 	const [stream] = (0, react.useState)(() => new _langchain_langgraph_sdk_ui.StreamManager(messageManager, {
@@ -87,7 +72,6 @@ function useStreamCustom(options) {
 				input: values,
 				context: submitOptions?.context,
 				command: submitOptions?.command,
-				streamSubgraphs: submitOptions?.streamSubgraphs,
 				signal,
 				config: {
 					...submitOptions?.config,
@@ -102,11 +86,7 @@ function useStreamCustom(options) {
 			setMessages,
 			initialValues: {},
 			callbacks: options,
-			onSuccess: () => {
-				if (!usableThreadId) return void 0;
-				const finalValues = stream.values ?? historyValues;
-				options.onFinish?.(createCustomTransportThreadState(finalValues, usableThreadId), void 0);
-			},
+			onSuccess: () => void 0,
 			onError(error) {
 				options.onError?.(error, void 0);
 				submitOptions?.onError?.(error, void 0);
@@ -116,24 +96,30 @@ function useStreamCustom(options) {
 	const submit = async (values, submitOptions) => {
 		await submitDirect(values, submitOptions);
 	};
-	const handledToolsRef = (0, react.useRef)(/* @__PURE__ */ new Set());
+	const toolsRef = (0, react.useRef)(options.tools);
+	toolsRef.current = options.tools;
+	const onToolRef = (0, react.useRef)(options.onTool);
+	onToolRef.current = options.onTool;
+	const handledBrowserToolsRef = (0, react.useRef)(/* @__PURE__ */ new Set());
 	(0, react.useEffect)(() => {
-		handledToolsRef.current.clear();
+		handledBrowserToolsRef.current.clear();
 	}, [threadId]);
 	(0, react.useEffect)(() => {
-		(0, _langchain_langgraph_sdk.flushPendingHeadlessToolInterrupts)(stream.values, options.tools, handledToolsRef.current, {
-			onTool: options.onTool,
-			defer: (run) => {
-				Promise.resolve().then(run);
-			},
-			resumeSubmit: (command) => submit(null, { command })
-		});
-	}, [
-		options.onTool,
-		options.tools,
-		stream.values,
-		submit
-	]);
+		const tools = toolsRef.current;
+		if (!tools?.length) return;
+		if (!stream.values) return;
+		const interrupts = stream.values.__interrupt__;
+		if (!Array.isArray(interrupts) || interrupts.length === 0) return;
+		for (const interrupt of interrupts) {
+			if (!(0, _langchain_langgraph_sdk.isHeadlessToolInterrupt)(interrupt.value)) continue;
+			const interruptId = interrupt.id ?? interrupt.value.toolCall.id ?? "";
+			if (handledBrowserToolsRef.current.has(interruptId)) continue;
+			handledBrowserToolsRef.current.add(interruptId);
+			(0, _langchain_langgraph_sdk.handleHeadlessToolInterrupt)(interrupt.value, tools, onToolRef.current).then((result) => {
+				submit(null, { command: { resume: result.toolCallId ? { [result.toolCallId]: result.value } : result.value } });
+			});
+		}
+	}, [stream.values, submit]);
 	return {
 		get values() {
 			return stream.values ?? {};
@@ -156,7 +142,11 @@ function useStreamCustom(options) {
 			};
 		},
 		get interrupts() {
-			if (stream.values != null && "__interrupt__" in stream.values && Array.isArray(stream.values.__interrupt__)) return (0, _langchain_langgraph_sdk_ui.userFacingInterruptsFromValuesArray)(stream.values.__interrupt__);
+			if (stream.values != null && "__interrupt__" in stream.values && Array.isArray(stream.values.__interrupt__)) {
+				const valueInterrupts = stream.values.__interrupt__;
+				if (valueInterrupts.length === 0) return [{ when: "breakpoint" }];
+				return valueInterrupts;
+			}
 			return [];
 		},
 		get interrupt() {

package/dist/stream.custom.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"stream.custom.cjs","names":["MessageTupleManager","StreamManager","toMessageClass","useControllableThreadId"],"sources":["../src/stream.custom.tsx"],"sourcesContent":["/* __LC_ALLOW_ENTRYPOINT_SIDE_EFFECTS__ */\n\n\"use client\";\n\nimport {\n  useCallback,\n  useEffect,\n  useRef,\n  useState,\n  useSyncExternalStore,\n} from \"react\";\nimport {\n  StreamManager,\n  MessageTupleManager,\n  extractInterrupts,\n  userFacingInterruptsFromValuesArray,\n  FetchStreamTransport,\n  toMessageClass,\n  ensureMessageInstances,\n  type EventStreamEvent,\n  type GetUpdateType,\n  type GetCustomEventType,\n  type GetInterruptType,\n  type GetToolCallsType,\n  type GetConfigurableType,\n  type AnyStreamCustomOptions,\n  type CustomSubmitOptions,\n  type MessageMetadata,\n} from \"@langchain/langgraph-sdk/ui\";\nimport { getToolCallsWithResults } from \"@langchain/langgraph-sdk/utils\";\nimport type { BaseMessage } from \"@langchain/core/messages\";\nimport type {\n  BagTemplate,\n  Message,\n  Interrupt,\n  ThreadState,\n} from \"@langchain/langgraph-sdk\";\nimport { flushPendingHeadlessToolInterrupts } from \"@langchain/langgraph-sdk\";\nimport { useControllableThreadId } from \"./thread.js\";\nimport type { UseStreamCustom } from \"./types.js\";\n\nexport { FetchStreamTransport };\n\nfunction createCustomTransportThreadState<\n  StateType extends Record<string, unknown>,\n>(values: StateType, threadId: string): ThreadState<StateType> {\n  return {\n    values,\n    next: [],\n    tasks: [],\n    metadata: undefined,\n    created_at: null,\n    checkpoint: {\n      thread_id: threadId,\n      checkpoint_id: null,\n      checkpoint_ns: \"\",\n      checkpoint_map: null,\n    },\n    parent_checkpoint: null,\n  };\n}\n\nexport function useStreamCustom<\n  StateType extends Record<string, unknown> = Record<string, unknown>,\n  Bag extends BagTemplate = BagTemplate,\n>(\n  options: AnyStreamCustomOptions<StateType, Bag>\n): UseStreamCustom<StateType, Bag> {\n  type UpdateType = GetUpdateType<Bag, StateType>;\n  type CustomType = GetCustomEventType<Bag>;\n  type InterruptType = GetInterruptType<Bag>;\n  type ConfigurableType = GetConfigurableType<Bag>;\n  type ToolCallType = GetToolCallsType<StateType>;\n\n  const [messageManager] = useState(() => new MessageTupleManager());\n  const [stream] = useState(\n    () =>\n      new StreamManager<StateType, Bag>(messageManager, {\n        throttle: options.throttle ?? false,\n        subagentToolNames: options.subagentToolNames,\n        filterSubagentMessages: options.filterSubagentMessages,\n        toMessage: options.toMessage ?? toMessageClass,\n      })\n  );\n\n  useSyncExternalStore(\n    stream.subscribe,\n    stream.getSnapshot,\n    stream.getSnapshot\n  );\n\n  const [branch, _setBranch] = useState(\"\");\n\n  const [threadId, onThreadId] = useControllableThreadId(options);\n  const threadIdRef = useRef<string | null>(threadId);\n\n  // Cancel the stream if thread ID has changed\n  useEffect(() => {\n    if (threadIdRef.current !== threadId) {\n      threadIdRef.current = threadId;\n      stream.clear();\n    }\n  }, [threadId, stream]);\n\n  const switchThread = useCallback(\n    (newThreadId: string | null) => {\n      if (newThreadId !== threadIdRef.current) {\n        threadIdRef.current = newThreadId;\n        stream.clear();\n      }\n    },\n    [stream]\n  );\n\n  const getMessages = (value: StateType): Message[] => {\n    const messagesKey = options.messagesKey ?? \"messages\";\n    return Array.isArray(value[messagesKey])\n      ? (value[messagesKey] as Message[])\n      : [];\n  };\n\n  const setMessages = (current: StateType, messages: Message[]): StateType => {\n    const messagesKey = options.messagesKey ?? \"messages\";\n    return { ...current, [messagesKey]: messages };\n  };\n\n  const historyValues = options.initialValues ?? ({} as StateType);\n\n  // Reconstruct subagents from initialValues when:\n  // 1. Subagent filtering is enabled\n  // 2. Not currently streaming\n  // 3. initialValues has messages\n  // This ensures subagent visualization works with cached/persisted state\n  const historyMessages = getMessages(historyValues);\n  const shouldReconstructSubagents =\n    options.filterSubagentMessages &&\n    !stream.isLoading &&\n    historyMessages.length > 0;\n\n  useEffect(() => {\n    if (shouldReconstructSubagents) {\n      // skipIfPopulated: true ensures we don't overwrite subagents from active streaming\n      stream.reconstructSubagents(historyMessages, { skipIfPopulated: true });\n    }\n    // We intentionally only run this when shouldReconstructSubagents changes\n    // to avoid unnecessary reconstructions during streaming\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n  }, [shouldReconstructSubagents, historyMessages.length]);\n\n  const stop = () => stream.stop(historyValues, { onStop: options.onStop });\n\n  const submitDirect = async (\n    values: UpdateType | null | undefined,\n    submitOptions?: CustomSubmitOptions<StateType, ConfigurableType>\n  ) => {\n    if (threadId !== threadIdRef.current) {\n      threadIdRef.current = threadId;\n      stream.clear();\n    }\n\n    let usableThreadId = threadIdRef.current ?? submitOptions?.threadId;\n\n    stream.setStreamValues(() => {\n      if (submitOptions?.optimisticValues != null) {\n        return {\n          ...historyValues,\n          ...(typeof submitOptions.optimisticValues === \"function\"\n            ? submitOptions.optimisticValues(historyValues)\n            : submitOptions.optimisticValues),\n        };\n      }\n\n      return { ...historyValues };\n    });\n\n    await stream.start(\n      async (signal: AbortSignal) => {\n        if (!usableThreadId) {\n          usableThreadId = crypto.randomUUID();\n          threadIdRef.current = usableThreadId;\n          onThreadId(usableThreadId);\n        }\n\n        if (!usableThreadId) {\n          throw new Error(\"Failed to obtain valid thread ID.\");\n        }\n\n        return options.transport.stream({\n          input: values,\n          context: submitOptions?.context,\n          command: submitOptions?.command,\n          streamSubgraphs: submitOptions?.streamSubgraphs,\n          signal,\n          config: {\n            ...submitOptions?.config,\n            configurable: {\n              thread_id: usableThreadId,\n              ...submitOptions?.config?.configurable,\n            } as unknown as GetConfigurableType<Bag>,\n          },\n        }) as Promise<\n          AsyncGenerator<EventStreamEvent<StateType, UpdateType, CustomType>>\n        >;\n      },\n      {\n        getMessages,\n        setMessages,\n\n        initialValues: {} as StateType,\n        callbacks: options,\n\n        onSuccess: () => {\n          if (!usableThreadId) return undefined;\n\n          const finalValues = stream.values ?? historyValues;\n          options.onFinish?.(\n            createCustomTransportThreadState(finalValues, usableThreadId),\n            undefined\n          );\n\n          return undefined;\n        },\n        onError(error) {\n          options.onError?.(error, undefined);\n          submitOptions?.onError?.(error, undefined);\n        },\n      }\n    );\n  };\n\n  const submit = async (\n    values: UpdateType | null | undefined,\n    submitOptions?: CustomSubmitOptions<StateType, ConfigurableType>\n  ) => {\n    await submitDirect(values, submitOptions);\n  };\n\n  const handledToolsRef = useRef<Set<string>>(new Set());\n  useEffect(() => {\n    handledToolsRef.current.clear();\n  }, [threadId]);\n\n  useEffect(() => {\n    flushPendingHeadlessToolInterrupts(\n      stream.values as Record<string, unknown> | null,\n      options.tools,\n      handledToolsRef.current,\n      {\n        onTool: options.onTool,\n        defer: (run) => {\n          void Promise.resolve().then(run);\n        },\n        resumeSubmit: (command) =>\n          submit(null, {\n            command,\n          }),\n      }\n    );\n  }, [options.onTool, options.tools, stream.values, submit]);\n\n  return {\n    get values() {\n      return stream.values ?? ({} as StateType);\n    },\n\n    error: stream.error,\n    isLoading: stream.isLoading,\n\n    stop,\n    submit,\n    switchThread,\n\n    branch,\n    setBranch: _setBranch,\n\n    getMessagesMetadata(\n      message: BaseMessage,\n      index?: number\n    ): MessageMetadata<StateType> | undefined {\n      const streamMetadata = messageManager.get(message.id)?.metadata;\n      if (streamMetadata != null) {\n        return {\n          messageId: message.id ?? String(index),\n          firstSeenState: undefined,\n          branch: undefined,\n          branchOptions: undefined,\n          streamMetadata,\n        } as MessageMetadata<StateType>;\n      }\n      return undefined;\n    },\n\n    get interrupts(): Interrupt<InterruptType>[] {\n      if (\n        stream.values != null &&\n        \"__interrupt__\" in stream.values &&\n        Array.isArray(stream.values.__interrupt__)\n      ) {\n        return userFacingInterruptsFromValuesArray<InterruptType>(\n          stream.values.__interrupt__ as Interrupt<InterruptType>[]\n        );\n      }\n\n      return [];\n    },\n\n    get interrupt(): Interrupt<InterruptType> | undefined {\n      return extractInterrupts<InterruptType>(stream.values);\n    },\n\n    get messages(): BaseMessage[] {\n      if (!stream.values) return [];\n      return ensureMessageInstances(\n        getMessages(stream.values)\n      ) as BaseMessage[];\n    },\n\n    get toolCalls() {\n      if (!stream.values) return [];\n      const msgs = getMessages(stream.values);\n      return getToolCallsWithResults<ToolCallType>(msgs);\n    },\n\n    getToolCalls(message) {\n      if (!stream.values) return [];\n      const msgs = getMessages(stream.values);\n      const allToolCalls = getToolCallsWithResults<ToolCallType>(msgs);\n      return allToolCalls.filter((tc) => tc.aiMessage.id === message.id);\n    },\n\n    get subagents() {\n      return stream.getSubagents();\n    },\n\n    get activeSubagents() {\n      return stream.getActiveSubagents();\n    },\n\n    getSubagent(toolCallId: string) {\n      return stream.getSubagent(toolCallId);\n    },\n\n    getSubagentsByType(type: string) {\n      return stream.getSubagentsByType(type);\n    },\n\n    getSubagentsByMessage(messageId: string) {\n      return stream.getSubagentsByMessage(messageId);\n    },\n\n    queue: {\n      get entries() {\n        return [];\n      },\n      get size() {\n        return 0;\n      },\n      async cancel() {\n        return false;\n      },\n      async clear() {},\n    },\n  };\n}\n"],"mappings":";;;;;;;AA2CA,SAAS,iCAEP,QAAmB,UAA0C;AAC7D,QAAO;EACL;EACA,MAAM,EAAE;EACR,OAAO,EAAE;EACT,UAAU,KAAA;EACV,YAAY;EACZ,YAAY;GACV,WAAW;GACX,eAAe;GACf,eAAe;GACf,gBAAgB;GACjB;EACD,mBAAmB;EACpB;;AAGH,SAAgB,gBAId,SACiC;CAOjC,MAAM,CAAC,mBAAA,GAAA,MAAA,gBAAiC,IAAIA,4BAAAA,qBAAqB,CAAC;CAClE,MAAM,CAAC,WAAA,GAAA,MAAA,gBAEH,IAAIC,4BAAAA,cAA8B,gBAAgB;EAChD,UAAU,QAAQ,YAAY;EAC9B,mBAAmB,QAAQ;EAC3B,wBAAwB,QAAQ;EAChC,WAAW,QAAQ,aAAaC,4BAAAA;EACjC,CAAC,CACL;AAED,EAAA,GAAA,MAAA,sBACE,OAAO,WACP,OAAO,aACP,OAAO,YACR;CAED,MAAM,CAAC,QAAQ,eAAA,GAAA,MAAA,UAAuB,GAAG;CAEzC,MAAM,CAAC,UAAU,cAAcC,eAAAA,wBAAwB,QAAQ;CAC/D,MAAM,eAAA,GAAA,MAAA,QAAoC,SAAS;AAGnD,EAAA,GAAA,MAAA,iBAAgB;AACd,MAAI,YAAY,YAAY,UAAU;AACpC,eAAY,UAAU;AACtB,UAAO,OAAO;;IAEf,CAAC,UAAU,OAAO,CAAC;CAEtB,MAAM,gBAAA,GAAA,MAAA,cACH,gBAA+B;AAC9B,MAAI,gBAAgB,YAAY,SAAS;AACvC,eAAY,UAAU;AACtB,UAAO,OAAO;;IAGlB,CAAC,OAAO,CACT;CAED,MAAM,eAAe,UAAgC;EACnD,MAAM,cAAc,QAAQ,eAAe;AAC3C,SAAO,MAAM,QAAQ,MAAM,aAAa,GACnC,MAAM,eACP,EAAE;;CAGR,MAAM,eAAe,SAAoB,aAAmC;EAC1E,MAAM,cAAc,QAAQ,eAAe;AAC3C,SAAO;GAAE,GAAG;IAAU,cAAc;GAAU;;CAGhD,MAAM,gBAAgB,QAAQ,iBAAkB,EAAE;CAOlD,MAAM,kBAAkB,YAAY,cAAc;CAClD,MAAM,6BACJ,QAAQ,0BACR,CAAC,OAAO,aACR,gBAAgB,SAAS;AAE3B,EAAA,GAAA,MAAA,iBAAgB;AACd,MAAI,2BAEF,QAAO,qBAAqB,iBAAiB,EAAE,iBAAiB,MAAM,CAAC;IAKxE,CAAC,4BAA4B,gBAAgB,OAAO,CAAC;CAExD,MAAM,aAAa,OAAO,KAAK,eAAe,EAAE,QAAQ,QAAQ,QAAQ,CAAC;CAEzE,MAAM,eAAe,OACnB,QACA,kBACG;AACH,MAAI,aAAa,YAAY,SAAS;AACpC,eAAY,UAAU;AACtB,UAAO,OAAO;;EAGhB,IAAI,iBAAiB,YAAY,WAAW,eAAe;AAE3D,SAAO,sBAAsB;AAC3B,OAAI,eAAe,oBAAoB,KACrC,QAAO;IACL,GAAG;IACH,GAAI,OAAO,cAAc,qBAAqB,aAC1C,cAAc,iBAAiB,cAAc,GAC7C,cAAc;IACnB;AAGH,UAAO,EAAE,GAAG,eAAe;IAC3B;AAEF,QAAM,OAAO,MACX,OAAO,WAAwB;AAC7B,OAAI,CAAC,gBAAgB;AACnB,qBAAiB,OAAO,YAAY;AACpC,gBAAY,UAAU;AACtB,eAAW,eAAe;;AAG5B,OAAI,CAAC,eACH,OAAM,IAAI,MAAM,oCAAoC;AAGtD,UAAO,QAAQ,UAAU,OAAO;IAC9B,OAAO;IACP,SAAS,eAAe;IACxB,SAAS,eAAe;IACxB,iBAAiB,eAAe;IAChC;IACA,QAAQ;KACN,GAAG,eAAe;KAClB,cAAc;MACZ,WAAW;MACX,GAAG,eAAe,QAAQ;MAC3B;KACF;IACF,CAAC;KAIJ;GACE;GACA;GAEA,eAAe,EAAE;GACjB,WAAW;GAEX,iBAAiB;AACf,QAAI,CAAC,eAAgB,QAAO,KAAA;IAE5B,MAAM,cAAc,OAAO,UAAU;AACrC,YAAQ,WACN,iCAAiC,aAAa,eAAe,EAC7D,KAAA,EACD;;GAIH,QAAQ,OAAO;AACb,YAAQ,UAAU,OAAO,KAAA,EAAU;AACnC,mBAAe,UAAU,OAAO,KAAA,EAAU;;GAE7C,CACF;;CAGH,MAAM,SAAS,OACb,QACA,kBACG;AACH,QAAM,aAAa,QAAQ,cAAc;;CAG3C,MAAM,mBAAA,GAAA,MAAA,wBAAsC,IAAI,KAAK,CAAC;AACtD,EAAA,GAAA,MAAA,iBAAgB;AACd,kBAAgB,QAAQ,OAAO;IAC9B,CAAC,SAAS,CAAC;AAEd,EAAA,GAAA,MAAA,iBAAgB;AACd,GAAA,GAAA,yBAAA,oCACE,OAAO,QACP,QAAQ,OACR,gBAAgB,SAChB;GACE,QAAQ,QAAQ;GAChB,QAAQ,QAAQ;AACT,YAAQ,SAAS,CAAC,KAAK,IAAI;;GAElC,eAAe,YACb,OAAO,MAAM,EACX,SACD,CAAC;GACL,CACF;IACA;EAAC,QAAQ;EAAQ,QAAQ;EAAO,OAAO;EAAQ;EAAO,CAAC;AAE1D,QAAO;EACL,IAAI,SAAS;AACX,UAAO,OAAO,UAAW,EAAE;;EAG7B,OAAO,OAAO;EACd,WAAW,OAAO;EAElB;EACA;EACA;EAEA;EACA,WAAW;EAEX,oBACE,SACA,OACwC;GACxC,MAAM,iBAAiB,eAAe,IAAI,QAAQ,GAAG,EAAE;AACvD,OAAI,kBAAkB,KACpB,QAAO;IACL,WAAW,QAAQ,MAAM,OAAO,MAAM;IACtC,gBAAgB,KAAA;IAChB,QAAQ,KAAA;IACR,eAAe,KAAA;IACf;IACD;;EAKL,IAAI,aAAyC;AAC3C,OACE,OAAO,UAAU,QACjB,mBAAmB,OAAO,UAC1B,MAAM,QAAQ,OAAO,OAAO,cAAc,CAE1C,SAAA,GAAA,4BAAA,qCACE,OAAO,OAAO,cACf;AAGH,UAAO,EAAE;;EAGX,IAAI,YAAkD;AACpD,WAAA,GAAA,4BAAA,mBAAwC,OAAO,OAAO;;EAGxD,IAAI,WAA0B;AAC5B,OAAI,CAAC,OAAO,OAAQ,QAAO,EAAE;AAC7B,WAAA,GAAA,4BAAA,wBACE,YAAY,OAAO,OAAO,CAC3B;;EAGH,IAAI,YAAY;AACd,OAAI,CAAC,OAAO,OAAQ,QAAO,EAAE;AAE7B,WAAA,GAAA,+BAAA,yBADa,YAAY,OAAO,OAAO,CACW;;EAGpD,aAAa,SAAS;AACpB,OAAI,CAAC,OAAO,OAAQ,QAAO,EAAE;AAG7B,WAAA,GAAA,+BAAA,yBAFa,YAAY,OAAO,OAAO,CACyB,CAC5C,QAAQ,OAAO,GAAG,UAAU,OAAO,QAAQ,GAAG;;EAGpE,IAAI,YAAY;AACd,UAAO,OAAO,cAAc;;EAG9B,IAAI,kBAAkB;AACpB,UAAO,OAAO,oBAAoB;;EAGpC,YAAY,YAAoB;AAC9B,UAAO,OAAO,YAAY,WAAW;;EAGvC,mBAAmB,MAAc;AAC/B,UAAO,OAAO,mBAAmB,KAAK;;EAGxC,sBAAsB,WAAmB;AACvC,UAAO,OAAO,sBAAsB,UAAU;;EAGhD,OAAO;GACL,IAAI,UAAU;AACZ,WAAO,EAAE;;GAEX,IAAI,OAAO;AACT,WAAO;;GAET,MAAM,SAAS;AACb,WAAO;;GAET,MAAM,QAAQ;GACf;EACF"}
+{"version":3,"file":"stream.custom.cjs","names":["MessageTupleManager","StreamManager","toMessageClass","useControllableThreadId"],"sources":["../src/stream.custom.tsx"],"sourcesContent":["/* __LC_ALLOW_ENTRYPOINT_SIDE_EFFECTS__ */\n\n\"use client\";\n\nimport {\n  useCallback,\n  useEffect,\n  useRef,\n  useState,\n  useSyncExternalStore,\n} from \"react\";\nimport {\n  StreamManager,\n  MessageTupleManager,\n  extractInterrupts,\n  FetchStreamTransport,\n  toMessageClass,\n  ensureMessageInstances,\n  type EventStreamEvent,\n  type GetUpdateType,\n  type GetCustomEventType,\n  type GetInterruptType,\n  type GetToolCallsType,\n  type GetConfigurableType,\n  type AnyStreamCustomOptions,\n  type CustomSubmitOptions,\n  type MessageMetadata,\n} from \"@langchain/langgraph-sdk/ui\";\nimport { getToolCallsWithResults } from \"@langchain/langgraph-sdk/utils\";\nimport type { BagTemplate, Message, Interrupt } from \"@langchain/langgraph-sdk\";\nimport {\n  isHeadlessToolInterrupt,\n  handleHeadlessToolInterrupt,\n} from \"@langchain/langgraph-sdk\";\nimport type { BaseMessage } from \"@langchain/core/messages\";\nimport { useControllableThreadId } from \"./thread.js\";\nimport type { UseStreamCustom } from \"./types.js\";\n\nexport { FetchStreamTransport };\n\nexport function useStreamCustom<\n  StateType extends Record<string, unknown> = Record<string, unknown>,\n  Bag extends BagTemplate = BagTemplate,\n>(\n  options: AnyStreamCustomOptions<StateType, Bag>,\n): UseStreamCustom<StateType, Bag> {\n  type UpdateType = GetUpdateType<Bag, StateType>;\n  type CustomType = GetCustomEventType<Bag>;\n  type InterruptType = GetInterruptType<Bag>;\n  type ConfigurableType = GetConfigurableType<Bag>;\n  type ToolCallType = GetToolCallsType<StateType>;\n\n  const [messageManager] = useState(() => new MessageTupleManager());\n  const [stream] = useState(\n    () =>\n      new StreamManager<StateType, Bag>(messageManager, {\n        throttle: options.throttle ?? false,\n        subagentToolNames: options.subagentToolNames,\n        filterSubagentMessages: options.filterSubagentMessages,\n        toMessage: options.toMessage ?? toMessageClass,\n      }),\n  );\n\n  useSyncExternalStore(\n    stream.subscribe,\n    stream.getSnapshot,\n    stream.getSnapshot,\n  );\n\n  const [branch, _setBranch] = useState(\"\");\n\n  const [threadId, onThreadId] = useControllableThreadId(options);\n  const threadIdRef = useRef<string | null>(threadId);\n\n  // Cancel the stream if thread ID has changed\n  useEffect(() => {\n    if (threadIdRef.current !== threadId) {\n      threadIdRef.current = threadId;\n      stream.clear();\n    }\n  }, [threadId, stream]);\n\n  const switchThread = useCallback(\n    (newThreadId: string | null) => {\n      if (newThreadId !== threadIdRef.current) {\n        threadIdRef.current = newThreadId;\n        stream.clear();\n      }\n    },\n    [stream],\n  );\n\n  const getMessages = (value: StateType): Message[] => {\n    const messagesKey = options.messagesKey ?? \"messages\";\n    return Array.isArray(value[messagesKey])\n      ? (value[messagesKey] as Message[])\n      : [];\n  };\n\n  const setMessages = (current: StateType, messages: Message[]): StateType => {\n    const messagesKey = options.messagesKey ?? \"messages\";\n    return { ...current, [messagesKey]: messages };\n  };\n\n  const historyValues = options.initialValues ?? ({} as StateType);\n\n  // Reconstruct subagents from initialValues when:\n  // 1. Subagent filtering is enabled\n  // 2. Not currently streaming\n  // 3. initialValues has messages\n  // This ensures subagent visualization works with cached/persisted state\n  const historyMessages = getMessages(historyValues);\n  const shouldReconstructSubagents =\n    options.filterSubagentMessages &&\n    !stream.isLoading &&\n    historyMessages.length > 0;\n\n  useEffect(() => {\n    if (shouldReconstructSubagents) {\n      // skipIfPopulated: true ensures we don't overwrite subagents from active streaming\n      stream.reconstructSubagents(historyMessages, { skipIfPopulated: true });\n    }\n    // We intentionally only run this when shouldReconstructSubagents changes\n    // to avoid unnecessary reconstructions during streaming\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n  }, [shouldReconstructSubagents, historyMessages.length]);\n\n  const stop = () => stream.stop(historyValues, { onStop: options.onStop });\n\n  const submitDirect = async (\n    values: UpdateType | null | undefined,\n    submitOptions?: CustomSubmitOptions<StateType, ConfigurableType>,\n  ) => {\n    if (threadId !== threadIdRef.current) {\n      threadIdRef.current = threadId;\n      stream.clear();\n    }\n\n    let usableThreadId = threadIdRef.current ?? submitOptions?.threadId;\n\n    stream.setStreamValues(() => {\n      if (submitOptions?.optimisticValues != null) {\n        return {\n          ...historyValues,\n          ...(typeof submitOptions.optimisticValues === \"function\"\n            ? submitOptions.optimisticValues(historyValues)\n            : submitOptions.optimisticValues),\n        };\n      }\n\n      return { ...historyValues };\n    });\n\n    await stream.start(\n      async (signal: AbortSignal) => {\n        if (!usableThreadId) {\n          usableThreadId = crypto.randomUUID();\n          threadIdRef.current = usableThreadId;\n          onThreadId(usableThreadId);\n        }\n\n        if (!usableThreadId) {\n          throw new Error(\"Failed to obtain valid thread ID.\");\n        }\n\n        return options.transport.stream({\n          input: values,\n          context: submitOptions?.context,\n          command: submitOptions?.command,\n          signal,\n          config: {\n            ...submitOptions?.config,\n            configurable: {\n              thread_id: usableThreadId,\n              ...submitOptions?.config?.configurable,\n            } as unknown as GetConfigurableType<Bag>,\n          },\n        }) as Promise<\n          AsyncGenerator<EventStreamEvent<StateType, UpdateType, CustomType>>\n        >;\n      },\n      {\n        getMessages,\n        setMessages,\n\n        initialValues: {} as StateType,\n        callbacks: options,\n\n        onSuccess: () => undefined,\n        onError(error) {\n          options.onError?.(error, undefined);\n          submitOptions?.onError?.(error, undefined);\n        },\n      },\n    );\n  };\n\n  const submit = async (\n    values: UpdateType | null | undefined,\n    submitOptions?: CustomSubmitOptions<StateType, ConfigurableType>,\n  ) => {\n    await submitDirect(values, submitOptions);\n  };\n\n  // Headless tools handling\n  const toolsRef = useRef(options.tools);\n  toolsRef.current = options.tools;\n\n  const onToolRef = useRef(options.onTool);\n  onToolRef.current = options.onTool;\n\n  const handledBrowserToolsRef = useRef<Set<string>>(new Set());\n\n  useEffect(() => {\n    handledBrowserToolsRef.current.clear();\n  }, [threadId]);\n\n  useEffect(() => {\n    const tools = toolsRef.current;\n    if (!tools?.length) return;\n    if (!stream.values) return;\n\n    const interrupts = stream.values.__interrupt__;\n    if (!Array.isArray(interrupts) || interrupts.length === 0) return;\n\n    for (const interrupt of interrupts) {\n      if (!isHeadlessToolInterrupt(interrupt.value)) continue;\n\n      const interruptId = interrupt.id ?? interrupt.value.toolCall.id ?? \"\";\n      if (handledBrowserToolsRef.current.has(interruptId)) continue;\n      handledBrowserToolsRef.current.add(interruptId);\n\n      void handleHeadlessToolInterrupt(\n        interrupt.value,\n        tools,\n        onToolRef.current,\n      ).then((result) => {\n        void submit(null, {\n          command: {\n            resume: result.toolCallId\n              ? { [result.toolCallId]: result.value }\n              : result.value,\n          },\n        });\n      });\n    }\n  }, [stream.values, submit]);\n\n  return {\n    get values() {\n      return stream.values ?? ({} as StateType);\n    },\n\n    error: stream.error,\n    isLoading: stream.isLoading,\n\n    stop,\n    submit,\n    switchThread,\n\n    branch,\n    setBranch: _setBranch,\n\n    getMessagesMetadata(\n      message: BaseMessage,\n      index?: number,\n    ): MessageMetadata<StateType> | undefined {\n      const streamMetadata = messageManager.get(message.id)?.metadata;\n      if (streamMetadata != null) {\n        return {\n          messageId: message.id ?? String(index),\n          firstSeenState: undefined,\n          branch: undefined,\n          branchOptions: undefined,\n          streamMetadata,\n        } as MessageMetadata<StateType>;\n      }\n      return undefined;\n    },\n\n    get interrupts(): Interrupt<InterruptType>[] {\n      if (\n        stream.values != null &&\n        \"__interrupt__\" in stream.values &&\n        Array.isArray(stream.values.__interrupt__)\n      ) {\n        const valueInterrupts = stream.values.__interrupt__;\n        if (valueInterrupts.length === 0) return [{ when: \"breakpoint\" }];\n        return valueInterrupts;\n      }\n\n      return [];\n    },\n\n    get interrupt(): Interrupt<InterruptType> | undefined {\n      return extractInterrupts<InterruptType>(stream.values);\n    },\n\n    get messages(): BaseMessage[] {\n      if (!stream.values) return [];\n      return ensureMessageInstances(\n        getMessages(stream.values),\n      ) as BaseMessage[];\n    },\n\n    get toolCalls() {\n      if (!stream.values) return [];\n      const msgs = getMessages(stream.values);\n      return getToolCallsWithResults<ToolCallType>(msgs);\n    },\n\n    getToolCalls(message) {\n      if (!stream.values) return [];\n      const msgs = getMessages(stream.values);\n      const allToolCalls = getToolCallsWithResults<ToolCallType>(msgs);\n      return allToolCalls.filter((tc) => tc.aiMessage.id === message.id);\n    },\n\n    get subagents() {\n      return stream.getSubagents();\n    },\n\n    get activeSubagents() {\n      return stream.getActiveSubagents();\n    },\n\n    getSubagent(toolCallId: string) {\n      return stream.getSubagent(toolCallId);\n    },\n\n    getSubagentsByType(type: string) {\n      return stream.getSubagentsByType(type);\n    },\n\n    getSubagentsByMessage(messageId: string) {\n      return stream.getSubagentsByMessage(messageId);\n    },\n\n    queue: {\n      get entries() {\n        return [];\n      },\n      get size() {\n        return 0;\n      },\n      async cancel() {\n        return false;\n      },\n      async clear() {},\n    },\n  };\n}\n"],"mappings":";;;;;;;;AAwCA,SAAgB,gBAId,SACiC;CAOjC,MAAM,CAAC,mBAAA,GAAA,MAAA,gBAAiC,IAAIA,4BAAAA,qBAAqB,CAAC;CAClE,MAAM,CAAC,WAAA,GAAA,MAAA,gBAEH,IAAIC,4BAAAA,cAA8B,gBAAgB;EAChD,UAAU,QAAQ,YAAY;EAC9B,mBAAmB,QAAQ;EAC3B,wBAAwB,QAAQ;EAChC,WAAW,QAAQ,aAAaC,4BAAAA;EACjC,CAAC,CACL;AAED,EAAA,GAAA,MAAA,sBACE,OAAO,WACP,OAAO,aACP,OAAO,YACR;CAED,MAAM,CAAC,QAAQ,eAAA,GAAA,MAAA,UAAuB,GAAG;CAEzC,MAAM,CAAC,UAAU,cAAcC,eAAAA,wBAAwB,QAAQ;CAC/D,MAAM,eAAA,GAAA,MAAA,QAAoC,SAAS;AAGnD,EAAA,GAAA,MAAA,iBAAgB;AACd,MAAI,YAAY,YAAY,UAAU;AACpC,eAAY,UAAU;AACtB,UAAO,OAAO;;IAEf,CAAC,UAAU,OAAO,CAAC;CAEtB,MAAM,gBAAA,GAAA,MAAA,cACH,gBAA+B;AAC9B,MAAI,gBAAgB,YAAY,SAAS;AACvC,eAAY,UAAU;AACtB,UAAO,OAAO;;IAGlB,CAAC,OAAO,CACT;CAED,MAAM,eAAe,UAAgC;EACnD,MAAM,cAAc,QAAQ,eAAe;AAC3C,SAAO,MAAM,QAAQ,MAAM,aAAa,GACnC,MAAM,eACP,EAAE;;CAGR,MAAM,eAAe,SAAoB,aAAmC;EAC1E,MAAM,cAAc,QAAQ,eAAe;AAC3C,SAAO;GAAE,GAAG;IAAU,cAAc;GAAU;;CAGhD,MAAM,gBAAgB,QAAQ,iBAAkB,EAAE;CAOlD,MAAM,kBAAkB,YAAY,cAAc;CAClD,MAAM,6BACJ,QAAQ,0BACR,CAAC,OAAO,aACR,gBAAgB,SAAS;AAE3B,EAAA,GAAA,MAAA,iBAAgB;AACd,MAAI,2BAEF,QAAO,qBAAqB,iBAAiB,EAAE,iBAAiB,MAAM,CAAC;IAKxE,CAAC,4BAA4B,gBAAgB,OAAO,CAAC;CAExD,MAAM,aAAa,OAAO,KAAK,eAAe,EAAE,QAAQ,QAAQ,QAAQ,CAAC;CAEzE,MAAM,eAAe,OACnB,QACA,kBACG;AACH,MAAI,aAAa,YAAY,SAAS;AACpC,eAAY,UAAU;AACtB,UAAO,OAAO;;EAGhB,IAAI,iBAAiB,YAAY,WAAW,eAAe;AAE3D,SAAO,sBAAsB;AAC3B,OAAI,eAAe,oBAAoB,KACrC,QAAO;IACL,GAAG;IACH,GAAI,OAAO,cAAc,qBAAqB,aAC1C,cAAc,iBAAiB,cAAc,GAC7C,cAAc;IACnB;AAGH,UAAO,EAAE,GAAG,eAAe;IAC3B;AAEF,QAAM,OAAO,MACX,OAAO,WAAwB;AAC7B,OAAI,CAAC,gBAAgB;AACnB,qBAAiB,OAAO,YAAY;AACpC,gBAAY,UAAU;AACtB,eAAW,eAAe;;AAG5B,OAAI,CAAC,eACH,OAAM,IAAI,MAAM,oCAAoC;AAGtD,UAAO,QAAQ,UAAU,OAAO;IAC9B,OAAO;IACP,SAAS,eAAe;IACxB,SAAS,eAAe;IACxB;IACA,QAAQ;KACN,GAAG,eAAe;KAClB,cAAc;MACZ,WAAW;MACX,GAAG,eAAe,QAAQ;MAC3B;KACF;IACF,CAAC;KAIJ;GACE;GACA;GAEA,eAAe,EAAE;GACjB,WAAW;GAEX,iBAAiB,KAAA;GACjB,QAAQ,OAAO;AACb,YAAQ,UAAU,OAAO,KAAA,EAAU;AACnC,mBAAe,UAAU,OAAO,KAAA,EAAU;;GAE7C,CACF;;CAGH,MAAM,SAAS,OACb,QACA,kBACG;AACH,QAAM,aAAa,QAAQ,cAAc;;CAI3C,MAAM,YAAA,GAAA,MAAA,QAAkB,QAAQ,MAAM;AACtC,UAAS,UAAU,QAAQ;CAE3B,MAAM,aAAA,GAAA,MAAA,QAAmB,QAAQ,OAAO;AACxC,WAAU,UAAU,QAAQ;CAE5B,MAAM,0BAAA,GAAA,MAAA,wBAA6C,IAAI,KAAK,CAAC;AAE7D,EAAA,GAAA,MAAA,iBAAgB;AACd,yBAAuB,QAAQ,OAAO;IACrC,CAAC,SAAS,CAAC;AAEd,EAAA,GAAA,MAAA,iBAAgB;EACd,MAAM,QAAQ,SAAS;AACvB,MAAI,CAAC,OAAO,OAAQ;AACpB,MAAI,CAAC,OAAO,OAAQ;EAEpB,MAAM,aAAa,OAAO,OAAO;AACjC,MAAI,CAAC,MAAM,QAAQ,WAAW,IAAI,WAAW,WAAW,EAAG;AAE3D,OAAK,MAAM,aAAa,YAAY;AAClC,OAAI,EAAA,GAAA,yBAAA,yBAAyB,UAAU,MAAM,CAAE;GAE/C,MAAM,cAAc,UAAU,MAAM,UAAU,MAAM,SAAS,MAAM;AACnE,OAAI,uBAAuB,QAAQ,IAAI,YAAY,CAAE;AACrD,0BAAuB,QAAQ,IAAI,YAAY;AAE/C,IAAA,GAAA,yBAAA,6BACE,UAAU,OACV,OACA,UAAU,QACX,CAAC,MAAM,WAAW;AACZ,WAAO,MAAM,EAChB,SAAS,EACP,QAAQ,OAAO,aACX,GAAG,OAAO,aAAa,OAAO,OAAO,GACrC,OAAO,OACZ,EACF,CAAC;KACF;;IAEH,CAAC,OAAO,QAAQ,OAAO,CAAC;AAE3B,QAAO;EACL,IAAI,SAAS;AACX,UAAO,OAAO,UAAW,EAAE;;EAG7B,OAAO,OAAO;EACd,WAAW,OAAO;EAElB;EACA;EACA;EAEA;EACA,WAAW;EAEX,oBACE,SACA,OACwC;GACxC,MAAM,iBAAiB,eAAe,IAAI,QAAQ,GAAG,EAAE;AACvD,OAAI,kBAAkB,KACpB,QAAO;IACL,WAAW,QAAQ,MAAM,OAAO,MAAM;IACtC,gBAAgB,KAAA;IAChB,QAAQ,KAAA;IACR,eAAe,KAAA;IACf;IACD;;EAKL,IAAI,aAAyC;AAC3C,OACE,OAAO,UAAU,QACjB,mBAAmB,OAAO,UAC1B,MAAM,QAAQ,OAAO,OAAO,cAAc,EAC1C;IACA,MAAM,kBAAkB,OAAO,OAAO;AACtC,QAAI,gBAAgB,WAAW,EAAG,QAAO,CAAC,EAAE,MAAM,cAAc,CAAC;AACjE,WAAO;;AAGT,UAAO,EAAE;;EAGX,IAAI,YAAkD;AACpD,WAAA,GAAA,4BAAA,mBAAwC,OAAO,OAAO;;EAGxD,IAAI,WAA0B;AAC5B,OAAI,CAAC,OAAO,OAAQ,QAAO,EAAE;AAC7B,WAAA,GAAA,4BAAA,wBACE,YAAY,OAAO,OAAO,CAC3B;;EAGH,IAAI,YAAY;AACd,OAAI,CAAC,OAAO,OAAQ,QAAO,EAAE;AAE7B,WAAA,GAAA,+BAAA,yBADa,YAAY,OAAO,OAAO,CACW;;EAGpD,aAAa,SAAS;AACpB,OAAI,CAAC,OAAO,OAAQ,QAAO,EAAE;AAG7B,WAAA,GAAA,+BAAA,yBAFa,YAAY,OAAO,OAAO,CACyB,CAC5C,QAAQ,OAAO,GAAG,UAAU,OAAO,QAAQ,GAAG;;EAGpE,IAAI,YAAY;AACd,UAAO,OAAO,cAAc;;EAG9B,IAAI,kBAAkB;AACpB,UAAO,OAAO,oBAAoB;;EAGpC,YAAY,YAAoB;AAC9B,UAAO,OAAO,YAAY,WAAW;;EAGvC,mBAAmB,MAAc;AAC/B,UAAO,OAAO,mBAAmB,KAAK;;EAGxC,sBAAsB,WAAmB;AACvC,UAAO,OAAO,sBAAsB,UAAU;;EAGhD,OAAO;GACL,IAAI,UAAU;AACZ,WAAO,EAAE;;GAEX,IAAI,OAAO;AACT,WAAO;;GAET,MAAM,SAAS;AACb,WAAO;;GAET,MAAM,QAAQ;GACf;EACF"}