@ai-sdk/langchain 3.0.0-beta.17 → 3.0.0-beta.178

package/dist/index.js

@@ -1,38 +1,18 @@
-"use strict";
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
-};
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
-      if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
-  }
-  return to;
-};
-var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-
-// src/index.ts
-var src_exports = {};
-__export(src_exports, {
-  LangSmithDeploymentTransport: () => LangSmithDeploymentTransport,
-  convertModelMessages: () => convertModelMessages,
-  toBaseMessages: () => toBaseMessages,
-  toUIMessageStream: () => toUIMessageStream
-});
-module.exports = __toCommonJS(src_exports);
-
 // src/adapter.ts
-var import_messages2 = require("@langchain/core/messages");
-var import_ai = require("ai");
+import {
+  SystemMessage
+} from "@langchain/core/messages";
+import {
+  convertToModelMessages
+} from "ai";
 
 // src/utils.ts
-var import_messages = require("@langchain/core/messages");
+import {
+  AIMessage,
+  HumanMessage,
+  ToolMessage,
+  AIMessageChunk
+} from "@langchain/core/messages";
 function parseLangGraphEvent(event) {
   return event.length === 3 ? [event[1], event[2]] : [event[0], event[1]];
 }
@@ -54,14 +34,14 @@ function convertToolResultPart(block) {
     }
     return "";
   })();
-  return new import_messages.ToolMessage({
+  return new ToolMessage({
     tool_call_id: block.toolCallId,
     content
   });
 }
 function convertAssistantContent(content) {
   if (typeof content === "string") {
-    return new import_messages.AIMessage({ content });
+    return new AIMessage({ content });
   }
   const textParts = [];
   const toolCalls = [];
@@ -76,19 +56,19 @@ function convertAssistantContent(content) {
       });
     }
   }
