@assistant-ui/react-langgraph 0.1.8 → 0.1.10

package/dist/index.d.mts

@@ -55,17 +55,47 @@ type LangChainEvent = {
     data: LangChainMessage[];
 };
 
-declare const useLangGraphRuntime: ({ threadId, autoCancelPendingToolCalls, unstable_allowImageAttachments, stream, onSwitchToNewThread, onSwitchToThread, adapters: { attachments }, }: {
+type LangGraphCommand = {
+    resume: string;
+};
+type LangGraphSendMessageConfig = {
+    command?: LangGraphCommand;
+    runConfig?: unknown;
+};
+declare const useLangGraphMessages: <TMessage>({ stream, }: {
+    stream: (messages: TMessage[], config: LangGraphSendMessageConfig & {
+        abortSignal: AbortSignal;
+    }) => Promise<AsyncGenerator<{
+        event: string;
+        data: any;
+    }>>;
+}) => {
+    messages: TMessage[];
+    sendMessage: (newMessages: TMessage[], config: LangGraphSendMessageConfig) => Promise<void>;
+    cancel: () => void;
+    setMessages: react.Dispatch<react.SetStateAction<TMessage[]>>;
+};
+
+declare const useLangGraphRuntimeSend: () => (messages: LangChainMessage[], config: LangGraphSendMessageConfig) => Promise<void>;
+declare const useLangGraphRuntimeSendCommand: (command: LangGraphCommand) => () => Promise<void>;
+declare const useLangGraphRuntime: ({ autoCancelPendingToolCalls, adapters: { attachments }, unstable_allowImageAttachments, unstable_allowCancellation, stream, threadId, onSwitchToNewThread, onSwitchToThread, }: {
+    /**
+     * @deprecated For thread management use `useCloudThreadListRuntime` instead. This option will be removed in a future version.
+     */
     threadId?: string | undefined;
     autoCancelPendingToolCalls?: boolean | undefined;
     /**
      * @deprecated Use `adapters: { attachments: new SimpleImageAttachmentAdapter() }` instead. This option will be removed in a future version.
      */
     unstable_allowImageAttachments?: boolean | undefined;
-    stream: (messages: LangChainMessage[]) => Promise<AsyncGenerator<{
+    unstable_allowCancellation?: boolean | undefined;
+    stream: (messages: LangChainMessage[], config: LangGraphSendMessageConfig) => Promise<AsyncGenerator<{
         event: string;
         data: any;
     }>>;
+    /**
+     * @deprecated For thread management use `useCloudThreadListRuntime` instead. This option will be removed in a future version.
+     */
     onSwitchToNewThread?: () => Promise<void> | void;
     onSwitchToThread?: (threadId: string) => Promise<{
         messages: LangChainMessage[];
@@ -75,17 +105,6 @@ declare const useLangGraphRuntime: ({ threadId, autoCancelPendingToolCalls, unst
     } | undefined;
 }) => _assistant_ui_react.AssistantRuntime;
 
-declare const useLangGraphMessages: <TMessage>({ stream, }: {
-    stream: (messages: TMessage[]) => Promise<AsyncGenerator<{
-        event: string;
-        data: any;
-    }>>;
-}) => {
-    messages: TMessage[];
-    sendMessage: (newMessages: TMessage[]) => Promise<void>;
-    setMessages: react.Dispatch<react.SetStateAction<TMessage[]>>;
-};
-
 declare const convertLangchainMessages: useExternalMessageConverter.Callback<LangChainMessage>;
 
-export { type LangChainEvent, type LangChainMessage, type LangChainToolCall, type LangChainToolCallChunk, convertLangchainMessages, useLangGraphMessages, useLangGraphRuntime };
+export { type LangChainEvent, type LangChainMessage, type LangChainToolCall, type LangChainToolCallChunk, convertLangchainMessages, useLangGraphMessages, useLangGraphRuntime, useLangGraphRuntimeSend, useLangGraphRuntimeSendCommand };

package/dist/index.d.ts

@@ -55,17 +55,47 @@ type LangChainEvent = {
     data: LangChainMessage[];
 };
 
-declare const useLangGraphRuntime: ({ threadId, autoCancelPendingToolCalls, unstable_allowImageAttachments, stream, onSwitchToNewThread, onSwitchToThread, adapters: { attachments }, }: {
+type LangGraphCommand = {
+    resume: string;
+};
+type LangGraphSendMessageConfig = {
+    command?: LangGraphCommand;
+    runConfig?: unknown;
+};
+declare const useLangGraphMessages: <TMessage>({ stream, }: {
+    stream: (messages: TMessage[], config: LangGraphSendMessageConfig & {
+        abortSignal: AbortSignal;
+    }) => Promise<AsyncGenerator<{
+        event: string;
+        data: any;
+    }>>;
+}) => {
+    messages: TMessage[];
+    sendMessage: (newMessages: TMessage[], config: LangGraphSendMessageConfig) => Promise<void>;
+    cancel: () => void;
+    setMessages: react.Dispatch<react.SetStateAction<TMessage[]>>;
+};
+
+declare const useLangGraphRuntimeSend: () => (messages: LangChainMessage[], config: LangGraphSendMessageConfig) => Promise<void>;
+declare const useLangGraphRuntimeSendCommand: (command: LangGraphCommand) => () => Promise<void>;
+declare const useLangGraphRuntime: ({ autoCancelPendingToolCalls, adapters: { attachments }, unstable_allowImageAttachments, unstable_allowCancellation, stream, threadId, onSwitchToNewThread, onSwitchToThread, }: {
+    /**
+     * @deprecated For thread management use `useCloudThreadListRuntime` instead. This option will be removed in a future version.
+     */
     threadId?: string | undefined;
     autoCancelPendingToolCalls?: boolean | undefined;
     /**
      * @deprecated Use `adapters: { attachments: new SimpleImageAttachmentAdapter() }` instead. This option will be removed in a future version.
      */
     unstable_allowImageAttachments?: boolean | undefined;
-    stream: (messages: LangChainMessage[]) => Promise<AsyncGenerator<{
+    unstable_allowCancellation?: boolean | undefined;
+    stream: (messages: LangChainMessage[], config: LangGraphSendMessageConfig) => Promise<AsyncGenerator<{
         event: string;
         data: any;
     }>>;
+    /**
+     * @deprecated For thread management use `useCloudThreadListRuntime` instead. This option will be removed in a future version.
+     */
     onSwitchToNewThread?: () => Promise<void> | void;
     onSwitchToThread?: (threadId: string) => Promise<{
         messages: LangChainMessage[];
@@ -75,17 +105,6 @@ declare const useLangGraphRuntime: ({ threadId, autoCancelPendingToolCalls, unst
     } | undefined;
 }) => _assistant_ui_react.AssistantRuntime;
 
-declare const useLangGraphMessages: <TMessage>({ stream, }: {
-    stream: (messages: TMessage[]) => Promise<AsyncGenerator<{
-        event: string;
-        data: any;
-    }>>;
-}) => {
-    messages: TMessage[];
-    sendMessage: (newMessages: TMessage[]) => Promise<void>;
-    setMessages: react.Dispatch<react.SetStateAction<TMessage[]>>;
-};
-
 declare const convertLangchainMessages: useExternalMessageConverter.Callback<LangChainMessage>;
 
-export { type LangChainEvent, type LangChainMessage, type LangChainToolCall, type LangChainToolCallChunk, convertLangchainMessages, useLangGraphMessages, useLangGraphRuntime };
+export { type LangChainEvent, type LangChainMessage, type LangChainToolCall, type LangChainToolCallChunk, convertLangchainMessages, useLangGraphMessages, useLangGraphRuntime, useLangGraphRuntimeSend, useLangGraphRuntimeSendCommand };

package/dist/index.js

@@ -22,7 +22,9 @@ var index_exports = {};
 __export(index_exports, {
   convertLangchainMessages: () => convertLangchainMessages,
   useLangGraphMessages: () => useLangGraphMessages,
-  useLangGraphRuntime: () => useLangGraphRuntime
+  useLangGraphRuntime: () => useLangGraphRuntime,
+  useLangGraphRuntimeSend: () => useLangGraphRuntimeSend,
+  useLangGraphRuntimeSendCommand: () => useLangGraphRuntimeSendCommand
 });
 module.exports = __toCommonJS(index_exports);
 
@@ -105,13 +107,19 @@ var useLangGraphMessages = ({
   stream
 }) => {
   const [messages, setMessages] = (0, import_react.useState)([]);
+  const abortControllerRef = (0, import_react.useRef)(null);
   const sendMessage = (0, import_react.useCallback)(
-    async (newMessages) => {
+    async (newMessages, config) => {
       const optimisticMessages = [...messages, ...newMessages];
       if (newMessages.length > 0) {
         setMessages(optimisticMessages);
       }
-      const response = await stream(newMessages);
+      const abortController = new AbortController();
+      abortControllerRef.current = abortController;
+      const response = await stream(newMessages, {
+        ...config,
+        abortSignal: abortController.signal
+      });
       const completeMessages = [];
       let partialMessages = /* @__PURE__ */ new Map();
       for await (const chunk of response) {
@@ -142,11 +150,17 @@ var useLangGraphMessages = ({
     },
     [messages, stream]
   );
-  return { messages, sendMessage, setMessages };
+  const cancel = (0, import_react.useCallback)(() => {
+    if (abortControllerRef.current) {
+      abortControllerRef.current.abort();
+    }
+  }, [abortControllerRef]);
+  return { messages, sendMessage, cancel, setMessages };
 };
 
 // src/useLangGraphRuntime.ts
 var import_react5 = require("@assistant-ui/react");
+var import_ThreadListItemContext = require("@assistant-ui/react/context/react/ThreadListItemContext");
 var getPendingToolCalls = (messages) => {
   const pendingToolCalls = /* @__PURE__ */ new Map();
   for (const message of messages) {
@@ -187,23 +201,40 @@ var getMessageContent = (msg) => {
   }
   return content;
 };
+var symbolLangGraphRuntimeExtras = Symbol("langgraph-runtime-extras");
+var asLangGraphRuntimeExtras = (extras) => {
+  if (typeof extras !== "object" || extras == null || !(symbolLangGraphRuntimeExtras in extras))
+    throw new Error(
+      "This method can only be called when you are using useLangGraphRuntime"
+    );
+  return extras;
+};
+var useLangGraphRuntimeSend = () => {
+  const { send } = (0, import_react4.useThread)((t) => asLangGraphRuntimeExtras(t.extras));
+  return send;
+};
+var useLangGraphRuntimeSendCommand = (command) => {
+  const send = useLangGraphRuntimeSend();
+  return () => send([], { command });
+};
 var useLangGraphRuntime = ({
-  threadId,
   autoCancelPendingToolCalls,
+  adapters: { attachments } = {},
   unstable_allowImageAttachments,
+  unstable_allowCancellation,
   stream,
+  threadId,
   onSwitchToNewThread,
-  onSwitchToThread,
-  adapters: { attachments } = {}
+  onSwitchToThread
 }) => {
-  const { messages, sendMessage, setMessages } = useLangGraphMessages({
+  const { messages, sendMessage, cancel, setMessages } = useLangGraphMessages({
     stream
   });
   const [isRunning, setIsRunning] = (0, import_react3.useState)(false);
-  const handleSendMessage = async (messages2) => {
+  const handleSendMessage = async (messages2, config) => {
     try {
       setIsRunning(true);
-      await sendMessage(messages2);
+      await sendMessage(messages2, config);
     } catch (error) {
       console.error("Error streaming messages:", error);
     } finally {
@@ -221,17 +252,30 @@ var useLangGraphRuntime = ({
     );
   if (unstable_allowImageAttachments)
     attachments = new import_react5.SimpleImageAttachmentAdapter();
+  const switchToThread = !onSwitchToThread ? void 0 : async (threadId2) => {
+    const { messages: messages2 } = await onSwitchToThread(threadId2);
+    setMessages(messages2);
+  };
   const threadList = {
     threadId,
     onSwitchToNewThread: !onSwitchToNewThread ? void 0 : async () => {
       await onSwitchToNewThread();
       setMessages([]);
     },
-    onSwitchToThread: !onSwitchToThread ? void 0 : async (threadId2) => {
-      const { messages: messages2 } = await onSwitchToThread(threadId2);
-      setMessages(messages2);
-    }
+    onSwitchToThread: switchToThread
   };
+  const loadingRef = (0, import_react3.useRef)(false);
+  const threadListItemRuntime = (0, import_ThreadListItemContext.useThreadListItemRuntime)({ optional: true });
+  (0, import_react3.useEffect)(() => {
+    if (!threadListItemRuntime || !switchToThread || loadingRef.current) return;
+    const externalId = threadListItemRuntime.getState().externalId;
+    if (externalId) {
+      loadingRef.current = true;
+      switchToThread(externalId).finally(() => {
+        loadingRef.current = false;
+      });
+    }
+  }, []);
   return (0, import_react4.useExternalStoreRuntime)({
     isRunning,
     messages: threadMessages,
@@ -239,6 +283,10 @@ var useLangGraphRuntime = ({
       attachments,
       threadList
     },
+    extras: {
+      [symbolLangGraphRuntimeExtras]: true,
+      send: handleSendMessage
+    },
     onNew: (msg) => {
       const cancellations = autoCancelPendingToolCalls !== false ? getPendingToolCalls(messages).map(
         (t) => ({
@@ -248,30 +296,44 @@ var useLangGraphRuntime = ({
           content: JSON.stringify({ cancelled: true })
         })
       ) : [];
-      return handleSendMessage([
-        ...cancellations,
+      return handleSendMessage(
+        [
+          ...cancellations,
+          {
+            type: "human",
+            content: getMessageContent(msg)
+          }
+        ],
         {
-          type: "human",
-          content: getMessageContent(msg)
+          runConfig: msg.runConfig
         }
-      ]);
+      );
     },
     onAddToolResult: async ({ toolCallId, toolName, result }) => {
-      await handleSendMessage([
-        {
-          type: "tool",
-          name: toolName,
-          tool_call_id: toolCallId,
-          content: JSON.stringify(result)
-        }
-      ]);
-    }
+      await handleSendMessage(
+        [
+          {
+            type: "tool",
+            name: toolName,
+            tool_call_id: toolCallId,
+            content: JSON.stringify(result)
+          }
+        ],
+        // TODO reuse runconfig here!
+        {}
+      );
+    },
+    onCancel: unstable_allowCancellation ? async () => {
+      cancel();
+    } : void 0
   });
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   convertLangchainMessages,
   useLangGraphMessages,
-  useLangGraphRuntime
+  useLangGraphRuntime,
+  useLangGraphRuntimeSend,
+  useLangGraphRuntimeSendCommand
 });
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts","../src/useLangGraphRuntime.ts","../src/convertLangchainMessages.ts","../src/useLangGraphMessages.ts"],"sourcesContent":["export { useLangGraphRuntime } from \"./useLangGraphRuntime\";\n\nexport { useLangGraphMessages } from \"./useLangGraphMessages\";\nexport { convertLangchainMessages } from \"./convertLangchainMessages\";\nexport type {\n  LangChainMessage,\n  LangChainEvent,\n  LangChainToolCall,\n  LangChainToolCallChunk,\n} from \"./types\";\n","import { useState } from \"react\";\nimport { LangChainMessage, LangChainToolCall } from \"./types\";\nimport {\n  useExternalMessageConverter,\n  useExternalStoreRuntime,\n} from \"@assistant-ui/react\";\nimport { convertLangchainMessages } from \"./convertLangchainMessages\";\nimport { useLangGraphMessages } from \"./useLangGraphMessages\";\nimport { SimpleImageAttachmentAdapter } from \"@assistant-ui/react\";\nimport { AttachmentAdapter } from \"@assistant-ui/react\";\nimport { AppendMessage } from \"@assistant-ui/react\";\nimport { ExternalStoreAdapter } from \"@assistant-ui/react\";\n\nconst getPendingToolCalls = (messages: LangChainMessage[]) => {\n  const pendingToolCalls = new Map<string, LangChainToolCall>();\n  for (const message of messages) {\n    if (message.type === \"ai\") {\n      for (const toolCall of message.tool_calls ?? []) {\n        pendingToolCalls.set(toolCall.id, toolCall);\n      }\n    }\n    if (message.type === \"tool\") {\n      pendingToolCalls.delete(message.tool_call_id);\n    }\n  }\n\n  return [...pendingToolCalls.values()];\n};\n\nconst getMessageContent = (msg: AppendMessage) => {\n  const allContent = [\n    ...msg.content,\n    ...msg.attachments.flatMap((a) => a.content),\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\n      case \"audio\":\n        throw new Error(\"Audio appends are not supported yet.\");\n      case \"tool-call\":\n        throw new Error(\"Tool call appends are not supported yet.\");\n\n      default:\n        const _exhaustiveCheck: never = type;\n        throw new Error(`Unknown content part type: ${_exhaustiveCheck}`);\n    }\n  });\n\n  if (content.length === 1 && content[0]?.type === \"text\") {\n    return content[0].text ?? \"\";\n  }\n\n  return content;\n};\n\nexport const useLangGraphRuntime = ({\n  threadId,\n  autoCancelPendingToolCalls,\n  unstable_allowImageAttachments,\n  stream,\n  onSwitchToNewThread,\n  onSwitchToThread,\n  adapters: { attachments } = {},\n}: {\n  threadId?: string | undefined;\n  autoCancelPendingToolCalls?: boolean | undefined;\n  /**\n   * @deprecated Use `adapters: { attachments: new SimpleImageAttachmentAdapter() }` instead. This option will be removed in a future version.\n   */\n  unstable_allowImageAttachments?: boolean | undefined;\n  stream: (messages: LangChainMessage[]) => Promise<\n    AsyncGenerator<{\n      event: string;\n      data: any;\n    }>\n  >;\n  onSwitchToNewThread?: () => Promise<void> | void;\n  onSwitchToThread?: (\n    threadId: string,\n  ) => Promise<{ messages: LangChainMessage[] }>;\n  adapters?:\n    | {\n        attachments?: AttachmentAdapter;\n      }\n    | undefined;\n}) => {\n  const { messages, sendMessage, setMessages } = useLangGraphMessages({\n    stream,\n  });\n\n  const [isRunning, setIsRunning] = useState(false);\n  const handleSendMessage = async (messages: LangChainMessage[]) => {\n    try {\n      setIsRunning(true);\n      await sendMessage(messages);\n    } catch (error) {\n      console.error(\"Error streaming messages:\", error);\n    } finally {\n      setIsRunning(false);\n    }\n  };\n\n  const threadMessages = useExternalMessageConverter({\n    callback: convertLangchainMessages,\n    messages,\n    isRunning,\n  });\n\n  if (attachments && unstable_allowImageAttachments)\n    throw new Error(\n      \"Replace unstable_allowImageAttachments with `adapters: { attachments: new SimpleImageAttachmentAdapter() }`.\",\n    );\n  if (unstable_allowImageAttachments)\n    attachments = new SimpleImageAttachmentAdapter();\n\n  const threadList: NonNullable<\n    ExternalStoreAdapter[\"adapters\"]\n  >[\"threadList\"] = {\n    threadId,\n    onSwitchToNewThread: !onSwitchToNewThread\n      ? undefined\n      : async () => {\n          await onSwitchToNewThread();\n          setMessages([]);\n        },\n    onSwitchToThread: !onSwitchToThread\n      ? undefined\n      : async (threadId) => {\n          const { messages } = await onSwitchToThread(threadId);\n          setMessages(messages);\n        },\n  };\n\n  return useExternalStoreRuntime({\n    isRunning,\n    messages: threadMessages,\n    adapters: {\n      attachments,\n      threadList,\n    },\n    onNew: (msg) => {\n      const cancellations =\n        autoCancelPendingToolCalls !== false\n          ? getPendingToolCalls(messages).map(\n              (t) =>\n                ({\n                  type: \"tool\",\n                  name: t.name,\n                  tool_call_id: t.id,\n                  content: JSON.stringify({ cancelled: true }),\n                }) satisfies LangChainMessage & { type: \"tool\" },\n            )\n          : [];\n\n      return handleSendMessage([\n        ...cancellations,\n        {\n          type: \"human\",\n          content: getMessageContent(msg),\n        },\n      ]);\n    },\n    onAddToolResult: async ({ toolCallId, toolName, result }) => {\n      await handleSendMessage([\n        {\n          type: \"tool\",\n          name: toolName,\n          tool_call_id: toolCallId,\n          content: JSON.stringify(result),\n        },\n      ]);\n    },\n  });\n};\n","\"use client\";\n\nimport { useExternalMessageConverter } from \"@assistant-ui/react\";\nimport { LangChainMessage } from \"./types\";\nimport { ToolCallContentPart } from \"@assistant-ui/react\";\nimport { ThreadUserMessage } from \"@assistant-ui/react\";\n\nconst contentToParts = (\n  content: LangChainMessage[\"content\"],\n) => {\n  if (typeof content === \"string\")\n    return [{ type: \"text\" as const, text: content }];\n  return content\n    .map((part): ThreadUserMessage[\"content\"][number] | null => {\n      const type = part.type;\n      switch (type) {\n        case \"text\":\n          return { type: \"text\", text: part.text };\n        case \"image_url\":\n          if (typeof part.image_url === \"string\") {\n            return { type: \"image\", image: part.image_url };\n          } else {\n            return {\n              type: \"image\",\n              image: part.image_url.url,\n            };\n          }\n\n        case \"tool_use\":\n          return null;\n        default:\n          const _exhaustiveCheck: never = type;\n          throw new Error(`Unknown content part type: ${_exhaustiveCheck}`);\n      }\n    })\n    .filter((a) => a !== null);\n};\n\nexport const convertLangchainMessages: useExternalMessageConverter.Callback<\n  LangChainMessage\n> = (message) => {\n  switch (message.type) {\n    case \"system\":\n      return {\n        role: \"system\",\n        id: message.id,\n        content: [{ type: \"text\", text: message.content }],\n      };\n    case \"human\":\n      return {\n        role: \"user\",\n        id: message.id,\n        content: contentToParts(message.content),\n      };\n    case \"ai\":\n      return {\n        role: \"assistant\",\n        id: message.id,\n        content: [\n          ...contentToParts(message.content),\n          ...(message.tool_calls?.map(\n            (chunk): ToolCallContentPart => ({\n              type: \"tool-call\",\n              toolCallId: chunk.id,\n              toolName: chunk.name,\n              args: chunk.args,\n              argsText:\n                message.tool_call_chunks?.find((c) => c.id === chunk.id)\n                  ?.args ?? JSON.stringify(chunk.args),\n            }),\n          ) ?? []),\n        ],\n      };\n    case \"tool\":\n      return {\n        role: \"tool\",\n        toolName: message.name,\n        toolCallId: message.tool_call_id,\n        result: message.content,\n      };\n  }\n};\n","import { useState, useCallback } from \"react\";\nimport { INTERNAL } from \"@assistant-ui/react\";\n\nconst { generateId } = INTERNAL;\n\nexport const useLangGraphMessages = <TMessage>({\n  stream,\n}: {\n  stream: (messages: TMessage[]) => Promise<\n    AsyncGenerator<{\n      event: string;\n      data: any;\n    }>\n  >;\n}) => {\n  const [messages, setMessages] = useState<TMessage[]>([]);\n\n  const sendMessage = useCallback(\n    async (newMessages: TMessage[]) => {\n      const optimisticMessages = [...messages, ...newMessages];\n      if (newMessages.length > 0) {\n        setMessages(optimisticMessages);\n      }\n\n      const response = await stream(newMessages);\n\n      const completeMessages: TMessage[] = [];\n      let partialMessages: Map<string, TMessage> = new Map();\n      for await (const chunk of response) {\n        if (chunk.event === \"messages/partial\") {\n          // bug fix: messages/complete is not sent for some python langchain backends\n          if (completeMessages.length === 0) {\n            completeMessages.push(...optimisticMessages);\n          }\n\n          for (const message of chunk.data) {\n            if (!message.id) throw new Error(\"Partial message missing id\");\n\n            partialMessages.set(message.id, message);\n          }\n        } else if (chunk.event === \"messages/complete\") {\n          for (const message of chunk.data) {\n            if (message.id) {\n              partialMessages.delete(message.id);\n            }\n\n            // bug fix: tool results can arrive before the message is complete ?\n            // if we have partial messages left, we add complete message to partial messages\n            if (partialMessages.size > 0) {\n              partialMessages.set(message.id ?? generateId(), message);\n            } else {\n              completeMessages.push(message);\n            }\n          }\n        } else {\n          continue;\n        }\n\n        setMessages([...completeMessages, ...partialMessages.values()]);\n      }\n      // if (partialMessages.size > 0) {\n      //   throw new Error(\"A partial message was not marked as complete\");\n      // }\n    },\n    [messages, stream],\n  );\n\n  return { messages, sendMessage, setMessages };\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,IAAAA,gBAAyB;AAEzB,IAAAA,gBAGO;;;ACEP,IAAM,iBAAiB,CACrB,YACG;AACH,MAAI,OAAO,YAAY;AACrB,WAAO,CAAC,EAAE,MAAM,QAAiB,MAAM,QAAQ,CAAC;AAClD,SAAO,QACJ,IAAI,CAAC,SAAsD;AAC1D,UAAM,OAAO,KAAK;AAClB,YAAQ,MAAM;AAAA,MACZ,KAAK;AACH,eAAO,EAAE,MAAM,QAAQ,MAAM,KAAK,KAAK;AAAA,MACzC,KAAK;AACH,YAAI,OAAO,KAAK,cAAc,UAAU;AACtC,iBAAO,EAAE,MAAM,SAAS,OAAO,KAAK,UAAU;AAAA,QAChD,OAAO;AACL,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,OAAO,KAAK,UAAU;AAAA,UACxB;AAAA,QACF;AAAA,MAEF,KAAK;AACH,eAAO;AAAA,MACT;AACE,cAAM,mBAA0B;AAChC,cAAM,IAAI,MAAM,8BAA8B,gBAAgB,EAAE;AAAA,IACpE;AAAA,EACF,CAAC,EACA,OAAO,CAAC,MAAM,MAAM,IAAI;AAC7B;AAEO,IAAM,2BAET,CAAC,YAAY;AACf,UAAQ,QAAQ,MAAM;AAAA,IACpB,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,IAAI,QAAQ;AAAA,QACZ,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,QAAQ,QAAQ,CAAC;AAAA,MACnD;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,IAAI,QAAQ;AAAA,QACZ,SAAS,eAAe,QAAQ,OAAO;AAAA,MACzC;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,IAAI,QAAQ;AAAA,QACZ,SAAS;AAAA,UACP,GAAG,eAAe,QAAQ,OAAO;AAAA,UACjC,GAAI,QAAQ,YAAY;AAAA,YACtB,CAAC,WAAgC;AAAA,cAC/B,MAAM;AAAA,cACN,YAAY,MAAM;AAAA,cAClB,UAAU,MAAM;AAAA,cAChB,MAAM,MAAM;AAAA,cACZ,UACE,QAAQ,kBAAkB,KAAK,CAAC,MAAM,EAAE,OAAO,MAAM,EAAE,GACnD,QAAQ,KAAK,UAAU,MAAM,IAAI;AAAA,YACzC;AAAA,UACF,KAAK,CAAC;AAAA,QACR;AAAA,MACF;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,UAAU,QAAQ;AAAA,QAClB,YAAY,QAAQ;AAAA,QACpB,QAAQ,QAAQ;AAAA,MAClB;AAAA,EACJ;AACF;;;ACjFA,mBAAsC;AACtC,IAAAC,gBAAyB;AAEzB,IAAM,EAAE,WAAW,IAAI;AAEhB,IAAM,uBAAuB,CAAW;AAAA,EAC7C;AACF,MAOM;AACJ,QAAM,CAAC,UAAU,WAAW,QAAI,uBAAqB,CAAC,CAAC;AAEvD,QAAM,kBAAc;AAAA,IAClB,OAAO,gBAA4B;AACjC,YAAM,qBAAqB,CAAC,GAAG,UAAU,GAAG,WAAW;AACvD,UAAI,YAAY,SAAS,GAAG;AAC1B,oBAAY,kBAAkB;AAAA,MAChC;AAEA,YAAM,WAAW,MAAM,OAAO,WAAW;AAEzC,YAAM,mBAA+B,CAAC;AACtC,UAAI,kBAAyC,oBAAI,IAAI;AACrD,uBAAiB,SAAS,UAAU;AAClC,YAAI,MAAM,UAAU,oBAAoB;AAEtC,cAAI,iBAAiB,WAAW,GAAG;AACjC,6BAAiB,KAAK,GAAG,kBAAkB;AAAA,UAC7C;AAEA,qBAAW,WAAW,MAAM,MAAM;AAChC,gBAAI,CAAC,QAAQ,GAAI,OAAM,IAAI,MAAM,4BAA4B;AAE7D,4BAAgB,IAAI,QAAQ,IAAI,OAAO;AAAA,UACzC;AAAA,QACF,WAAW,MAAM,UAAU,qBAAqB;AAC9C,qBAAW,WAAW,MAAM,MAAM;AAChC,gBAAI,QAAQ,IAAI;AACd,8BAAgB,OAAO,QAAQ,EAAE;AAAA,YACnC;AAIA,gBAAI,gBAAgB,OAAO,GAAG;AAC5B,8BAAgB,IAAI,QAAQ,MAAM,WAAW,GAAG,OAAO;AAAA,YACzD,OAAO;AACL,+BAAiB,KAAK,OAAO;AAAA,YAC/B;AAAA,UACF;AAAA,QACF,OAAO;AACL;AAAA,QACF;AAEA,oBAAY,CAAC,GAAG,kBAAkB,GAAG,gBAAgB,OAAO,CAAC,CAAC;AAAA,MAChE;AAAA,IAIF;AAAA,IACA,CAAC,UAAU,MAAM;AAAA,EACnB;AAEA,SAAO,EAAE,UAAU,aAAa,YAAY;AAC9C;;;AF5DA,IAAAC,gBAA6C;AAK7C,IAAM,sBAAsB,CAAC,aAAiC;AAC5D,QAAM,mBAAmB,oBAAI,IAA+B;AAC5D,aAAW,WAAW,UAAU;AAC9B,QAAI,QAAQ,SAAS,MAAM;AACzB,iBAAW,YAAY,QAAQ,cAAc,CAAC,GAAG;AAC/C,yBAAiB,IAAI,SAAS,IAAI,QAAQ;AAAA,MAC5C;AAAA,IACF;AACA,QAAI,QAAQ,SAAS,QAAQ;AAC3B,uBAAiB,OAAO,QAAQ,YAAY;AAAA,IAC9C;AAAA,EACF;AAEA,SAAO,CAAC,GAAG,iBAAiB,OAAO,CAAC;AACtC;AAEA,IAAM,oBAAoB,CAAC,QAAuB;AAChD,QAAM,aAAa;AAAA,IACjB,GAAG,IAAI;AAAA,IACP,GAAG,IAAI,YAAY,QAAQ,CAAC,MAAM,EAAE,OAAO;AAAA,EAC7C;AACA,QAAM,UAAU,WAAW,IAAI,CAAC,SAAS;AACvC,UAAM,OAAO,KAAK;AAClB,YAAQ,MAAM;AAAA,MACZ,KAAK;AACH,eAAO,EAAE,MAAM,QAAiB,MAAM,KAAK,KAAK;AAAA,MAClD,KAAK;AACH,eAAO,EAAE,MAAM,aAAsB,WAAW,EAAE,KAAK,KAAK,MAAM,EAAE;AAAA,MAEtE,KAAK;AACH,cAAM,IAAI,MAAM,sCAAsC;AAAA,MACxD,KAAK;AACH,cAAM,IAAI,MAAM,0CAA0C;AAAA,MAE5D;AACE,cAAM,mBAA0B;AAChC,cAAM,IAAI,MAAM,8BAA8B,gBAAgB,EAAE;AAAA,IACpE;AAAA,EACF,CAAC;AAED,MAAI,QAAQ,WAAW,KAAK,QAAQ,CAAC,GAAG,SAAS,QAAQ;AACvD,WAAO,QAAQ,CAAC,EAAE,QAAQ;AAAA,EAC5B;AAEA,SAAO;AACT;AAEO,IAAM,sBAAsB,CAAC;AAAA,EAClC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA,UAAU,EAAE,YAAY,IAAI,CAAC;AAC/B,MAsBM;AACJ,QAAM,EAAE,UAAU,aAAa,YAAY,IAAI,qBAAqB;AAAA,IAClE;AAAA,EACF,CAAC;AAED,QAAM,CAAC,WAAW,YAAY,QAAI,wBAAS,KAAK;AAChD,QAAM,oBAAoB,OAAOC,cAAiC;AAChE,QAAI;AACF,mBAAa,IAAI;AACjB,YAAM,YAAYA,SAAQ;AAAA,IAC5B,SAAS,OAAO;AACd,cAAQ,MAAM,6BAA6B,KAAK;AAAA,IAClD,UAAE;AACA,mBAAa,KAAK;AAAA,IACpB;AAAA,EACF;AAEA,QAAM,qBAAiB,2CAA4B;AAAA,IACjD,UAAU;AAAA,IACV;AAAA,IACA;AAAA,EACF,CAAC;AAED,MAAI,eAAe;AACjB,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AACF,MAAI;AACF,kBAAc,IAAI,2CAA6B;AAEjD,QAAM,aAEY;AAAA,IAChB;AAAA,IACA,qBAAqB,CAAC,sBAClB,SACA,YAAY;AACV,YAAM,oBAAoB;AAC1B,kBAAY,CAAC,CAAC;AAAA,IAChB;AAAA,IACJ,kBAAkB,CAAC,mBACf,SACA,OAAOC,cAAa;AAClB,YAAM,EAAE,UAAAD,UAAS,IAAI,MAAM,iBAAiBC,SAAQ;AACpD,kBAAYD,SAAQ;AAAA,IACtB;AAAA,EACN;AAEA,aAAO,uCAAwB;AAAA,IAC7B;AAAA,IACA,UAAU;AAAA,IACV,UAAU;AAAA,MACR;AAAA,MACA;AAAA,IACF;AAAA,IACA,OAAO,CAAC,QAAQ;AACd,YAAM,gBACJ,+BAA+B,QAC3B,oBAAoB,QAAQ,EAAE;AAAA,QAC5B,CAAC,OACE;AAAA,UACC,MAAM;AAAA,UACN,MAAM,EAAE;AAAA,UACR,cAAc,EAAE;AAAA,UAChB,SAAS,KAAK,UAAU,EAAE,WAAW,KAAK,CAAC;AAAA,QAC7C;AAAA,MACJ,IACA,CAAC;AAEP,aAAO,kBAAkB;AAAA,QACvB,GAAG;AAAA,QACH;AAAA,UACE,MAAM;AAAA,UACN,SAAS,kBAAkB,GAAG;AAAA,QAChC;AAAA,MACF,CAAC;AAAA,IACH;AAAA,IACA,iBAAiB,OAAO,EAAE,YAAY,UAAU,OAAO,MAAM;AAC3D,YAAM,kBAAkB;AAAA,QACtB;AAAA,UACE,MAAM;AAAA,UACN,MAAM;AAAA,UACN,cAAc;AAAA,UACd,SAAS,KAAK,UAAU,MAAM;AAAA,QAChC;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF,CAAC;AACH;","names":["import_react","import_react","import_react","messages","threadId"]}
+{"version":3,"sources":["../src/index.ts","../src/useLangGraphRuntime.ts","../src/convertLangchainMessages.ts","../src/useLangGraphMessages.ts"],"sourcesContent":["export {\n  useLangGraphRuntime,\n  useLangGraphRuntimeSend,\n  useLangGraphRuntimeSendCommand,\n} from \"./useLangGraphRuntime\";\n\nexport { useLangGraphMessages } from \"./useLangGraphMessages\";\nexport { convertLangchainMessages } from \"./convertLangchainMessages\";\nexport type {\n  LangChainMessage,\n  LangChainEvent,\n  LangChainToolCall,\n  LangChainToolCallChunk,\n} from \"./types\";\n","import { useEffect, useRef, useState } from \"react\";\nimport { LangChainMessage, LangChainToolCall } from \"./types\";\nimport {\n  useExternalMessageConverter,\n  useExternalStoreRuntime,\n  useThread,\n} from \"@assistant-ui/react\";\nimport { convertLangchainMessages } from \"./convertLangchainMessages\";\nimport {\n  LangGraphCommand,\n  LangGraphSendMessageConfig,\n  useLangGraphMessages,\n} from \"./useLangGraphMessages\";\nimport { SimpleImageAttachmentAdapter } from \"@assistant-ui/react\";\nimport { AttachmentAdapter } from \"@assistant-ui/react\";\nimport { AppendMessage } from \"@assistant-ui/react\";\nimport { ExternalStoreAdapter } from \"@assistant-ui/react\";\nimport { useThreadListItemRuntime } from \"@assistant-ui/react/context/react/ThreadListItemContext\";\n\nconst getPendingToolCalls = (messages: LangChainMessage[]) => {\n  const pendingToolCalls = new Map<string, LangChainToolCall>();\n  for (const message of messages) {\n    if (message.type === \"ai\") {\n      for (const toolCall of message.tool_calls ?? []) {\n        pendingToolCalls.set(toolCall.id, toolCall);\n      }\n    }\n    if (message.type === \"tool\") {\n      pendingToolCalls.delete(message.tool_call_id);\n    }\n  }\n\n  return [...pendingToolCalls.values()];\n};\n\nconst getMessageContent = (msg: AppendMessage) => {\n  const allContent = [\n    ...msg.content,\n    ...msg.attachments.flatMap((a) => a.content),\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\n      case \"audio\":\n        throw new Error(\"Audio appends are not supported yet.\");\n      case \"tool-call\":\n        throw new Error(\"Tool call appends are not supported yet.\");\n\n      default:\n        const _exhaustiveCheck: never = type;\n        throw new Error(`Unknown content part type: ${_exhaustiveCheck}`);\n    }\n  });\n\n  if (content.length === 1 && content[0]?.type === \"text\") {\n    return content[0].text ?? \"\";\n  }\n\n  return content;\n};\n\nconst symbolLangGraphRuntimeExtras = Symbol(\"langgraph-runtime-extras\");\ntype LangGraphRuntimeExtras = {\n  [symbolLangGraphRuntimeExtras]: true;\n  send: (\n    messages: LangChainMessage[],\n    config: LangGraphSendMessageConfig,\n  ) => Promise<void>;\n};\n\nconst asLangGraphRuntimeExtras = (extras: unknown): LangGraphRuntimeExtras => {\n  if (\n    typeof extras !== \"object\" ||\n    extras == null ||\n    !(symbolLangGraphRuntimeExtras in extras)\n  )\n    throw new Error(\n      \"This method can only be called when you are using useLangGraphRuntime\",\n    );\n\n  return extras as LangGraphRuntimeExtras;\n};\n\nexport const useLangGraphRuntimeSend = () => {\n  const { send } = useThread((t) => asLangGraphRuntimeExtras(t.extras));\n  return send;\n};\n\nexport const useLangGraphRuntimeSendCommand = (command: LangGraphCommand) => {\n  const send = useLangGraphRuntimeSend();\n  return () => send([], { command });\n};\n\nexport const useLangGraphRuntime = ({\n  autoCancelPendingToolCalls,\n  adapters: { attachments } = {},\n  unstable_allowImageAttachments,\n  unstable_allowCancellation,\n  stream,\n  threadId,\n  onSwitchToNewThread,\n  onSwitchToThread,\n}: {\n  /**\n   * @deprecated For thread management use `useCloudThreadListRuntime` instead. This option will be removed in a future version.\n   */\n  threadId?: string | undefined;\n  autoCancelPendingToolCalls?: boolean | undefined;\n  /**\n   * @deprecated Use `adapters: { attachments: new SimpleImageAttachmentAdapter() }` instead. This option will be removed in a future version.\n   */\n  unstable_allowImageAttachments?: boolean | undefined;\n  unstable_allowCancellation?: boolean | undefined;\n  stream: (\n    messages: LangChainMessage[],\n    config: LangGraphSendMessageConfig,\n  ) => Promise<\n    AsyncGenerator<{\n      event: string;\n      data: any;\n    }>\n  >;\n  /**\n   * @deprecated For thread management use `useCloudThreadListRuntime` instead. This option will be removed in a future version.\n   */\n  onSwitchToNewThread?: () => Promise<void> | void;\n  onSwitchToThread?: (\n    threadId: string,\n  ) => Promise<{ messages: LangChainMessage[] }>;\n  adapters?:\n    | {\n        attachments?: AttachmentAdapter;\n      }\n    | undefined;\n}) => {\n  const { messages, sendMessage, cancel, setMessages } = useLangGraphMessages({\n    stream,\n  });\n\n  const [isRunning, setIsRunning] = useState(false);\n  const handleSendMessage = async (\n    messages: LangChainMessage[],\n    config: LangGraphSendMessageConfig,\n  ) => {\n    try {\n      setIsRunning(true);\n      await sendMessage(messages, config);\n    } catch (error) {\n      console.error(\"Error streaming messages:\", error);\n    } finally {\n      setIsRunning(false);\n    }\n  };\n\n  const threadMessages = useExternalMessageConverter({\n    callback: convertLangchainMessages,\n    messages,\n    isRunning,\n  });\n\n  if (attachments && unstable_allowImageAttachments)\n    throw new Error(\n      \"Replace unstable_allowImageAttachments with `adapters: { attachments: new SimpleImageAttachmentAdapter() }`.\",\n    );\n  if (unstable_allowImageAttachments)\n    attachments = new SimpleImageAttachmentAdapter();\n\n  const switchToThread = !onSwitchToThread\n    ? undefined\n    : async (threadId: string) => {\n        const { messages } = await onSwitchToThread(threadId);\n        setMessages(messages);\n      };\n\n  const threadList: NonNullable<\n    ExternalStoreAdapter[\"adapters\"]\n  >[\"threadList\"] = {\n    threadId,\n    onSwitchToNewThread: !onSwitchToNewThread\n      ? undefined\n      : async () => {\n          await onSwitchToNewThread();\n          setMessages([]);\n        },\n    onSwitchToThread: switchToThread,\n  };\n\n  const loadingRef = useRef(false);\n  const threadListItemRuntime = useThreadListItemRuntime({ optional: true });\n  useEffect(() => {\n    if (!threadListItemRuntime || !switchToThread || loadingRef.current) return;\n\n    const externalId = threadListItemRuntime.getState().externalId;\n    if (externalId) {\n      loadingRef.current = true;\n      switchToThread(externalId).finally(() => {\n        loadingRef.current = false;\n      });\n    }\n  }, []);\n\n  return useExternalStoreRuntime({\n    isRunning,\n    messages: threadMessages,\n    adapters: {\n      attachments,\n      threadList,\n    },\n    extras: {\n      [symbolLangGraphRuntimeExtras]: true,\n      send: handleSendMessage,\n    } satisfies LangGraphRuntimeExtras,\n    onNew: (msg) => {\n      const cancellations =\n        autoCancelPendingToolCalls !== false\n          ? getPendingToolCalls(messages).map(\n              (t) =>\n                ({\n                  type: \"tool\",\n                  name: t.name,\n                  tool_call_id: t.id,\n                  content: JSON.stringify({ cancelled: true }),\n                }) satisfies LangChainMessage & { type: \"tool\" },\n            )\n          : [];\n\n      return handleSendMessage(\n        [\n          ...cancellations,\n          {\n            type: \"human\",\n            content: getMessageContent(msg),\n          },\n        ],\n        {\n          runConfig: msg.runConfig,\n        },\n      );\n    },\n    onAddToolResult: async ({ toolCallId, toolName, result }) => {\n      // TODO parallel human in the loop calls\n      await handleSendMessage(\n        [\n          {\n            type: \"tool\",\n            name: toolName,\n            tool_call_id: toolCallId,\n            content: JSON.stringify(result),\n          },\n        ],\n        // TODO reuse runconfig here!\n        {},\n      );\n    },\n    onCancel: unstable_allowCancellation\n      ? async () => {\n          cancel();\n        }\n      : undefined,\n  });\n};\n","\"use client\";\n\nimport { useExternalMessageConverter } from \"@assistant-ui/react\";\nimport { LangChainMessage } from \"./types\";\nimport { ToolCallContentPart } from \"@assistant-ui/react\";\nimport { ThreadUserMessage } from \"@assistant-ui/react\";\n\nconst contentToParts = (\n  content: LangChainMessage[\"content\"],\n) => {\n  if (typeof content === \"string\")\n    return [{ type: \"text\" as const, text: content }];\n  return content\n    .map((part): ThreadUserMessage[\"content\"][number] | null => {\n      const type = part.type;\n      switch (type) {\n        case \"text\":\n          return { type: \"text\", text: part.text };\n        case \"image_url\":\n          if (typeof part.image_url === \"string\") {\n            return { type: \"image\", image: part.image_url };\n          } else {\n            return {\n              type: \"image\",\n              image: part.image_url.url,\n            };\n          }\n\n        case \"tool_use\":\n          return null;\n        default:\n          const _exhaustiveCheck: never = type;\n          throw new Error(`Unknown content part type: ${_exhaustiveCheck}`);\n      }\n    })\n    .filter((a) => a !== null);\n};\n\nexport const convertLangchainMessages: useExternalMessageConverter.Callback<\n  LangChainMessage\n> = (message) => {\n  switch (message.type) {\n    case \"system\":\n      return {\n        role: \"system\",\n        id: message.id,\n        content: [{ type: \"text\", text: message.content }],\n      };\n    case \"human\":\n      return {\n        role: \"user\",\n        id: message.id,\n        content: contentToParts(message.content),\n      };\n    case \"ai\":\n      return {\n        role: \"assistant\",\n        id: message.id,\n        content: [\n          ...contentToParts(message.content),\n          ...(message.tool_calls?.map(\n            (chunk): ToolCallContentPart => ({\n              type: \"tool-call\",\n              toolCallId: chunk.id,\n              toolName: chunk.name,\n              args: chunk.args,\n              argsText:\n                message.tool_call_chunks?.find((c) => c.id === chunk.id)\n                  ?.args ?? JSON.stringify(chunk.args),\n            }),\n          ) ?? []),\n        ],\n      };\n    case \"tool\":\n      return {\n        role: \"tool\",\n        toolName: message.name,\n        toolCallId: message.tool_call_id,\n        result: message.content,\n      };\n  }\n};\n","import { useState, useCallback, useRef } from \"react\";\nimport { INTERNAL } from \"@assistant-ui/react\";\n\nconst { generateId } = INTERNAL;\n\nexport type LangGraphCommand = {\n  resume: string;\n};\n\nexport type LangGraphSendMessageConfig = {\n  command?: LangGraphCommand;\n  runConfig?: unknown;\n};\n\nexport const useLangGraphMessages = <TMessage>({\n  stream,\n}: {\n  stream: (\n    messages: TMessage[],\n    config: LangGraphSendMessageConfig & { abortSignal: AbortSignal },\n  ) => Promise<\n    AsyncGenerator<{\n      event: string;\n      data: any;\n    }>\n  >;\n}) => {\n  const [messages, setMessages] = useState<TMessage[]>([]);\n  const abortControllerRef = useRef<AbortController | null>(null);\n\n  const sendMessage = useCallback(\n    async (newMessages: TMessage[], config: LangGraphSendMessageConfig) => {\n      const optimisticMessages = [...messages, ...newMessages];\n      if (newMessages.length > 0) {\n        setMessages(optimisticMessages);\n      }\n\n      const abortController = new AbortController();\n      abortControllerRef.current = abortController;\n      const response = await stream(newMessages, {\n        ...config,\n        abortSignal: abortController.signal,\n      });\n\n      const completeMessages: TMessage[] = [];\n      let partialMessages: Map<string, TMessage> = new Map();\n      for await (const chunk of response) {\n        if (chunk.event === \"messages/partial\") {\n          // bug fix: messages/complete is not sent for some python langchain backends\n          if (completeMessages.length === 0) {\n            completeMessages.push(...optimisticMessages);\n          }\n\n          for (const message of chunk.data) {\n            if (!message.id) throw new Error(\"Partial message missing id\");\n\n            partialMessages.set(message.id, message);\n          }\n        } else if (chunk.event === \"messages/complete\") {\n          for (const message of chunk.data) {\n            if (message.id) {\n              partialMessages.delete(message.id);\n            }\n\n            // bug fix: tool results can arrive before the message is complete ?\n            // if we have partial messages left, we add complete message to partial messages\n            if (partialMessages.size > 0) {\n              partialMessages.set(message.id ?? generateId(), message);\n            } else {\n              completeMessages.push(message);\n            }\n          }\n        } else {\n          continue;\n        }\n\n        setMessages([...completeMessages, ...partialMessages.values()]);\n      }\n      // if (partialMessages.size > 0) {\n      //   throw new Error(\"A partial message was not marked as complete\");\n      // }\n    },\n    [messages, stream],\n  );\n\n  const cancel = useCallback(() => {\n    if (abortControllerRef.current) {\n      abortControllerRef.current.abort();\n    }\n  }, [abortControllerRef]);\n\n  return { messages, sendMessage, cancel, setMessages };\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,IAAAA,gBAA4C;AAE5C,IAAAA,gBAIO;;;ACCP,IAAM,iBAAiB,CACrB,YACG;AACH,MAAI,OAAO,YAAY;AACrB,WAAO,CAAC,EAAE,MAAM,QAAiB,MAAM,QAAQ,CAAC;AAClD,SAAO,QACJ,IAAI,CAAC,SAAsD;AAC1D,UAAM,OAAO,KAAK;AAClB,YAAQ,MAAM;AAAA,MACZ,KAAK;AACH,eAAO,EAAE,MAAM,QAAQ,MAAM,KAAK,KAAK;AAAA,MACzC,KAAK;AACH,YAAI,OAAO,KAAK,cAAc,UAAU;AACtC,iBAAO,EAAE,MAAM,SAAS,OAAO,KAAK,UAAU;AAAA,QAChD,OAAO;AACL,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,OAAO,KAAK,UAAU;AAAA,UACxB;AAAA,QACF;AAAA,MAEF,KAAK;AACH,eAAO;AAAA,MACT;AACE,cAAM,mBAA0B;AAChC,cAAM,IAAI,MAAM,8BAA8B,gBAAgB,EAAE;AAAA,IACpE;AAAA,EACF,CAAC,EACA,OAAO,CAAC,MAAM,MAAM,IAAI;AAC7B;AAEO,IAAM,2BAET,CAAC,YAAY;AACf,UAAQ,QAAQ,MAAM;AAAA,IACpB,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,IAAI,QAAQ;AAAA,QACZ,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,QAAQ,QAAQ,CAAC;AAAA,MACnD;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,IAAI,QAAQ;AAAA,QACZ,SAAS,eAAe,QAAQ,OAAO;AAAA,MACzC;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,IAAI,QAAQ;AAAA,QACZ,SAAS;AAAA,UACP,GAAG,eAAe,QAAQ,OAAO;AAAA,UACjC,GAAI,QAAQ,YAAY;AAAA,YACtB,CAAC,WAAgC;AAAA,cAC/B,MAAM;AAAA,cACN,YAAY,MAAM;AAAA,cAClB,UAAU,MAAM;AAAA,cAChB,MAAM,MAAM;AAAA,cACZ,UACE,QAAQ,kBAAkB,KAAK,CAAC,MAAM,EAAE,OAAO,MAAM,EAAE,GACnD,QAAQ,KAAK,UAAU,MAAM,IAAI;AAAA,YACzC;AAAA,UACF,KAAK,CAAC;AAAA,QACR;AAAA,MACF;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,UAAU,QAAQ;AAAA,QAClB,YAAY,QAAQ;AAAA,QACpB,QAAQ,QAAQ;AAAA,MAClB;AAAA,EACJ;AACF;;;ACjFA,mBAA8C;AAC9C,IAAAC,gBAAyB;AAEzB,IAAM,EAAE,WAAW,IAAI;AAWhB,IAAM,uBAAuB,CAAW;AAAA,EAC7C;AACF,MAUM;AACJ,QAAM,CAAC,UAAU,WAAW,QAAI,uBAAqB,CAAC,CAAC;AACvD,QAAM,yBAAqB,qBAA+B,IAAI;AAE9D,QAAM,kBAAc;AAAA,IAClB,OAAO,aAAyB,WAAuC;AACrE,YAAM,qBAAqB,CAAC,GAAG,UAAU,GAAG,WAAW;AACvD,UAAI,YAAY,SAAS,GAAG;AAC1B,oBAAY,kBAAkB;AAAA,MAChC;AAEA,YAAM,kBAAkB,IAAI,gBAAgB;AAC5C,yBAAmB,UAAU;AAC7B,YAAM,WAAW,MAAM,OAAO,aAAa;AAAA,QACzC,GAAG;AAAA,QACH,aAAa,gBAAgB;AAAA,MAC/B,CAAC;AAED,YAAM,mBAA+B,CAAC;AACtC,UAAI,kBAAyC,oBAAI,IAAI;AACrD,uBAAiB,SAAS,UAAU;AAClC,YAAI,MAAM,UAAU,oBAAoB;AAEtC,cAAI,iBAAiB,WAAW,GAAG;AACjC,6BAAiB,KAAK,GAAG,kBAAkB;AAAA,UAC7C;AAEA,qBAAW,WAAW,MAAM,MAAM;AAChC,gBAAI,CAAC,QAAQ,GAAI,OAAM,IAAI,MAAM,4BAA4B;AAE7D,4BAAgB,IAAI,QAAQ,IAAI,OAAO;AAAA,UACzC;AAAA,QACF,WAAW,MAAM,UAAU,qBAAqB;AAC9C,qBAAW,WAAW,MAAM,MAAM;AAChC,gBAAI,QAAQ,IAAI;AACd,8BAAgB,OAAO,QAAQ,EAAE;AAAA,YACnC;AAIA,gBAAI,gBAAgB,OAAO,GAAG;AAC5B,8BAAgB,IAAI,QAAQ,MAAM,WAAW,GAAG,OAAO;AAAA,YACzD,OAAO;AACL,+BAAiB,KAAK,OAAO;AAAA,YAC/B;AAAA,UACF;AAAA,QACF,OAAO;AACL;AAAA,QACF;AAEA,oBAAY,CAAC,GAAG,kBAAkB,GAAG,gBAAgB,OAAO,CAAC,CAAC;AAAA,MAChE;AAAA,IAIF;AAAA,IACA,CAAC,UAAU,MAAM;AAAA,EACnB;AAEA,QAAM,aAAS,0BAAY,MAAM;AAC/B,QAAI,mBAAmB,SAAS;AAC9B,yBAAmB,QAAQ,MAAM;AAAA,IACnC;AAAA,EACF,GAAG,CAAC,kBAAkB,CAAC;AAEvB,SAAO,EAAE,UAAU,aAAa,QAAQ,YAAY;AACtD;;;AF/EA,IAAAC,gBAA6C;AAI7C,mCAAyC;AAEzC,IAAM,sBAAsB,CAAC,aAAiC;AAC5D,QAAM,mBAAmB,oBAAI,IAA+B;AAC5D,aAAW,WAAW,UAAU;AAC9B,QAAI,QAAQ,SAAS,MAAM;AACzB,iBAAW,YAAY,QAAQ,cAAc,CAAC,GAAG;AAC/C,yBAAiB,IAAI,SAAS,IAAI,QAAQ;AAAA,MAC5C;AAAA,IACF;AACA,QAAI,QAAQ,SAAS,QAAQ;AAC3B,uBAAiB,OAAO,QAAQ,YAAY;AAAA,IAC9C;AAAA,EACF;AAEA,SAAO,CAAC,GAAG,iBAAiB,OAAO,CAAC;AACtC;AAEA,IAAM,oBAAoB,CAAC,QAAuB;AAChD,QAAM,aAAa;AAAA,IACjB,GAAG,IAAI;AAAA,IACP,GAAG,IAAI,YAAY,QAAQ,CAAC,MAAM,EAAE,OAAO;AAAA,EAC7C;AACA,QAAM,UAAU,WAAW,IAAI,CAAC,SAAS;AACvC,UAAM,OAAO,KAAK;AAClB,YAAQ,MAAM;AAAA,MACZ,KAAK;AACH,eAAO,EAAE,MAAM,QAAiB,MAAM,KAAK,KAAK;AAAA,MAClD,KAAK;AACH,eAAO,EAAE,MAAM,aAAsB,WAAW,EAAE,KAAK,KAAK,MAAM,EAAE;AAAA,MAEtE,KAAK;AACH,cAAM,IAAI,MAAM,sCAAsC;AAAA,MACxD,KAAK;AACH,cAAM,IAAI,MAAM,0CAA0C;AAAA,MAE5D;AACE,cAAM,mBAA0B;AAChC,cAAM,IAAI,MAAM,8BAA8B,gBAAgB,EAAE;AAAA,IACpE;AAAA,EACF,CAAC;AAED,MAAI,QAAQ,WAAW,KAAK,QAAQ,CAAC,GAAG,SAAS,QAAQ;AACvD,WAAO,QAAQ,CAAC,EAAE,QAAQ;AAAA,EAC5B;AAEA,SAAO;AACT;AAEA,IAAM,+BAA+B,OAAO,0BAA0B;AAStE,IAAM,2BAA2B,CAAC,WAA4C;AAC5E,MACE,OAAO,WAAW,YAClB,UAAU,QACV,EAAE,gCAAgC;AAElC,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAEF,SAAO;AACT;AAEO,IAAM,0BAA0B,MAAM;AAC3C,QAAM,EAAE,KAAK,QAAI,yBAAU,CAAC,MAAM,yBAAyB,EAAE,MAAM,CAAC;AACpE,SAAO;AACT;AAEO,IAAM,iCAAiC,CAAC,YAA8B;AAC3E,QAAM,OAAO,wBAAwB;AACrC,SAAO,MAAM,KAAK,CAAC,GAAG,EAAE,QAAQ,CAAC;AACnC;AAEO,IAAM,sBAAsB,CAAC;AAAA,EAClC;AAAA,EACA,UAAU,EAAE,YAAY,IAAI,CAAC;AAAA,EAC7B;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,MAgCM;AACJ,QAAM,EAAE,UAAU,aAAa,QAAQ,YAAY,IAAI,qBAAqB;AAAA,IAC1E;AAAA,EACF,CAAC;AAED,QAAM,CAAC,WAAW,YAAY,QAAI,wBAAS,KAAK;AAChD,QAAM,oBAAoB,OACxBC,WACA,WACG;AACH,QAAI;AACF,mBAAa,IAAI;AACjB,YAAM,YAAYA,WAAU,MAAM;AAAA,IACpC,SAAS,OAAO;AACd,cAAQ,MAAM,6BAA6B,KAAK;AAAA,IAClD,UAAE;AACA,mBAAa,KAAK;AAAA,IACpB;AAAA,EACF;AAEA,QAAM,qBAAiB,2CAA4B;AAAA,IACjD,UAAU;AAAA,IACV;AAAA,IACA;AAAA,EACF,CAAC;AAED,MAAI,eAAe;AACjB,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AACF,MAAI;AACF,kBAAc,IAAI,2CAA6B;AAEjD,QAAM,iBAAiB,CAAC,mBACpB,SACA,OAAOC,cAAqB;AAC1B,UAAM,EAAE,UAAAD,UAAS,IAAI,MAAM,iBAAiBC,SAAQ;AACpD,gBAAYD,SAAQ;AAAA,EACtB;AAEJ,QAAM,aAEY;AAAA,IAChB;AAAA,IACA,qBAAqB,CAAC,sBAClB,SACA,YAAY;AACV,YAAM,oBAAoB;AAC1B,kBAAY,CAAC,CAAC;AAAA,IAChB;AAAA,IACJ,kBAAkB;AAAA,EACpB;AAEA,QAAM,iBAAa,sBAAO,KAAK;AAC/B,QAAM,4BAAwB,uDAAyB,EAAE,UAAU,KAAK,CAAC;AACzE,+BAAU,MAAM;AACd,QAAI,CAAC,yBAAyB,CAAC,kBAAkB,WAAW,QAAS;AAErE,UAAM,aAAa,sBAAsB,SAAS,EAAE;AACpD,QAAI,YAAY;AACd,iBAAW,UAAU;AACrB,qBAAe,UAAU,EAAE,QAAQ,MAAM;AACvC,mBAAW,UAAU;AAAA,MACvB,CAAC;AAAA,IACH;AAAA,EACF,GAAG,CAAC,CAAC;AAEL,aAAO,uCAAwB;AAAA,IAC7B;AAAA,IACA,UAAU;AAAA,IACV,UAAU;AAAA,MACR;AAAA,MACA;AAAA,IACF;AAAA,IACA,QAAQ;AAAA,MACN,CAAC,4BAA4B,GAAG;AAAA,MAChC,MAAM;AAAA,IACR;AAAA,IACA,OAAO,CAAC,QAAQ;AACd,YAAM,gBACJ,+BAA+B,QAC3B,oBAAoB,QAAQ,EAAE;AAAA,QAC5B,CAAC,OACE;AAAA,UACC,MAAM;AAAA,UACN,MAAM,EAAE;AAAA,UACR,cAAc,EAAE;AAAA,UAChB,SAAS,KAAK,UAAU,EAAE,WAAW,KAAK,CAAC;AAAA,QAC7C;AAAA,MACJ,IACA,CAAC;AAEP,aAAO;AAAA,QACL;AAAA,UACE,GAAG;AAAA,UACH;AAAA,YACE,MAAM;AAAA,YACN,SAAS,kBAAkB,GAAG;AAAA,UAChC;AAAA,QACF;AAAA,QACA;AAAA,UACE,WAAW,IAAI;AAAA,QACjB;AAAA,MACF;AAAA,IACF;AAAA,IACA,iBAAiB,OAAO,EAAE,YAAY,UAAU,OAAO,MAAM;AAE3D,YAAM;AAAA,QACJ;AAAA,UACE;AAAA,YACE,MAAM;AAAA,YACN,MAAM;AAAA,YACN,cAAc;AAAA,YACd,SAAS,KAAK,UAAU,MAAM;AAAA,UAChC;AAAA,QACF;AAAA;AAAA,QAEA,CAAC;AAAA,MACH;AAAA,IACF;AAAA,IACA,UAAU,6BACN,YAAY;AACV,aAAO;AAAA,IACT,IACA;AAAA,EACN,CAAC;AACH;","names":["import_react","import_react","import_react","messages","threadId"]}

package/dist/index.mjs

@@ -1,8 +1,9 @@
 // src/useLangGraphRuntime.ts
-import { useState as useState2 } from "react";
+import { useEffect, useRef as useRef2, useState as useState2 } from "react";
 import {
   useExternalMessageConverter,
-  useExternalStoreRuntime
+  useExternalStoreRuntime,
+  useThread
 } from "@assistant-ui/react";
 
 // src/convertLangchainMessages.ts
@@ -73,20 +74,26 @@ var convertLangchainMessages = (message) => {
 };
 
 // src/useLangGraphMessages.ts
-import { useState, useCallback } from "react";
+import { useState, useCallback, useRef } from "react";
 import { INTERNAL } from "@assistant-ui/react";
 var { generateId } = INTERNAL;
 var useLangGraphMessages = ({
   stream
 }) => {
   const [messages, setMessages] = useState([]);
+  const abortControllerRef = useRef(null);
   const sendMessage = useCallback(
-    async (newMessages) => {
+    async (newMessages, config) => {
       const optimisticMessages = [...messages, ...newMessages];
       if (newMessages.length > 0) {
         setMessages(optimisticMessages);
       }
-      const response = await stream(newMessages);
+      const abortController = new AbortController();
+      abortControllerRef.current = abortController;
+      const response = await stream(newMessages, {
+        ...config,
+        abortSignal: abortController.signal
+      });
       const completeMessages = [];
       let partialMessages = /* @__PURE__ */ new Map();
       for await (const chunk of response) {
@@ -117,11 +124,17 @@ var useLangGraphMessages = ({
     },
     [messages, stream]
   );
-  return { messages, sendMessage, setMessages };
+  const cancel = useCallback(() => {
+    if (abortControllerRef.current) {
+      abortControllerRef.current.abort();
+    }
+  }, [abortControllerRef]);
+  return { messages, sendMessage, cancel, setMessages };
 };
 
 // src/useLangGraphRuntime.ts
 import { SimpleImageAttachmentAdapter } from "@assistant-ui/react";
+import { useThreadListItemRuntime } from "@assistant-ui/react/context/react/ThreadListItemContext";
 var getPendingToolCalls = (messages) => {
   const pendingToolCalls = /* @__PURE__ */ new Map();
   for (const message of messages) {
@@ -162,23 +175,40 @@ var getMessageContent = (msg) => {
   }
   return content;
 };
+var symbolLangGraphRuntimeExtras = Symbol("langgraph-runtime-extras");
+var asLangGraphRuntimeExtras = (extras) => {
+  if (typeof extras !== "object" || extras == null || !(symbolLangGraphRuntimeExtras in extras))
+    throw new Error(
+      "This method can only be called when you are using useLangGraphRuntime"
+    );
+  return extras;
+};
+var useLangGraphRuntimeSend = () => {
+  const { send } = useThread((t) => asLangGraphRuntimeExtras(t.extras));
+  return send;
+};
+var useLangGraphRuntimeSendCommand = (command) => {
+  const send = useLangGraphRuntimeSend();
+  return () => send([], { command });
+};
 var useLangGraphRuntime = ({
-  threadId,
   autoCancelPendingToolCalls,
+  adapters: { attachments } = {},
   unstable_allowImageAttachments,
+  unstable_allowCancellation,
   stream,
+  threadId,
   onSwitchToNewThread,
-  onSwitchToThread,
-  adapters: { attachments } = {}
+  onSwitchToThread
 }) => {
-  const { messages, sendMessage, setMessages } = useLangGraphMessages({
+  const { messages, sendMessage, cancel, setMessages } = useLangGraphMessages({
     stream
   });
   const [isRunning, setIsRunning] = useState2(false);
-  const handleSendMessage = async (messages2) => {
+  const handleSendMessage = async (messages2, config) => {
     try {
       setIsRunning(true);
-      await sendMessage(messages2);
+      await sendMessage(messages2, config);
     } catch (error) {
       console.error("Error streaming messages:", error);
     } finally {
@@ -196,17 +226,30 @@ var useLangGraphRuntime = ({
     );
   if (unstable_allowImageAttachments)
     attachments = new SimpleImageAttachmentAdapter();
+  const switchToThread = !onSwitchToThread ? void 0 : async (threadId2) => {
+    const { messages: messages2 } = await onSwitchToThread(threadId2);
+    setMessages(messages2);
+  };
   const threadList = {
     threadId,
     onSwitchToNewThread: !onSwitchToNewThread ? void 0 : async () => {
       await onSwitchToNewThread();
       setMessages([]);
     },
-    onSwitchToThread: !onSwitchToThread ? void 0 : async (threadId2) => {
-      const { messages: messages2 } = await onSwitchToThread(threadId2);
-      setMessages(messages2);
-    }
+    onSwitchToThread: switchToThread
   };
+  const loadingRef = useRef2(false);
+  const threadListItemRuntime = useThreadListItemRuntime({ optional: true });
+  useEffect(() => {
+    if (!threadListItemRuntime || !switchToThread || loadingRef.current) return;
+    const externalId = threadListItemRuntime.getState().externalId;
+    if (externalId) {
+      loadingRef.current = true;
+      switchToThread(externalId).finally(() => {
+        loadingRef.current = false;
+      });
+    }
+  }, []);
   return useExternalStoreRuntime({
     isRunning,
     messages: threadMessages,
@@ -214,6 +257,10 @@ var useLangGraphRuntime = ({
       attachments,
       threadList
     },
+    extras: {
+      [symbolLangGraphRuntimeExtras]: true,
+      send: handleSendMessage
+    },
     onNew: (msg) => {
       const cancellations = autoCancelPendingToolCalls !== false ? getPendingToolCalls(messages).map(
         (t) => ({
@@ -223,29 +270,43 @@ var useLangGraphRuntime = ({
           content: JSON.stringify({ cancelled: true })
         })
       ) : [];
-      return handleSendMessage([
-        ...cancellations,
+      return handleSendMessage(
+        [
+          ...cancellations,
+          {
+            type: "human",
+            content: getMessageContent(msg)
+          }
+        ],
         {
-          type: "human",
-          content: getMessageContent(msg)
+          runConfig: msg.runConfig
         }
-      ]);
+      );
     },
     onAddToolResult: async ({ toolCallId, toolName, result }) => {
-      await handleSendMessage([
-        {
-          type: "tool",
-          name: toolName,
-          tool_call_id: toolCallId,
-          content: JSON.stringify(result)
-        }
-      ]);
-    }
+      await handleSendMessage(
+        [
+          {
+            type: "tool",
+            name: toolName,
+            tool_call_id: toolCallId,
+            content: JSON.stringify(result)
+          }
+        ],
+        // TODO reuse runconfig here!
+        {}
+      );
+    },
+    onCancel: unstable_allowCancellation ? async () => {
+      cancel();
+    } : void 0
   });
 };
 export {
   convertLangchainMessages,
   useLangGraphMessages,
-  useLangGraphRuntime
+  useLangGraphRuntime,
+  useLangGraphRuntimeSend,
+  useLangGraphRuntimeSendCommand
 };
 //# sourceMappingURL=index.mjs.map

package/dist/index.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/useLangGraphRuntime.ts","../src/convertLangchainMessages.ts","../src/useLangGraphMessages.ts"],"sourcesContent":["import { useState } from \"react\";\nimport { LangChainMessage, LangChainToolCall } from \"./types\";\nimport {\n  useExternalMessageConverter,\n  useExternalStoreRuntime,\n} from \"@assistant-ui/react\";\nimport { convertLangchainMessages } from \"./convertLangchainMessages\";\nimport { useLangGraphMessages } from \"./useLangGraphMessages\";\nimport { SimpleImageAttachmentAdapter } from \"@assistant-ui/react\";\nimport { AttachmentAdapter } from \"@assistant-ui/react\";\nimport { AppendMessage } from \"@assistant-ui/react\";\nimport { ExternalStoreAdapter } from \"@assistant-ui/react\";\n\nconst getPendingToolCalls = (messages: LangChainMessage[]) => {\n  const pendingToolCalls = new Map<string, LangChainToolCall>();\n  for (const message of messages) {\n    if (message.type === \"ai\") {\n      for (const toolCall of message.tool_calls ?? []) {\n        pendingToolCalls.set(toolCall.id, toolCall);\n      }\n    }\n    if (message.type === \"tool\") {\n      pendingToolCalls.delete(message.tool_call_id);\n    }\n  }\n\n  return [...pendingToolCalls.values()];\n};\n\nconst getMessageContent = (msg: AppendMessage) => {\n  const allContent = [\n    ...msg.content,\n    ...msg.attachments.flatMap((a) => a.content),\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\n      case \"audio\":\n        throw new Error(\"Audio appends are not supported yet.\");\n      case \"tool-call\":\n        throw new Error(\"Tool call appends are not supported yet.\");\n\n      default:\n        const _exhaustiveCheck: never = type;\n        throw new Error(`Unknown content part type: ${_exhaustiveCheck}`);\n    }\n  });\n\n  if (content.length === 1 && content[0]?.type === \"text\") {\n    return content[0].text ?? \"\";\n  }\n\n  return content;\n};\n\nexport const useLangGraphRuntime = ({\n  threadId,\n  autoCancelPendingToolCalls,\n  unstable_allowImageAttachments,\n  stream,\n  onSwitchToNewThread,\n  onSwitchToThread,\n  adapters: { attachments } = {},\n}: {\n  threadId?: string | undefined;\n  autoCancelPendingToolCalls?: boolean | undefined;\n  /**\n   * @deprecated Use `adapters: { attachments: new SimpleImageAttachmentAdapter() }` instead. This option will be removed in a future version.\n   */\n  unstable_allowImageAttachments?: boolean | undefined;\n  stream: (messages: LangChainMessage[]) => Promise<\n    AsyncGenerator<{\n      event: string;\n      data: any;\n    }>\n  >;\n  onSwitchToNewThread?: () => Promise<void> | void;\n  onSwitchToThread?: (\n    threadId: string,\n  ) => Promise<{ messages: LangChainMessage[] }>;\n  adapters?:\n    | {\n        attachments?: AttachmentAdapter;\n      }\n    | undefined;\n}) => {\n  const { messages, sendMessage, setMessages } = useLangGraphMessages({\n    stream,\n  });\n\n  const [isRunning, setIsRunning] = useState(false);\n  const handleSendMessage = async (messages: LangChainMessage[]) => {\n    try {\n      setIsRunning(true);\n      await sendMessage(messages);\n    } catch (error) {\n      console.error(\"Error streaming messages:\", error);\n    } finally {\n      setIsRunning(false);\n    }\n  };\n\n  const threadMessages = useExternalMessageConverter({\n    callback: convertLangchainMessages,\n    messages,\n    isRunning,\n  });\n\n  if (attachments && unstable_allowImageAttachments)\n    throw new Error(\n      \"Replace unstable_allowImageAttachments with `adapters: { attachments: new SimpleImageAttachmentAdapter() }`.\",\n    );\n  if (unstable_allowImageAttachments)\n    attachments = new SimpleImageAttachmentAdapter();\n\n  const threadList: NonNullable<\n    ExternalStoreAdapter[\"adapters\"]\n  >[\"threadList\"] = {\n    threadId,\n    onSwitchToNewThread: !onSwitchToNewThread\n      ? undefined\n      : async () => {\n          await onSwitchToNewThread();\n          setMessages([]);\n        },\n    onSwitchToThread: !onSwitchToThread\n      ? undefined\n      : async (threadId) => {\n          const { messages } = await onSwitchToThread(threadId);\n          setMessages(messages);\n        },\n  };\n\n  return useExternalStoreRuntime({\n    isRunning,\n    messages: threadMessages,\n    adapters: {\n      attachments,\n      threadList,\n    },\n    onNew: (msg) => {\n      const cancellations =\n        autoCancelPendingToolCalls !== false\n          ? getPendingToolCalls(messages).map(\n              (t) =>\n                ({\n                  type: \"tool\",\n                  name: t.name,\n                  tool_call_id: t.id,\n                  content: JSON.stringify({ cancelled: true }),\n                }) satisfies LangChainMessage & { type: \"tool\" },\n            )\n          : [];\n\n      return handleSendMessage([\n        ...cancellations,\n        {\n          type: \"human\",\n          content: getMessageContent(msg),\n        },\n      ]);\n    },\n    onAddToolResult: async ({ toolCallId, toolName, result }) => {\n      await handleSendMessage([\n        {\n          type: \"tool\",\n          name: toolName,\n          tool_call_id: toolCallId,\n          content: JSON.stringify(result),\n        },\n      ]);\n    },\n  });\n};\n","\"use client\";\n\nimport { useExternalMessageConverter } from \"@assistant-ui/react\";\nimport { LangChainMessage } from \"./types\";\nimport { ToolCallContentPart } from \"@assistant-ui/react\";\nimport { ThreadUserMessage } from \"@assistant-ui/react\";\n\nconst contentToParts = (\n  content: LangChainMessage[\"content\"],\n) => {\n  if (typeof content === \"string\")\n    return [{ type: \"text\" as const, text: content }];\n  return content\n    .map((part): ThreadUserMessage[\"content\"][number] | null => {\n      const type = part.type;\n      switch (type) {\n        case \"text\":\n          return { type: \"text\", text: part.text };\n        case \"image_url\":\n          if (typeof part.image_url === \"string\") {\n            return { type: \"image\", image: part.image_url };\n          } else {\n            return {\n              type: \"image\",\n              image: part.image_url.url,\n            };\n          }\n\n        case \"tool_use\":\n          return null;\n        default:\n          const _exhaustiveCheck: never = type;\n          throw new Error(`Unknown content part type: ${_exhaustiveCheck}`);\n      }\n    })\n    .filter((a) => a !== null);\n};\n\nexport const convertLangchainMessages: useExternalMessageConverter.Callback<\n  LangChainMessage\n> = (message) => {\n  switch (message.type) {\n    case \"system\":\n      return {\n        role: \"system\",\n        id: message.id,\n        content: [{ type: \"text\", text: message.content }],\n      };\n    case \"human\":\n      return {\n        role: \"user\",\n        id: message.id,\n        content: contentToParts(message.content),\n      };\n    case \"ai\":\n      return {\n        role: \"assistant\",\n        id: message.id,\n        content: [\n          ...contentToParts(message.content),\n          ...(message.tool_calls?.map(\n            (chunk): ToolCallContentPart => ({\n              type: \"tool-call\",\n              toolCallId: chunk.id,\n              toolName: chunk.name,\n              args: chunk.args,\n              argsText:\n                message.tool_call_chunks?.find((c) => c.id === chunk.id)\n                  ?.args ?? JSON.stringify(chunk.args),\n            }),\n          ) ?? []),\n        ],\n      };\n    case \"tool\":\n      return {\n        role: \"tool\",\n        toolName: message.name,\n        toolCallId: message.tool_call_id,\n        result: message.content,\n      };\n  }\n};\n","import { useState, useCallback } from \"react\";\nimport { INTERNAL } from \"@assistant-ui/react\";\n\nconst { generateId } = INTERNAL;\n\nexport const useLangGraphMessages = <TMessage>({\n  stream,\n}: {\n  stream: (messages: TMessage[]) => Promise<\n    AsyncGenerator<{\n      event: string;\n      data: any;\n    }>\n  >;\n}) => {\n  const [messages, setMessages] = useState<TMessage[]>([]);\n\n  const sendMessage = useCallback(\n    async (newMessages: TMessage[]) => {\n      const optimisticMessages = [...messages, ...newMessages];\n      if (newMessages.length > 0) {\n        setMessages(optimisticMessages);\n      }\n\n      const response = await stream(newMessages);\n\n      const completeMessages: TMessage[] = [];\n      let partialMessages: Map<string, TMessage> = new Map();\n      for await (const chunk of response) {\n        if (chunk.event === \"messages/partial\") {\n          // bug fix: messages/complete is not sent for some python langchain backends\n          if (completeMessages.length === 0) {\n            completeMessages.push(...optimisticMessages);\n          }\n\n          for (const message of chunk.data) {\n            if (!message.id) throw new Error(\"Partial message missing id\");\n\n            partialMessages.set(message.id, message);\n          }\n        } else if (chunk.event === \"messages/complete\") {\n          for (const message of chunk.data) {\n            if (message.id) {\n              partialMessages.delete(message.id);\n            }\n\n            // bug fix: tool results can arrive before the message is complete ?\n            // if we have partial messages left, we add complete message to partial messages\n            if (partialMessages.size > 0) {\n              partialMessages.set(message.id ?? generateId(), message);\n            } else {\n              completeMessages.push(message);\n            }\n          }\n        } else {\n          continue;\n        }\n\n        setMessages([...completeMessages, ...partialMessages.values()]);\n      }\n      // if (partialMessages.size > 0) {\n      //   throw new Error(\"A partial message was not marked as complete\");\n      // }\n    },\n    [messages, stream],\n  );\n\n  return { messages, sendMessage, setMessages };\n};\n"],"mappings":";AAAA,SAAS,YAAAA,iBAAgB;AAEzB;AAAA,EACE;AAAA,EACA;AAAA,OACK;;;ACEP,IAAM,iBAAiB,CACrB,YACG;AACH,MAAI,OAAO,YAAY;AACrB,WAAO,CAAC,EAAE,MAAM,QAAiB,MAAM,QAAQ,CAAC;AAClD,SAAO,QACJ,IAAI,CAAC,SAAsD;AAC1D,UAAM,OAAO,KAAK;AAClB,YAAQ,MAAM;AAAA,MACZ,KAAK;AACH,eAAO,EAAE,MAAM,QAAQ,MAAM,KAAK,KAAK;AAAA,MACzC,KAAK;AACH,YAAI,OAAO,KAAK,cAAc,UAAU;AACtC,iBAAO,EAAE,MAAM,SAAS,OAAO,KAAK,UAAU;AAAA,QAChD,OAAO;AACL,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,OAAO,KAAK,UAAU;AAAA,UACxB;AAAA,QACF;AAAA,MAEF,KAAK;AACH,eAAO;AAAA,MACT;AACE,cAAM,mBAA0B;AAChC,cAAM,IAAI,MAAM,8BAA8B,gBAAgB,EAAE;AAAA,IACpE;AAAA,EACF,CAAC,EACA,OAAO,CAAC,MAAM,MAAM,IAAI;AAC7B;AAEO,IAAM,2BAET,CAAC,YAAY;AACf,UAAQ,QAAQ,MAAM;AAAA,IACpB,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,IAAI,QAAQ;AAAA,QACZ,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,QAAQ,QAAQ,CAAC;AAAA,MACnD;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,IAAI,QAAQ;AAAA,QACZ,SAAS,eAAe,QAAQ,OAAO;AAAA,MACzC;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,IAAI,QAAQ;AAAA,QACZ,SAAS;AAAA,UACP,GAAG,eAAe,QAAQ,OAAO;AAAA,UACjC,GAAI,QAAQ,YAAY;AAAA,YACtB,CAAC,WAAgC;AAAA,cAC/B,MAAM;AAAA,cACN,YAAY,MAAM;AAAA,cAClB,UAAU,MAAM;AAAA,cAChB,MAAM,MAAM;AAAA,cACZ,UACE,QAAQ,kBAAkB,KAAK,CAAC,MAAM,EAAE,OAAO,MAAM,EAAE,GACnD,QAAQ,KAAK,UAAU,MAAM,IAAI;AAAA,YACzC;AAAA,UACF,KAAK,CAAC;AAAA,QACR;AAAA,MACF;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,UAAU,QAAQ;AAAA,QAClB,YAAY,QAAQ;AAAA,QACpB,QAAQ,QAAQ;AAAA,MAClB;AAAA,EACJ;AACF;;;ACjFA,SAAS,UAAU,mBAAmB;AACtC,SAAS,gBAAgB;AAEzB,IAAM,EAAE,WAAW,IAAI;AAEhB,IAAM,uBAAuB,CAAW;AAAA,EAC7C;AACF,MAOM;AACJ,QAAM,CAAC,UAAU,WAAW,IAAI,SAAqB,CAAC,CAAC;AAEvD,QAAM,cAAc;AAAA,IAClB,OAAO,gBAA4B;AACjC,YAAM,qBAAqB,CAAC,GAAG,UAAU,GAAG,WAAW;AACvD,UAAI,YAAY,SAAS,GAAG;AAC1B,oBAAY,kBAAkB;AAAA,MAChC;AAEA,YAAM,WAAW,MAAM,OAAO,WAAW;AAEzC,YAAM,mBAA+B,CAAC;AACtC,UAAI,kBAAyC,oBAAI,IAAI;AACrD,uBAAiB,SAAS,UAAU;AAClC,YAAI,MAAM,UAAU,oBAAoB;AAEtC,cAAI,iBAAiB,WAAW,GAAG;AACjC,6BAAiB,KAAK,GAAG,kBAAkB;AAAA,UAC7C;AAEA,qBAAW,WAAW,MAAM,MAAM;AAChC,gBAAI,CAAC,QAAQ,GAAI,OAAM,IAAI,MAAM,4BAA4B;AAE7D,4BAAgB,IAAI,QAAQ,IAAI,OAAO;AAAA,UACzC;AAAA,QACF,WAAW,MAAM,UAAU,qBAAqB;AAC9C,qBAAW,WAAW,MAAM,MAAM;AAChC,gBAAI,QAAQ,IAAI;AACd,8BAAgB,OAAO,QAAQ,EAAE;AAAA,YACnC;AAIA,gBAAI,gBAAgB,OAAO,GAAG;AAC5B,8BAAgB,IAAI,QAAQ,MAAM,WAAW,GAAG,OAAO;AAAA,YACzD,OAAO;AACL,+BAAiB,KAAK,OAAO;AAAA,YAC/B;AAAA,UACF;AAAA,QACF,OAAO;AACL;AAAA,QACF;AAEA,oBAAY,CAAC,GAAG,kBAAkB,GAAG,gBAAgB,OAAO,CAAC,CAAC;AAAA,MAChE;AAAA,IAIF;AAAA,IACA,CAAC,UAAU,MAAM;AAAA,EACnB;AAEA,SAAO,EAAE,UAAU,aAAa,YAAY;AAC9C;;;AF5DA,SAAS,oCAAoC;AAK7C,IAAM,sBAAsB,CAAC,aAAiC;AAC5D,QAAM,mBAAmB,oBAAI,IAA+B;AAC5D,aAAW,WAAW,UAAU;AAC9B,QAAI,QAAQ,SAAS,MAAM;AACzB,iBAAW,YAAY,QAAQ,cAAc,CAAC,GAAG;AAC/C,yBAAiB,IAAI,SAAS,IAAI,QAAQ;AAAA,MAC5C;AAAA,IACF;AACA,QAAI,QAAQ,SAAS,QAAQ;AAC3B,uBAAiB,OAAO,QAAQ,YAAY;AAAA,IAC9C;AAAA,EACF;AAEA,SAAO,CAAC,GAAG,iBAAiB,OAAO,CAAC;AACtC;AAEA,IAAM,oBAAoB,CAAC,QAAuB;AAChD,QAAM,aAAa;AAAA,IACjB,GAAG,IAAI;AAAA,IACP,GAAG,IAAI,YAAY,QAAQ,CAAC,MAAM,EAAE,OAAO;AAAA,EAC7C;AACA,QAAM,UAAU,WAAW,IAAI,CAAC,SAAS;AACvC,UAAM,OAAO,KAAK;AAClB,YAAQ,MAAM;AAAA,MACZ,KAAK;AACH,eAAO,EAAE,MAAM,QAAiB,MAAM,KAAK,KAAK;AAAA,MAClD,KAAK;AACH,eAAO,EAAE,MAAM,aAAsB,WAAW,EAAE,KAAK,KAAK,MAAM,EAAE;AAAA,MAEtE,KAAK;AACH,cAAM,IAAI,MAAM,sCAAsC;AAAA,MACxD,KAAK;AACH,cAAM,IAAI,MAAM,0CAA0C;AAAA,MAE5D;AACE,cAAM,mBAA0B;AAChC,cAAM,IAAI,MAAM,8BAA8B,gBAAgB,EAAE;AAAA,IACpE;AAAA,EACF,CAAC;AAED,MAAI,QAAQ,WAAW,KAAK,QAAQ,CAAC,GAAG,SAAS,QAAQ;AACvD,WAAO,QAAQ,CAAC,EAAE,QAAQ;AAAA,EAC5B;AAEA,SAAO;AACT;AAEO,IAAM,sBAAsB,CAAC;AAAA,EAClC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA,UAAU,EAAE,YAAY,IAAI,CAAC;AAC/B,MAsBM;AACJ,QAAM,EAAE,UAAU,aAAa,YAAY,IAAI,qBAAqB;AAAA,IAClE;AAAA,EACF,CAAC;AAED,QAAM,CAAC,WAAW,YAAY,IAAIC,UAAS,KAAK;AAChD,QAAM,oBAAoB,OAAOC,cAAiC;AAChE,QAAI;AACF,mBAAa,IAAI;AACjB,YAAM,YAAYA,SAAQ;AAAA,IAC5B,SAAS,OAAO;AACd,cAAQ,MAAM,6BAA6B,KAAK;AAAA,IAClD,UAAE;AACA,mBAAa,KAAK;AAAA,IACpB;AAAA,EACF;AAEA,QAAM,iBAAiB,4BAA4B;AAAA,IACjD,UAAU;AAAA,IACV;AAAA,IACA;AAAA,EACF,CAAC;AAED,MAAI,eAAe;AACjB,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AACF,MAAI;AACF,kBAAc,IAAI,6BAA6B;AAEjD,QAAM,aAEY;AAAA,IAChB;AAAA,IACA,qBAAqB,CAAC,sBAClB,SACA,YAAY;AACV,YAAM,oBAAoB;AAC1B,kBAAY,CAAC,CAAC;AAAA,IAChB;AAAA,IACJ,kBAAkB,CAAC,mBACf,SACA,OAAOC,cAAa;AAClB,YAAM,EAAE,UAAAD,UAAS,IAAI,MAAM,iBAAiBC,SAAQ;AACpD,kBAAYD,SAAQ;AAAA,IACtB;AAAA,EACN;AAEA,SAAO,wBAAwB;AAAA,IAC7B;AAAA,IACA,UAAU;AAAA,IACV,UAAU;AAAA,MACR;AAAA,MACA;AAAA,IACF;AAAA,IACA,OAAO,CAAC,QAAQ;AACd,YAAM,gBACJ,+BAA+B,QAC3B,oBAAoB,QAAQ,EAAE;AAAA,QAC5B,CAAC,OACE;AAAA,UACC,MAAM;AAAA,UACN,MAAM,EAAE;AAAA,UACR,cAAc,EAAE;AAAA,UAChB,SAAS,KAAK,UAAU,EAAE,WAAW,KAAK,CAAC;AAAA,QAC7C;AAAA,MACJ,IACA,CAAC;AAEP,aAAO,kBAAkB;AAAA,QACvB,GAAG;AAAA,QACH;AAAA,UACE,MAAM;AAAA,UACN,SAAS,kBAAkB,GAAG;AAAA,QAChC;AAAA,MACF,CAAC;AAAA,IACH;AAAA,IACA,iBAAiB,OAAO,EAAE,YAAY,UAAU,OAAO,MAAM;AAC3D,YAAM,kBAAkB;AAAA,QACtB;AAAA,UACE,MAAM;AAAA,UACN,MAAM;AAAA,UACN,cAAc;AAAA,UACd,SAAS,KAAK,UAAU,MAAM;AAAA,QAChC;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF,CAAC;AACH;","names":["useState","useState","messages","threadId"]}
+{"version":3,"sources":["../src/useLangGraphRuntime.ts","../src/convertLangchainMessages.ts","../src/useLangGraphMessages.ts"],"sourcesContent":["import { useEffect, useRef, useState } from \"react\";\nimport { LangChainMessage, LangChainToolCall } from \"./types\";\nimport {\n  useExternalMessageConverter,\n  useExternalStoreRuntime,\n  useThread,\n} from \"@assistant-ui/react\";\nimport { convertLangchainMessages } from \"./convertLangchainMessages\";\nimport {\n  LangGraphCommand,\n  LangGraphSendMessageConfig,\n  useLangGraphMessages,\n} from \"./useLangGraphMessages\";\nimport { SimpleImageAttachmentAdapter } from \"@assistant-ui/react\";\nimport { AttachmentAdapter } from \"@assistant-ui/react\";\nimport { AppendMessage } from \"@assistant-ui/react\";\nimport { ExternalStoreAdapter } from \"@assistant-ui/react\";\nimport { useThreadListItemRuntime } from \"@assistant-ui/react/context/react/ThreadListItemContext\";\n\nconst getPendingToolCalls = (messages: LangChainMessage[]) => {\n  const pendingToolCalls = new Map<string, LangChainToolCall>();\n  for (const message of messages) {\n    if (message.type === \"ai\") {\n      for (const toolCall of message.tool_calls ?? []) {\n        pendingToolCalls.set(toolCall.id, toolCall);\n      }\n    }\n    if (message.type === \"tool\") {\n      pendingToolCalls.delete(message.tool_call_id);\n    }\n  }\n\n  return [...pendingToolCalls.values()];\n};\n\nconst getMessageContent = (msg: AppendMessage) => {\n  const allContent = [\n    ...msg.content,\n    ...msg.attachments.flatMap((a) => a.content),\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\n      case \"audio\":\n        throw new Error(\"Audio appends are not supported yet.\");\n      case \"tool-call\":\n        throw new Error(\"Tool call appends are not supported yet.\");\n\n      default:\n        const _exhaustiveCheck: never = type;\n        throw new Error(`Unknown content part type: ${_exhaustiveCheck}`);\n    }\n  });\n\n  if (content.length === 1 && content[0]?.type === \"text\") {\n    return content[0].text ?? \"\";\n  }\n\n  return content;\n};\n\nconst symbolLangGraphRuntimeExtras = Symbol(\"langgraph-runtime-extras\");\ntype LangGraphRuntimeExtras = {\n  [symbolLangGraphRuntimeExtras]: true;\n  send: (\n    messages: LangChainMessage[],\n    config: LangGraphSendMessageConfig,\n  ) => Promise<void>;\n};\n\nconst asLangGraphRuntimeExtras = (extras: unknown): LangGraphRuntimeExtras => {\n  if (\n    typeof extras !== \"object\" ||\n    extras == null ||\n    !(symbolLangGraphRuntimeExtras in extras)\n  )\n    throw new Error(\n      \"This method can only be called when you are using useLangGraphRuntime\",\n    );\n\n  return extras as LangGraphRuntimeExtras;\n};\n\nexport const useLangGraphRuntimeSend = () => {\n  const { send } = useThread((t) => asLangGraphRuntimeExtras(t.extras));\n  return send;\n};\n\nexport const useLangGraphRuntimeSendCommand = (command: LangGraphCommand) => {\n  const send = useLangGraphRuntimeSend();\n  return () => send([], { command });\n};\n\nexport const useLangGraphRuntime = ({\n  autoCancelPendingToolCalls,\n  adapters: { attachments } = {},\n  unstable_allowImageAttachments,\n  unstable_allowCancellation,\n  stream,\n  threadId,\n  onSwitchToNewThread,\n  onSwitchToThread,\n}: {\n  /**\n   * @deprecated For thread management use `useCloudThreadListRuntime` instead. This option will be removed in a future version.\n   */\n  threadId?: string | undefined;\n  autoCancelPendingToolCalls?: boolean | undefined;\n  /**\n   * @deprecated Use `adapters: { attachments: new SimpleImageAttachmentAdapter() }` instead. This option will be removed in a future version.\n   */\n  unstable_allowImageAttachments?: boolean | undefined;\n  unstable_allowCancellation?: boolean | undefined;\n  stream: (\n    messages: LangChainMessage[],\n    config: LangGraphSendMessageConfig,\n  ) => Promise<\n    AsyncGenerator<{\n      event: string;\n      data: any;\n    }>\n  >;\n  /**\n   * @deprecated For thread management use `useCloudThreadListRuntime` instead. This option will be removed in a future version.\n   */\n  onSwitchToNewThread?: () => Promise<void> | void;\n  onSwitchToThread?: (\n    threadId: string,\n  ) => Promise<{ messages: LangChainMessage[] }>;\n  adapters?:\n    | {\n        attachments?: AttachmentAdapter;\n      }\n    | undefined;\n}) => {\n  const { messages, sendMessage, cancel, setMessages } = useLangGraphMessages({\n    stream,\n  });\n\n  const [isRunning, setIsRunning] = useState(false);\n  const handleSendMessage = async (\n    messages: LangChainMessage[],\n    config: LangGraphSendMessageConfig,\n  ) => {\n    try {\n      setIsRunning(true);\n      await sendMessage(messages, config);\n    } catch (error) {\n      console.error(\"Error streaming messages:\", error);\n    } finally {\n      setIsRunning(false);\n    }\n  };\n\n  const threadMessages = useExternalMessageConverter({\n    callback: convertLangchainMessages,\n    messages,\n    isRunning,\n  });\n\n  if (attachments && unstable_allowImageAttachments)\n    throw new Error(\n      \"Replace unstable_allowImageAttachments with `adapters: { attachments: new SimpleImageAttachmentAdapter() }`.\",\n    );\n  if (unstable_allowImageAttachments)\n    attachments = new SimpleImageAttachmentAdapter();\n\n  const switchToThread = !onSwitchToThread\n    ? undefined\n    : async (threadId: string) => {\n        const { messages } = await onSwitchToThread(threadId);\n        setMessages(messages);\n      };\n\n  const threadList: NonNullable<\n    ExternalStoreAdapter[\"adapters\"]\n  >[\"threadList\"] = {\n    threadId,\n    onSwitchToNewThread: !onSwitchToNewThread\n      ? undefined\n      : async () => {\n          await onSwitchToNewThread();\n          setMessages([]);\n        },\n    onSwitchToThread: switchToThread,\n  };\n\n  const loadingRef = useRef(false);\n  const threadListItemRuntime = useThreadListItemRuntime({ optional: true });\n  useEffect(() => {\n    if (!threadListItemRuntime || !switchToThread || loadingRef.current) return;\n\n    const externalId = threadListItemRuntime.getState().externalId;\n    if (externalId) {\n      loadingRef.current = true;\n      switchToThread(externalId).finally(() => {\n        loadingRef.current = false;\n      });\n    }\n  }, []);\n\n  return useExternalStoreRuntime({\n    isRunning,\n    messages: threadMessages,\n    adapters: {\n      attachments,\n      threadList,\n    },\n    extras: {\n      [symbolLangGraphRuntimeExtras]: true,\n      send: handleSendMessage,\n    } satisfies LangGraphRuntimeExtras,\n    onNew: (msg) => {\n      const cancellations =\n        autoCancelPendingToolCalls !== false\n          ? getPendingToolCalls(messages).map(\n              (t) =>\n                ({\n                  type: \"tool\",\n                  name: t.name,\n                  tool_call_id: t.id,\n                  content: JSON.stringify({ cancelled: true }),\n                }) satisfies LangChainMessage & { type: \"tool\" },\n            )\n          : [];\n\n      return handleSendMessage(\n        [\n          ...cancellations,\n          {\n            type: \"human\",\n            content: getMessageContent(msg),\n          },\n        ],\n        {\n          runConfig: msg.runConfig,\n        },\n      );\n    },\n    onAddToolResult: async ({ toolCallId, toolName, result }) => {\n      // TODO parallel human in the loop calls\n      await handleSendMessage(\n        [\n          {\n            type: \"tool\",\n            name: toolName,\n            tool_call_id: toolCallId,\n            content: JSON.stringify(result),\n          },\n        ],\n        // TODO reuse runconfig here!\n        {},\n      );\n    },\n    onCancel: unstable_allowCancellation\n      ? async () => {\n          cancel();\n        }\n      : undefined,\n  });\n};\n","\"use client\";\n\nimport { useExternalMessageConverter } from \"@assistant-ui/react\";\nimport { LangChainMessage } from \"./types\";\nimport { ToolCallContentPart } from \"@assistant-ui/react\";\nimport { ThreadUserMessage } from \"@assistant-ui/react\";\n\nconst contentToParts = (\n  content: LangChainMessage[\"content\"],\n) => {\n  if (typeof content === \"string\")\n    return [{ type: \"text\" as const, text: content }];\n  return content\n    .map((part): ThreadUserMessage[\"content\"][number] | null => {\n      const type = part.type;\n      switch (type) {\n        case \"text\":\n          return { type: \"text\", text: part.text };\n        case \"image_url\":\n          if (typeof part.image_url === \"string\") {\n            return { type: \"image\", image: part.image_url };\n          } else {\n            return {\n              type: \"image\",\n              image: part.image_url.url,\n            };\n          }\n\n        case \"tool_use\":\n          return null;\n        default:\n          const _exhaustiveCheck: never = type;\n          throw new Error(`Unknown content part type: ${_exhaustiveCheck}`);\n      }\n    })\n    .filter((a) => a !== null);\n};\n\nexport const convertLangchainMessages: useExternalMessageConverter.Callback<\n  LangChainMessage\n> = (message) => {\n  switch (message.type) {\n    case \"system\":\n      return {\n        role: \"system\",\n        id: message.id,\n        content: [{ type: \"text\", text: message.content }],\n      };\n    case \"human\":\n      return {\n        role: \"user\",\n        id: message.id,\n        content: contentToParts(message.content),\n      };\n    case \"ai\":\n      return {\n        role: \"assistant\",\n        id: message.id,\n        content: [\n          ...contentToParts(message.content),\n          ...(message.tool_calls?.map(\n            (chunk): ToolCallContentPart => ({\n              type: \"tool-call\",\n              toolCallId: chunk.id,\n              toolName: chunk.name,\n              args: chunk.args,\n              argsText:\n                message.tool_call_chunks?.find((c) => c.id === chunk.id)\n                  ?.args ?? JSON.stringify(chunk.args),\n            }),\n          ) ?? []),\n        ],\n      };\n    case \"tool\":\n      return {\n        role: \"tool\",\n        toolName: message.name,\n        toolCallId: message.tool_call_id,\n        result: message.content,\n      };\n  }\n};\n","import { useState, useCallback, useRef } from \"react\";\nimport { INTERNAL } from \"@assistant-ui/react\";\n\nconst { generateId } = INTERNAL;\n\nexport type LangGraphCommand = {\n  resume: string;\n};\n\nexport type LangGraphSendMessageConfig = {\n  command?: LangGraphCommand;\n  runConfig?: unknown;\n};\n\nexport const useLangGraphMessages = <TMessage>({\n  stream,\n}: {\n  stream: (\n    messages: TMessage[],\n    config: LangGraphSendMessageConfig & { abortSignal: AbortSignal },\n  ) => Promise<\n    AsyncGenerator<{\n      event: string;\n      data: any;\n    }>\n  >;\n}) => {\n  const [messages, setMessages] = useState<TMessage[]>([]);\n  const abortControllerRef = useRef<AbortController | null>(null);\n\n  const sendMessage = useCallback(\n    async (newMessages: TMessage[], config: LangGraphSendMessageConfig) => {\n      const optimisticMessages = [...messages, ...newMessages];\n      if (newMessages.length > 0) {\n        setMessages(optimisticMessages);\n      }\n\n      const abortController = new AbortController();\n      abortControllerRef.current = abortController;\n      const response = await stream(newMessages, {\n        ...config,\n        abortSignal: abortController.signal,\n      });\n\n      const completeMessages: TMessage[] = [];\n      let partialMessages: Map<string, TMessage> = new Map();\n      for await (const chunk of response) {\n        if (chunk.event === \"messages/partial\") {\n          // bug fix: messages/complete is not sent for some python langchain backends\n          if (completeMessages.length === 0) {\n            completeMessages.push(...optimisticMessages);\n          }\n\n          for (const message of chunk.data) {\n            if (!message.id) throw new Error(\"Partial message missing id\");\n\n            partialMessages.set(message.id, message);\n          }\n        } else if (chunk.event === \"messages/complete\") {\n          for (const message of chunk.data) {\n            if (message.id) {\n              partialMessages.delete(message.id);\n            }\n\n            // bug fix: tool results can arrive before the message is complete ?\n            // if we have partial messages left, we add complete message to partial messages\n            if (partialMessages.size > 0) {\n              partialMessages.set(message.id ?? generateId(), message);\n            } else {\n              completeMessages.push(message);\n            }\n          }\n        } else {\n          continue;\n        }\n\n        setMessages([...completeMessages, ...partialMessages.values()]);\n      }\n      // if (partialMessages.size > 0) {\n      //   throw new Error(\"A partial message was not marked as complete\");\n      // }\n    },\n    [messages, stream],\n  );\n\n  const cancel = useCallback(() => {\n    if (abortControllerRef.current) {\n      abortControllerRef.current.abort();\n    }\n  }, [abortControllerRef]);\n\n  return { messages, sendMessage, cancel, setMessages };\n};\n"],"mappings":";AAAA,SAAS,WAAW,UAAAA,SAAQ,YAAAC,iBAAgB;AAE5C;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OACK;;;ACCP,IAAM,iBAAiB,CACrB,YACG;AACH,MAAI,OAAO,YAAY;AACrB,WAAO,CAAC,EAAE,MAAM,QAAiB,MAAM,QAAQ,CAAC;AAClD,SAAO,QACJ,IAAI,CAAC,SAAsD;AAC1D,UAAM,OAAO,KAAK;AAClB,YAAQ,MAAM;AAAA,MACZ,KAAK;AACH,eAAO,EAAE,MAAM,QAAQ,MAAM,KAAK,KAAK;AAAA,MACzC,KAAK;AACH,YAAI,OAAO,KAAK,cAAc,UAAU;AACtC,iBAAO,EAAE,MAAM,SAAS,OAAO,KAAK,UAAU;AAAA,QAChD,OAAO;AACL,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,OAAO,KAAK,UAAU;AAAA,UACxB;AAAA,QACF;AAAA,MAEF,KAAK;AACH,eAAO;AAAA,MACT;AACE,cAAM,mBAA0B;AAChC,cAAM,IAAI,MAAM,8BAA8B,gBAAgB,EAAE;AAAA,IACpE;AAAA,EACF,CAAC,EACA,OAAO,CAAC,MAAM,MAAM,IAAI;AAC7B;AAEO,IAAM,2BAET,CAAC,YAAY;AACf,UAAQ,QAAQ,MAAM;AAAA,IACpB,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,IAAI,QAAQ;AAAA,QACZ,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,QAAQ,QAAQ,CAAC;AAAA,MACnD;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,IAAI,QAAQ;AAAA,QACZ,SAAS,eAAe,QAAQ,OAAO;AAAA,MACzC;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,IAAI,QAAQ;AAAA,QACZ,SAAS;AAAA,UACP,GAAG,eAAe,QAAQ,OAAO;AAAA,UACjC,GAAI,QAAQ,YAAY;AAAA,YACtB,CAAC,WAAgC;AAAA,cAC/B,MAAM;AAAA,cACN,YAAY,MAAM;AAAA,cAClB,UAAU,MAAM;AAAA,cAChB,MAAM,MAAM;AAAA,cACZ,UACE,QAAQ,kBAAkB,KAAK,CAAC,MAAM,EAAE,OAAO,MAAM,EAAE,GACnD,QAAQ,KAAK,UAAU,MAAM,IAAI;AAAA,YACzC;AAAA,UACF,KAAK,CAAC;AAAA,QACR;AAAA,MACF;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,UAAU,QAAQ;AAAA,QAClB,YAAY,QAAQ;AAAA,QACpB,QAAQ,QAAQ;AAAA,MAClB;AAAA,EACJ;AACF;;;ACjFA,SAAS,UAAU,aAAa,cAAc;AAC9C,SAAS,gBAAgB;AAEzB,IAAM,EAAE,WAAW,IAAI;AAWhB,IAAM,uBAAuB,CAAW;AAAA,EAC7C;AACF,MAUM;AACJ,QAAM,CAAC,UAAU,WAAW,IAAI,SAAqB,CAAC,CAAC;AACvD,QAAM,qBAAqB,OAA+B,IAAI;AAE9D,QAAM,cAAc;AAAA,IAClB,OAAO,aAAyB,WAAuC;AACrE,YAAM,qBAAqB,CAAC,GAAG,UAAU,GAAG,WAAW;AACvD,UAAI,YAAY,SAAS,GAAG;AAC1B,oBAAY,kBAAkB;AAAA,MAChC;AAEA,YAAM,kBAAkB,IAAI,gBAAgB;AAC5C,yBAAmB,UAAU;AAC7B,YAAM,WAAW,MAAM,OAAO,aAAa;AAAA,QACzC,GAAG;AAAA,QACH,aAAa,gBAAgB;AAAA,MAC/B,CAAC;AAED,YAAM,mBAA+B,CAAC;AACtC,UAAI,kBAAyC,oBAAI,IAAI;AACrD,uBAAiB,SAAS,UAAU;AAClC,YAAI,MAAM,UAAU,oBAAoB;AAEtC,cAAI,iBAAiB,WAAW,GAAG;AACjC,6BAAiB,KAAK,GAAG,kBAAkB;AAAA,UAC7C;AAEA,qBAAW,WAAW,MAAM,MAAM;AAChC,gBAAI,CAAC,QAAQ,GAAI,OAAM,IAAI,MAAM,4BAA4B;AAE7D,4BAAgB,IAAI,QAAQ,IAAI,OAAO;AAAA,UACzC;AAAA,QACF,WAAW,MAAM,UAAU,qBAAqB;AAC9C,qBAAW,WAAW,MAAM,MAAM;AAChC,gBAAI,QAAQ,IAAI;AACd,8BAAgB,OAAO,QAAQ,EAAE;AAAA,YACnC;AAIA,gBAAI,gBAAgB,OAAO,GAAG;AAC5B,8BAAgB,IAAI,QAAQ,MAAM,WAAW,GAAG,OAAO;AAAA,YACzD,OAAO;AACL,+BAAiB,KAAK,OAAO;AAAA,YAC/B;AAAA,UACF;AAAA,QACF,OAAO;AACL;AAAA,QACF;AAEA,oBAAY,CAAC,GAAG,kBAAkB,GAAG,gBAAgB,OAAO,CAAC,CAAC;AAAA,MAChE;AAAA,IAIF;AAAA,IACA,CAAC,UAAU,MAAM;AAAA,EACnB;AAEA,QAAM,SAAS,YAAY,MAAM;AAC/B,QAAI,mBAAmB,SAAS;AAC9B,yBAAmB,QAAQ,MAAM;AAAA,IACnC;AAAA,EACF,GAAG,CAAC,kBAAkB,CAAC;AAEvB,SAAO,EAAE,UAAU,aAAa,QAAQ,YAAY;AACtD;;;AF/EA,SAAS,oCAAoC;AAI7C,SAAS,gCAAgC;AAEzC,IAAM,sBAAsB,CAAC,aAAiC;AAC5D,QAAM,mBAAmB,oBAAI,IAA+B;AAC5D,aAAW,WAAW,UAAU;AAC9B,QAAI,QAAQ,SAAS,MAAM;AACzB,iBAAW,YAAY,QAAQ,cAAc,CAAC,GAAG;AAC/C,yBAAiB,IAAI,SAAS,IAAI,QAAQ;AAAA,MAC5C;AAAA,IACF;AACA,QAAI,QAAQ,SAAS,QAAQ;AAC3B,uBAAiB,OAAO,QAAQ,YAAY;AAAA,IAC9C;AAAA,EACF;AAEA,SAAO,CAAC,GAAG,iBAAiB,OAAO,CAAC;AACtC;AAEA,IAAM,oBAAoB,CAAC,QAAuB;AAChD,QAAM,aAAa;AAAA,IACjB,GAAG,IAAI;AAAA,IACP,GAAG,IAAI,YAAY,QAAQ,CAAC,MAAM,EAAE,OAAO;AAAA,EAC7C;AACA,QAAM,UAAU,WAAW,IAAI,CAAC,SAAS;AACvC,UAAM,OAAO,KAAK;AAClB,YAAQ,MAAM;AAAA,MACZ,KAAK;AACH,eAAO,EAAE,MAAM,QAAiB,MAAM,KAAK,KAAK;AAAA,MAClD,KAAK;AACH,eAAO,EAAE,MAAM,aAAsB,WAAW,EAAE,KAAK,KAAK,MAAM,EAAE;AAAA,MAEtE,KAAK;AACH,cAAM,IAAI,MAAM,sCAAsC;AAAA,MACxD,KAAK;AACH,cAAM,IAAI,MAAM,0CAA0C;AAAA,MAE5D;AACE,cAAM,mBAA0B;AAChC,cAAM,IAAI,MAAM,8BAA8B,gBAAgB,EAAE;AAAA,IACpE;AAAA,EACF,CAAC;AAED,MAAI,QAAQ,WAAW,KAAK,QAAQ,CAAC,GAAG,SAAS,QAAQ;AACvD,WAAO,QAAQ,CAAC,EAAE,QAAQ;AAAA,EAC5B;AAEA,SAAO;AACT;AAEA,IAAM,+BAA+B,OAAO,0BAA0B;AAStE,IAAM,2BAA2B,CAAC,WAA4C;AAC5E,MACE,OAAO,WAAW,YAClB,UAAU,QACV,EAAE,gCAAgC;AAElC,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAEF,SAAO;AACT;AAEO,IAAM,0BAA0B,MAAM;AAC3C,QAAM,EAAE,KAAK,IAAI,UAAU,CAAC,MAAM,yBAAyB,EAAE,MAAM,CAAC;AACpE,SAAO;AACT;AAEO,IAAM,iCAAiC,CAAC,YAA8B;AAC3E,QAAM,OAAO,wBAAwB;AACrC,SAAO,MAAM,KAAK,CAAC,GAAG,EAAE,QAAQ,CAAC;AACnC;AAEO,IAAM,sBAAsB,CAAC;AAAA,EAClC;AAAA,EACA,UAAU,EAAE,YAAY,IAAI,CAAC;AAAA,EAC7B;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,MAgCM;AACJ,QAAM,EAAE,UAAU,aAAa,QAAQ,YAAY,IAAI,qBAAqB;AAAA,IAC1E;AAAA,EACF,CAAC;AAED,QAAM,CAAC,WAAW,YAAY,IAAIC,UAAS,KAAK;AAChD,QAAM,oBAAoB,OACxBC,WACA,WACG;AACH,QAAI;AACF,mBAAa,IAAI;AACjB,YAAM,YAAYA,WAAU,MAAM;AAAA,IACpC,SAAS,OAAO;AACd,cAAQ,MAAM,6BAA6B,KAAK;AAAA,IAClD,UAAE;AACA,mBAAa,KAAK;AAAA,IACpB;AAAA,EACF;AAEA,QAAM,iBAAiB,4BAA4B;AAAA,IACjD,UAAU;AAAA,IACV;AAAA,IACA;AAAA,EACF,CAAC;AAED,MAAI,eAAe;AACjB,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AACF,MAAI;AACF,kBAAc,IAAI,6BAA6B;AAEjD,QAAM,iBAAiB,CAAC,mBACpB,SACA,OAAOC,cAAqB;AAC1B,UAAM,EAAE,UAAAD,UAAS,IAAI,MAAM,iBAAiBC,SAAQ;AACpD,gBAAYD,SAAQ;AAAA,EACtB;AAEJ,QAAM,aAEY;AAAA,IAChB;AAAA,IACA,qBAAqB,CAAC,sBAClB,SACA,YAAY;AACV,YAAM,oBAAoB;AAC1B,kBAAY,CAAC,CAAC;AAAA,IAChB;AAAA,IACJ,kBAAkB;AAAA,EACpB;AAEA,QAAM,aAAaE,QAAO,KAAK;AAC/B,QAAM,wBAAwB,yBAAyB,EAAE,UAAU,KAAK,CAAC;AACzE,YAAU,MAAM;AACd,QAAI,CAAC,yBAAyB,CAAC,kBAAkB,WAAW,QAAS;AAErE,UAAM,aAAa,sBAAsB,SAAS,EAAE;AACpD,QAAI,YAAY;AACd,iBAAW,UAAU;AACrB,qBAAe,UAAU,EAAE,QAAQ,MAAM;AACvC,mBAAW,UAAU;AAAA,MACvB,CAAC;AAAA,IACH;AAAA,EACF,GAAG,CAAC,CAAC;AAEL,SAAO,wBAAwB;AAAA,IAC7B;AAAA,IACA,UAAU;AAAA,IACV,UAAU;AAAA,MACR;AAAA,MACA;AAAA,IACF;AAAA,IACA,QAAQ;AAAA,MACN,CAAC,4BAA4B,GAAG;AAAA,MAChC,MAAM;AAAA,IACR;AAAA,IACA,OAAO,CAAC,QAAQ;AACd,YAAM,gBACJ,+BAA+B,QAC3B,oBAAoB,QAAQ,EAAE;AAAA,QAC5B,CAAC,OACE;AAAA,UACC,MAAM;AAAA,UACN,MAAM,EAAE;AAAA,UACR,cAAc,EAAE;AAAA,UAChB,SAAS,KAAK,UAAU,EAAE,WAAW,KAAK,CAAC;AAAA,QAC7C;AAAA,MACJ,IACA,CAAC;AAEP,aAAO;AAAA,QACL;AAAA,UACE,GAAG;AAAA,UACH;AAAA,YACE,MAAM;AAAA,YACN,SAAS,kBAAkB,GAAG;AAAA,UAChC;AAAA,QACF;AAAA,QACA;AAAA,UACE,WAAW,IAAI;AAAA,QACjB;AAAA,MACF;AAAA,IACF;AAAA,IACA,iBAAiB,OAAO,EAAE,YAAY,UAAU,OAAO,MAAM;AAE3D,YAAM;AAAA,QACJ;AAAA,UACE;AAAA,YACE,MAAM;AAAA,YACN,MAAM;AAAA,YACN,cAAc;AAAA,YACd,SAAS,KAAK,UAAU,MAAM;AAAA,UAChC;AAAA,QACF;AAAA;AAAA,QAEA,CAAC;AAAA,MACH;AAAA,IACF;AAAA,IACA,UAAU,6BACN,YAAY;AACV,aAAO;AAAA,IACT,IACA;AAAA,EACN,CAAC;AACH;","names":["useRef","useState","useState","messages","threadId","useRef"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-langgraph",
-  "version": "0.1.8",
+  "version": "0.1.10",
   "license": "MIT",
   "exports": {
     ".": {
@@ -27,7 +27,7 @@
     "zod": "^3.24.1"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.7.19",
+    "@assistant-ui/react": "^0.7.27",
     "@types/react": "*",
     "react": "^18 || ^19 || ^19.0.0-rc"
   },
@@ -39,7 +39,7 @@
   "devDependencies": {
     "eslint-config-next": "15.1.2",
     "tsup": "8.3.5",
-    "@assistant-ui/react": "0.7.19",
+    "@assistant-ui/react": "0.7.27",
     "@assistant-ui/tsconfig": "0.0.0"
   },
   "publishConfig": {