@ai-sdk/langchain 2.0.214 → 2.0.216

package/dist/index.mjs

@@ -209,6 +209,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;
   }
@@ -264,6 +267,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;
@@ -426,6 +438,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";
 }
@@ -616,6 +707,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;
@@ -763,6 +863,15 @@ function processLangGraphEvent(event, state, controller) {
                 emittedReasoningIds.add(reasoningId);
               }
             }
+            const valuesCitations = extractCitationsFromContentBlocks(msg);
+            if (valuesCitations.length > 0) {
+              emitSourceChunks(
+                valuesCitations,
+                msgId,
+                state.emittedSourceIds,
+                controller
+              );
+            }
           }
         }
       }
@@ -912,6 +1021,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;
     }
@@ -959,7 +1077,8 @@ 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: /* @__PURE__ */ new Map(),
@@ -970,7 +1089,8 @@ function toUIMessageStream(stream, callbacks) {
     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 = () => {