ai 4.1.26 → 4.1.28

package/dist/index.mjs

@@ -3832,7 +3832,7 @@ async function generateText({
     }),
     tracer,
     fn: async (span) => {
-      var _a16, _b, _c, _d, _e, _f, _g;
+      var _a16, _b, _c, _d, _e, _f, _g, _h, _i;
       const mode = {
         type: "regular",
         ...prepareToolsAndToolChoice({ tools, toolChoice, activeTools })
@@ -3844,6 +3844,7 @@ async function generateText({
       let stepCount = 0;
       const responseMessages = [];
       let text2 = "";
+      const sources = [];
       const steps = [];
       let usage = {
         completionTokens: 0,
@@ -3992,6 +3993,7 @@ async function generateText({
         text2.trimEnd() !== text2 ? originalText.trimStart() : originalText;
         const stepText = nextStepType === "continue" ? removeTextAfterLastWhitespace(stepTextLeadingWhitespaceTrimmed) : stepTextLeadingWhitespaceTrimmed;
         text2 = nextStepType === "continue" || stepType === "continue" ? text2 + stepText : stepText;
+        sources.push(...(_d = currentModelResponse.sources) != null ? _d : []);
         if (stepType === "continue") {
           const lastMessage = responseMessages[responseMessages.length - 1];
           if (typeof lastMessage.content === "string") {
@@ -4018,16 +4020,17 @@ async function generateText({
           stepType,
           text: stepText,
           reasoning: currentModelResponse.reasoning,
+          sources: (_e = currentModelResponse.sources) != null ? _e : [],
           toolCalls: currentToolCalls,
           toolResults: currentToolResults,
           finishReason: currentModelResponse.finishReason,
           usage: currentUsage,
           warnings: currentModelResponse.warnings,
           logprobs: currentModelResponse.logprobs,
-          request: (_d = currentModelResponse.request) != null ? _d : {},
+          request: (_f = currentModelResponse.request) != null ? _f : {},
           response: {
             ...currentModelResponse.response,
-            headers: (_e = currentModelResponse.rawResponse) == null ? void 0 : _e.headers,
+            headers: (_g = currentModelResponse.rawResponse) == null ? void 0 : _g.headers,
             // deep clone msgs to avoid mutating past messages in multi-step:
             messages: structuredClone(responseMessages)
           },
@@ -4057,6 +4060,7 @@ async function generateText({
       return new DefaultGenerateTextResult({
         text: text2,
         reasoning: currentModelResponse.reasoning,
+        sources,
         outputResolver: () => {
           if (output == null) {
             throw new NoOutputSpecifiedError();
@@ -4071,10 +4075,10 @@ async function generateText({
         finishReason: currentModelResponse.finishReason,
         usage,
         warnings: currentModelResponse.warnings,
-        request: (_f = currentModelResponse.request) != null ? _f : {},
+        request: (_h = currentModelResponse.request) != null ? _h : {},
         response: {
           ...currentModelResponse.response,
-          headers: (_g = currentModelResponse.rawResponse) == null ? void 0 : _g.headers,
+          headers: (_i = currentModelResponse.rawResponse) == null ? void 0 : _i.headers,
           messages: responseMessages
         },
         logprobs: currentModelResponse.logprobs,
@@ -4174,6 +4178,7 @@ var DefaultGenerateTextResult = class {
     this.experimental_providerMetadata = options.providerMetadata;
     this.logprobs = options.logprobs;
     this.outputResolver = options.outputResolver;
+    this.sources = options.sources;
   }
   get experimental_output() {
     return this.outputResolver();
@@ -4466,6 +4471,7 @@ function runToolsTransformation({
       switch (chunkType) {
         case "text-delta":
         case "reasoning":
+        case "source":
         case "response-metadata":
         case "error": {
           controller.enqueue(chunk);
@@ -4779,6 +4785,7 @@ var DefaultStreamTextResult = class {
     this.providerMetadataPromise = new DelayedPromise();
     this.textPromise = new DelayedPromise();
     this.reasoningPromise = new DelayedPromise();
+    this.sourcesPromise = new DelayedPromise();
     this.toolCallsPromise = new DelayedPromise();
     this.toolResultsPromise = new DelayedPromise();
     this.requestPromise = new DelayedPromise();
@@ -4797,6 +4804,8 @@ var DefaultStreamTextResult = class {
     let recordedContinuationText = "";
     let recordedFullText = "";
     let recordedReasoningText = void 0;
+    let recordedStepSources = [];
+    const recordedSources = [];
     const recordedResponse = {
       id: generateId3(),
       timestamp: currentDate(),
@@ -4814,7 +4823,7 @@ var DefaultStreamTextResult = class {
       async transform(chunk, controller) {
         controller.enqueue(chunk);
         const { part } = chunk;
-        if (part.type === "text-delta" || part.type === "reasoning" || part.type === "tool-call" || part.type === "tool-result" || part.type === "tool-call-streaming-start" || part.type === "tool-call-delta") {
+        if (part.type === "text-delta" || part.type === "reasoning" || part.type === "source" || part.type === "tool-call" || part.type === "tool-result" || part.type === "tool-call-streaming-start" || part.type === "tool-call-delta") {
           await (onChunk == null ? void 0 : onChunk({ chunk: part }));
         }
         if (part.type === "error") {
@@ -4828,6 +4837,10 @@ var DefaultStreamTextResult = class {
         if (part.type === "reasoning") {
           recordedReasoningText = (recordedReasoningText != null ? recordedReasoningText : "") + part.textDelta;
         }
+        if (part.type === "source") {
+          recordedSources.push(part.source);
+          recordedStepSources.push(part.source);
+        }
         if (part.type === "tool-call") {
           recordedToolCalls.push(part);
         }
@@ -4861,6 +4874,7 @@ var DefaultStreamTextResult = class {
             stepType,
             text: recordedStepText,
             reasoning: recordedReasoningText,
+            sources: recordedStepSources,
             toolCalls: recordedToolCalls,
             toolResults: recordedToolResults,
             finishReason: part.finishReason,
@@ -4880,6 +4894,7 @@ var DefaultStreamTextResult = class {
           recordedToolCalls = [];
           recordedToolResults = [];
           recordedStepText = "";
+          recordedStepSources = [];
           if (nextStepType !== "done") {
             stepType = nextStepType;
           }
@@ -4922,13 +4937,15 @@ var DefaultStreamTextResult = class {
           self.usagePromise.resolve(usage);
           self.textPromise.resolve(recordedFullText);
           self.reasoningPromise.resolve(recordedReasoningText);
+          self.sourcesPromise.resolve(recordedSources);
           self.stepsPromise.resolve(recordedSteps);
           await (onFinish == null ? void 0 : onFinish({
             finishReason,
             logprobs: void 0,
             usage,
             text: recordedFullText,
-            reasoning: recordedReasoningText,
+            reasoning: lastStep.reasoning,
+            sources: lastStep.sources,
             toolCalls: lastStep.toolCalls,
             toolResults: lastStep.toolResults,
             request: (_a16 = lastStep.request) != null ? _a16 : {},
@@ -5204,6 +5221,10 @@ var DefaultStreamTextResult = class {
                       stepReasoning += chunk.textDelta;
                       break;
                     }
+                    case "source": {
+                      controller.enqueue(chunk);
+                      break;
+                    }
                     case "tool-call": {
                       controller.enqueue(chunk);
                       stepToolCalls.push(chunk);
@@ -5420,6 +5441,9 @@ var DefaultStreamTextResult = class {
   get reasoning() {
     return this.reasoningPromise.value;
   }
+  get sources() {
+    return this.sourcesPromise.value;
+  }
   get toolCalls() {
     return this.toolCallsPromise.value;
   }
@@ -5524,6 +5548,9 @@ var DefaultStreamTextResult = class {
             }
             break;
           }
+          case "source": {
+            break;
+          }
           case "tool-call-streaming-start": {
             controller.enqueue(
               formatDataStreamPart2("tool_call_streaming_start", {