@assistant-ui/react-langchain 0.0.8 → 0.0.10

package/dist/useStreamRuntime.d.ts

@@ -1,3 +1,4 @@
+/// <reference types="@assistant-ui/core/store" />
 import { LangChainBaseMessage } from "./types.js";
 import { UseStreamOptions } from "@langchain/react";
 import { AttachmentAdapter, FeedbackAdapter, RemoteThreadListAdapter, SpeechSynthesisAdapter } from "@assistant-ui/core";

package/dist/useStreamRuntime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useStreamRuntime.d.ts","names":[],"sources":["../src/useStreamRuntime.ts"],"mappings":";;;;;;KAkDK,4BAAA;EACH,KAAA,GAAQ,cAAA;EACR,QAAA;IAEM,WAAA,GAAc,iBAAA;IACd,MAAA,GAAS,sBAAA;IACT,QAAA,GAAW,eAAA;EAAA;EADF;;;;;;EAUf,0BAAA;EAdA;;;;EAmBA,0BAAA;EAfM;;;;EAoBN,0BAAA,GAA6B,uBAAA,cAL7B;EAOA,MAAA,UAAgB,OAAA;IAAU,UAAA;EAAA,iBAAV;EAEhB,MAAA,KAAW,QAAA,aAAqB,OAAA;AAAA;AAAA,KAsBtB,uBAAA,GAA0B,gBAAA,mBAClC,CAAA,SAAU,gBAAA,GACR,CAAA,GAAI,4BAAA;;;AAxB+B;AAsBzC;;;;;;;;;;;;;;;;AAEsC;AA4OtC;;;;cAAa,gBAAA,GAAoB,UAAA,EAAY,uBAAuB,kCAAA,gBAAA;;;;cAgCvD,0BAAA;EAMZ,KAAA;AAAA;;;;AAOD;;cAAa,kBAAA,SAGT,MAAA,EAAQ,MAAA,sCACR,OAAA,GAAU,MAAA,sBAAuB,OAAA;;;;;;cAaxB,gBAAA,SAGT,QAAA,WAAmB,oBAAA,IACnB,OAAA,GAAU,MAAA,sBAAuB,OAAA;;;;;;AAjBA;AAarC;cAoBa,uBAAA,SAEH,OAAA,EAAS,MAAA,sBAAuB,OAAA;;;;;;;;;;;;iBAc1B,iBAAA,GAAA,CAAqB,GAAA,WAAc,CAAC;AAAA,iBACpC,iBAAA,GAAA,CAAqB,GAAA,UAAa,YAAA,EAAc,CAAA,GAAI,CAAC"}
+{"version":3,"file":"useStreamRuntime.d.ts","names":[],"sources":["../src/useStreamRuntime.ts"],"mappings":";;;;;;KAkDK,4BAAA;EACH,KAAA,GAAQ,cAAA;EACR,QAAA;IAEM,WAAA,GAAc,iBAAA;IACd,MAAA,GAAS,sBAAA;IACT,QAAA,GAAW,eAAA;EAAA;EADF;;;;;;EAUf,0BAAA;EAdA;;;;EAmBA,0BAAA;EAfM;;;;EAoBN,0BAAA,GAA6B,uBAAA,cAL7B;EAOA,MAAA,UAAgB,OAAA;IAAU,UAAA;EAAA,iBAAV;EAEhB,MAAA,KAAW,QAAA,aAAqB,OAAA;AAAA;AAAA,KAsBtB,uBAAA,GAA0B,gBAAA,mBAClC,CAAA,SAAU,gBAAA,GACR,CAAA,GAAI,4BAAA;;;AAxB+B;AAsBzC;;;;;;;;;;;;;;;;AAEsC;AAyMtC;;;;cAAa,gBAAA,GAAoB,UAAA,EAAY,uBAAuB,kCAAA,gBAAA;;;;cAgCvD,0BAAA;EAMZ,KAAA;AAAA;;;;AAOD;;cAAa,kBAAA,SAGT,MAAA,EAAQ,MAAA,sCACR,OAAA,GAAU,MAAA,sBAAuB,OAAA;;;;;;cAaxB,gBAAA,SAGT,QAAA,WAAmB,oBAAA,IACnB,OAAA,GAAU,MAAA,sBAAuB,OAAA;;;;;;AAjBA;AAarC;cAoBa,uBAAA,SAEH,OAAA,EAAS,MAAA,sBAAuB,OAAA;;;;;;;;;;;;iBAc1B,iBAAA,GAAA,CAAqB,GAAA,WAAc,CAAC;AAAA,iBACpC,iBAAA,GAAA,CAAqB,GAAA,UAAa,YAAA,EAAc,CAAA,GAAI,CAAC"}

package/dist/useStreamRuntime.js

@@ -1,7 +1,7 @@
 "use client";
 import { convertLangChainBaseMessage, getMessageType } from "./convertMessages.js";
 import { useMemo, useRef, useState } from "react";
-import { useCloudThreadListAdapter, useExternalMessageConverter, useExternalStoreRuntime, useRemoteThreadListRuntime, useToolInvocations } from "@assistant-ui/core/react";
+import { useCloudThreadListAdapter, useExternalMessageConverter, useExternalStoreRuntime, useRemoteThreadListRuntime } from "@assistant-ui/core/react";
 import { useAui, useAuiState } from "@assistant-ui/store";
 import { useStream } from "@langchain/react";
 //#region src/useStreamRuntime.ts
@@ -65,29 +65,9 @@ const useStreamThreadRuntime = (options) => {
 		messages: stream.messages,
 		isRunning: effectiveIsRunning
 	});
-	const [runtimeRef] = useState(() => ({ get current() {
-		return runtime;
-	} }));
 	const streamRef = useRef(stream);
 	streamRef.current = stream;
