@assistant-ui/react-langgraph 0.7.0 → 0.7.2

package/dist/LangGraphMessageAccumulator.d.ts

@@ -11,6 +11,7 @@ export declare class LangGraphMessageAccumulator<TMessage extends {
     private ensureMessageId;
     addMessages(newMessages: TMessage[]): TMessage[];
     getMessages(): TMessage[];
+    replaceMessages(newMessages: TMessage[]): TMessage[];
     clear(): void;
 }
 //# sourceMappingURL=LangGraphMessageAccumulator.d.ts.map

package/dist/LangGraphMessageAccumulator.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"LangGraphMessageAccumulator.d.ts","sourceRoot":"","sources":["../src/LangGraphMessageAccumulator.ts"],"names":[],"mappings":"AAEA,MAAM,MAAM,+BAA+B,CAAC,QAAQ,IAAI;IACtD,eAAe,CAAC,EAAE,QAAQ,EAAE,CAAC;IAC7B,aAAa,CAAC,EAAE,CAAC,IAAI,EAAE,QAAQ,GAAG,SAAS,EAAE,IAAI,EAAE,QAAQ,KAAK,QAAQ,CAAC;CAC1E,CAAC;AAEF,qBAAa,2BAA2B,CAAC,QAAQ,SAAS;IAAE,EAAE,CAAC,EAAE,MAAM,CAAA;CAAE;IACvE,OAAO,CAAC,WAAW,CAA+B;IAClD,OAAO,CAAC,aAAa,CAGP;gBAEF,EACV,eAAoB,EACpB,aAGa,GACd,GAAE,+BAA+B,CAAC,QAAQ,CAAM;IAKjD,OAAO,CAAC,eAAe;IAIhB,WAAW,CAAC,WAAW,EAAE,QAAQ,EAAE;IAWnC,WAAW,IAAI,QAAQ,EAAE;IAIzB,KAAK;CAGb"}
+{"version":3,"file":"LangGraphMessageAccumulator.d.ts","sourceRoot":"","sources":["../src/LangGraphMessageAccumulator.ts"],"names":[],"mappings":"AAEA,MAAM,MAAM,+BAA+B,CAAC,QAAQ,IAAI;IACtD,eAAe,CAAC,EAAE,QAAQ,EAAE,CAAC;IAC7B,aAAa,CAAC,EAAE,CAAC,IAAI,EAAE,QAAQ,GAAG,SAAS,EAAE,IAAI,EAAE,QAAQ,KAAK,QAAQ,CAAC;CAC1E,CAAC;AAEF,qBAAa,2BAA2B,CAAC,QAAQ,SAAS;IAAE,EAAE,CAAC,EAAE,MAAM,CAAA;CAAE;IACvE,OAAO,CAAC,WAAW,CAA+B;IAClD,OAAO,CAAC,aAAa,CAGP;gBAEF,EACV,eAAoB,EACpB,aAGa,GACd,GAAE,+BAA+B,CAAC,QAAQ,CAAM;IAKjD,OAAO,CAAC,eAAe;IAIhB,WAAW,CAAC,WAAW,EAAE,QAAQ,EAAE;IAWnC,WAAW,IAAI,QAAQ,EAAE;IAIzB,eAAe,CAAC,WAAW,EAAE,QAAQ,EAAE,GAAG,QAAQ,EAAE;IASpD,KAAK;CAGb"}

package/dist/LangGraphMessageAccumulator.js

