@ai-sdk/langchain 2.0.300 → 2.0.302

package/CHANGELOG.md

@@ -1,5 +1,20 @@
 # @ai-sdk/langchain
 
+## 2.0.302
+
+### Patch Changes
+
+- e33fecc: fix(langchain): start bare messages-mode tool lifecycles before emitting outputs
+- Updated dependencies [16a04d7]
+  - ai@6.0.294
+
+## 2.0.301
+
+### Patch Changes
+
+- Updated dependencies [3983fea]
+  - ai@6.0.293
+
 ## 2.0.300
 
 ### Patch Changes

package/dist/index.js

@@ -636,6 +636,16 @@ function markToolCallEmitted(state, toolCallId, namespace) {
     ).add(toolCallId);
   }
 }
+function markToolCallStarted(state, toolCallId, namespace) {
+  markToolCallEmitted(state, toolCallId, namespace);
+  getOrCreateNamespaceSet(state.unfinishedToolCallsByNamespace, namespace).add(
+    toolCallId
+  );
+}
+function hasUnfinishedToolCall(state, toolCallId, namespace) {
+  var _a;
+  return ((_a = state.unfinishedToolCallsByNamespace.get(namespace)) == null ? void 0 : _a.has(toolCallId)) === true;
+}
 function findMessageCurrentStepNamespace(state, messageId) {
   for (const [
     namespace,
@@ -673,7 +683,7 @@ function closeStepNamespaceMessages(state, stepNamespace, controller) {
   return hasConcurrentMessageParts;
 }
 function processLangGraphEvent(event, state, controller) {
-  var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n;
+  var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o;
   const {
     messageSeen,
     messageConcat,
@@ -717,12 +727,14 @@ function processLangGraphEvent(event, state, controller) {
       if (namespace !== void 0) {
         state.messageNamespaces.set(msgId, namespace);
       }
+      let startedUIReducerStep = false;
       const langgraphStep = typeof (metadata == null ? void 0 : metadata.langgraph_step) === "number" ? metadata.langgraph_step : null;
       if (langgraphStep !== null) {
         const currentStep = (_a = state.currentStepsByNamespace.get(eventNamespace)) != null ? _a : null;
         if (currentStep === null) {
           if (state.currentStepsByNamespace.size === 0) {
             controller.enqueue({ type: "start-step" });
+            startedUIReducerStep = true;
           }
           state.currentStepsByNamespace.set(eventNamespace, langgraphStep);
           state.messageIdsInCurrentStepByNamespace.set(
@@ -743,6 +755,7 @@ function processLangGraphEvent(event, state, controller) {
             controller.enqueue({ type: "finish-step" });
           }
           controller.enqueue({ type: "start-step" });
+          startedUIReducerStep = true;
           state.currentStepsByNamespace.set(eventNamespace, langgraphStep);
           state.messageIdsInCurrentStepByNamespace.set(
             eventNamespace,
@@ -817,7 +830,7 @@ function processLangGraphEvent(event, state, controller) {
                 toolCallId,
                 eventNamespace
               )) {
-                markToolCallEmitted(state, toolCallId, eventNamespace);
+                markToolCallStarted(state, toolCallId, eventNamespace);
                 controller.enqueue({
                   type: "tool-input-start",
                   toolCallId,
@@ -890,6 +903,25 @@ function processLangGraphEvent(event, state, controller) {
         const toolCallId = dataSource.tool_call_id;
         const status = dataSource.status;
         if (toolCallId) {
+          const wasEmittedInCurrentStep = hasEmittedToolCallInCurrentStep(
+            state,
+            toolCallId,
+            eventNamespace
+          );
+          const isDelayedOutputForPreviousLifecycle = !wasEmittedInCurrentStep && hasUnfinishedToolCall(state, toolCallId, eventNamespace);
+          if (!wasEmittedInCurrentStep && !isDelayedOutputForPreviousLifecycle) {
+            if (!startedUIReducerStep && state.currentStepsByNamespace.has(eventNamespace) && emittedToolCalls.has(toolCallId)) {
+              controller.enqueue({ type: "start-step" });
+            }
+            markToolCallStarted(state, toolCallId, eventNamespace);
+            controller.enqueue({
+              type: "tool-input-start",
+              toolCallId,
+              toolName: typeof dataSource.name === "string" ? dataSource.name : "unknown",
+              dynamic: true
+            });
+          }
+          (_n = state.unfinishedToolCallsByNamespace.get(eventNamespace)) == null ? void 0 : _n.delete(toolCallId);
           if (status === "error") {
             controller.enqueue({
               type: "tool-output-error",
@@ -916,7 +948,7 @@ function processLangGraphEvent(event, state, controller) {
         if (seen.tool) {
           for (const toolCallId of seen.tool) {
             const concatMsg = messageConcat.get(id);
-            const toolCall = (_n = concatMsg == null ? void 0 : concatMsg.tool_calls) == null ? void 0 : _n.find(
+            const toolCall = (_o = concatMsg == null ? void 0 : concatMsg.tool_calls) == null ? void 0 : _o.find(
               (call) => call.id === toolCallId
             );
             if (toolCall) {
@@ -1009,7 +1041,7 @@ function processLangGraphEvent(event, state, controller) {
                   toolCall.id,
                   lifecycleNamespace
                 ) && (wasObservedInCurrentStep || !emittedToolCalls.has(toolCall.id) && !completedToolCallIds.has(toolCall.id))) {
-                  markToolCallEmitted(state, toolCall.id, lifecycleNamespace);
+                  markToolCallStarted(state, toolCall.id, lifecycleNamespace);
                   const toolCallKey = `${toolCall.name}:${JSON.stringify(toolCall.args)}`;
                   emittedToolCallsByKey.set(toolCallKey, toolCall.id);
                   controller.enqueue({
@@ -1074,7 +1106,7 @@ function processLangGraphEvent(event, state, controller) {
               const toolCallKey = `${toolName}:${JSON.stringify(input)}`;
               const toolCallId = emittedToolCallsByKey.get(toolCallKey) || actionRequest.id || `hitl-${toolName}-${Date.now()}`;
               if (!emittedToolCalls.has(toolCallId)) {
-                markToolCallEmitted(state, toolCallId, eventNamespace);
+                markToolCallStarted(state, toolCallId, eventNamespace);
                 emittedToolCallsByKey.set(toolCallKey, toolCallId);
                 controller.enqueue({
                   type: "tool-input-start",
@@ -1288,6 +1320,7 @@ function toUIMessageStream(stream, callbacks) {
     messageIdsInCurrentStepByNamespace: /* @__PURE__ */ new Map(),
     emittedToolCalls: /* @__PURE__ */ new Set(),
     emittedToolCallsInCurrentStepByNamespace: /* @__PURE__ */ new Map(),
+    unfinishedToolCallsByNamespace: /* @__PURE__ */ new Map(),
     emittedImages: /* @__PURE__ */ new Set(),
     emittedReasoningIds: /* @__PURE__ */ new Set(),
     messageReasoningIds: /* @__PURE__ */ new Map(),