@assistant-ui/react-langgraph 0.5.5 → 0.5.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-langgraph",
-  "version": "0.5.5",
+  "version": "0.5.7",
   "license": "MIT",
   "type": "module",
   "exports": {
@@ -20,10 +20,10 @@
   "dependencies": {
     "assistant-stream": "^0.2.14",
     "uuid": "^11.1.0",
-    "zod": "^3.25.28"
+    "zod": "^3.25.49"
   },
   "peerDependencies": {
-    "@assistant-ui/react": "^0.10.19",
+    "@assistant-ui/react": "^0.10.21",
     "@types/react": "*",
     "react": "^18 || ^19 || ^19.0.0-rc"
   },
@@ -33,14 +33,18 @@
     }
   },
   "devDependencies": {
-    "@types/node": "^22.15.21",
-    "@types/react": "^19.1.5",
+    "@testing-library/dom": "^10.4.0",
+    "@testing-library/react": "^16.3.0",
+    "@types/node": "^22.15.29",
+    "@types/react": "^19.1.6",
     "@types/uuid": "^10.0.0",
     "eslint": "^9",
-    "eslint-config-next": "15.3.2",
+    "eslint-config-next": "15.3.3",
+    "jsdom": "^26.1.0",
     "react": "^19.1.0",
     "tsx": "^4.19.4",
-    "@assistant-ui/react": "0.10.19",
+    "vitest": "^3.2.0",
+    "@assistant-ui/react": "0.10.21",
     "@assistant-ui/x-buildutils": "0.0.1"
   },
   "publishConfig": {
@@ -57,6 +61,8 @@
   },
   "scripts": {
     "build": "tsx scripts/build.mts",
-    "lint": "eslint ."
+    "lint": "eslint .",
+    "test": "vitest run",
+    "test:watch": "vitest"
   }
 }

package/src/LangGraphMessageAccumulator.ts

@@ -1,16 +1,5 @@
 import { v4 as uuidv4 } from "uuid";
 