-	const toolInvocations = useToolInvocations({
-		state: {
-			messages: threadMessages,
-			isRunning: effectiveIsRunning
-		},
-		getTools: () => runtimeRef.current.thread.getModelContext().tools,
-		onResult: (command) => {
-			if (command.type === "add-tool-result") streamRef.current.submit({ [messagesKey]: [{
-				type: "tool",
-				name: command.toolName,
-				tool_call_id: command.toolCallId,
-				content: JSON.stringify(command.result),
-				status: command.isError ? "error" : "success"
-			}] }, {});
-		},
-		setToolStatuses
-	});
-	const runtime = useExternalStoreRuntime({
+	return useExternalStoreRuntime({
 		isRunning: effectiveIsRunning,
 		messages: threadMessages,
 		adapters,
@@ -105,8 +85,9 @@ const useStreamThreadRuntime = (options) => {
 			stream.values,
 			messagesKey
 		]),
+		unstable_enableToolInvocations: true,
+		setToolStatuses,
 		onNew: async (msg) => {
-			await toolInvocations.abort();
 			const content = getMessageContent(msg);
 			const cancellations = autoCancelPendingToolCalls !== false ? getPendingToolCalls(streamRef.current.messages).map((t) => ({
 				type: "tool",
@@ -132,10 +113,8 @@ const useStreamThreadRuntime = (options) => {
 		},
 		onCancel: unstable_allowCancellation !== false ? async () => {
 			await stream.stop();
-			await toolInvocations.abort();
 		} : void 0
 	});
-	return runtime;
 };
 /**
 * Creates an assistant-ui runtime backed by LangChain's `useStream` hook.

package/dist/useStreamRuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useStreamRuntime.js","names":["_exhaustiveCheck"],"sources":["../src/useStreamRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport { useMemo, useRef, useState } from \"react\";\nimport type {\n  AppendMessage,\n  AttachmentAdapter,\n  FeedbackAdapter,\n  RemoteThreadListAdapter,\n  SpeechSynthesisAdapter,\n} from \"@assistant-ui/core\";\nimport {\n  useCloudThreadListAdapter,\n  useExternalStoreRuntime,\n  useExternalMessageConverter,\n  useRemoteThreadListRuntime,\n  useToolInvocations,\n  type ToolExecutionStatus,\n} from \"@assistant-ui/core/react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport type { AssistantCloud } from \"assistant-cloud\";\nimport { useStream, type UseStreamOptions } from \"@langchain/react\";\nimport type { LangChainBaseMessage, LangChainToolCall } from \"./types\";\nimport { convertLangChainBaseMessage, getMessageType } from \"./convertMessages\";\n\nconst symbolLangChainRuntimeExtras = Symbol(\"langchain-runtime-extras\");\n\ntype LangChainRuntimeExtras = {\n  [symbolLangChainRuntimeExtras]: true;\n  interrupt: { value?: unknown } | undefined;\n  interrupts: readonly { value?: unknown }[];\n  submit: (\n    values: Record<string, unknown> | null | undefined,\n    options?: Record<string, unknown>,\n  ) => Promise<void>;\n  values: Record<string, unknown>;\n  messagesKey: string;\n};\n\nconst asLangChainRuntimeExtras = (extras: unknown): LangChainRuntimeExtras => {\n  if (\n    typeof extras !== \"object\" ||\n    extras == null ||\n    !(symbolLangChainRuntimeExtras in extras)\n  )\n    throw new Error(\n      \"This method can only be called when you are using useStreamRuntime\",\n    );\n  return extras as LangChainRuntimeExtras;\n};\n\ntype LangChainRuntimeExtraOptions = {\n  cloud?: AssistantCloud | undefined;\n  adapters?:\n    | {\n        attachments?: AttachmentAdapter | undefined;\n        speech?: SpeechSynthesisAdapter | undefined;\n        feedback?: FeedbackAdapter | undefined;\n      }\n    | undefined;\n  /**\n   * When the user sends a new message while previous tool calls are\n   * still pending, automatically submit `tool` messages that cancel\n   * them so the agent's tool-call accounting stays consistent.\n   * Defaults to `true`.\n   */\n  autoCancelPendingToolCalls?: boolean | undefined;\n  /**\n   * Routes the Cancel button's click to `useStream().stop()`. On by\n   * default. Pass `false` to disable the Cancel button.\n   */\n  unstable_allowCancellation?: boolean | undefined;\n  /**\n   * Custom `RemoteThreadListAdapter`. When provided, replaces the\n   * cloud-backed thread list adapter.\n   */\n  unstable_threadListAdapter?: RemoteThreadListAdapter | undefined;\n  /** Custom thread-creation hook, forwarded to the cloud adapter. */\n  create?: (() => Promise<{ externalId: string | undefined }>) | undefined;\n  /** Custom thread-deletion hook, forwarded to the cloud adapter. */\n  delete?: ((threadId: string) => Promise<void>) | undefined;\n};\n\nconst getPendingToolCalls = (\n  messages: readonly LangChainBaseMessage[],\n): LangChainToolCall[] => {\n  const pending = new Map<string, LangChainToolCall>();\n  for (const m of messages) {\n    const type = getMessageType(m);\n    if (type === \"ai\") {\n      for (const tc of m.tool_calls ?? []) pending.set(tc.id, tc);\n    } else if (type === \"tool\" && m.tool_call_id) {\n      pending.delete(m.tool_call_id);\n    }\n  }\n  return [...pending.values()];\n};\n\n// Distribute the intersection through the union arms of `UseStreamOptions`\n// (`AgentServerOptions | CustomAdapterOptions`). Writing `UseStreamOptions & X`\n// directly collapses arm tracking, so `Omit<…, \"cloud\">` and the like would\n// produce a flattened structural type that no longer matches either arm.\nexport type UseStreamRuntimeOptions = UseStreamOptions extends infer O\n  ? O extends UseStreamOptions\n    ? O & LangChainRuntimeExtraOptions\n    : never\n  : never;\n\nconst getMessageContent = (msg: AppendMessage) => {\n  const allContent = [\n    ...msg.content,\n    ...(msg.attachments?.flatMap((a) => a.content) ?? []),\n  ];\n\n  const hasNonText = allContent.some(\n    (part) => part.type === \"file\" || part.type === \"image\",\n  );\n  const hasText = allContent.some((part) => part.type === \"text\");\n  if (hasNonText && !hasText) {\n    allContent.unshift({ type: \"text\", text: \" \" });\n  }\n\n  const content = allContent.map((part) => {\n    const type = part.type;\n    switch (type) {\n      case \"text\":\n        return { type: \"text\" as const, text: part.text };\n      case \"image\":\n        return { type: \"image_url\" as const, image_url: { url: part.image } };\n      case \"file\":\n        return {\n          type: \"file\" as const,\n          data: part.data,\n          mime_type: part.mimeType,\n          metadata: { filename: part.filename ?? \"file\" },\n          source_type: \"base64\" as const,\n        };\n      case \"tool-call\":\n        throw new Error(\"Tool call appends are not supported.\");\n      default: {\n        const _exhaustiveCheck:\n          | \"reasoning\"\n          | \"source\"\n          | \"audio\"\n          | \"data\"\n          | \"generative-ui\" = type;\n        throw new Error(\n          `Unsupported append message part type: ${_exhaustiveCheck}`,\n        );\n      }\n    }\n  });\n\n  if (content.length === 1 && content[0]?.type === \"text\") {\n    return content[0].text ?? \"\";\n  }\n  return content;\n};\n\ntype DistributiveOmit<T, K extends keyof any> = T extends unknown\n  ? Omit<T, K>\n  : never;\n\nconst useStreamThreadRuntime = (\n  options: DistributiveOmit<\n    UseStreamRuntimeOptions,\n    \"cloud\" | \"unstable_threadListAdapter\" | \"create\" | \"delete\"\n  >,\n) => {\n  const { adapters, autoCancelPendingToolCalls, unstable_allowCancellation } =\n    options;\n  const messagesKey = options.messagesKey ?? \"messages\";\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const externalId = useAuiState((s) => s.threadListItem.externalId) as\n    | string\n    | null;\n  // Mutate in place rather than `{ ...options, threadId }`: spreading\n  // `UseStreamOptions` (a discriminated union on `transport`) into an object\n  // literal merges both arms' transport types, breaking arm assignment.\n  options.threadId = externalId;\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const stream = useStream(options);\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n  const hasExecutingTools = Object.values(toolStatuses).some(\n    (s) => s?.type === \"executing\",\n  );\n  const effectiveIsRunning = stream.isLoading || hasExecutingTools;\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const threadMessages = useExternalMessageConverter({\n    callback: convertLangChainBaseMessage,\n    messages: stream.messages as LangChainBaseMessage[],\n    isRunning: effectiveIsRunning,\n  });\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const [runtimeRef] = useState(() => ({\n    get current() {\n      return runtime;\n    },\n  }));\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const streamRef = useRef(stream);\n  streamRef.current = stream;\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const toolInvocations = useToolInvocations({\n    state: {\n      messages: threadMessages,\n      isRunning: effectiveIsRunning,\n    },\n    getTools: () => runtimeRef.current.thread.getModelContext().tools,\n    onResult: (command) => {\n      if (command.type === \"add-tool-result\") {\n        void streamRef.current.submit(\n          {\n            [messagesKey]: [\n              {\n                type: \"tool\",\n                name: command.toolName,\n                tool_call_id: command.toolCallId,\n                content: JSON.stringify(command.result),\n                status: command.isError ? \"error\" : \"success\",\n              },\n            ],\n          },\n          {},\n        );\n      }\n    },\n    setToolStatuses,\n  });\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const extras = useMemo(\n    (): LangChainRuntimeExtras => ({\n      [symbolLangChainRuntimeExtras]: true,\n      interrupt: stream.interrupt,\n      interrupts: stream.interrupts,\n      submit: stream.submit,\n      values: stream.values,\n      messagesKey,\n    }),\n    [\n      stream.interrupt,\n      stream.interrupts,\n      stream.submit,\n      stream.values,\n      messagesKey,\n    ],\n  );\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const runtime = useExternalStoreRuntime({\n    isRunning: effectiveIsRunning,\n    messages: threadMessages,\n    adapters,\n    extras,\n    onNew: async (msg) => {\n      await toolInvocations.abort();\n      const content = getMessageContent(msg);\n      const cancellations =\n        autoCancelPendingToolCalls !== false\n          ? getPendingToolCalls(\n              streamRef.current.messages as readonly LangChainBaseMessage[],\n            ).map((t) => ({\n              type: \"tool\" as const,\n              name: t.name,\n              tool_call_id: t.id,\n              content: JSON.stringify({ cancelled: true }),\n              status: \"error\" as const,\n            }))\n          : [];\n      await stream.submit({\n        [messagesKey]: [...cancellations, { type: \"human\", content }],\n      });\n    },\n    onAddToolResult: async ({\n      toolCallId,\n      toolName,\n      result,\n      isError,\n      artifact,\n    }) => {\n      await stream.submit({\n        [messagesKey]: [\n          {\n            type: \"tool\",\n            name: toolName,\n            tool_call_id: toolCallId,\n            content: JSON.stringify(result),\n            ...(artifact !== undefined && { artifact }),\n            status: isError ? \"error\" : \"success\",\n          },\n        ],\n      });\n    },\n    onCancel:\n      unstable_allowCancellation !== false\n        ? async () => {\n            await stream.stop();\n            await toolInvocations.abort();\n          }\n        : undefined,\n  });\n\n  return runtime;\n};\n\n/**\n * Creates an assistant-ui runtime backed by LangChain's `useStream` hook.\n * Accepts the same options as `useStream` from `@langchain/react`, plus\n * `cloud` and `adapters`.\n *\n * @example\n * ```tsx\n * import { useStreamRuntime } from \"@assistant-ui/react-langchain\";\n * import { AssistantRuntimeProvider, Thread } from \"@assistant-ui/react\";\n *\n * function App() {\n *   const runtime = useStreamRuntime({\n *     assistantId: \"agent\",\n *     apiUrl: \"http://localhost:2024\",\n *   });\n *\n *   return (\n *     <AssistantRuntimeProvider runtime={runtime}>\n *       <Thread />\n *     </AssistantRuntimeProvider>\n *   );\n * }\n * ```\n */\nexport const useStreamRuntime = (rawOptions: UseStreamRuntimeOptions) => {\n  const {\n    cloud,\n    unstable_threadListAdapter,\n    create,\n    delete: deleteFn,\n    ...options\n  } = rawOptions;\n\n  const optionsRef = useRef(options);\n  optionsRef.current = options;\n\n  const cloudAdapter = useCloudThreadListAdapter({\n    cloud,\n    create,\n    delete: deleteFn,\n  });\n  const adapter = unstable_threadListAdapter ?? cloudAdapter;\n\n  return useRemoteThreadListRuntime({\n    runtimeHook: function RuntimeHook() {\n      // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n      return useStreamThreadRuntime(optionsRef.current);\n    },\n    adapter,\n    allowNesting: true,\n  });\n};\n\n/**\n * Read the current LangGraph interrupt state from the runtime extras.\n */\nexport const useLangChainInterruptState = () => {\n  return useAuiState((s) => {\n    const extras = s.thread.extras;\n    if (!extras) return undefined;\n    return asLangChainRuntimeExtras(extras).interrupt;\n  });\n};\n\n/**\n * Returns a function to submit raw state updates to the LangGraph agent,\n * bypassing the normal message flow. Useful for sending interrupt resume\n * commands.\n */\nexport const useLangChainSubmit = () => {\n  const aui = useAui();\n  return (\n    values: Record<string, unknown> | null | undefined,\n    options?: Record<string, unknown>,\n  ) => {\n    const extras = aui.thread().getState().extras;\n    const { submit } = asLangChainRuntimeExtras(extras);\n    return submit(values, options);\n  };\n};\n\n/**\n * Submit a list of LangChain-shaped messages on the current thread.\n * Parity helper for migrating from `useLangGraphSend`. Routes to\n * `useStream().submit({ [messagesKey]: messages }, options)`.\n */\nexport const useLangChainSend = () => {\n  const aui = useAui();\n  return (\n    messages: readonly LangChainBaseMessage[],\n    options?: Record<string, unknown>,\n  ) => {\n    const { submit, messagesKey } = asLangChainRuntimeExtras(\n      aui.thread().getState().extras,\n    );\n    return submit({ [messagesKey]: messages }, options);\n  };\n};\n\n/**\n * Submit a `useStream` command (e.g. interrupt resume). Parity helper\n * for migrating from `useLangGraphSendCommand`. Note that v1's command\n * shape (`{ resume?, goto?, update? }`) differs from the legacy\n * `{ resume: string }` form — to carry a payload, use the input or\n * `stream.respond` instead.\n */\nexport const useLangChainSendCommand = () => {\n  const submit = useLangChainSubmit();\n  return (command: Record<string, unknown>) => submit(null, { command });\n};\n\n/**\n * Read a custom LangGraph state key from the current thread. Mirrors\n * `useStream().values[key]` from `@langchain/react` and updates when the\n * stream emits new state.\n *\n * @example\n * ```tsx\n * const todos = useLangChainState<Todo[]>(\"todos\");\n * const files = useLangChainState<Record<string, string>>(\"files\", {});\n * ```\n */\nexport function useLangChainState<T>(key: string): T | undefined;\nexport function useLangChainState<T>(key: string, defaultValue: T): T;\nexport function useLangChainState<T>(\n  key: string,\n  defaultValue?: T,\n): T | undefined {\n  return useAuiState((s) => {\n    const extras = s.thread.extras;\n    if (!extras) return defaultValue;\n    const value = asLangChainRuntimeExtras(extras).values[key] as T | undefined;\n    return value !== undefined ? value : defaultValue;\n  });\n}\n"],"mappings":";;;;;;;AAwBA,MAAM,+BAA+B,OAAO,0BAA0B;AActE,MAAM,4BAA4B,WAA4C;CAC5E,IACE,OAAO,WAAW,YAClB,UAAU,QACV,EAAE,gCAAgC,SAElC,MAAM,IAAI,MACR,oEACF;CACF,OAAO;AACT;AAkCA,MAAM,uBACJ,aACwB;CACxB,MAAM,0BAAU,IAAI,IAA+B;CACnD,KAAK,MAAM,KAAK,UAAU;EACxB,MAAM,OAAO,eAAe,CAAC;EAC7B,IAAI,SAAS,MACX,KAAK,MAAM,MAAM,EAAE,cAAc,CAAC,GAAG,QAAQ,IAAI,GAAG,IAAI,EAAE;OACrD,IAAI,SAAS,UAAU,EAAE,cAC9B,QAAQ,OAAO,EAAE,YAAY;CAEjC;CACA,OAAO,CAAC,GAAG,QAAQ,OAAO,CAAC;AAC7B;AAYA,MAAM,qBAAqB,QAAuB;CAChD,MAAM,aAAa,CACjB,GAAG,IAAI,SACP,GAAI,IAAI,aAAa,SAAS,MAAM,EAAE,OAAO,KAAK,CAAC,CACrD;CAEA,MAAM,aAAa,WAAW,MAC3B,SAAS,KAAK,SAAS,UAAU,KAAK,SAAS,OAClD;CACA,MAAM,UAAU,WAAW,MAAM,SAAS,KAAK,SAAS,MAAM;CAC9D,IAAI,cAAc,CAAC,SACjB,WAAW,QAAQ;EAAE,MAAM;EAAQ,MAAM;CAAI,CAAC;CAGhD,MAAM,UAAU,WAAW,KAAK,SAAS;EACvC,MAAM,OAAO,KAAK;EAClB,QAAQ,MAAR;GACE,KAAK,QACH,OAAO;IAAE,MAAM;IAAiB,MAAM,KAAK;GAAK;GAClD,KAAK,SACH,OAAO;IAAE,MAAM;IAAsB,WAAW,EAAE,KAAK,KAAK,MAAM;GAAE;GACtE,KAAK,QACH,OAAO;IACL,MAAM;IACN,MAAM,KAAK;IACX,WAAW,KAAK;IAChB,UAAU,EAAE,UAAU,KAAK,YAAY,OAAO;IAC9C,aAAa;GACf;GACF,KAAK,aACH,MAAM,IAAI,MAAM,sCAAsC;GACxD,SAOE,MAAM,IAAI,MACR,yCAAyCA,MAC3C;EAEJ;CACF,CAAC;CAED,IAAI,QAAQ,WAAW,KAAK,QAAQ,IAAI,SAAS,QAC/C,OAAO,QAAQ,GAAG,QAAQ;CAE5B,OAAO;AACT;AAMA,MAAM,0BACJ,YAIG;CACH,MAAM,EAAE,UAAU,4BAA4B,+BAC5C;CACF,MAAM,cAAc,QAAQ,eAAe;CAS3C,QAAQ,WANW,aAAa,MAAM,EAAE,eAAe,UAM3B;CAG5B,MAAM,SAAS,UAAU,OAAO;CAGhC,MAAM,CAAC,cAAc,mBAAmB,SAEtC,CAAC,CAAC;CACJ,MAAM,oBAAoB,OAAO,OAAO,YAAY,EAAE,MACnD,MAAM,GAAG,SAAS,WACrB;CACA,MAAM,qBAAqB,OAAO,aAAa;CAG/C,MAAM,iBAAiB,4BAA4B;EACjD,UAAU;EACV,UAAU,OAAO;EACjB,WAAW;CACb,CAAC;CAGD,MAAM,CAAC,cAAc,gBAAgB,EACnC,IAAI,UAAU;EACZ,OAAO;CACT,EACF,EAAE;CAGF,MAAM,YAAY,OAAO,MAAM;CAC/B,UAAU,UAAU;CAGpB,MAAM,kBAAkB,mBAAmB;EACzC,OAAO;GACL,UAAU;GACV,WAAW;EACb;EACA,gBAAgB,WAAW,QAAQ,OAAO,gBAAgB,EAAE;EAC5D,WAAW,YAAY;GACrB,IAAI,QAAQ,SAAS,mBACnB,UAAe,QAAQ,OACrB,GACG,cAAc,CACb;IACE,MAAM;IACN,MAAM,QAAQ;IACd,cAAc,QAAQ;IACtB,SAAS,KAAK,UAAU,QAAQ,MAAM;IACtC,QAAQ,QAAQ,UAAU,UAAU;GACtC,CACF,EACF,GACA,CAAC,CACH;EAEJ;EACA;CACF,CAAC;CAsBD,MAAM,UAAU,wBAAwB;EACtC,WAAW;EACX,UAAU;EACV;EACA,QAvBa,eACkB;IAC5B,+BAA+B;GAChC,WAAW,OAAO;GAClB,YAAY,OAAO;GACnB,QAAQ,OAAO;GACf,QAAQ,OAAO;GACf;EACF,IACA;GACE,OAAO;GACP,OAAO;GACP,OAAO;GACP,OAAO;GACP;EACF,CAQK;EACL,OAAO,OAAO,QAAQ;GACpB,MAAM,gBAAgB,MAAM;GAC5B,MAAM,UAAU,kBAAkB,GAAG;GACrC,MAAM,gBACJ,+BAA+B,QAC3B,oBACE,UAAU,QAAQ,QACpB,EAAE,KAAK,OAAO;IACZ,MAAM;IACN,MAAM,EAAE;IACR,cAAc,EAAE;IAChB,SAAS,KAAK,UAAU,EAAE,WAAW,KAAK,CAAC;IAC3C,QAAQ;GACV,EAAE,IACF,CAAC;GACP,MAAM,OAAO,OAAO,GACjB,cAAc,CAAC,GAAG,eAAe;IAAE,MAAM;IAAS;GAAQ,CAAC,EAC9D,CAAC;EACH;EACA,iBAAiB,OAAO,EACtB,YACA,UACA,QACA,SACA,eACI;GACJ,MAAM,OAAO,OAAO,GACjB,cAAc,CACb;IACE,MAAM;IACN,MAAM;IACN,cAAc;IACd,SAAS,KAAK,UAAU,MAAM;IAC9B,GAAI,aAAa,KAAA,KAAa,EAAE,SAAS;IACzC,QAAQ,UAAU,UAAU;GAC9B,CACF,EACF,CAAC;EACH;EACA,UACE,+BAA+B,QAC3B,YAAY;GACV,MAAM,OAAO,KAAK;GAClB,MAAM,gBAAgB,MAAM;EAC9B,IACA,KAAA;CACR,CAAC;CAED,OAAO;AACT;;;;;;;;;;;;;;;;;;;;;;;;;AA0BA,MAAa,oBAAoB,eAAwC;CACvE,MAAM,EACJ,OACA,4BACA,QACA,QAAQ,UACR,GAAG,YACD;CAEJ,MAAM,aAAa,OAAO,OAAO;CACjC,WAAW,UAAU;CAErB,MAAM,eAAe,0BAA0B;EAC7C;EACA;EACA,QAAQ;CACV,CAAC;CAGD,OAAO,2BAA2B;EAChC,aAAa,SAAS,cAAc;GAElC,OAAO,uBAAuB,WAAW,OAAO;EAClD;EACA,SAPc,8BAA8B;EAQ5C,cAAc;CAChB,CAAC;AACH;;;;AAKA,MAAa,mCAAmC;CAC9C,OAAO,aAAa,MAAM;EACxB,MAAM,SAAS,EAAE,OAAO;EACxB,IAAI,CAAC,QAAQ,OAAO,KAAA;EACpB,OAAO,yBAAyB,MAAM,EAAE;CAC1C,CAAC;AACH;;;;;;AAOA,MAAa,2BAA2B;CACtC,MAAM,MAAM,OAAO;CACnB,QACE,QACA,YACG;EACH,MAAM,SAAS,IAAI,OAAO,EAAE,SAAS,EAAE;EACvC,MAAM,EAAE,WAAW,yBAAyB,MAAM;EAClD,OAAO,OAAO,QAAQ,OAAO;CAC/B;AACF;;;;;;AAOA,MAAa,yBAAyB;CACpC,MAAM,MAAM,OAAO;CACnB,QACE,UACA,YACG;EACH,MAAM,EAAE,QAAQ,gBAAgB,yBAC9B,IAAI,OAAO,EAAE,SAAS,EAAE,MAC1B;EACA,OAAO,OAAO,GAAG,cAAc,SAAS,GAAG,OAAO;CACpD;AACF;;;;;;;;AASA,MAAa,gCAAgC;CAC3C,MAAM,SAAS,mBAAmB;CAClC,QAAQ,YAAqC,OAAO,MAAM,EAAE,QAAQ,CAAC;AACvE;AAeA,SAAgB,kBACd,KACA,cACe;CACf,OAAO,aAAa,MAAM;EACxB,MAAM,SAAS,EAAE,OAAO;EACxB,IAAI,CAAC,QAAQ,OAAO;EACpB,MAAM,QAAQ,yBAAyB,MAAM,EAAE,OAAO;EACtD,OAAO,UAAU,KAAA,IAAY,QAAQ;CACvC,CAAC;AACH"}
+{"version":3,"file":"useStreamRuntime.js","names":["_exhaustiveCheck"],"sources":["../src/useStreamRuntime.ts"],"sourcesContent":["/// <reference types=\"@assistant-ui/core/store\" />\n\"use client\";\n\nimport { useMemo, useRef, useState } from \"react\";\nimport type {\n  AppendMessage,\n  AttachmentAdapter,\n  FeedbackAdapter,\n  RemoteThreadListAdapter,\n  SpeechSynthesisAdapter,\n  ToolExecutionStatus,\n} from \"@assistant-ui/core\";\nimport {\n  useCloudThreadListAdapter,\n  useExternalStoreRuntime,\n  useExternalMessageConverter,\n  useRemoteThreadListRuntime,\n} from \"@assistant-ui/core/react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport type { AssistantCloud } from \"assistant-cloud\";\nimport { useStream, type UseStreamOptions } from \"@langchain/react\";\nimport type { LangChainBaseMessage, LangChainToolCall } from \"./types\";\nimport { convertLangChainBaseMessage, getMessageType } from \"./convertMessages\";\n\nconst symbolLangChainRuntimeExtras = Symbol(\"langchain-runtime-extras\");\n\ntype LangChainRuntimeExtras = {\n  [symbolLangChainRuntimeExtras]: true;\n  interrupt: { value?: unknown } | undefined;\n  interrupts: readonly { value?: unknown }[];\n  submit: (\n    values: Record<string, unknown> | null | undefined,\n    options?: Record<string, unknown>,\n  ) => Promise<void>;\n  values: Record<string, unknown>;\n  messagesKey: string;\n};\n\nconst asLangChainRuntimeExtras = (extras: unknown): LangChainRuntimeExtras => {\n  if (\n    typeof extras !== \"object\" ||\n    extras == null ||\n    !(symbolLangChainRuntimeExtras in extras)\n  )\n    throw new Error(\n      \"This method can only be called when you are using useStreamRuntime\",\n    );\n  return extras as LangChainRuntimeExtras;\n};\n\ntype LangChainRuntimeExtraOptions = {\n  cloud?: AssistantCloud | undefined;\n  adapters?:\n    | {\n        attachments?: AttachmentAdapter | undefined;\n        speech?: SpeechSynthesisAdapter | undefined;\n        feedback?: FeedbackAdapter | undefined;\n      }\n    | undefined;\n  /**\n   * When the user sends a new message while previous tool calls are\n   * still pending, automatically submit `tool` messages that cancel\n   * them so the agent's tool-call accounting stays consistent.\n   * Defaults to `true`.\n   */\n  autoCancelPendingToolCalls?: boolean | undefined;\n  /**\n   * Routes the Cancel button's click to `useStream().stop()`. On by\n   * default. Pass `false` to disable the Cancel button.\n   */\n  unstable_allowCancellation?: boolean | undefined;\n  /**\n   * Custom `RemoteThreadListAdapter`. When provided, replaces the\n   * cloud-backed thread list adapter.\n   */\n  unstable_threadListAdapter?: RemoteThreadListAdapter | undefined;\n  /** Custom thread-creation hook, forwarded to the cloud adapter. */\n  create?: (() => Promise<{ externalId: string | undefined }>) | undefined;\n  /** Custom thread-deletion hook, forwarded to the cloud adapter. */\n  delete?: ((threadId: string) => Promise<void>) | undefined;\n};\n\nconst getPendingToolCalls = (\n  messages: readonly LangChainBaseMessage[],\n): LangChainToolCall[] => {\n  const pending = new Map<string, LangChainToolCall>();\n  for (const m of messages) {\n    const type = getMessageType(m);\n    if (type === \"ai\") {\n      for (const tc of m.tool_calls ?? []) pending.set(tc.id, tc);\n    } else if (type === \"tool\" && m.tool_call_id) {\n      pending.delete(m.tool_call_id);\n    }\n  }\n  return [...pending.values()];\n};\n\n// Distribute the intersection through the union arms of `UseStreamOptions`\n// (`AgentServerOptions | CustomAdapterOptions`). Writing `UseStreamOptions & X`\n// directly collapses arm tracking, so `Omit<…, \"cloud\">` and the like would\n// produce a flattened structural type that no longer matches either arm.\nexport type UseStreamRuntimeOptions = UseStreamOptions extends infer O\n  ? O extends UseStreamOptions\n    ? O & LangChainRuntimeExtraOptions\n    : never\n  : never;\n\nconst getMessageContent = (msg: AppendMessage) => {\n  const allContent = [\n    ...msg.content,\n    ...(msg.attachments?.flatMap((a) => a.content) ?? []),\n  ];\n\n  const hasNonText = allContent.some(\n    (part) => part.type === \"file\" || part.type === \"image\",\n  );\n  const hasText = allContent.some((part) => part.type === \"text\");\n  if (hasNonText && !hasText) {\n    allContent.unshift({ type: \"text\", text: \" \" });\n  }\n\n  const content = allContent.map((part) => {\n    const type = part.type;\n    switch (type) {\n      case \"text\":\n        return { type: \"text\" as const, text: part.text };\n      case \"image\":\n        return { type: \"image_url\" as const, image_url: { url: part.image } };\n      case \"file\":\n        return {\n          type: \"file\" as const,\n          data: part.data,\n          mime_type: part.mimeType,\n          metadata: { filename: part.filename ?? \"file\" },\n          source_type: \"base64\" as const,\n        };\n      case \"tool-call\":\n        throw new Error(\"Tool call appends are not supported.\");\n      default: {\n        const _exhaustiveCheck:\n          | \"reasoning\"\n          | \"source\"\n          | \"audio\"\n          | \"data\"\n          | \"generative-ui\" = type;\n        throw new Error(\n          `Unsupported append message part type: ${_exhaustiveCheck}`,\n        );\n      }\n    }\n  });\n\n  if (content.length === 1 && content[0]?.type === \"text\") {\n    return content[0].text ?? \"\";\n  }\n  return content;\n};\n\ntype DistributiveOmit<T, K extends keyof any> = T extends unknown\n  ? Omit<T, K>\n  : never;\n\nconst useStreamThreadRuntime = (\n  options: DistributiveOmit<\n    UseStreamRuntimeOptions,\n    \"cloud\" | \"unstable_threadListAdapter\" | \"create\" | \"delete\"\n  >,\n) => {\n  const { adapters, autoCancelPendingToolCalls, unstable_allowCancellation } =\n    options;\n  const messagesKey = options.messagesKey ?? \"messages\";\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const externalId = useAuiState((s) => s.threadListItem.externalId) as\n    | string\n    | null;\n  // Mutate in place rather than `{ ...options, threadId }`: spreading\n  // `UseStreamOptions` (a discriminated union on `transport`) into an object\n  // literal merges both arms' transport types, breaking arm assignment.\n  options.threadId = externalId;\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const stream = useStream(options);\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n  const hasExecutingTools = Object.values(toolStatuses).some(\n    (s) => s?.type === \"executing\",\n  );\n  const effectiveIsRunning = stream.isLoading || hasExecutingTools;\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const threadMessages = useExternalMessageConverter({\n    callback: convertLangChainBaseMessage,\n    messages: stream.messages as LangChainBaseMessage[],\n    isRunning: effectiveIsRunning,\n  });\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const streamRef = useRef(stream);\n  streamRef.current = stream;\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const extras = useMemo(\n    (): LangChainRuntimeExtras => ({\n      [symbolLangChainRuntimeExtras]: true,\n      interrupt: stream.interrupt,\n      interrupts: stream.interrupts,\n      submit: stream.submit,\n      values: stream.values,\n      messagesKey,\n    }),\n    [\n      stream.interrupt,\n      stream.interrupts,\n      stream.submit,\n      stream.values,\n      messagesKey,\n    ],\n  );\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const runtime = useExternalStoreRuntime({\n    isRunning: effectiveIsRunning,\n    messages: threadMessages,\n    adapters,\n    extras,\n    unstable_enableToolInvocations: true,\n    setToolStatuses,\n    onNew: async (msg) => {\n      const content = getMessageContent(msg);\n      const cancellations =\n        autoCancelPendingToolCalls !== false\n          ? getPendingToolCalls(\n              streamRef.current.messages as readonly LangChainBaseMessage[],\n            ).map((t) => ({\n              type: \"tool\" as const,\n              name: t.name,\n              tool_call_id: t.id,\n              content: JSON.stringify({ cancelled: true }),\n              status: \"error\" as const,\n            }))\n          : [];\n      await stream.submit({\n        [messagesKey]: [...cancellations, { type: \"human\", content }],\n      });\n    },\n    onAddToolResult: async ({\n      toolCallId,\n      toolName,\n      result,\n      isError,\n      artifact,\n    }) => {\n      await stream.submit({\n        [messagesKey]: [\n          {\n            type: \"tool\",\n            name: toolName,\n            tool_call_id: toolCallId,\n            content: JSON.stringify(result),\n            ...(artifact !== undefined && { artifact }),\n            status: isError ? \"error\" : \"success\",\n          },\n        ],\n      });\n    },\n    onCancel:\n      unstable_allowCancellation !== false\n        ? async () => {\n            await stream.stop();\n          }\n        : undefined,\n  });\n\n  return runtime;\n};\n\n/**\n * Creates an assistant-ui runtime backed by LangChain's `useStream` hook.\n * Accepts the same options as `useStream` from `@langchain/react`, plus\n * `cloud` and `adapters`.\n *\n * @example\n * ```tsx\n * import { useStreamRuntime } from \"@assistant-ui/react-langchain\";\n * import { AssistantRuntimeProvider, Thread } from \"@assistant-ui/react\";\n *\n * function App() {\n *   const runtime = useStreamRuntime({\n *     assistantId: \"agent\",\n *     apiUrl: \"http://localhost:2024\",\n *   });\n *\n *   return (\n *     <AssistantRuntimeProvider runtime={runtime}>\n *       <Thread />\n *     </AssistantRuntimeProvider>\n *   );\n * }\n * ```\n */\nexport const useStreamRuntime = (rawOptions: UseStreamRuntimeOptions) => {\n  const {\n    cloud,\n    unstable_threadListAdapter,\n    create,\n    delete: deleteFn,\n    ...options\n  } = rawOptions;\n\n  const optionsRef = useRef(options);\n  optionsRef.current = options;\n\n  const cloudAdapter = useCloudThreadListAdapter({\n    cloud,\n    create,\n    delete: deleteFn,\n  });\n  const adapter = unstable_threadListAdapter ?? cloudAdapter;\n\n  return useRemoteThreadListRuntime({\n    runtimeHook: function RuntimeHook() {\n      // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n      return useStreamThreadRuntime(optionsRef.current);\n    },\n    adapter,\n    allowNesting: true,\n  });\n};\n\n/**\n * Read the current LangGraph interrupt state from the runtime extras.\n */\nexport const useLangChainInterruptState = () => {\n  return useAuiState((s) => {\n    const extras = s.thread.extras;\n    if (!extras) return undefined;\n    return asLangChainRuntimeExtras(extras).interrupt;\n  });\n};\n\n/**\n * Returns a function to submit raw state updates to the LangGraph agent,\n * bypassing the normal message flow. Useful for sending interrupt resume\n * commands.\n */\nexport const useLangChainSubmit = () => {\n  const aui = useAui();\n  return (\n    values: Record<string, unknown> | null | undefined,\n    options?: Record<string, unknown>,\n  ) => {\n    const extras = aui.thread().getState().extras;\n    const { submit } = asLangChainRuntimeExtras(extras);\n    return submit(values, options);\n  };\n};\n\n/**\n * Submit a list of LangChain-shaped messages on the current thread.\n * Parity helper for migrating from `useLangGraphSend`. Routes to\n * `useStream().submit({ [messagesKey]: messages }, options)`.\n */\nexport const useLangChainSend = () => {\n  const aui = useAui();\n  return (\n    messages: readonly LangChainBaseMessage[],\n    options?: Record<string, unknown>,\n  ) => {\n    const { submit, messagesKey } = asLangChainRuntimeExtras(\n      aui.thread().getState().extras,\n    );\n    return submit({ [messagesKey]: messages }, options);\n  };\n};\n\n/**\n * Submit a `useStream` command (e.g. interrupt resume). Parity helper\n * for migrating from `useLangGraphSendCommand`. Note that v1's command\n * shape (`{ resume?, goto?, update? }`) differs from the legacy\n * `{ resume: string }` form — to carry a payload, use the input or\n * `stream.respond` instead.\n */\nexport const useLangChainSendCommand = () => {\n  const submit = useLangChainSubmit();\n  return (command: Record<string, unknown>) => submit(null, { command });\n};\n\n/**\n * Read a custom LangGraph state key from the current thread. Mirrors\n * `useStream().values[key]` from `@langchain/react` and updates when the\n * stream emits new state.\n *\n * @example\n * ```tsx\n * const todos = useLangChainState<Todo[]>(\"todos\");\n * const files = useLangChainState<Record<string, string>>(\"files\", {});\n * ```\n */\nexport function useLangChainState<T>(key: string): T | undefined;\nexport function useLangChainState<T>(key: string, defaultValue: T): T;\nexport function useLangChainState<T>(\n  key: string,\n  defaultValue?: T,\n): T | undefined {\n  return useAuiState((s) => {\n    const extras = s.thread.extras;\n    if (!extras) return defaultValue;\n    const value = asLangChainRuntimeExtras(extras).values[key] as T | undefined;\n    return value !== undefined ? value : defaultValue;\n  });\n}\n"],"mappings":";;;;;;;AAwBA,MAAM,+BAA+B,OAAO,0BAA0B;AActE,MAAM,4BAA4B,WAA4C;CAC5E,IACE,OAAO,WAAW,YAClB,UAAU,QACV,EAAE,gCAAgC,SAElC,MAAM,IAAI,MACR,oEACF;CACF,OAAO;AACT;AAkCA,MAAM,uBACJ,aACwB;CACxB,MAAM,0BAAU,IAAI,IAA+B;CACnD,KAAK,MAAM,KAAK,UAAU;EACxB,MAAM,OAAO,eAAe,CAAC;EAC7B,IAAI,SAAS,MACX,KAAK,MAAM,MAAM,EAAE,cAAc,CAAC,GAAG,QAAQ,IAAI,GAAG,IAAI,EAAE;OACrD,IAAI,SAAS,UAAU,EAAE,cAC9B,QAAQ,OAAO,EAAE,YAAY;CAEjC;CACA,OAAO,CAAC,GAAG,QAAQ,OAAO,CAAC;AAC7B;AAYA,MAAM,qBAAqB,QAAuB;CAChD,MAAM,aAAa,CACjB,GAAG,IAAI,SACP,GAAI,IAAI,aAAa,SAAS,MAAM,EAAE,OAAO,KAAK,CAAC,CACrD;CAEA,MAAM,aAAa,WAAW,MAC3B,SAAS,KAAK,SAAS,UAAU,KAAK,SAAS,OAClD;CACA,MAAM,UAAU,WAAW,MAAM,SAAS,KAAK,SAAS,MAAM;CAC9D,IAAI,cAAc,CAAC,SACjB,WAAW,QAAQ;EAAE,MAAM;EAAQ,MAAM;CAAI,CAAC;CAGhD,MAAM,UAAU,WAAW,KAAK,SAAS;EACvC,MAAM,OAAO,KAAK;EAClB,QAAQ,MAAR;GACE,KAAK,QACH,OAAO;IAAE,MAAM;IAAiB,MAAM,KAAK;GAAK;GAClD,KAAK,SACH,OAAO;IAAE,MAAM;IAAsB,WAAW,EAAE,KAAK,KAAK,MAAM;GAAE;GACtE,KAAK,QACH,OAAO;IACL,MAAM;IACN,MAAM,KAAK;IACX,WAAW,KAAK;IAChB,UAAU,EAAE,UAAU,KAAK,YAAY,OAAO;IAC9C,aAAa;GACf;GACF,KAAK,aACH,MAAM,IAAI,MAAM,sCAAsC;GACxD,SAOE,MAAM,IAAI,MACR,yCAAyCA,MAC3C;EAEJ;CACF,CAAC;CAED,IAAI,QAAQ,WAAW,KAAK,QAAQ,IAAI,SAAS,QAC/C,OAAO,QAAQ,GAAG,QAAQ;CAE5B,OAAO;AACT;AAMA,MAAM,0BACJ,YAIG;CACH,MAAM,EAAE,UAAU,4BAA4B,+BAC5C;CACF,MAAM,cAAc,QAAQ,eAAe;CAS3C,QAAQ,WANW,aAAa,MAAM,EAAE,eAAe,UAM3B;CAG5B,MAAM,SAAS,UAAU,OAAO;CAGhC,MAAM,CAAC,cAAc,mBAAmB,SAEtC,CAAC,CAAC;CACJ,MAAM,oBAAoB,OAAO,OAAO,YAAY,EAAE,MACnD,MAAM,GAAG,SAAS,WACrB;CACA,MAAM,qBAAqB,OAAO,aAAa;CAG/C,MAAM,iBAAiB,4BAA4B;EACjD,UAAU;EACV,UAAU,OAAO;EACjB,WAAW;CACb,CAAC;CAGD,MAAM,YAAY,OAAO,MAAM;CAC/B,UAAU,UAAU;CA2EpB,OArDgB,wBAAwB;EACtC,WAAW;EACX,UAAU;EACV;EACA,QAvBa,eACkB;IAC5B,+BAA+B;GAChC,WAAW,OAAO;GAClB,YAAY,OAAO;GACnB,QAAQ,OAAO;GACf,QAAQ,OAAO;GACf;EACF,IACA;GACE,OAAO;GACP,OAAO;GACP,OAAO;GACP,OAAO;GACP;EACF,CAQK;EACL,gCAAgC;EAChC;EACA,OAAO,OAAO,QAAQ;GACpB,MAAM,UAAU,kBAAkB,GAAG;GACrC,MAAM,gBACJ,+BAA+B,QAC3B,oBACE,UAAU,QAAQ,QACpB,EAAE,KAAK,OAAO;IACZ,MAAM;IACN,MAAM,EAAE;IACR,cAAc,EAAE;IAChB,SAAS,KAAK,UAAU,EAAE,WAAW,KAAK,CAAC;IAC3C,QAAQ;GACV,EAAE,IACF,CAAC;GACP,MAAM,OAAO,OAAO,GACjB,cAAc,CAAC,GAAG,eAAe;IAAE,MAAM;IAAS;GAAQ,CAAC,EAC9D,CAAC;EACH;EACA,iBAAiB,OAAO,EACtB,YACA,UACA,QACA,SACA,eACI;GACJ,MAAM,OAAO,OAAO,GACjB,cAAc,CACb;IACE,MAAM;IACN,MAAM;IACN,cAAc;IACd,SAAS,KAAK,UAAU,MAAM;IAC9B,GAAI,aAAa,KAAA,KAAa,EAAE,SAAS;IACzC,QAAQ,UAAU,UAAU;GAC9B,CACF,EACF,CAAC;EACH;EACA,UACE,+BAA+B,QAC3B,YAAY;GACV,MAAM,OAAO,KAAK;EACpB,IACA,KAAA;CACR,CAEa;AACf;;;;;;;;;;;;;;;;;;;;;;;;;AA0BA,MAAa,oBAAoB,eAAwC;CACvE,MAAM,EACJ,OACA,4BACA,QACA,QAAQ,UACR,GAAG,YACD;CAEJ,MAAM,aAAa,OAAO,OAAO;CACjC,WAAW,UAAU;CAErB,MAAM,eAAe,0BAA0B;EAC7C;EACA;EACA,QAAQ;CACV,CAAC;CAGD,OAAO,2BAA2B;EAChC,aAAa,SAAS,cAAc;GAElC,OAAO,uBAAuB,WAAW,OAAO;EAClD;EACA,SAPc,8BAA8B;EAQ5C,cAAc;CAChB,CAAC;AACH;;;;AAKA,MAAa,mCAAmC;CAC9C,OAAO,aAAa,MAAM;EACxB,MAAM,SAAS,EAAE,OAAO;EACxB,IAAI,CAAC,QAAQ,OAAO,KAAA;EACpB,OAAO,yBAAyB,MAAM,EAAE;CAC1C,CAAC;AACH;;;;;;AAOA,MAAa,2BAA2B;CACtC,MAAM,MAAM,OAAO;CACnB,QACE,QACA,YACG;EACH,MAAM,SAAS,IAAI,OAAO,EAAE,SAAS,EAAE;EACvC,MAAM,EAAE,WAAW,yBAAyB,MAAM;EAClD,OAAO,OAAO,QAAQ,OAAO;CAC/B;AACF;;;;;;AAOA,MAAa,yBAAyB;CACpC,MAAM,MAAM,OAAO;CACnB,QACE,UACA,YACG;EACH,MAAM,EAAE,QAAQ,gBAAgB,yBAC9B,IAAI,OAAO,EAAE,SAAS,EAAE,MAC1B;EACA,OAAO,OAAO,GAAG,cAAc,SAAS,GAAG,OAAO;CACpD;AACF;;;;;;;;AASA,MAAa,gCAAgC;CAC3C,MAAM,SAAS,mBAAmB;CAClC,QAAQ,YAAqC,OAAO,MAAM,EAAE,QAAQ,CAAC;AACvE;AAeA,SAAgB,kBACd,KACA,cACe;CACf,OAAO,aAAa,MAAM;EACxB,MAAM,SAAS,EAAE,OAAO;EACxB,IAAI,CAAC,QAAQ,OAAO;EACpB,MAAM,QAAQ,yBAAyB,MAAM,EAAE,OAAO;EACtD,OAAO,UAAU,KAAA,IAAY,QAAQ;CACvC,CAAC;AACH"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-langchain",
-  "version": "0.0.8",
+  "version": "0.0.10",
   "description": "LangChain useStream adapter for assistant-ui",
   "keywords": [
     "langchain",
@@ -29,7 +29,7 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@assistant-ui/core": "^0.2.5",
+    "@assistant-ui/core": "^0.2.7",
     "@assistant-ui/store": "^0.2.12",
     "assistant-cloud": "*",
     "assistant-stream": "^0.3.16"
@@ -45,9 +45,9 @@
     }
   },
   "devDependencies": {
-    "@langchain/core": "^1.1.47",
-    "@langchain/langgraph-sdk": "^1.9.4",
-    "@langchain/react": "^1.0.4",
+    "@langchain/core": "^1.1.48",
+    "@langchain/langgraph-sdk": "^1.9.6",
+    "@langchain/react": "^1.0.6",
     "@testing-library/react": "^16.3.2",
     "@types/react": "^19.2.15",
     "jsdom": "^29.1.1",

package/src/useStreamRuntime.ts

@@ -1,3 +1,4 @@
+/// <reference types="@assistant-ui/core/store" />
 "use client";
 
 import { useMemo, useRef, useState } from "react";
@@ -7,14 +8,13 @@ import type {
   FeedbackAdapter,
   RemoteThreadListAdapter,
   SpeechSynthesisAdapter,
+  ToolExecutionStatus,
 } from "@assistant-ui/core";
 import {
   useCloudThreadListAdapter,
   useExternalStoreRuntime,
   useExternalMessageConverter,
   useRemoteThreadListRuntime,
-  useToolInvocations,
-  type ToolExecutionStatus,
 } from "@assistant-ui/core/react";
 import { useAui, useAuiState } from "@assistant-ui/store";
 import type { AssistantCloud } from "assistant-cloud";
@@ -198,45 +198,10 @@ const useStreamThreadRuntime = (
     isRunning: effectiveIsRunning,
   });
 
-  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-  const [runtimeRef] = useState(() => ({
-    get current() {
-      return runtime;
-    },
-  }));
-
   // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
   const streamRef = useRef(stream);
   streamRef.current = stream;
 
-  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-  const toolInvocations = useToolInvocations({
-    state: {
-      messages: threadMessages,
-      isRunning: effectiveIsRunning,
-    },
-    getTools: () => runtimeRef.current.thread.getModelContext().tools,
-    onResult: (command) => {
-      if (command.type === "add-tool-result") {
-        void streamRef.current.submit(
-          {
-            [messagesKey]: [
-              {
-                type: "tool",
-                name: command.toolName,
-                tool_call_id: command.toolCallId,
-                content: JSON.stringify(command.result),
-                status: command.isError ? "error" : "success",
-              },
-            ],
-          },
-          {},
-        );
-      }
-    },
-    setToolStatuses,
-  });
-
   // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
   const extras = useMemo(
     (): LangChainRuntimeExtras => ({
@@ -262,8 +227,9 @@ const useStreamThreadRuntime = (
     messages: threadMessages,
     adapters,
     extras,
+    unstable_enableToolInvocations: true,
+    setToolStatuses,
     onNew: async (msg) => {
-      await toolInvocations.abort();
       const content = getMessageContent(msg);
       const cancellations =
         autoCancelPendingToolCalls !== false
@@ -305,7 +271,6 @@ const useStreamThreadRuntime = (
       unstable_allowCancellation !== false
         ? async () => {
             await stream.stop();
-            await toolInvocations.abort();
           }
         : undefined,
   });