-  return new import_messages.AIMessage({
+  return new AIMessage({
     content: textParts.join(""),
     tool_calls: toolCalls.length > 0 ? toolCalls : void 0
   });
 }
 function getDefaultFilename(mediaType, prefix = "file") {
-  const ext = mediaType.split("/")[1] || "bin";
-  return `${prefix}.${ext}`;
+  const fileExtension = mediaType.split("/")[1] || "bin";
+  return `${prefix}.${fileExtension}`;
 }
 function convertUserContent(content) {
   var _a;
   if (typeof content === "string") {
-    return new import_messages.HumanMessage({ content });
+    return new HumanMessage({ content });
   }
   const contentBlocks = [];
   for (const part of content) {
@@ -129,7 +109,29 @@ function convertUserContent(content) {
         });
       }
     } else if (part.type === "file") {
-      const filePart = part;
+      const rawFilePart = part;
+      const normalizedData = (() => {
+        const data = rawFilePart.data;
+        if (typeof data === "object" && data !== null && !(data instanceof URL) && !(data instanceof Uint8Array) && !(data instanceof ArrayBuffer) && "type" in data) {
+          switch (data.type) {
+            case "data":
+              return data.data;
+            case "url":
+              return data.url;
+            case "text":
+              return data.text;
+            default:
+              return "";
+          }
+        }
+        return data;
+      })();
+      const filePart = {
+        type: "file",
+        data: normalizedData,
+        mediaType: rawFilePart.mediaType,
+        filename: rawFilePart.filename
+      };
       const isImage = (_a = filePart.mediaType) == null ? void 0 : _a.startsWith("image/");
       if (isImage) {
         if (filePart.data instanceof URL) {
@@ -215,11 +217,11 @@ function convertUserContent(content) {
     }
   }
   if (contentBlocks.every((block) => block.type === "text")) {
-    return new import_messages.HumanMessage({
+    return new HumanMessage({
       content: contentBlocks.map((block) => block.text).join("")
     });
   }
-  return new import_messages.HumanMessage({ content: contentBlocks });
+  return new HumanMessage({ content: contentBlocks });
 }
 function isToolResultPart(item) {
   return item != null && typeof item === "object" && "type" in item && item.type === "tool-result";
@@ -229,6 +231,9 @@ function processModelChunk(chunk, state, controller) {
   if (!state.emittedImages) {
     state.emittedImages = /* @__PURE__ */ new Set();
   }
+  if (!state.emittedSourceIds) {
+    state.emittedSourceIds = /* @__PURE__ */ new Set();
+  }
   if (chunk.id) {
     state.messageId = chunk.id;
   }
@@ -284,6 +289,15 @@ function processModelChunk(chunk, state, controller) {
       id: (_c = state.textMessageId) != null ? _c : state.messageId
     });
   }
+  const citations = extractCitationsFromContentBlocks(chunk);
+  if (citations.length > 0) {
+    emitSourceChunks(
+      citations,
+      state.messageId,
+      state.emittedSourceIds,
+      controller
+    );
+  }
 }
 function isPlainMessageObject(msg) {
   if (msg == null || typeof msg !== "object") return false;
@@ -304,22 +318,24 @@ function getMessageId(msg) {
   return void 0;
 }
 function isAIMessageChunk(msg) {
-  if (import_messages.AIMessageChunk.isInstance(msg)) return true;
+  if (AIMessageChunk.isInstance(msg)) return true;
   if (isPlainMessageObject(msg)) {
-    const obj = msg;
-    if ("type" in obj && obj.type === "ai") return true;
-    if (obj.type === "constructor" && Array.isArray(obj.id) && (obj.id.includes("AIMessageChunk") || obj.id.includes("AIMessage"))) {
+    const messageRecord = msg;
+    if ("type" in messageRecord && (messageRecord.type === "ai" || messageRecord.type === "AIMessageChunk")) {
+      return true;
+    }
+    if (messageRecord.type === "constructor" && Array.isArray(messageRecord.id) && (messageRecord.id.includes("AIMessageChunk") || messageRecord.id.includes("AIMessage"))) {
       return true;
     }
   }
   return false;
 }
 function isToolMessageType(msg) {
-  if (import_messages.ToolMessage.isInstance(msg)) return true;
+  if (ToolMessage.isInstance(msg)) return true;
   if (isPlainMessageObject(msg)) {
-    const obj = msg;
-    if ("type" in obj && obj.type === "tool") return true;
-    if (obj.type === "constructor" && Array.isArray(obj.id) && obj.id.includes("ToolMessage")) {
+    const messageRecord = msg;
+    if ("type" in messageRecord && messageRecord.type === "tool") return true;
+    if (messageRecord.type === "constructor" && Array.isArray(messageRecord.id) && messageRecord.id.includes("ToolMessage")) {
       return true;
     }
   }
@@ -327,7 +343,7 @@ function isToolMessageType(msg) {
 }
 function getMessageText(msg) {
   var _a;
-  if (import_messages.AIMessageChunk.isInstance(msg)) {
+  if (AIMessageChunk.isInstance(msg)) {
     return (_a = msg.text) != null ? _a : "";
   }
   if (msg == null || typeof msg !== "object") return "";
@@ -444,6 +460,85 @@ function extractReasoningFromValuesMessage(msg) {
   }
   return void 0;
 }
+function isCitationContentBlock(obj) {
+  return obj != null && typeof obj === "object" && "type" in obj && obj.type === "citation";
+}
+function extractCitationsFromContentBlocks(msg) {
+  if (msg == null || typeof msg !== "object") return [];
+  const msgObj = msg;
+  const kwargs = msgObj.kwargs && typeof msgObj.kwargs === "object" ? msgObj.kwargs : msgObj;
+  const contentBlocks = kwargs.contentBlocks;
+  const content = kwargs.content;
+  const blockSources = Array.isArray(contentBlocks) ? contentBlocks : Array.isArray(content) ? content : [];
+  const citations = [];
+  for (const block of blockSources) {
+    if (block == null || typeof block !== "object") continue;
+    const annotations = block.annotations;
+    if (!Array.isArray(annotations)) continue;
+    for (const annotation of annotations) {
+      if (!isCitationContentBlock(annotation)) continue;
+      citations.push({
+        url: annotation.url,
+        title: annotation.title,
+        source: annotation.source,
+        citedText: annotation.citedText,
+        startIndex: annotation.startIndex,
+        endIndex: annotation.endIndex
+      });
+    }
+  }
+  return citations;
+}
+function buildSourceProviderMetadata(citation) {
+  const langchain = {};
+  if (typeof citation.citedText === "string") {
+    langchain.citedText = citation.citedText;
+  }
+  if (typeof citation.startIndex === "number") {
+    langchain.startIndex = citation.startIndex;
+  }
+  if (typeof citation.endIndex === "number") {
+    langchain.endIndex = citation.endIndex;
+  }
+  if (typeof citation.source === "string") {
+    langchain.source = citation.source;
+  }
+  if (Object.keys(langchain).length === 0) {
+    return void 0;
+  }
+  return { langchain };
+}
+function emitSourceChunks(citations, messageId, emittedSourceIds, controller) {
+  var _a, _b, _c, _d;
+  for (const citation of citations) {
+    if (citation.url) {
+      if (emittedSourceIds.has(citation.url)) continue;
+      emittedSourceIds.add(citation.url);
+      const providerMetadata2 = buildSourceProviderMetadata(citation);
+      controller.enqueue({
+        type: "source-url",
+        sourceId: citation.url,
+        url: citation.url,
+        ...citation.title ? { title: citation.title } : {},
+        ...providerMetadata2 ? { providerMetadata: providerMetadata2 } : {}
+      });
+      continue;
+    }
+    const title = (_a = citation.title) != null ? _a : citation.source;
+    if (!title) continue;
+    const sourceId = `${messageId}:${title}:${(_b = citation.citedText) != null ? _b : ""}:${(_c = citation.startIndex) != null ? _c : ""}:${(_d = citation.endIndex) != null ? _d : ""}`;
+    if (emittedSourceIds.has(sourceId)) continue;
+    emittedSourceIds.add(sourceId);
+    const providerMetadata = buildSourceProviderMetadata(citation);
+    controller.enqueue({
+      type: "source-document",
+      sourceId,
+      mediaType: "text/plain",
+      title,
+      ...providerMetadata ? { providerMetadata } : {}
+    });
+  }
+}
 function isImageGenerationOutput(obj) {
   return obj != null && typeof obj === "object" && "type" in obj && obj.type === "image_generation_call";
 }
@@ -453,8 +548,34 @@ function extractImageOutputs(additionalKwargs) {
   if (!Array.isArray(toolOutputs)) return [];
   return toolOutputs.filter(isImageGenerationOutput);
 }
+function formatToolError(error) {
+  if (error instanceof Error) return error.message;
+  if (typeof error === "string") return error;
+  try {
+    const serialized = JSON.stringify(error);
+    return serialized != null ? serialized : String(error);
+  } catch (e) {
+    return String(error);
+  }
+}
+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,
@@ -463,7 +584,8 @@ function processLangGraphEvent(event, state, controller) {
     emittedReasoningIds,
     messageReasoningIds,
     toolCallInfoByIndex,
-    emittedToolCallsByKey
+    emittedToolCallsByKey,
+    emittedToolInputs
   } = state;
   const [type, data] = parseLangGraphEvent(event);
   switch (type) {
@@ -495,22 +617,35 @@ 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 messageSeen) {
+            if (seen.text) {
+              controller.enqueue({ type: "text-end", id });
+            }
+            if (seen.reasoning) {
+              controller.enqueue({ type: "reasoning-end", id });
+            }
+            messageSeen.delete(id);
+            messageConcat.delete(id);
+            messageReasoningIds.delete(id);
+          }
           controller.enqueue({ type: "finish-step" });
         }
         controller.enqueue({ type: "start-step" });
         state.currentStep = langgraphStep;
       }
-      if (import_messages.AIMessageChunk.isInstance(msg)) {
-        if (messageConcat[msgId]) {
-          messageConcat[msgId] = messageConcat[msgId].concat(
-            msg
+      if (AIMessageChunk.isInstance(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;
@@ -529,30 +664,36 @@ function processLangGraphEvent(event, state, controller) {
         const toolCallChunks = dataSource.tool_call_chunks;
         if (toolCallChunks == null ? void 0 : toolCallChunks.length) {
           for (const toolCallChunk of toolCallChunks) {
-            const idx = (_a = toolCallChunk.index) != null ? _a : 0;
+            const toolCallIndex = (_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(
+                toolCallIndex,
+                {
+                  id: toolCallChunk.id,
+                  name: toolCallChunk.name || ((_c = (_b = concatChunk == null ? void 0 : concatChunk.tool_call_chunks) == null ? void 0 : _b[toolCallIndex]) == 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(toolCallIndex);
+            const toolCallId = toolCallChunk.id || (storedToolCallInfo == null ? void 0 : storedToolCallInfo.id) || ((_f = (_e = concatChunk == null ? void 0 : concatChunk.tool_call_chunks) == null ? void 0 : _e[toolCallIndex]) == 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[toolCallIndex]) == 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({
@@ -566,18 +707,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",
@@ -588,10 +729,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",
@@ -599,6 +740,15 @@ function processLangGraphEvent(event, state, controller) {
             id: msgId
           });
         }
+        const citations = extractCitationsFromContentBlocks(msg);
+        if (citations.length > 0) {
+          emitSourceChunks(
+            citations,
+            msgId,
+            state.emittedSourceIds,
+            controller
+          );
+        }
       } else if (isToolMessageType(msg)) {
         const msgObj = msg;
         const dataSource = msgObj.type === "constructor" && msgObj.kwargs && typeof msgObj.kwargs === "object" ? msgObj.kwargs : msgObj;
@@ -622,35 +772,105 @@ function processLangGraphEvent(event, state, controller) {
       }
       return;
     }
+    case "tools": {
+      if (data == null || typeof data !== "object" || Array.isArray(data)) {
+        return;
+      }
+      const payload = data;
+      const toolCallId = typeof payload.toolCallId === "string" ? payload.toolCallId : void 0;
+      const toolName = typeof payload.name === "string" ? payload.name : "unknown";
+      if (!toolCallId) return;
+      const ensureToolInputLifecycle = () => {
+        if (!emittedToolCalls.has(toolCallId)) {
+          emittedToolCalls.add(toolCallId);
+          controller.enqueue({
+            type: "tool-input-start",
+            toolCallId,
+            toolName,
+            dynamic: true
+          });
+        }
+        if (!emittedToolInputs.has(toolCallId)) {
+          emittedToolInputs.add(toolCallId);
+          controller.enqueue({
+            type: "tool-input-available",
+            toolCallId,
+            toolName,
+            input: payload.input,
+            dynamic: true
+          });
+        }
+      };
+      switch (payload.event) {
+        case "on_tool_start": {
+          const toolCallKey = `${toolName}:${JSON.stringify(payload.input)}`;
+          emittedToolCallsByKey.set(toolCallKey, toolCallId);
+          ensureToolInputLifecycle();
+          break;
+        }
+        case "on_tool_event": {
+          ensureToolInputLifecycle();
+          controller.enqueue({
+            type: "tool-output-available",
+            toolCallId,
+            output: payload.data,
+            preliminary: true
+          });
+          break;
+        }
+        case "on_tool_end": {
+          ensureToolInputLifecycle();
+          controller.enqueue({
+            type: "tool-output-available",
+            toolCallId,
+            output: payload.output
+          });
+          break;
+        }
+        case "on_tool_error": {
+          ensureToolInputLifecycle();
+          controller.enqueue({
+            type: "tool-output-error",
+            toolCallId,
+            errorText: formatToolError(payload.error)
+          });
+          break;
+        }
+      }
+      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);
-              controller.enqueue({
-                type: "tool-input-available",
-                toolCallId,
-                toolName: toolCall.name,
-                input: toolCall.args,
-                dynamic: true
-              });
+              if (!emittedToolInputs.has(toolCallId)) {
+                emittedToolInputs.add(toolCallId);
+                controller.enqueue({
+                  type: "tool-input-available",
+                  toolCallId,
+                  toolName: toolCall.name,
+                  input: toolCall.args,
+                  dynamic: true
+                });
+              }
             }
           }
         }
         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;
@@ -672,13 +892,13 @@ function processLangGraphEvent(event, state, controller) {
             const msgId = getMessageId(msg);
             if (!msgId) continue;
             let toolCalls;
-            if (import_messages.AIMessageChunk.isInstance(msg) || import_messages.AIMessage.isInstance(msg)) {
+            if (AIMessageChunk.isInstance(msg) || AIMessage.isInstance(msg)) {
               toolCalls = msg.tool_calls;
             } else if (isPlainMessageObject(msg)) {
-              const obj = msg;
-              const isSerializedFormat = obj.type === "constructor" && Array.isArray(obj.id) && (obj.id.includes("AIMessageChunk") || obj.id.includes("AIMessage"));
-              const dataSource = isSerializedFormat ? obj.kwargs : obj;
-              if (obj.type === "ai" || isSerializedFormat) {
+              const messageRecord = msg;
+              const isSerializedFormat = messageRecord.type === "constructor" && Array.isArray(messageRecord.id) && (messageRecord.id.includes("AIMessageChunk") || messageRecord.id.includes("AIMessage"));
+              const dataSource = isSerializedFormat ? messageRecord.kwargs : messageRecord;
+              if (messageRecord.type === "ai" || messageRecord.type === "AIMessageChunk" || isSerializedFormat) {
                 if (Array.isArray(dataSource == null ? void 0 : dataSource.tool_calls)) {
                   toolCalls = dataSource.tool_calls;
                 } else if (
@@ -719,6 +939,7 @@ function processLangGraphEvent(event, state, controller) {
                     toolName: toolCall.name,
                     dynamic: true
                   });
+                  emittedToolInputs.add(toolCall.id);
                   controller.enqueue({
                     type: "tool-input-available",
                     toolCallId: toolCall.id,
@@ -726,11 +947,14 @@ function processLangGraphEvent(event, state, controller) {
                     input: toolCall.args,
                     dynamic: true
                   });
+                } else if (toolCall.id && emittedToolCalls.has(toolCall.id)) {
+                  const toolCallKey = `${toolCall.name}:${JSON.stringify(toolCall.args)}`;
+                  emittedToolCallsByKey.set(toolCallKey, toolCall.id);
                 }
               }
             }
             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) {
@@ -746,6 +970,15 @@ function processLangGraphEvent(event, state, controller) {
                 emittedReasoningIds.add(reasoningId);
               }
             }
+            const valuesCitations = extractCitationsFromContentBlocks(msg);
+            if (valuesCitations.length > 0) {
+              emitSourceChunks(
+                valuesCitations,
+                msgId,
+                state.emittedSourceIds,
+                controller
+              );
+            }
           }
         }
       }
@@ -771,6 +1004,7 @@ function processLangGraphEvent(event, state, controller) {
                   toolName,
                   dynamic: true
                 });
+                emittedToolInputs.add(toolCallId);
                 controller.enqueue({
                   type: "tool-input-available",
                   toolCallId,
@@ -795,7 +1029,7 @@ function processLangGraphEvent(event, state, controller) {
 
 // src/adapter.ts
 async function toBaseMessages(messages) {
-  const modelMessages = await (0, import_ai.convertToModelMessages)(messages);
+  const modelMessages = await convertToModelMessages(messages);
   return convertModelMessages(modelMessages);
 }
 function convertModelMessages(modelMessages) {
@@ -815,7 +1049,7 @@ function convertModelMessages(modelMessages) {
         break;
       }
       case "system": {
-        result.push(new import_messages2.SystemMessage({ content: message.content }));
+        result.push(new SystemMessage({ content: message.content }));
         break;
       }
       case "user": {
@@ -841,6 +1075,22 @@ function processStreamEventsEvent(event, state, controller) {
   if (!event.data) return;
   switch (event.event) {
     case "on_chat_model_start": {
+      if (state.reasoningStarted) {
+        controller.enqueue({
+          type: "reasoning-end",
+          id: state.reasoningMessageId != null ? state.reasoningMessageId : state.messageId
+        });
+        state.reasoningStarted = false;
+        state.reasoningMessageId = null;
+      }
+      if (state.textStarted) {
+        controller.enqueue({
+          type: "text-end",
+          id: state.textMessageId != null ? state.textMessageId : state.messageId
+        });
+        state.textStarted = false;
+        state.textMessageId = null;
+      }
       const runId = event.run_id || event.data.run_id;
       if (runId) {
         state.messageId = runId;
@@ -895,6 +1145,15 @@ function processStreamEventsEvent(event, state, controller) {
             id: (_c = state.textMessageId) != null ? _c : state.messageId
           });
         }
+        const citations = extractCitationsFromContentBlocks(chunk);
+        if (citations.length > 0) {
+          emitSourceChunks(
+            citations,
+            state.messageId,
+            state.emittedSourceIds,
+            controller
+          );
+        }
       }
       break;
     }
@@ -942,18 +1201,21 @@ function toUIMessageStream(stream, callbacks) {
     /** Track the ID used for text-start to ensure text-end uses the same ID */
     textMessageId: null,
     /** Track the ID used for reasoning-start to ensure reasoning-end uses the same ID */
-    reasoningMessageId: null
+    reasoningMessageId: null,
+    emittedSourceIds: /* @__PURE__ */ new Set()
   };
   const langGraphState = {
-    messageSeen: {},
-    messageConcat: {},
+    messageSeen: /* @__PURE__ */ new Map(),
+    messageConcat: /* @__PURE__ */ new Map(),
     emittedToolCalls: /* @__PURE__ */ new Set(),
+    emittedToolInputs: /* @__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()
+    emittedToolCallsByKey: /* @__PURE__ */ new Map(),
+    emittedSourceIds: /* @__PURE__ */ new Set()
   };
   let streamType = null;
   const getAsyncIterator = () => {
@@ -1046,6 +1308,14 @@ function toUIMessageStream(stream, callbacks) {
           }
           controller.enqueue({ type: "finish" });
         } else if (streamType === "langgraph") {
+          for (const [id, seen] of langGraphState.messageSeen) {
+            if (seen.text) {
+              controller.enqueue({ type: "text-end", id });
+            }
+            if (seen.reasoning) {
+              controller.enqueue({ type: "reasoning-end", id });
+            }
+          }
           if (langGraphState.currentStep !== null) {
             controller.enqueue({ type: "finish-step" });
           }
@@ -1073,11 +1343,13 @@ function toUIMessageStream(stream, callbacks) {
 }
 
 // src/transport.ts
-var import_remote = require("@langchain/langgraph/remote");
+import {
+  RemoteGraph
+} from "@langchain/langgraph/remote";
 var LangSmithDeploymentTransport = class {
   constructor(options) {
     var _a;
-    this.graph = new import_remote.RemoteGraph({
+    this.graph = new RemoteGraph({
       ...options,
       graphId: (_a = options.graphId) != null ? _a : "agent"
     });
@@ -1096,11 +1368,10 @@ var LangSmithDeploymentTransport = class {
     throw new Error("Method not implemented.");
   }
 };
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
+export {
   LangSmithDeploymentTransport,
   convertModelMessages,
   toBaseMessages,
   toUIMessageStream
-});
+};
 //# sourceMappingURL=index.js.map