@@ -25,6 +25,13 @@ var LangGraphMessageAccumulator = class {
   getMessages() {
     return [...this.messagesMap.values()];
   }
+  replaceMessages(newMessages) {
+    this.messagesMap.clear();
+    for (const message of newMessages.map(this.ensureMessageId)) {
+      this.messagesMap.set(message.id, message);
+    }
+    return this.getMessages();
+  }
   clear() {
     this.messagesMap.clear();
   }

package/dist/LangGraphMessageAccumulator.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/LangGraphMessageAccumulator.ts"],"sourcesContent":["import { v4 as uuidv4 } from \"uuid\";\n\nexport type LangGraphStateAccumulatorConfig<TMessage> = {\n  initialMessages?: TMessage[];\n  appendMessage?: (prev: TMessage | undefined, curr: TMessage) => TMessage;\n};\n\nexport class LangGraphMessageAccumulator<TMessage extends { id?: string }> {\n  private messagesMap = new Map<string, TMessage>();\n  private appendMessage: (\n    prev: TMessage | undefined,\n    curr: TMessage,\n  ) => TMessage;\n\n  constructor({\n    initialMessages = [],\n    appendMessage = ((_: TMessage | undefined, curr: TMessage) => curr) as (\n      prev: TMessage | undefined,\n      curr: TMessage,\n    ) => TMessage,\n  }: LangGraphStateAccumulatorConfig<TMessage> = {}) {\n    this.appendMessage = appendMessage;\n    this.addMessages(initialMessages);\n  }\n\n  private ensureMessageId(message: TMessage): TMessage {\n    return message.id ? message : { ...message, id: uuidv4() };\n  }\n\n  public addMessages(newMessages: TMessage[]) {\n    if (newMessages.length === 0) return this.getMessages();\n\n    for (const message of newMessages.map(this.ensureMessageId)) {\n      const messageId = message.id!; // ensureMessageId guarantees id exists\n      const previous = this.messagesMap.get(messageId);\n      this.messagesMap.set(messageId, this.appendMessage(previous, message));\n    }\n    return this.getMessages();\n  }\n\n  public getMessages(): TMessage[] {\n    return [...this.messagesMap.values()];\n  }\n\n  public clear() {\n    this.messagesMap.clear();\n  }\n}\n"],"mappings":";AAAA,SAAS,MAAM,cAAc;AAOtB,IAAM,8BAAN,MAAoE;AAAA,EACjE,cAAc,oBAAI,IAAsB;AAAA,EACxC;AAAA,EAKR,YAAY;AAAA,IACV,kBAAkB,CAAC;AAAA,IACnB,iBAAiB,CAAC,GAAyB,SAAmB;AAAA,EAIhE,IAA+C,CAAC,GAAG;AACjD,SAAK,gBAAgB;AACrB,SAAK,YAAY,eAAe;AAAA,EAClC;AAAA,EAEQ,gBAAgB,SAA6B;AACnD,WAAO,QAAQ,KAAK,UAAU,EAAE,GAAG,SAAS,IAAI,OAAO,EAAE;AAAA,EAC3D;AAAA,EAEO,YAAY,aAAyB;AAC1C,QAAI,YAAY,WAAW,EAAG,QAAO,KAAK,YAAY;AAEtD,eAAW,WAAW,YAAY,IAAI,KAAK,eAAe,GAAG;AAC3D,YAAM,YAAY,QAAQ;AAC1B,YAAM,WAAW,KAAK,YAAY,IAAI,SAAS;AAC/C,WAAK,YAAY,IAAI,WAAW,KAAK,cAAc,UAAU,OAAO,CAAC;AAAA,IACvE;AACA,WAAO,KAAK,YAAY;AAAA,EAC1B;AAAA,EAEO,cAA0B;AAC/B,WAAO,CAAC,GAAG,KAAK,YAAY,OAAO,CAAC;AAAA,EACtC;AAAA,EAEO,QAAQ;AACb,SAAK,YAAY,MAAM;AAAA,EACzB;AACF;","names":[]}
+{"version":3,"sources":["../src/LangGraphMessageAccumulator.ts"],"sourcesContent":["import { v4 as uuidv4 } from \"uuid\";\n\nexport type LangGraphStateAccumulatorConfig<TMessage> = {\n  initialMessages?: TMessage[];\n  appendMessage?: (prev: TMessage | undefined, curr: TMessage) => TMessage;\n};\n\nexport class LangGraphMessageAccumulator<TMessage extends { id?: string }> {\n  private messagesMap = new Map<string, TMessage>();\n  private appendMessage: (\n    prev: TMessage | undefined,\n    curr: TMessage,\n  ) => TMessage;\n\n  constructor({\n    initialMessages = [],\n    appendMessage = ((_: TMessage | undefined, curr: TMessage) => curr) as (\n      prev: TMessage | undefined,\n      curr: TMessage,\n    ) => TMessage,\n  }: LangGraphStateAccumulatorConfig<TMessage> = {}) {\n    this.appendMessage = appendMessage;\n    this.addMessages(initialMessages);\n  }\n\n  private ensureMessageId(message: TMessage): TMessage {\n    return message.id ? message : { ...message, id: uuidv4() };\n  }\n\n  public addMessages(newMessages: TMessage[]) {\n    if (newMessages.length === 0) return this.getMessages();\n\n    for (const message of newMessages.map(this.ensureMessageId)) {\n      const messageId = message.id!; // ensureMessageId guarantees id exists\n      const previous = this.messagesMap.get(messageId);\n      this.messagesMap.set(messageId, this.appendMessage(previous, message));\n    }\n    return this.getMessages();\n  }\n\n  public getMessages(): TMessage[] {\n    return [...this.messagesMap.values()];\n  }\n\n  public replaceMessages(newMessages: TMessage[]): TMessage[] {\n    this.messagesMap.clear();\n\n    for (const message of newMessages.map(this.ensureMessageId)) {\n      this.messagesMap.set(message.id!, message);\n    }\n    return this.getMessages();\n  }\n\n  public clear() {\n    this.messagesMap.clear();\n  }\n}\n"],"mappings":";AAAA,SAAS,MAAM,cAAc;AAOtB,IAAM,8BAAN,MAAoE;AAAA,EACjE,cAAc,oBAAI,IAAsB;AAAA,EACxC;AAAA,EAKR,YAAY;AAAA,IACV,kBAAkB,CAAC;AAAA,IACnB,iBAAiB,CAAC,GAAyB,SAAmB;AAAA,EAIhE,IAA+C,CAAC,GAAG;AACjD,SAAK,gBAAgB;AACrB,SAAK,YAAY,eAAe;AAAA,EAClC;AAAA,EAEQ,gBAAgB,SAA6B;AACnD,WAAO,QAAQ,KAAK,UAAU,EAAE,GAAG,SAAS,IAAI,OAAO,EAAE;AAAA,EAC3D;AAAA,EAEO,YAAY,aAAyB;AAC1C,QAAI,YAAY,WAAW,EAAG,QAAO,KAAK,YAAY;AAEtD,eAAW,WAAW,YAAY,IAAI,KAAK,eAAe,GAAG;AAC3D,YAAM,YAAY,QAAQ;AAC1B,YAAM,WAAW,KAAK,YAAY,IAAI,SAAS;AAC/C,WAAK,YAAY,IAAI,WAAW,KAAK,cAAc,UAAU,OAAO,CAAC;AAAA,IACvE;AACA,WAAO,KAAK,YAAY;AAAA,EAC1B;AAAA,EAEO,cAA0B;AAC/B,WAAO,CAAC,GAAG,KAAK,YAAY,OAAO,CAAC;AAAA,EACtC;AAAA,EAEO,gBAAgB,aAAqC;AAC1D,SAAK,YAAY,MAAM;AAEvB,eAAW,WAAW,YAAY,IAAI,KAAK,eAAe,GAAG;AAC3D,WAAK,YAAY,IAAI,QAAQ,IAAK,OAAO;AAAA,IAC3C;AACA,WAAO,KAAK,YAAY;AAAA,EAC1B;AAAA,EAEO,QAAQ;AACb,SAAK,YAAY,MAAM;AAAA,EACzB;AACF;","names":[]}