-export type LangGraphMessagesEvent<TMessage> = {
-  event:
-    | "messages"
-    | "messages/partial"
-    | "messages/complete"
-    | "metadata"
-    | "updates"
-    | string;
-  data: TMessage[] | any;
-};
-
 export type LangGraphStateAccumulatorConfig<TMessage> = {
   initialMessages?: TMessage[];
   appendMessage?: (prev: TMessage | undefined, curr: TMessage) => TMessage;

package/src/appendLangChainChunk.ts

@@ -1,4 +1,8 @@
-import { LangChainMessage, LangChainMessageChunk } from "./types";
+import {
+  LangChainMessage,
+  LangChainMessageChunk,
+  MessageContentText,
+} from "./types";
 import { parsePartialJsonObject } from "assistant-stream/utils";
 
 export const appendLangChainChunk = (
@@ -21,20 +25,36 @@ export const appendLangChainChunk = (
       ? [{ type: "text" as const, text: prev.content }]
       : [...prev.content];
 
-  for (const chunk of curr.content) {
-    if (chunk.type === "text") {
-      const existing = newContent[chunk.index] ?? { type: "text", text: "" };
-      if (existing.type !== "text") throw new Error("");
-      newContent[chunk.index] = {
-        ...existing,
-        ...chunk,
-        text: existing.text + chunk.text,
-      };
+  if (typeof curr?.content === "string") {
+    const lastIndex = newContent.length - 1;
+    if (newContent[lastIndex]?.type === "text") {
+      (newContent[lastIndex] as MessageContentText).text =
+        (newContent[lastIndex] as MessageContentText).text + curr.content;
+    } else {
+      newContent.push({ type: "text", text: curr.content });
+    }
+  } else if (Array.isArray(curr.content)) {
+    const lastIndex = newContent.length - 1;
+    for (const item of curr.content) {
+      if (!("type" in item)) {
+        continue;
+      }
+
+      if (item.type === "text") {
+        if (newContent[lastIndex]?.type === "text") {
+          (newContent[lastIndex] as MessageContentText).text =
+            (newContent[lastIndex] as MessageContentText).text + item.text;
+        } else {
+          newContent.push({ type: "text", text: item.text });
+        }
+      } else if (item.type === "image_url") {
+        newContent.push(item);
+      }
     }
   }
 
   const newToolCalls = [...(prev.tool_calls ?? [])];
-  for (const chunk of curr.tool_call_chunks) {
+  for (const chunk of curr.tool_call_chunks ?? []) {
     const existing = newToolCalls[chunk.index - 1] ?? { argsText: "" };
     const newArgsText = existing.argsText + chunk.args;
     newToolCalls[chunk.index - 1] = {

package/src/convertLangChainMessages.ts

@@ -14,6 +14,8 @@ const contentToParts = (content: LangChainMessage["content"]) => {
       switch (type) {
         case "text":
           return { type: "text", text: part.text };
+        case "text_delta":
+          return { type: "text", text: part.text };
         case "image_url":
           if (typeof part.image_url === "string") {
             return { type: "image", image: part.image_url };
@@ -26,6 +28,8 @@ const contentToParts = (content: LangChainMessage["content"]) => {
 
         case "tool_use":
           return null;
+        case "input_json_delta":
+          return null;
         default:
           const _exhaustiveCheck: never = type;
           throw new Error(`Unknown content part type: ${_exhaustiveCheck}`);

package/src/testUtils.ts

@@ -0,0 +1,11 @@
+import { LangChainMessage } from "./types";
+import { LangGraphMessagesEvent } from "./useLangGraphMessages";
+
+export const mockStreamCallbackFactory = (
+  events: Array<LangGraphMessagesEvent<LangChainMessage>>,
+) =>
+  async function* () {
+    for (const event of events) {
+      yield event;
+    }
+  };

package/src/types.ts

@@ -14,23 +14,38 @@ export type LangChainToolCall = {
   args: ReadonlyJSONObject;
 };
 
-type MessageContentText = {
-  type: "text";
+export type MessageContentText = {
+  type: "text" | "text_delta";
   text: string;
 };
 
-type MessageContentImageUrl = {
+export type MessageContentImageUrl = {
   type: "image_url";
   image_url: string | { url: string };
 };
 
 type MessageContentToolUse = {
-  type: "tool_use";
+  type: "tool_use" | "input_json_delta";
 };
 
+export enum LangGraphKnownEventTypes {
+  Messages = "messages",
+  MessagesPartial = "messages/partial",
+  MessagesComplete = "messages/complete",
+  Metadata = "metadata",
+  Updates = "updates",
+  Info = "info",
+  Error = "error",
+}
+
+type CustomEventType = string;
+
+export type EventType = LangGraphKnownEventTypes | CustomEventType;
+
 type UserMessageContentComplex = MessageContentText | MessageContentImageUrl;
 type AssistantMessageContentComplex =
   | MessageContentText
+  | MessageContentImageUrl
   | MessageContentToolUse;
 
 type UserMessageContent = string | UserMessageContentComplex[];
@@ -65,13 +80,32 @@ export type LangChainMessage =
     };
 
 export type LangChainMessageChunk = {
-  id: string;
+  id?: string | undefined;
   type: "AIMessageChunk";
-  content: (AssistantMessageContentComplex & { index: number })[];
-  tool_call_chunks: LangChainToolCallChunk[];
+  content?: AssistantMessageContent | undefined;
+  tool_call_chunks?: LangChainToolCallChunk[] | undefined;
 };
 
 export type LangChainEvent = {
-  event: "messages/partial" | "messages/complete";
+  event:
+    | LangGraphKnownEventTypes.MessagesPartial
+    | LangGraphKnownEventTypes.MessagesComplete;
   data: LangChainMessage[];
 };
+
+type LangGraphTupleMetadata = Record<string, unknown>;
+
+export type LangChainMessageTupleEvent = {
+  event: LangGraphKnownEventTypes.Messages;
+  data: [LangChainMessageChunk, LangGraphTupleMetadata];
+};
+
+export type OnMetadataEventCallback = (
+  metadata: unknown,
+) => void | Promise<void>;
+export type OnInfoEventCallback = (info: unknown) => void | Promise<void>;
+export type OnErrorEventCallback = (error: unknown) => void | Promise<void>;
+export type OnCustomEventCallback = (
+  type: string,
+  data: unknown,
+) => void | Promise<void>;