@ai-sdk/langchain 2.0.210 → 2.0.212

package/dist/index.mjs

@@ -435,8 +435,24 @@ function extractImageOutputs(additionalKwargs) {
   if (!Array.isArray(toolOutputs)) return [];
   return toolOutputs.filter(isImageGenerationOutput);
 }
+function getOrCreateMessageSeen(messageSeen, msgId) {
+  let seen = messageSeen.get(msgId);
+  if (!seen) {
+    seen = {};
+    messageSeen.set(msgId, seen);
+  }
+  return seen;
+}
+function getOrCreateToolCallInfoByIndex(toolCallInfoByIndex, msgId) {
+  let toolCallInfo = toolCallInfoByIndex.get(msgId);
+  if (!toolCallInfo) {
+    toolCallInfo = /* @__PURE__ */ new Map();
+    toolCallInfoByIndex.set(msgId, toolCallInfo);
+  }
+  return toolCallInfo;
+}
 function processLangGraphEvent(event, state, controller) {
-  var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s, _t, _u, _v, _w, _x;
+  var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m;
   const {
     messageSeen,
     messageConcat,
@@ -477,16 +493,16 @@ function processLangGraphEvent(event, state, controller) {
       const langgraphStep = typeof (metadata == null ? void 0 : metadata.langgraph_step) === "number" ? metadata.langgraph_step : null;
       if (langgraphStep !== null && langgraphStep !== state.currentStep) {
         if (state.currentStep !== null) {
-          for (const [id, seen] of Object.entries(messageSeen)) {
+          for (const [id, seen] of messageSeen) {
             if (seen.text) {
               controller.enqueue({ type: "text-end", id });
             }
             if (seen.reasoning) {
               controller.enqueue({ type: "reasoning-end", id });
             }
-            delete messageSeen[id];
-            delete messageConcat[id];
-            delete messageReasoningIds[id];
+            messageSeen.delete(id);
+            messageConcat.delete(id);
+            messageReasoningIds.delete(id);
           }
           controller.enqueue({ type: "finish-step" });
         }
@@ -494,16 +510,18 @@ function processLangGraphEvent(event, state, controller) {
         state.currentStep = langgraphStep;
       }
       if (AIMessageChunk.isInstance(msg)) {
-        if (messageConcat[msgId]) {
-          messageConcat[msgId] = messageConcat[msgId].concat(
-            msg
+        const existingMessage = messageConcat.get(msgId);
+        if (existingMessage) {
+          messageConcat.set(
+            msgId,
+            existingMessage.concat(msg)
           );
         } else {
-          messageConcat[msgId] = msg;
+          messageConcat.set(msgId, msg);
         }
       }
       if (isAIMessageChunk(msg)) {
-        const concatChunk = messageConcat[msgId];
+        const concatChunk = messageConcat.get(msgId);
         const msgObj = msg;
         const dataSource = msgObj.type === "constructor" && msgObj.kwargs && typeof msgObj.kwargs === "object" ? msgObj.kwargs : msgObj;
         const additionalKwargs = dataSource.additional_kwargs;
@@ -524,28 +542,34 @@ function processLangGraphEvent(event, state, controller) {
           for (const toolCallChunk of toolCallChunks) {
             const idx = (_a = toolCallChunk.index) != null ? _a : 0;
             if (toolCallChunk.id) {
-              (_b = toolCallInfoByIndex[msgId]) != null ? _b : toolCallInfoByIndex[msgId] = {};
-              toolCallInfoByIndex[msgId][idx] = {
-                id: toolCallChunk.id,
-                name: toolCallChunk.name || ((_d = (_c = concatChunk == null ? void 0 : concatChunk.tool_call_chunks) == null ? void 0 : _c[idx]) == null ? void 0 : _d.name) || "unknown"
-              };
+              getOrCreateToolCallInfoByIndex(toolCallInfoByIndex, msgId).set(
+                idx,
+                {
+                  id: toolCallChunk.id,
+                  name: toolCallChunk.name || ((_c = (_b = concatChunk == null ? void 0 : concatChunk.tool_call_chunks) == null ? void 0 : _b[idx]) == null ? void 0 : _c.name) || "unknown"
+                }
+              );
             }
-            const toolCallId = toolCallChunk.id || ((_f = (_e = toolCallInfoByIndex[msgId]) == null ? void 0 : _e[idx]) == null ? void 0 : _f.id) || ((_h = (_g = concatChunk == null ? void 0 : concatChunk.tool_call_chunks) == null ? void 0 : _g[idx]) == null ? void 0 : _h.id);
+            const storedToolCallInfo = (_d = toolCallInfoByIndex.get(msgId)) == null ? void 0 : _d.get(idx);
+            const toolCallId = toolCallChunk.id || (storedToolCallInfo == null ? void 0 : storedToolCallInfo.id) || ((_f = (_e = concatChunk == null ? void 0 : concatChunk.tool_call_chunks) == null ? void 0 : _e[idx]) == null ? void 0 : _f.id);
             if (!toolCallId) {
               continue;
             }
-            const toolName = toolCallChunk.name || ((_j = (_i = toolCallInfoByIndex[msgId]) == null ? void 0 : _i[idx]) == null ? void 0 : _j.name) || ((_l = (_k = concatChunk == null ? void 0 : concatChunk.tool_call_chunks) == null ? void 0 : _k[idx]) == null ? void 0 : _l.name) || "unknown";
-            if (!((_n = (_m = messageSeen[msgId]) == null ? void 0 : _m.tool) == null ? void 0 : _n[toolCallId])) {
-              controller.enqueue({
-                type: "tool-input-start",
-                toolCallId,
-                toolName,
-                dynamic: true
-              });
-              (_o = messageSeen[msgId]) != null ? _o : messageSeen[msgId] = {};
-              (_q = (_p = messageSeen[msgId]).tool) != null ? _q : _p.tool = {};
-              messageSeen[msgId].tool[toolCallId] = true;
-              emittedToolCalls.add(toolCallId);
+            const toolName = toolCallChunk.name || (storedToolCallInfo == null ? void 0 : storedToolCallInfo.name) || ((_h = (_g = concatChunk == null ? void 0 : concatChunk.tool_call_chunks) == null ? void 0 : _g[idx]) == null ? void 0 : _h.name) || "unknown";
+            const seen = messageSeen.get(msgId);
+            if (!((_i = seen == null ? void 0 : seen.tool) == null ? void 0 : _i.has(toolCallId))) {
+              const updatedSeen = getOrCreateMessageSeen(messageSeen, msgId);
+              (_j = updatedSeen.tool) != null ? _j : updatedSeen.tool = /* @__PURE__ */ new Set();
+              updatedSeen.tool.add(toolCallId);
+              if (!emittedToolCalls.has(toolCallId)) {
+                emittedToolCalls.add(toolCallId);
+                controller.enqueue({
+                  type: "tool-input-start",
+                  toolCallId,
+                  toolName,
+                  dynamic: true
+                });
+              }
             }
             if (toolCallChunk.args) {
               controller.enqueue({
@@ -559,18 +583,18 @@ function processLangGraphEvent(event, state, controller) {
         }
         const chunkReasoningId = extractReasoningId(msg);
         if (chunkReasoningId) {
-          if (!messageReasoningIds[msgId]) {
-            messageReasoningIds[msgId] = chunkReasoningId;
+          if (!messageReasoningIds.has(msgId)) {
+            messageReasoningIds.set(msgId, chunkReasoningId);
           }
           emittedReasoningIds.add(chunkReasoningId);
         }
         const reasoning = extractReasoningFromContentBlocks(msg);
         if (reasoning) {
-          const reasoningId = (_s = (_r = messageReasoningIds[msgId]) != null ? _r : chunkReasoningId) != null ? _s : msgId;
-          if (!((_t = messageSeen[msgId]) == null ? void 0 : _t.reasoning)) {
+          const reasoningId = (_l = (_k = messageReasoningIds.get(msgId)) != null ? _k : chunkReasoningId) != null ? _l : msgId;
+          const seen = messageSeen.get(msgId);
+          if (!(seen == null ? void 0 : seen.reasoning)) {
             controller.enqueue({ type: "reasoning-start", id: msgId });
-            (_u = messageSeen[msgId]) != null ? _u : messageSeen[msgId] = {};
-            messageSeen[msgId].reasoning = true;
+            getOrCreateMessageSeen(messageSeen, msgId).reasoning = true;
           }
           controller.enqueue({
             type: "reasoning-delta",
@@ -581,10 +605,10 @@ function processLangGraphEvent(event, state, controller) {
         }
         const text = getMessageText(msg);
         if (text) {
-          if (!((_v = messageSeen[msgId]) == null ? void 0 : _v.text)) {
+          const seen = messageSeen.get(msgId);
+          if (!(seen == null ? void 0 : seen.text)) {
             controller.enqueue({ type: "text-start", id: msgId });
-            (_w = messageSeen[msgId]) != null ? _w : messageSeen[msgId] = {};
-            messageSeen[msgId].text = true;
+            getOrCreateMessageSeen(messageSeen, msgId).text = true;
           }
           controller.enqueue({
             type: "text-delta",
@@ -616,15 +640,15 @@ function processLangGraphEvent(event, state, controller) {
       return;
     }
     case "values": {
-      for (const [id, seen] of Object.entries(messageSeen)) {
+      for (const [id, seen] of messageSeen) {
         if (seen.text) controller.enqueue({ type: "text-end", id });
         if (seen.tool) {
-          for (const [toolCallId, toolCallSeen] of Object.entries(seen.tool)) {
-            const concatMsg = messageConcat[id];
-            const toolCall = (_x = concatMsg == null ? void 0 : concatMsg.tool_calls) == null ? void 0 : _x.find(
+          for (const toolCallId of seen.tool) {
+            const concatMsg = messageConcat.get(id);
+            const toolCall = (_m = concatMsg == null ? void 0 : concatMsg.tool_calls) == null ? void 0 : _m.find(
               (call) => call.id === toolCallId
             );
-            if (toolCallSeen && toolCall) {
+            if (toolCall) {
               emittedToolCalls.add(toolCallId);
               const toolCallKey = `${toolCall.name}:${JSON.stringify(toolCall.args)}`;
               emittedToolCallsByKey.set(toolCallKey, toolCallId);
@@ -641,9 +665,9 @@ function processLangGraphEvent(event, state, controller) {
         if (seen.reasoning) {
           controller.enqueue({ type: "reasoning-end", id });
         }
-        delete messageSeen[id];
-        delete messageConcat[id];
-        delete messageReasoningIds[id];
+        messageSeen.delete(id);
+        messageConcat.delete(id);
+        messageReasoningIds.delete(id);
       }
       if (data != null && typeof data === "object" && "messages" in data) {
         const messages = data.messages;
@@ -723,7 +747,7 @@ function processLangGraphEvent(event, state, controller) {
               }
             }
             const reasoningId = extractReasoningId(msg);
-            const wasStreamedThisRequest = !!messageSeen[msgId];
+            const wasStreamedThisRequest = messageSeen.has(msgId);
             const hasToolCalls = toolCalls && toolCalls.length > 0;
             const shouldEmitReasoning = reasoningId && !emittedReasoningIds.has(reasoningId) && (wasStreamedThisRequest || !hasToolCalls);
             if (shouldEmitReasoning) {
@@ -938,13 +962,13 @@ function toUIMessageStream(stream, callbacks) {
     reasoningMessageId: null
   };
   const langGraphState = {
-    messageSeen: {},
-    messageConcat: {},
+    messageSeen: /* @__PURE__ */ new Map(),
+    messageConcat: /* @__PURE__ */ new Map(),
     emittedToolCalls: /* @__PURE__ */ new Set(),
     emittedImages: /* @__PURE__ */ new Set(),
     emittedReasoningIds: /* @__PURE__ */ new Set(),
-    messageReasoningIds: {},
-    toolCallInfoByIndex: {},
+    messageReasoningIds: /* @__PURE__ */ new Map(),
+    toolCallInfoByIndex: /* @__PURE__ */ new Map(),
     currentStep: null,
     emittedToolCallsByKey: /* @__PURE__ */ new Map()
   };
@@ -1039,7 +1063,7 @@ function toUIMessageStream(stream, callbacks) {
           }
           controller.enqueue({ type: "finish" });
         } else if (streamType === "langgraph") {
-          for (const [id, seen] of Object.entries(langGraphState.messageSeen)) {
+          for (const [id, seen] of langGraphState.messageSeen) {
             if (seen.text) {
               controller.enqueue({ type: "text-end", id });
             }