ai 7.0.141 → 7.0.143

package/src/generate-text/stream-language-model-call.ts

@@ -225,6 +225,7 @@ export async function streamLanguageModelCall<
     generateId = originalGenerateId,
     generateCallId = originalGenerateCallId,
     now = originalNow,
+    collectContent = true,
   } = {},
   onStart,
   onLanguageModelCallStart,
@@ -258,6 +259,8 @@ export async function streamLanguageModelCall<
     generateId?: IdGenerator;
     generateCallId?: IdGenerator;
     now?: () => number;
+    /** Internal: content is only needed by collectors and full-content callbacks. */
+    collectContent?: boolean;
   };
   onLanguageModelCallStart?: Arrayable<OnLanguageModelCallStartCallback>;
   onLanguageModelCallEnd?: Arrayable<OnLanguageModelCallEndCallback<TOOLS>>;
@@ -383,6 +386,7 @@ export async function streamLanguageModelCall<
       now,
       callStartTimestampMs,
       onLanguageModelCallEnd,
+      collectContent,
     }),
   );
 
@@ -411,6 +415,7 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform<
   now,
   callStartTimestampMs,
   onLanguageModelCallEnd,
+  collectContent,
 }: {
   tools: TOOLS | undefined;
   instructions: Instructions | undefined;
@@ -426,6 +431,7 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform<
   now: () => number;
   callStartTimestampMs: number;
   onLanguageModelCallEnd?: Arrayable<OnLanguageModelCallEndCallback<TOOLS>>;
+  collectContent: boolean;
 }) {
   // keep track of parsed tool calls so provider-emitted approval requests can reference them
   // keep track of tool inputs for provider-side tool results
@@ -471,6 +477,7 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform<
 
         case 'text-start':
           upsertTextContentPart({
+            enabled: collectContent,
             content: modelCallContent,
             rawContent: rawModelCallContent,
             partIndexes: textPartIndexes,
@@ -484,6 +491,7 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform<
 
         case 'text-delta':
           upsertTextContentPart({
+            enabled: collectContent,
             content: modelCallContent,
             rawContent: rawModelCallContent,
             partIndexes: textPartIndexes,
@@ -503,6 +511,7 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform<
 
         case 'text-end':
           upsertTextContentPart({
+            enabled: collectContent,
             content: modelCallContent,
             rawContent: rawModelCallContent,
             partIndexes: textPartIndexes,
@@ -518,6 +527,7 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform<
 
         case 'reasoning-start':
           upsertTextContentPart({
+            enabled: collectContent,
             content: modelCallContent,
             rawContent: rawModelCallContent,
             partIndexes: reasoningPartIndexes,
@@ -531,6 +541,7 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform<
 
         case 'reasoning-delta':
           upsertTextContentPart({
+            enabled: collectContent,
             content: modelCallContent,
             rawContent: rawModelCallContent,
             partIndexes: reasoningPartIndexes,
@@ -550,6 +561,7 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform<
 
         case 'reasoning-end':
           upsertTextContentPart({
+            enabled: collectContent,
             content: modelCallContent,
             rawContent: rawModelCallContent,
             partIndexes: reasoningPartIndexes,
@@ -573,14 +585,15 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform<
             mediaType: chunk.mediaType,
           });
 
-          modelCallContent.push({
-            type: chunk.type,
-            file,
-            ...(chunk.providerMetadata != null
-              ? { providerMetadata: chunk.providerMetadata }
-              : {}),
-          });
-          rawModelCallContent.push(chunk);
+          if (collectContent)
+            modelCallContent.push({
+              type: chunk.type,
+              file,
+              ...(chunk.providerMetadata != null
+                ? { providerMetadata: chunk.providerMetadata }
+                : {}),
+            });
+          if (collectContent) rawModelCallContent.push(chunk);
 
           controller.enqueue({
             type: chunk.type,
@@ -689,7 +702,7 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform<
         }
 
         case 'tool-call': {
-          rawModelCallContent.push(chunk);
+          if (collectContent) rawModelCallContent.push(chunk);
 
           try {
             const toolCall = await parseToolCall({
@@ -704,7 +717,7 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform<
 
             toolCallsByToolCallId.set(toolCall.toolCallId, toolCall);
             controller.enqueue(toolCall);
-            modelCallContent.push(toolCall);
+            if (collectContent) modelCallContent.push(toolCall);
 
             if (toolCall.invalid) {
               if (!toolCall.providerExecuted) {
@@ -731,7 +744,7 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform<
         }
 
         case 'tool-approval-request': {
-          rawModelCallContent.push(chunk);
+          if (collectContent) rawModelCallContent.push(chunk);
 
           const toolCall = toolCallsByToolCallId.get(chunk.toolCallId);
 
@@ -753,12 +766,12 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform<
           } as const;
 
           controller.enqueue(toolApprovalRequest);
-          modelCallContent.push(toolApprovalRequest);
+          if (collectContent) modelCallContent.push(toolApprovalRequest);
           break;
         }
 
         case 'tool-result': {
-          rawModelCallContent.push(chunk);
+          if (collectContent) rawModelCallContent.push(chunk);
 
           const toolName = chunk.toolName as keyof TOOLS & string;
           const toolCall = toolCallsByToolCallId.get(chunk.toolCallId);
@@ -796,7 +809,7 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform<
               } as TypedToolResult<TOOLS>);
 
           controller.enqueue(toolResultPart);
-          modelCallContent.push(toolResultPart);
+          if (collectContent) modelCallContent.push(toolResultPart);
 
           break;
         }
@@ -836,8 +849,8 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform<
 
         default:
           if (chunk.type === 'custom' || chunk.type === 'source') {
-            modelCallContent.push(chunk);
-            rawModelCallContent.push(chunk);
+            if (collectContent) modelCallContent.push(chunk);
+            if (collectContent) rawModelCallContent.push(chunk);
           }
 
           controller.enqueue(chunk);
@@ -890,6 +903,7 @@ function calculateNearestRankPercentile(
  * Appends a text or reasoning content part into the content array and updates the part indexes.
  */
 function upsertTextContentPart<TOOLS extends ToolSet>({
+  enabled,
   content,
   rawContent,
   partIndexes,
@@ -899,6 +913,7 @@ function upsertTextContentPart<TOOLS extends ToolSet>({
   textDelta,
   providerMetadata,
 }: {
+  enabled: boolean;
   content: Array<ContentPart<TOOLS>>;
   rawContent: Array<LanguageModelV4Content>;
   partIndexes: Map<string, number>;
@@ -908,6 +923,7 @@ function upsertTextContentPart<TOOLS extends ToolSet>({
   textDelta?: string;
   providerMetadata?: ProviderMetadata;
 }) {
+  if (!enabled) return;
   let partIndex = partIndexes.get(id);
 
   if (partIndex == null) {

package/src/generate-text/stream-text-result.ts

@@ -493,6 +493,8 @@ export type TextStreamToolInputStartPart = {
   type: 'tool-input-start';
   id: string;
   toolName: string;
+  /** Encoding of the input deltas. Defaults to JSON when omitted. */
+  inputEncoding?: 'json' | 'text';
   providerMetadata?: ProviderMetadata;
   toolMetadata?: JSONObject;
   providerExecuted?: boolean;

package/src/generate-text/stream-text-single-consumer-result.ts

@@ -0,0 +1,75 @@
+import type { Context, ToolSet } from '@ai-sdk/provider-utils';
+import type { Output } from './output';
+import type { StreamTextResult } from './stream-text-result';
+
+/** A single-consumer view of a text generation, without replay or final content getters. */
+export type StreamTextSingleConsumerResult<
+  TOOLS extends ToolSet,
+  RUNTIME_CONTEXT extends Context,
+  OUTPUT extends Output,
+> = Pick<
+  StreamTextResult<TOOLS, RUNTIME_CONTEXT, OUTPUT>,
+  | 'textStream'
+  | 'stream'
+  | 'fullStream'
+  | 'partialOutputStream'
+  | 'elementStream'
+  | 'totalUsage'
+  | 'usage'
+  | 'finishReason'
+  | 'rawFinishReason'
+  | 'consumeStream'
+  | 'toUIMessageStream'
+  | 'toUIMessageStreamResponse'
+  | 'toTextStreamResponse'
+  | 'pipeUIMessageStreamToResponse'
+  | 'pipeTextStreamToResponse'
+>;
+
+/** Expose only the stream and completion metadata for a single consumer. */
+export function createSingleConsumerStreamTextResult<
+  TOOLS extends ToolSet,
+  RUNTIME_CONTEXT extends Context,
+  OUTPUT extends Output,
+>(
+  result: StreamTextResult<TOOLS, RUNTIME_CONTEXT, OUTPUT>,
+): StreamTextSingleConsumerResult<TOOLS, RUNTIME_CONTEXT, OUTPUT> {
+  // A concrete facade also prevents untyped callers from accidentally reading
+  // a final-content getter when no result collector was configured.
+  return {
+    get textStream() {
+      return result.textStream;
+    },
+    get stream() {
+      return result.stream;
+    },
+    get fullStream() {
+      return result.fullStream;
+    },
+    get partialOutputStream() {
+      return result.partialOutputStream;
+    },
+    get elementStream() {
+      return result.elementStream;
+    },
+    get totalUsage() {
+      return result.totalUsage;
+    },
+    get usage() {
+      return result.usage;
+    },
+    get finishReason() {
+      return result.finishReason;
+    },
+    get rawFinishReason() {
+      return result.rawFinishReason;
+    },
+    consumeStream: result.consumeStream.bind(result),
+    toUIMessageStream: result.toUIMessageStream.bind(result),
+    toUIMessageStreamResponse: result.toUIMessageStreamResponse.bind(result),
+    toTextStreamResponse: result.toTextStreamResponse.bind(result),
+    pipeUIMessageStreamToResponse:
+      result.pipeUIMessageStreamToResponse.bind(result),
+    pipeTextStreamToResponse: result.pipeTextStreamToResponse.bind(result),
+  };
+}