package/dist/useLangGraphMessages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useLangGraphMessages.d.ts","sourceRoot":"","sources":["../src/useLangGraphMessages.ts"],"names":[],"mappings":"AAGA,OAAO,EACL,SAAS,EAIT,qBAAqB,EACrB,oBAAoB,EACpB,mBAAmB,EACnB,uBAAuB,EACxB,MAAM,SAAS,CAAC;AAGjB,MAAM,MAAM,gBAAgB,GAAG;IAC7B,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,MAAM,MAAM,0BAA0B,GAAG;IACvC,OAAO,CAAC,EAAE,gBAAgB,CAAC;IAC3B,SAAS,CAAC,EAAE,OAAO,CAAC;CACrB,CAAC;AAEF,MAAM,MAAM,sBAAsB,CAAC,QAAQ,IAAI;IAC7C,KAAK,EAAE,SAAS,CAAC;IACjB,IAAI,EAAE,QAAQ,EAAE,GAAG,GAAG,CAAC;CACxB,CAAC;AAEF,MAAM,MAAM,uBAAuB,CAAC,QAAQ,IAAI,CAC9C,QAAQ,EAAE,QAAQ,EAAE,EACpB,MAAM,EAAE,0BAA0B,GAAG;IACnC,WAAW,EAAE,WAAW,CAAC;IACzB,UAAU,EAAE,MAAM,OAAO,CAAC;QACxB,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,GAAG,SAAS,CAAC;KAChC,CAAC,CAAC;CACJ,KAEC,OAAO,CAAC,cAAc,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC,CAAC,GACzD,cAAc,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC,CAAC;AAErD,MAAM,MAAM,uBAAuB,GAAG;IACpC,KAAK,CAAC,EAAE,GAAG,CAAC;IACZ,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC;CACf,CAAC;AAuBF,eAAO,MAAM,oBAAoB,GAAI,QAAQ,SAAS;IAAE,EAAE,CAAC,EAAE,MAAM,CAAA;CAAE,EAAE,2CAIpE;IACD,MAAM,EAAE,uBAAuB,CAAC,QAAQ,CAAC,CAAC;IAC1C,aAAa,CAAC,EAAE,CAAC,IAAI,EAAE,QAAQ,GAAG,SAAS,EAAE,IAAI,EAAE,QAAQ,KAAK,QAAQ,CAAC;IACzE,aAAa,CAAC,EAAE;QACd,UAAU,CAAC,EAAE,uBAAuB,CAAC;QACrC,MAAM,CAAC,EAAE,mBAAmB,CAAC;QAC7B,OAAO,CAAC,EAAE,oBAAoB,CAAC;QAC/B,aAAa,CAAC,EAAE,qBAAqB,CAAC;KACvC,CAAC;CACH;;;+BAcuB,QAAQ,EAAE,UAAU,0BAA0B;;;;CAiHrE,CAAC"}
+{"version":3,"file":"useLangGraphMessages.d.ts","sourceRoot":"","sources":["../src/useLangGraphMessages.ts"],"names":[],"mappings":"AAGA,OAAO,EACL,SAAS,EAIT,qBAAqB,EACrB,oBAAoB,EACpB,mBAAmB,EACnB,uBAAuB,EACxB,MAAM,SAAS,CAAC;AAGjB,MAAM,MAAM,gBAAgB,GAAG;IAC7B,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,MAAM,MAAM,0BAA0B,GAAG;IACvC,OAAO,CAAC,EAAE,gBAAgB,CAAC;IAC3B,SAAS,CAAC,EAAE,OAAO,CAAC;CACrB,CAAC;AAEF,MAAM,MAAM,sBAAsB,CAAC,QAAQ,IAAI;IAC7C,KAAK,EAAE,SAAS,CAAC;IACjB,IAAI,EAAE,QAAQ,EAAE,GAAG,GAAG,CAAC;CACxB,CAAC;AAEF,MAAM,MAAM,uBAAuB,CAAC,QAAQ,IAAI,CAC9C,QAAQ,EAAE,QAAQ,EAAE,EACpB,MAAM,EAAE,0BAA0B,GAAG;IACnC,WAAW,EAAE,WAAW,CAAC;IACzB,UAAU,EAAE,MAAM,OAAO,CAAC;QACxB,QAAQ,EAAE,MAAM,CAAC;QACjB,UAAU,EAAE,MAAM,GAAG,SAAS,CAAC;KAChC,CAAC,CAAC;CACJ,KAEC,OAAO,CAAC,cAAc,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC,CAAC,GACzD,cAAc,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC,CAAC;AAErD,MAAM,MAAM,uBAAuB,GAAG;IACpC,KAAK,CAAC,EAAE,GAAG,CAAC;IACZ,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC;CACf,CAAC;AAuBF,eAAO,MAAM,oBAAoB,GAAI,QAAQ,SAAS;IAAE,EAAE,CAAC,EAAE,MAAM,CAAA;CAAE,EAAE,2CAIpE;IACD,MAAM,EAAE,uBAAuB,CAAC,QAAQ,CAAC,CAAC;IAC1C,aAAa,CAAC,EAAE,CAAC,IAAI,EAAE,QAAQ,GAAG,SAAS,EAAE,IAAI,EAAE,QAAQ,KAAK,QAAQ,CAAC;IACzE,aAAa,CAAC,EAAE;QACd,UAAU,CAAC,EAAE,uBAAuB,CAAC;QACrC,MAAM,CAAC,EAAE,mBAAmB,CAAC;QAC7B,OAAO,CAAC,EAAE,oBAAoB,CAAC;QAC/B,aAAa,CAAC,EAAE,qBAAqB,CAAC;KACvC,CAAC;CACH;;;+BAcuB,QAAQ,EAAE,UAAU,0BAA0B;;;;CAoHrE,CAAC"}

package/dist/useLangGraphMessages.js

@@ -51,6 +51,9 @@ var useLangGraphMessages = ({
             setMessages(accumulator.addMessages(chunk.data));
             break;
           case LangGraphKnownEventTypes.Updates:
+            if (Array.isArray(chunk.data.messages)) {
+              setMessages(accumulator.replaceMessages(chunk.data.messages));
+            }
             setInterrupt(chunk.data.__interrupt__?.[0]);
             break;
           case LangGraphKnownEventTypes.Messages: {

package/dist/useLangGraphMessages.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/useLangGraphMessages.ts"],"sourcesContent":["import { useState, useCallback, useRef, useMemo } from \"react\";\nimport { v4 as uuidv4 } from \"uuid\";\nimport { LangGraphMessageAccumulator } from \"./LangGraphMessageAccumulator\";\nimport {\n  EventType,\n  LangChainMessageTupleEvent,\n  LangGraphKnownEventTypes,\n  LangChainMessageChunk,\n  OnCustomEventCallback,\n  OnErrorEventCallback,\n  OnInfoEventCallback,\n  OnMetadataEventCallback,\n} from \"./types\";\nimport { useAssistantApi } from \"@assistant-ui/react\";\n\nexport type LangGraphCommand = {\n  resume: string;\n};\n\nexport type LangGraphSendMessageConfig = {\n  command?: LangGraphCommand;\n  runConfig?: unknown;\n};\n\nexport type LangGraphMessagesEvent<TMessage> = {\n  event: EventType;\n  data: TMessage[] | any;\n};\n\nexport type LangGraphStreamCallback<TMessage> = (\n  messages: TMessage[],\n  config: LangGraphSendMessageConfig & {\n    abortSignal: AbortSignal;\n    initialize: () => Promise<{\n      remoteId: string;\n      externalId: string | undefined;\n    }>;\n  },\n) =>\n  | Promise<AsyncGenerator<LangGraphMessagesEvent<TMessage>>>\n  | AsyncGenerator<LangGraphMessagesEvent<TMessage>>;\n\nexport type LangGraphInterruptState = {\n  value?: any;\n  resumable?: boolean;\n  when?: string;\n  ns?: string[];\n};\n\nconst DEFAULT_APPEND_MESSAGE = <TMessage>(\n  _: TMessage | undefined,\n  curr: TMessage,\n) => curr;\n\nconst isLangChainMessageChunk = (\n  value: unknown,\n): value is LangChainMessageChunk => {\n  if (!value || typeof value !== \"object\") return false;\n  const chunk = value as any;\n  return (\n    \"type\" in chunk &&\n    chunk.type === \"AIMessageChunk\" &&\n    (chunk.content === undefined ||\n      typeof chunk.content === \"string\" ||\n      Array.isArray(chunk.content)) &&\n    (chunk.tool_call_chunks === undefined ||\n      Array.isArray(chunk.tool_call_chunks))\n  );\n};\n\nexport const useLangGraphMessages = <TMessage extends { id?: string }>({\n  stream,\n  appendMessage = DEFAULT_APPEND_MESSAGE,\n  eventHandlers,\n}: {\n  stream: LangGraphStreamCallback<TMessage>;\n  appendMessage?: (prev: TMessage | undefined, curr: TMessage) => TMessage;\n  eventHandlers?: {\n    onMetadata?: OnMetadataEventCallback;\n    onInfo?: OnInfoEventCallback;\n    onError?: OnErrorEventCallback;\n    onCustomEvent?: OnCustomEventCallback;\n  };\n}) => {\n  const [interrupt, setInterrupt] = useState<\n    LangGraphInterruptState | undefined\n  >();\n  const [messages, setMessages] = useState<TMessage[]>([]);\n  const abortControllerRef = useRef<AbortController | null>(null);\n\n  const { onMetadata, onInfo, onError, onCustomEvent } = useMemo(\n    () => eventHandlers ?? {},\n    [eventHandlers],\n  );\n\n  const api = useAssistantApi();\n  const sendMessage = useCallback(\n    async (newMessages: TMessage[], config: LangGraphSendMessageConfig) => {\n      // ensure all messages have an ID\n      const newMessagesWithId = newMessages.map((m) =>\n        m.id ? m : { ...m, id: uuidv4() },\n      );\n\n      const accumulator = new LangGraphMessageAccumulator({\n        initialMessages: messages,\n        appendMessage,\n      });\n      setMessages(accumulator.addMessages(newMessagesWithId));\n\n      const abortController = new AbortController();\n      abortControllerRef.current = abortController;\n      const response = await stream(newMessagesWithId, {\n        ...config,\n        abortSignal: abortController.signal,\n        initialize: async () => {\n          return await api.threadListItem().initialize();\n        },\n      });\n\n      for await (const chunk of response) {\n        switch (chunk.event) {\n          case LangGraphKnownEventTypes.MessagesPartial:\n          case LangGraphKnownEventTypes.MessagesComplete:\n            setMessages(accumulator.addMessages(chunk.data));\n            break;\n          case LangGraphKnownEventTypes.Updates:\n            setInterrupt(chunk.data.__interrupt__?.[0]);\n            break;\n          case LangGraphKnownEventTypes.Messages: {\n            const [messageChunk] = (chunk as LangChainMessageTupleEvent).data;\n            if (!isLangChainMessageChunk(messageChunk)) {\n              console.warn(\n                \"Received invalid message chunk format:\",\n                messageChunk,\n              );\n              break;\n            }\n            const updatedMessages = accumulator.addMessages([\n              messageChunk as unknown as TMessage,\n            ]);\n            setMessages(updatedMessages);\n            break;\n          }\n          case LangGraphKnownEventTypes.Metadata:\n            onMetadata?.(chunk.data);\n            break;\n          case LangGraphKnownEventTypes.Info:\n            onInfo?.(chunk.data);\n            break;\n          case LangGraphKnownEventTypes.Error: {\n            onError?.(chunk.data);\n            // Update the last AI message with error status\n            // Assumes last AI message is the one the error relates to\n            const messages = accumulator.getMessages();\n            const lastAiMessage = messages.findLast(\n              (m): m is TMessage & { type: string; id: string } =>\n                m != null && \"type\" in m && m.type === \"ai\" && m.id != null,\n            );\n            if (lastAiMessage) {\n              const errorMessage = {\n                ...lastAiMessage,\n                status: {\n                  type: \"incomplete\" as const,\n                  reason: \"error\" as const,\n                  error: chunk.data,\n                },\n              };\n              setMessages(accumulator.addMessages([errorMessage]));\n            }\n            break;\n          }\n          default:\n            if (onCustomEvent) {\n              onCustomEvent(chunk.event, chunk.data);\n            } else {\n              console.warn(\n                \"Unhandled event received:\",\n                chunk.event,\n                chunk.data,\n              );\n            }\n            break;\n        }\n      }\n    },\n    [\n      messages,\n      appendMessage,\n      stream,\n      onMetadata,\n      onInfo,\n      onError,\n      onCustomEvent,\n    ],\n  );\n\n  const cancel = useCallback(() => {\n    if (abortControllerRef.current) {\n      abortControllerRef.current.abort();\n    }\n  }, [abortControllerRef]);\n\n  return {\n    interrupt,\n    messages,\n    sendMessage,\n    cancel,\n    setInterrupt,\n    setMessages,\n  };\n};\n"],"mappings":";AAAA,SAAS,UAAU,aAAa,QAAQ,eAAe;AACvD,SAAS,MAAM,cAAc;AAC7B,SAAS,mCAAmC;AAC5C;AAAA,EAGE;AAAA,OAMK;AACP,SAAS,uBAAuB;AAoChC,IAAM,yBAAyB,CAC7B,GACA,SACG;AAEL,IAAM,0BAA0B,CAC9B,UACmC;AACnC,MAAI,CAAC,SAAS,OAAO,UAAU,SAAU,QAAO;AAChD,QAAM,QAAQ;AACd,SACE,UAAU,SACV,MAAM,SAAS,qBACd,MAAM,YAAY,UACjB,OAAO,MAAM,YAAY,YACzB,MAAM,QAAQ,MAAM,OAAO,OAC5B,MAAM,qBAAqB,UAC1B,MAAM,QAAQ,MAAM,gBAAgB;AAE1C;AAEO,IAAM,uBAAuB,CAAmC;AAAA,EACrE;AAAA,EACA,gBAAgB;AAAA,EAChB;AACF,MASM;AACJ,QAAM,CAAC,WAAW,YAAY,IAAI,SAEhC;AACF,QAAM,CAAC,UAAU,WAAW,IAAI,SAAqB,CAAC,CAAC;AACvD,QAAM,qBAAqB,OAA+B,IAAI;AAE9D,QAAM,EAAE,YAAY,QAAQ,SAAS,cAAc,IAAI;AAAA,IACrD,MAAM,iBAAiB,CAAC;AAAA,IACxB,CAAC,aAAa;AAAA,EAChB;AAEA,QAAM,MAAM,gBAAgB;AAC5B,QAAM,cAAc;AAAA,IAClB,OAAO,aAAyB,WAAuC;AAErE,YAAM,oBAAoB,YAAY;AAAA,QAAI,CAAC,MACzC,EAAE,KAAK,IAAI,EAAE,GAAG,GAAG,IAAI,OAAO,EAAE;AAAA,MAClC;AAEA,YAAM,cAAc,IAAI,4BAA4B;AAAA,QAClD,iBAAiB;AAAA,QACjB;AAAA,MACF,CAAC;AACD,kBAAY,YAAY,YAAY,iBAAiB,CAAC;AAEtD,YAAM,kBAAkB,IAAI,gBAAgB;AAC5C,yBAAmB,UAAU;AAC7B,YAAM,WAAW,MAAM,OAAO,mBAAmB;AAAA,QAC/C,GAAG;AAAA,QACH,aAAa,gBAAgB;AAAA,QAC7B,YAAY,YAAY;AACtB,iBAAO,MAAM,IAAI,eAAe,EAAE,WAAW;AAAA,QAC/C;AAAA,MACF,CAAC;AAED,uBAAiB,SAAS,UAAU;AAClC,gBAAQ,MAAM,OAAO;AAAA,UACnB,KAAK,yBAAyB;AAAA,UAC9B,KAAK,yBAAyB;AAC5B,wBAAY,YAAY,YAAY,MAAM,IAAI,CAAC;AAC/C;AAAA,UACF,KAAK,yBAAyB;AAC5B,yBAAa,MAAM,KAAK,gBAAgB,CAAC,CAAC;AAC1C;AAAA,UACF,KAAK,yBAAyB,UAAU;AACtC,kBAAM,CAAC,YAAY,IAAK,MAAqC;AAC7D,gBAAI,CAAC,wBAAwB,YAAY,GAAG;AAC1C,sBAAQ;AAAA,gBACN;AAAA,gBACA;AAAA,cACF;AACA;AAAA,YACF;AACA,kBAAM,kBAAkB,YAAY,YAAY;AAAA,cAC9C;AAAA,YACF,CAAC;AACD,wBAAY,eAAe;AAC3B;AAAA,UACF;AAAA,UACA,KAAK,yBAAyB;AAC5B,yBAAa,MAAM,IAAI;AACvB;AAAA,UACF,KAAK,yBAAyB;AAC5B,qBAAS,MAAM,IAAI;AACnB;AAAA,UACF,KAAK,yBAAyB,OAAO;AACnC,sBAAU,MAAM,IAAI;AAGpB,kBAAMA,YAAW,YAAY,YAAY;AACzC,kBAAM,gBAAgBA,UAAS;AAAA,cAC7B,CAAC,MACC,KAAK,QAAQ,UAAU,KAAK,EAAE,SAAS,QAAQ,EAAE,MAAM;AAAA,YAC3D;AACA,gBAAI,eAAe;AACjB,oBAAM,eAAe;AAAA,gBACnB,GAAG;AAAA,gBACH,QAAQ;AAAA,kBACN,MAAM;AAAA,kBACN,QAAQ;AAAA,kBACR,OAAO,MAAM;AAAA,gBACf;AAAA,cACF;AACA,0BAAY,YAAY,YAAY,CAAC,YAAY,CAAC,CAAC;AAAA,YACrD;AACA;AAAA,UACF;AAAA,UACA;AACE,gBAAI,eAAe;AACjB,4BAAc,MAAM,OAAO,MAAM,IAAI;AAAA,YACvC,OAAO;AACL,sBAAQ;AAAA,gBACN;AAAA,gBACA,MAAM;AAAA,gBACN,MAAM;AAAA,cACR;AAAA,YACF;AACA;AAAA,QACJ;AAAA,MACF;AAAA,IACF;AAAA,IACA;AAAA,MACE;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAEA,QAAM,SAAS,YAAY,MAAM;AAC/B,QAAI,mBAAmB,SAAS;AAC9B,yBAAmB,QAAQ,MAAM;AAAA,IACnC;AAAA,EACF,GAAG,CAAC,kBAAkB,CAAC;AAEvB,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;","names":["messages"]}
+{"version":3,"sources":["../src/useLangGraphMessages.ts"],"sourcesContent":["import { useState, useCallback, useRef, useMemo } from \"react\";\nimport { v4 as uuidv4 } from \"uuid\";\nimport { LangGraphMessageAccumulator } from \"./LangGraphMessageAccumulator\";\nimport {\n  EventType,\n  LangChainMessageTupleEvent,\n  LangGraphKnownEventTypes,\n  LangChainMessageChunk,\n  OnCustomEventCallback,\n  OnErrorEventCallback,\n  OnInfoEventCallback,\n  OnMetadataEventCallback,\n} from \"./types\";\nimport { useAssistantApi } from \"@assistant-ui/react\";\n\nexport type LangGraphCommand = {\n  resume: string;\n};\n\nexport type LangGraphSendMessageConfig = {\n  command?: LangGraphCommand;\n  runConfig?: unknown;\n};\n\nexport type LangGraphMessagesEvent<TMessage> = {\n  event: EventType;\n  data: TMessage[] | any;\n};\n\nexport type LangGraphStreamCallback<TMessage> = (\n  messages: TMessage[],\n  config: LangGraphSendMessageConfig & {\n    abortSignal: AbortSignal;\n    initialize: () => Promise<{\n      remoteId: string;\n      externalId: string | undefined;\n    }>;\n  },\n) =>\n  | Promise<AsyncGenerator<LangGraphMessagesEvent<TMessage>>>\n  | AsyncGenerator<LangGraphMessagesEvent<TMessage>>;\n\nexport type LangGraphInterruptState = {\n  value?: any;\n  resumable?: boolean;\n  when?: string;\n  ns?: string[];\n};\n\nconst DEFAULT_APPEND_MESSAGE = <TMessage>(\n  _: TMessage | undefined,\n  curr: TMessage,\n) => curr;\n\nconst isLangChainMessageChunk = (\n  value: unknown,\n): value is LangChainMessageChunk => {\n  if (!value || typeof value !== \"object\") return false;\n  const chunk = value as any;\n  return (\n    \"type\" in chunk &&\n    chunk.type === \"AIMessageChunk\" &&\n    (chunk.content === undefined ||\n      typeof chunk.content === \"string\" ||\n      Array.isArray(chunk.content)) &&\n    (chunk.tool_call_chunks === undefined ||\n      Array.isArray(chunk.tool_call_chunks))\n  );\n};\n\nexport const useLangGraphMessages = <TMessage extends { id?: string }>({\n  stream,\n  appendMessage = DEFAULT_APPEND_MESSAGE,\n  eventHandlers,\n}: {\n  stream: LangGraphStreamCallback<TMessage>;\n  appendMessage?: (prev: TMessage | undefined, curr: TMessage) => TMessage;\n  eventHandlers?: {\n    onMetadata?: OnMetadataEventCallback;\n    onInfo?: OnInfoEventCallback;\n    onError?: OnErrorEventCallback;\n    onCustomEvent?: OnCustomEventCallback;\n  };\n}) => {\n  const [interrupt, setInterrupt] = useState<\n    LangGraphInterruptState | undefined\n  >();\n  const [messages, setMessages] = useState<TMessage[]>([]);\n  const abortControllerRef = useRef<AbortController | null>(null);\n\n  const { onMetadata, onInfo, onError, onCustomEvent } = useMemo(\n    () => eventHandlers ?? {},\n    [eventHandlers],\n  );\n\n  const api = useAssistantApi();\n  const sendMessage = useCallback(\n    async (newMessages: TMessage[], config: LangGraphSendMessageConfig) => {\n      // ensure all messages have an ID\n      const newMessagesWithId = newMessages.map((m) =>\n        m.id ? m : { ...m, id: uuidv4() },\n      );\n\n      const accumulator = new LangGraphMessageAccumulator({\n        initialMessages: messages,\n        appendMessage,\n      });\n      setMessages(accumulator.addMessages(newMessagesWithId));\n\n      const abortController = new AbortController();\n      abortControllerRef.current = abortController;\n      const response = await stream(newMessagesWithId, {\n        ...config,\n        abortSignal: abortController.signal,\n        initialize: async () => {\n          return await api.threadListItem().initialize();\n        },\n      });\n\n      for await (const chunk of response) {\n        switch (chunk.event) {\n          case LangGraphKnownEventTypes.MessagesPartial:\n          case LangGraphKnownEventTypes.MessagesComplete:\n            setMessages(accumulator.addMessages(chunk.data));\n            break;\n          case LangGraphKnownEventTypes.Updates:\n            if (Array.isArray(chunk.data.messages)) {\n              setMessages(accumulator.replaceMessages(chunk.data.messages));\n            }\n            setInterrupt(chunk.data.__interrupt__?.[0]);\n            break;\n          case LangGraphKnownEventTypes.Messages: {\n            const [messageChunk] = (chunk as LangChainMessageTupleEvent).data;\n            if (!isLangChainMessageChunk(messageChunk)) {\n              console.warn(\n                \"Received invalid message chunk format:\",\n                messageChunk,\n              );\n              break;\n            }\n            const updatedMessages = accumulator.addMessages([\n              messageChunk as unknown as TMessage,\n            ]);\n            setMessages(updatedMessages);\n            break;\n          }\n          case LangGraphKnownEventTypes.Metadata:\n            onMetadata?.(chunk.data);\n            break;\n          case LangGraphKnownEventTypes.Info:\n            onInfo?.(chunk.data);\n            break;\n          case LangGraphKnownEventTypes.Error: {\n            onError?.(chunk.data);\n            // Update the last AI message with error status\n            // Assumes last AI message is the one the error relates to\n            const messages = accumulator.getMessages();\n            const lastAiMessage = messages.findLast(\n              (m): m is TMessage & { type: string; id: string } =>\n                m != null && \"type\" in m && m.type === \"ai\" && m.id != null,\n            );\n            if (lastAiMessage) {\n              const errorMessage = {\n                ...lastAiMessage,\n                status: {\n                  type: \"incomplete\" as const,\n                  reason: \"error\" as const,\n                  error: chunk.data,\n                },\n              };\n              setMessages(accumulator.addMessages([errorMessage]));\n            }\n            break;\n          }\n          default:\n            if (onCustomEvent) {\n              onCustomEvent(chunk.event, chunk.data);\n            } else {\n              console.warn(\n                \"Unhandled event received:\",\n                chunk.event,\n                chunk.data,\n              );\n            }\n            break;\n        }\n      }\n    },\n    [\n      messages,\n      appendMessage,\n      stream,\n      onMetadata,\n      onInfo,\n      onError,\n      onCustomEvent,\n    ],\n  );\n\n  const cancel = useCallback(() => {\n    if (abortControllerRef.current) {\n      abortControllerRef.current.abort();\n    }\n  }, [abortControllerRef]);\n\n  return {\n    interrupt,\n    messages,\n    sendMessage,\n    cancel,\n    setInterrupt,\n    setMessages,\n  };\n};\n"],"mappings":";AAAA,SAAS,UAAU,aAAa,QAAQ,eAAe;AACvD,SAAS,MAAM,cAAc;AAC7B,SAAS,mCAAmC;AAC5C;AAAA,EAGE;AAAA,OAMK;AACP,SAAS,uBAAuB;AAoChC,IAAM,yBAAyB,CAC7B,GACA,SACG;AAEL,IAAM,0BAA0B,CAC9B,UACmC;AACnC,MAAI,CAAC,SAAS,OAAO,UAAU,SAAU,QAAO;AAChD,QAAM,QAAQ;AACd,SACE,UAAU,SACV,MAAM,SAAS,qBACd,MAAM,YAAY,UACjB,OAAO,MAAM,YAAY,YACzB,MAAM,QAAQ,MAAM,OAAO,OAC5B,MAAM,qBAAqB,UAC1B,MAAM,QAAQ,MAAM,gBAAgB;AAE1C;AAEO,IAAM,uBAAuB,CAAmC;AAAA,EACrE;AAAA,EACA,gBAAgB;AAAA,EAChB;AACF,MASM;AACJ,QAAM,CAAC,WAAW,YAAY,IAAI,SAEhC;AACF,QAAM,CAAC,UAAU,WAAW,IAAI,SAAqB,CAAC,CAAC;AACvD,QAAM,qBAAqB,OAA+B,IAAI;AAE9D,QAAM,EAAE,YAAY,QAAQ,SAAS,cAAc,IAAI;AAAA,IACrD,MAAM,iBAAiB,CAAC;AAAA,IACxB,CAAC,aAAa;AAAA,EAChB;AAEA,QAAM,MAAM,gBAAgB;AAC5B,QAAM,cAAc;AAAA,IAClB,OAAO,aAAyB,WAAuC;AAErE,YAAM,oBAAoB,YAAY;AAAA,QAAI,CAAC,MACzC,EAAE,KAAK,IAAI,EAAE,GAAG,GAAG,IAAI,OAAO,EAAE;AAAA,MAClC;AAEA,YAAM,cAAc,IAAI,4BAA4B;AAAA,QAClD,iBAAiB;AAAA,QACjB;AAAA,MACF,CAAC;AACD,kBAAY,YAAY,YAAY,iBAAiB,CAAC;AAEtD,YAAM,kBAAkB,IAAI,gBAAgB;AAC5C,yBAAmB,UAAU;AAC7B,YAAM,WAAW,MAAM,OAAO,mBAAmB;AAAA,QAC/C,GAAG;AAAA,QACH,aAAa,gBAAgB;AAAA,QAC7B,YAAY,YAAY;AACtB,iBAAO,MAAM,IAAI,eAAe,EAAE,WAAW;AAAA,QAC/C;AAAA,MACF,CAAC;AAED,uBAAiB,SAAS,UAAU;AAClC,gBAAQ,MAAM,OAAO;AAAA,UACnB,KAAK,yBAAyB;AAAA,UAC9B,KAAK,yBAAyB;AAC5B,wBAAY,YAAY,YAAY,MAAM,IAAI,CAAC;AAC/C;AAAA,UACF,KAAK,yBAAyB;AAC5B,gBAAI,MAAM,QAAQ,MAAM,KAAK,QAAQ,GAAG;AACtC,0BAAY,YAAY,gBAAgB,MAAM,KAAK,QAAQ,CAAC;AAAA,YAC9D;AACA,yBAAa,MAAM,KAAK,gBAAgB,CAAC,CAAC;AAC1C;AAAA,UACF,KAAK,yBAAyB,UAAU;AACtC,kBAAM,CAAC,YAAY,IAAK,MAAqC;AAC7D,gBAAI,CAAC,wBAAwB,YAAY,GAAG;AAC1C,sBAAQ;AAAA,gBACN;AAAA,gBACA;AAAA,cACF;AACA;AAAA,YACF;AACA,kBAAM,kBAAkB,YAAY,YAAY;AAAA,cAC9C;AAAA,YACF,CAAC;AACD,wBAAY,eAAe;AAC3B;AAAA,UACF;AAAA,UACA,KAAK,yBAAyB;AAC5B,yBAAa,MAAM,IAAI;AACvB;AAAA,UACF,KAAK,yBAAyB;AAC5B,qBAAS,MAAM,IAAI;AACnB;AAAA,UACF,KAAK,yBAAyB,OAAO;AACnC,sBAAU,MAAM,IAAI;AAGpB,kBAAMA,YAAW,YAAY,YAAY;AACzC,kBAAM,gBAAgBA,UAAS;AAAA,cAC7B,CAAC,MACC,KAAK,QAAQ,UAAU,KAAK,EAAE,SAAS,QAAQ,EAAE,MAAM;AAAA,YAC3D;AACA,gBAAI,eAAe;AACjB,oBAAM,eAAe;AAAA,gBACnB,GAAG;AAAA,gBACH,QAAQ;AAAA,kBACN,MAAM;AAAA,kBACN,QAAQ;AAAA,kBACR,OAAO,MAAM;AAAA,gBACf;AAAA,cACF;AACA,0BAAY,YAAY,YAAY,CAAC,YAAY,CAAC,CAAC;AAAA,YACrD;AACA;AAAA,UACF;AAAA,UACA;AACE,gBAAI,eAAe;AACjB,4BAAc,MAAM,OAAO,MAAM,IAAI;AAAA,YACvC,OAAO;AACL,sBAAQ;AAAA,gBACN;AAAA,gBACA,MAAM;AAAA,gBACN,MAAM;AAAA,cACR;AAAA,YACF;AACA;AAAA,QACJ;AAAA,MACF;AAAA,IACF;AAAA,IACA;AAAA,MACE;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAEA,QAAM,SAAS,YAAY,MAAM;AAC/B,QAAI,mBAAmB,SAAS;AAC9B,yBAAmB,QAAQ,MAAM;AAAA,IACnC;AAAA,EACF,GAAG,CAAC,kBAAkB,CAAC;AAEvB,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;","names":["messages"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-langgraph",
-  "version": "0.7.0",
+  "version": "0.7.2",
   "license": "MIT",
   "type": "module",
   "exports": {
@@ -18,12 +18,12 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "assistant-stream": "^0.2.30",
+    "assistant-stream": "^0.2.34",
     "uuid": "^13.0.0",
     "zod": "^4.1.11"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.11.21",
+    "@assistant-ui/react": "^0.11.29",
     "@types/react": "*",
     "react": "^18 || ^19 || ^19.0.0-rc"
   },
@@ -35,16 +35,16 @@
   "devDependencies": {
     "@testing-library/dom": "^10.4.1",
     "@testing-library/react": "^16.3.0",
-    "@types/node": "^24.5.2",
-    "@types/react": "^19.1.15",
+    "@types/node": "^24.6.2",
+    "@types/react": "^19.2.0",
     "@types/uuid": "^11.0.0",
     "eslint": "^9",
     "eslint-config-next": "15.5.4",
     "jsdom": "^27.0.0",
-    "react": "19.1.1",
+    "react": "19.2.0",
     "tsx": "^4.20.6",
     "vitest": "^3.2.4",
-    "@assistant-ui/react": "0.11.21",
+    "@assistant-ui/react": "0.11.29",
     "@assistant-ui/x-buildutils": "0.0.1"
   },
   "publishConfig": {

package/src/LangGraphMessageAccumulator.ts

@@ -42,6 +42,15 @@ export class LangGraphMessageAccumulator<TMessage extends { id?: string }> {
     return [...this.messagesMap.values()];
   }
 
+  public replaceMessages(newMessages: TMessage[]): TMessage[] {
+    this.messagesMap.clear();
+
+    for (const message of newMessages.map(this.ensureMessageId)) {
+      this.messagesMap.set(message.id!, message);
+    }
+    return this.getMessages();
+  }
+
   public clear() {
     this.messagesMap.clear();
   }

package/src/useLangGraphMessages.test.ts

@@ -706,4 +706,68 @@ describe("useLangGraphMessages", {}, () => {
       ); // UUID v4 format
     });
   });
+
+  it("replaces messages with full list from updates event", async () => {
+    const initialHumanMessage = {
+      id: "user-1",
+      type: "human" as const,
+      content: "initial user message",
+    };
+
+    const manuallyAddedAIMessage = {
+      id: "ai-1",
+      type: "ai" as const,
+      content: "This is a manually added message from an Updates event",
+    };
+
+    const updatedMessagesFromBackend = [
+      initialHumanMessage,
+      manuallyAddedAIMessage,
+    ];
+
+    const mockStreamCallback = mockStreamCallbackFactory([
+      metadataEvent,
+      {
+        event: "updates",
+        data: {
+          messages: updatedMessagesFromBackend,
+        },
+      },
+      {
+        event: "messages",
+        data: [
+          {
+            id: "ai-2",
+            content: "This is a streamed AI response",
+            type: "AIMessageChunk",
+          },
+          { run_attempt: 1 },
+        ],
+      },
+    ]);
+
+    const { result } = renderHook(() =>
+      useLangGraphMessages({
+        stream: mockStreamCallback,
+        appendMessage: appendLangChainChunk,
+      }),
+    );
+
+    act(() => {
+      result.current.sendMessage([initialHumanMessage], {});
+    });
+
+    await waitFor(() => {
+      expect(result.current.messages).toHaveLength(3);
+      expect(result.current.messages[0].id).toEqual("user-1");
+      expect(result.current.messages[1].id).toEqual("ai-1");
+      expect(result.current.messages[1].content).toEqual(
+        "This is a manually added message from an Updates event",
+      );
+      expect(result.current.messages[2].id).toEqual("ai-2");
+      expect(result.current.messages[2].content).toEqual(
+        "This is a streamed AI response",
+      );
+    });
+  });
 });

package/src/useLangGraphMessages.ts

@@ -124,6 +124,9 @@ export const useLangGraphMessages = <TMessage extends { id?: string }>({
             setMessages(accumulator.addMessages(chunk.data));
             break;
           case LangGraphKnownEventTypes.Updates:
+            if (Array.isArray(chunk.data.messages)) {
+              setMessages(accumulator.replaceMessages(chunk.data.messages));
+            }
             setInterrupt(chunk.data.__interrupt__?.[0]);
             break;
           case LangGraphKnownEventTypes.Messages: {