ai 6.0.0-beta.62 → 6.0.0-beta.63

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # ai
 
+## 6.0.0-beta.63
+
+### Patch Changes
+
+- a7da2b6: feat(agent): change output generics
+
 ## 6.0.0-beta.62
 
 ### Patch Changes

package/dist/index.d.mts

@@ -1,6 +1,6 @@
 export { createGateway, gateway } from '@ai-sdk/gateway';
 import * as _ai_sdk_provider_utils from '@ai-sdk/provider-utils';
-import { Tool, InferToolInput, InferToolOutput, AssistantModelMessage, ToolModelMessage, ReasoningPart, ModelMessage, FlexibleSchema, InferSchema, SystemModelMessage, UserModelMessage, ProviderOptions, IdGenerator, ToolCall, Resolvable, FetchFunction, DataContent } from '@ai-sdk/provider-utils';
+import { Tool, InferToolInput, InferToolOutput, AssistantModelMessage, ToolModelMessage, ReasoningPart, FlexibleSchema, InferSchema, ModelMessage, SystemModelMessage, UserModelMessage, ProviderOptions, IdGenerator, ToolCall, Resolvable, FetchFunction, DataContent } from '@ai-sdk/provider-utils';
 export { AssistantContent, AssistantModelMessage, DataContent, FilePart, FlexibleSchema, IdGenerator, ImagePart, InferSchema, InferToolInput, InferToolOutput, ModelMessage, Schema, SystemModelMessage, TextPart, Tool, ToolApprovalRequest, ToolApprovalResponse, ToolCallOptions, ToolCallPart, ToolContent, ToolExecuteFunction, ToolModelMessage, ToolResultPart, UserContent, UserModelMessage, asSchema, createIdGenerator, dynamicTool, generateId, jsonSchema, parseJsonEventStream, tool, zodSchema } from '@ai-sdk/provider-utils';
 import * as _ai_sdk_provider from '@ai-sdk/provider';
 import { EmbeddingModelV3, EmbeddingModelV2, EmbeddingModelV3Embedding, ImageModelV3, ImageModelV3CallWarning, ImageModelV3ProviderMetadata, JSONValue as JSONValue$1, LanguageModelV3, LanguageModelV2, LanguageModelV3FinishReason, LanguageModelV3CallWarning, LanguageModelV3Source, LanguageModelV3Middleware, SharedV3ProviderMetadata, SpeechModelV3, SpeechModelV2, SpeechModelV3CallWarning, TranscriptionModelV3, TranscriptionModelV2, TranscriptionModelV3CallWarning, LanguageModelV3Usage, LanguageModelV3CallOptions, AISDKError, LanguageModelV3ToolCall, JSONSchema7, JSONParseError, TypeValidationError, ProviderV3, ProviderV2, NoSuchModelError, JSONObject } from '@ai-sdk/provider';
@@ -637,6 +637,61 @@ interface GenerateTextResult<TOOLS extends ToolSet, OUTPUT> {
     readonly experimental_output: OUTPUT;
 }
 
+/**
+Create a type from an object with all keys and nested keys set to optional.
+The helper supports normal objects and schemas (which are resolved automatically).
+It always recurses into arrays.
+
+Adopted from [type-fest](https://github.com/sindresorhus/type-fest/tree/main) PartialDeep.
+ */
+type DeepPartial<T> = T extends FlexibleSchema ? DeepPartialInternal<InferSchema<T>> : DeepPartialInternal<T>;
+type DeepPartialInternal<T> = T extends null | undefined | string | number | boolean | symbol | bigint | void | Date | RegExp | ((...arguments_: any[]) => unknown) | (new (...arguments_: any[]) => unknown) ? T : T extends Map<infer KeyType, infer ValueType> ? PartialMap<KeyType, ValueType> : T extends Set<infer ItemType> ? PartialSet<ItemType> : T extends ReadonlyMap<infer KeyType, infer ValueType> ? PartialReadonlyMap<KeyType, ValueType> : T extends ReadonlySet<infer ItemType> ? PartialReadonlySet<ItemType> : T extends object ? T extends ReadonlyArray<infer ItemType> ? ItemType[] extends T ? readonly ItemType[] extends T ? ReadonlyArray<DeepPartialInternal<ItemType | undefined>> : Array<DeepPartialInternal<ItemType | undefined>> : PartialObject<T> : PartialObject<T> : unknown;
+type PartialMap<KeyType, ValueType> = {} & Map<DeepPartialInternal<KeyType>, DeepPartialInternal<ValueType>>;
+type PartialSet<T> = {} & Set<DeepPartialInternal<T>>;
+type PartialReadonlyMap<KeyType, ValueType> = {} & ReadonlyMap<DeepPartialInternal<KeyType>, DeepPartialInternal<ValueType>>;
+type PartialReadonlySet<T> = {} & ReadonlySet<DeepPartialInternal<T>>;
+type PartialObject<ObjectType extends object> = {
+    [KeyType in keyof ObjectType]?: DeepPartialInternal<ObjectType[KeyType]>;
+};
+
+interface Output<OUTPUT = any, PARTIAL = any> {
+    readonly type: 'object' | 'text';
+    responseFormat: PromiseLike<LanguageModelV3CallOptions['responseFormat']>;
+    parsePartial(options: {
+        text: string;
+    }): Promise<{
+        partial: PARTIAL;
+    } | undefined>;
+    parseOutput(options: {
+        text: string;
+    }, context: {
+        response: LanguageModelResponseMetadata;
+        usage: LanguageModelUsage;
+        finishReason: FinishReason;
+    }): Promise<OUTPUT>;
+}
+declare const text: () => Output<string, string>;
+declare const object: <OUTPUT>({ schema: inputSchema, }: {
+    schema: FlexibleSchema<OUTPUT>;
+}) => Output<OUTPUT, DeepPartial<OUTPUT>>;
+type InferGenerateOutput<OUTPUT extends Output> = OUTPUT extends Output<infer T, any> ? T : never;
+type InferStreamOutput<OUTPUT extends Output> = OUTPUT extends Output<any, infer P> ? P : never;
+
+type output_InferGenerateOutput<OUTPUT extends Output> = InferGenerateOutput<OUTPUT>;
+type output_InferStreamOutput<OUTPUT extends Output> = InferStreamOutput<OUTPUT>;
+type output_Output<OUTPUT = any, PARTIAL = any> = Output<OUTPUT, PARTIAL>;
+declare const output_object: typeof object;
+declare const output_text: typeof text;
+declare namespace output {
+  export {
+    output_InferGenerateOutput as InferGenerateOutput,
+    output_InferStreamOutput as InferStreamOutput,
+    output_Output as Output,
+    output_object as object,
+    output_text as text,
+  };
+}
+
 type CallSettings = {
     /**
   Maximum number of tokens to generate.
@@ -806,55 +861,6 @@ type DownloadFunction = (options: Array<{
     mediaType: string | undefined;
 } | null>>;
 
-/**
-Create a type from an object with all keys and nested keys set to optional.
-The helper supports normal objects and schemas (which are resolved automatically).
-It always recurses into arrays.
-
-Adopted from [type-fest](https://github.com/sindresorhus/type-fest/tree/main) PartialDeep.
- */
-type DeepPartial<T> = T extends FlexibleSchema ? DeepPartialInternal<InferSchema<T>> : DeepPartialInternal<T>;
-type DeepPartialInternal<T> = T extends null | undefined | string | number | boolean | symbol | bigint | void | Date | RegExp | ((...arguments_: any[]) => unknown) | (new (...arguments_: any[]) => unknown) ? T : T extends Map<infer KeyType, infer ValueType> ? PartialMap<KeyType, ValueType> : T extends Set<infer ItemType> ? PartialSet<ItemType> : T extends ReadonlyMap<infer KeyType, infer ValueType> ? PartialReadonlyMap<KeyType, ValueType> : T extends ReadonlySet<infer ItemType> ? PartialReadonlySet<ItemType> : T extends object ? T extends ReadonlyArray<infer ItemType> ? ItemType[] extends T ? readonly ItemType[] extends T ? ReadonlyArray<DeepPartialInternal<ItemType | undefined>> : Array<DeepPartialInternal<ItemType | undefined>> : PartialObject<T> : PartialObject<T> : unknown;
-type PartialMap<KeyType, ValueType> = {} & Map<DeepPartialInternal<KeyType>, DeepPartialInternal<ValueType>>;
-type PartialSet<T> = {} & Set<DeepPartialInternal<T>>;
-type PartialReadonlyMap<KeyType, ValueType> = {} & ReadonlyMap<DeepPartialInternal<KeyType>, DeepPartialInternal<ValueType>>;
-type PartialReadonlySet<T> = {} & ReadonlySet<DeepPartialInternal<T>>;
-type PartialObject<ObjectType extends object> = {
-    [KeyType in keyof ObjectType]?: DeepPartialInternal<ObjectType[KeyType]>;
-};
-
-interface Output<OUTPUT, PARTIAL> {
-    readonly type: 'object' | 'text';
-    responseFormat: PromiseLike<LanguageModelV3CallOptions['responseFormat']>;
-    parsePartial(options: {
-        text: string;
-    }): Promise<{
-        partial: PARTIAL;
-    } | undefined>;
-    parseOutput(options: {
-        text: string;
-    }, context: {
-        response: LanguageModelResponseMetadata;
-        usage: LanguageModelUsage;
-        finishReason: FinishReason;
-    }): Promise<OUTPUT>;
-}
-declare const text: () => Output<string, string>;
-declare const object: <OUTPUT>({ schema: inputSchema, }: {
-    schema: FlexibleSchema<OUTPUT>;
-}) => Output<OUTPUT, DeepPartial<OUTPUT>>;
-
-type output_Output<OUTPUT, PARTIAL> = Output<OUTPUT, PARTIAL>;
-declare const output_object: typeof object;
-declare const output_text: typeof text;
-declare namespace output {
-  export {
-    output_Output as Output,
-    output_object as object,
-    output_text as text,
-  };
-}
-
 /**
 Function that you can use to provide different settings for a step.
 
@@ -2381,7 +2387,7 @@ type TextStreamPart<TOOLS extends ToolSet> = {
  * You can implement your own Agent by implementing the `Agent` interface,
  * or use the `ToolLoopAgent` class.
  */
-interface Agent<TOOLS extends ToolSet = {}, OUTPUT = never, OUTPUT_PARTIAL = never> {
+interface Agent<TOOLS extends ToolSet = {}, OUTPUT extends Output = never> {
     /**
      * The specification version of the agent interface. This will enable
      * us to evolve the agent interface and retain backwards compatibility.
@@ -2398,11 +2404,11 @@ interface Agent<TOOLS extends ToolSet = {}, OUTPUT = never, OUTPUT_PARTIAL = nev
     /**
      * Generates an output from the agent (non-streaming).
      */
-    generate(options: Prompt): PromiseLike<GenerateTextResult<TOOLS, OUTPUT>>;
+    generate(options: Prompt): PromiseLike<GenerateTextResult<TOOLS, InferGenerateOutput<OUTPUT>>>;
     /**
      * Streams an output from the agent (streaming).
      */
-    stream(options: Prompt): StreamTextResult<TOOLS, OUTPUT_PARTIAL>;
+    stream(options: Prompt): StreamTextResult<TOOLS, InferStreamOutput<OUTPUT>>;
 }
 
 /**
@@ -2431,7 +2437,7 @@ type ToolLoopAgentOnStepFinishCallback<TOOLS extends ToolSet = {}> = (stepResult
 /**
  * Configuration options for an agent.
  */
-type ToolLoopAgentSettings<TOOLS extends ToolSet = {}, OUTPUT = never, OUTPUT_PARTIAL = never> = CallSettings & {
+type ToolLoopAgentSettings<TOOLS extends ToolSet = {}, OUTPUT extends Output = never> = CallSettings & {
     /**
      * The id of the agent.
      */
@@ -2471,7 +2477,7 @@ type ToolLoopAgentSettings<TOOLS extends ToolSet = {}, OUTPUT = never, OUTPUT_PA
     /**
   Optional specification for parsing structured outputs from the LLM response.
      */
-    experimental_output?: Output<OUTPUT, OUTPUT_PARTIAL>;
+    experimental_output?: OUTPUT;
     /**
      * @deprecated Use `prepareStep` instead.
      */
@@ -2519,10 +2525,10 @@ type ToolLoopAgentSettings<TOOLS extends ToolSet = {}, OUTPUT = never, OUTPUT_PA
  * - A tool call needs approval, or
  * - A stop condition is met (default stop condition is stepCountIs(20))
  */
-declare class ToolLoopAgent<TOOLS extends ToolSet = {}, OUTPUT = never, OUTPUT_PARTIAL = never> implements Agent<TOOLS, OUTPUT, OUTPUT_PARTIAL> {
+declare class ToolLoopAgent<TOOLS extends ToolSet = {}, OUTPUT extends Output = never> implements Agent<TOOLS, OUTPUT> {
     readonly version = "agent-v1";
     private readonly settings;
-    constructor(settings: ToolLoopAgentSettings<TOOLS, OUTPUT, OUTPUT_PARTIAL>);
+    constructor(settings: ToolLoopAgentSettings<TOOLS, OUTPUT>);
     /**
      * The id of the agent.
      */
@@ -2534,17 +2540,17 @@ declare class ToolLoopAgent<TOOLS extends ToolSet = {}, OUTPUT = never, OUTPUT_P
     /**
      * Generates an output from the agent (non-streaming).
      */
-    generate(options: Prompt): Promise<GenerateTextResult<TOOLS, OUTPUT>>;
+    generate(options: Prompt): Promise<GenerateTextResult<TOOLS, InferGenerateOutput<OUTPUT>>>;
     /**
      * Streams an output from the agent (streaming).
      */
-    stream(options: Prompt): StreamTextResult<TOOLS, OUTPUT_PARTIAL>;
+    stream(options: Prompt): StreamTextResult<TOOLS, InferStreamOutput<OUTPUT>>;
 }
 
 /**
  * Infer the type of the tools of an agent.
  */
-type InferAgentTools<AGENT> = AGENT extends Agent<infer TOOLS, any, any> ? TOOLS : never;
+type InferAgentTools<AGENT> = AGENT extends Agent<infer TOOLS, never> ? TOOLS : never;
 
 /**
  * Infer the UI message type of an agent.
@@ -2559,8 +2565,8 @@ type InferAgentUIMessage<AGENT> = UIMessage<never, never, InferUITools<InferAgen
  *
  * @returns The response object.
  */
-declare function createAgentUIStreamResponse<TOOLS extends ToolSet = {}, OUTPUT = never, OUTPUT_PARTIAL = never>({ headers, status, statusText, consumeSseStream, ...options }: {
-    agent: Agent<TOOLS, OUTPUT, OUTPUT_PARTIAL>;
+declare function createAgentUIStreamResponse<TOOLS extends ToolSet = {}, OUTPUT extends Output = never>({ headers, status, statusText, consumeSseStream, ...options }: {
+    agent: Agent<TOOLS, OUTPUT>;
     messages: unknown[];
 } & UIMessageStreamResponseInit & UIMessageStreamOptions<UIMessage<never, never, InferUITools<TOOLS>>>): Promise<Response>;
 
@@ -3212,8 +3218,8 @@ declare const UI_MESSAGE_STREAM_HEADERS: {
  *
  * @returns The UI message stream.
  */
-declare function createAgentUIStream<TOOLS extends ToolSet = {}, OUTPUT = never, OUTPUT_PARTIAL = never>({ agent, messages, ...uiMessageStreamOptions }: {
-    agent: Agent<TOOLS, OUTPUT, OUTPUT_PARTIAL>;
+declare function createAgentUIStream<TOOLS extends ToolSet = {}, OUTPUT extends Output = never>({ agent, messages, ...uiMessageStreamOptions }: {
+    agent: Agent<TOOLS, OUTPUT>;
     messages: unknown[];
 } & UIMessageStreamOptions<UIMessage<never, never, InferUITools<TOOLS>>>): Promise<AsyncIterableStream<InferUIMessageChunk<UIMessage<never, never, InferUITools<TOOLS>>>>>;
 
@@ -3223,9 +3229,9 @@ declare function createAgentUIStream<TOOLS extends ToolSet = {}, OUTPUT = never,
  * @param agent - The agent to run.
  * @param messages - The input UI messages.
  */
-declare function pipeAgentUIStreamToResponse<TOOLS extends ToolSet = {}, OUTPUT = never, OUTPUT_PARTIAL = never>({ response, headers, status, statusText, consumeSseStream, ...options }: {
+declare function pipeAgentUIStreamToResponse<TOOLS extends ToolSet = {}, OUTPUT extends Output = never>({ response, headers, status, statusText, consumeSseStream, ...options }: {
     response: ServerResponse;
-    agent: Agent<TOOLS, OUTPUT, OUTPUT_PARTIAL>;
+    agent: Agent<TOOLS, OUTPUT>;
     messages: unknown[];
 } & UIMessageStreamResponseInit & UIMessageStreamOptions<UIMessage<never, never, InferUITools<TOOLS>>>): Promise<void>;
 

package/dist/index.d.ts

@@ -1,6 +1,6 @@
 export { createGateway, gateway } from '@ai-sdk/gateway';
 import * as _ai_sdk_provider_utils from '@ai-sdk/provider-utils';
-import { Tool, InferToolInput, InferToolOutput, AssistantModelMessage, ToolModelMessage, ReasoningPart, ModelMessage, FlexibleSchema, InferSchema, SystemModelMessage, UserModelMessage, ProviderOptions, IdGenerator, ToolCall, Resolvable, FetchFunction, DataContent } from '@ai-sdk/provider-utils';
+import { Tool, InferToolInput, InferToolOutput, AssistantModelMessage, ToolModelMessage, ReasoningPart, FlexibleSchema, InferSchema, ModelMessage, SystemModelMessage, UserModelMessage, ProviderOptions, IdGenerator, ToolCall, Resolvable, FetchFunction, DataContent } from '@ai-sdk/provider-utils';
 export { AssistantContent, AssistantModelMessage, DataContent, FilePart, FlexibleSchema, IdGenerator, ImagePart, InferSchema, InferToolInput, InferToolOutput, ModelMessage, Schema, SystemModelMessage, TextPart, Tool, ToolApprovalRequest, ToolApprovalResponse, ToolCallOptions, ToolCallPart, ToolContent, ToolExecuteFunction, ToolModelMessage, ToolResultPart, UserContent, UserModelMessage, asSchema, createIdGenerator, dynamicTool, generateId, jsonSchema, parseJsonEventStream, tool, zodSchema } from '@ai-sdk/provider-utils';
 import * as _ai_sdk_provider from '@ai-sdk/provider';
 import { EmbeddingModelV3, EmbeddingModelV2, EmbeddingModelV3Embedding, ImageModelV3, ImageModelV3CallWarning, ImageModelV3ProviderMetadata, JSONValue as JSONValue$1, LanguageModelV3, LanguageModelV2, LanguageModelV3FinishReason, LanguageModelV3CallWarning, LanguageModelV3Source, LanguageModelV3Middleware, SharedV3ProviderMetadata, SpeechModelV3, SpeechModelV2, SpeechModelV3CallWarning, TranscriptionModelV3, TranscriptionModelV2, TranscriptionModelV3CallWarning, LanguageModelV3Usage, LanguageModelV3CallOptions, AISDKError, LanguageModelV3ToolCall, JSONSchema7, JSONParseError, TypeValidationError, ProviderV3, ProviderV2, NoSuchModelError, JSONObject } from '@ai-sdk/provider';
@@ -637,6 +637,61 @@ interface GenerateTextResult<TOOLS extends ToolSet, OUTPUT> {
     readonly experimental_output: OUTPUT;
 }
 
+/**
+Create a type from an object with all keys and nested keys set to optional.
+The helper supports normal objects and schemas (which are resolved automatically).
+It always recurses into arrays.
+
+Adopted from [type-fest](https://github.com/sindresorhus/type-fest/tree/main) PartialDeep.
+ */
+type DeepPartial<T> = T extends FlexibleSchema ? DeepPartialInternal<InferSchema<T>> : DeepPartialInternal<T>;
+type DeepPartialInternal<T> = T extends null | undefined | string | number | boolean | symbol | bigint | void | Date | RegExp | ((...arguments_: any[]) => unknown) | (new (...arguments_: any[]) => unknown) ? T : T extends Map<infer KeyType, infer ValueType> ? PartialMap<KeyType, ValueType> : T extends Set<infer ItemType> ? PartialSet<ItemType> : T extends ReadonlyMap<infer KeyType, infer ValueType> ? PartialReadonlyMap<KeyType, ValueType> : T extends ReadonlySet<infer ItemType> ? PartialReadonlySet<ItemType> : T extends object ? T extends ReadonlyArray<infer ItemType> ? ItemType[] extends T ? readonly ItemType[] extends T ? ReadonlyArray<DeepPartialInternal<ItemType | undefined>> : Array<DeepPartialInternal<ItemType | undefined>> : PartialObject<T> : PartialObject<T> : unknown;
+type PartialMap<KeyType, ValueType> = {} & Map<DeepPartialInternal<KeyType>, DeepPartialInternal<ValueType>>;
+type PartialSet<T> = {} & Set<DeepPartialInternal<T>>;
+type PartialReadonlyMap<KeyType, ValueType> = {} & ReadonlyMap<DeepPartialInternal<KeyType>, DeepPartialInternal<ValueType>>;
+type PartialReadonlySet<T> = {} & ReadonlySet<DeepPartialInternal<T>>;
+type PartialObject<ObjectType extends object> = {
+    [KeyType in keyof ObjectType]?: DeepPartialInternal<ObjectType[KeyType]>;
+};
+
+interface Output<OUTPUT = any, PARTIAL = any> {
+    readonly type: 'object' | 'text';
+    responseFormat: PromiseLike<LanguageModelV3CallOptions['responseFormat']>;
+    parsePartial(options: {
+        text: string;
+    }): Promise<{
+        partial: PARTIAL;
+    } | undefined>;
+    parseOutput(options: {
+        text: string;
+    }, context: {
+        response: LanguageModelResponseMetadata;
+        usage: LanguageModelUsage;
+        finishReason: FinishReason;
+    }): Promise<OUTPUT>;
+}
+declare const text: () => Output<string, string>;
+declare const object: <OUTPUT>({ schema: inputSchema, }: {
+    schema: FlexibleSchema<OUTPUT>;
+}) => Output<OUTPUT, DeepPartial<OUTPUT>>;
+type InferGenerateOutput<OUTPUT extends Output> = OUTPUT extends Output<infer T, any> ? T : never;
+type InferStreamOutput<OUTPUT extends Output> = OUTPUT extends Output<any, infer P> ? P : never;
+
+type output_InferGenerateOutput<OUTPUT extends Output> = InferGenerateOutput<OUTPUT>;
+type output_InferStreamOutput<OUTPUT extends Output> = InferStreamOutput<OUTPUT>;
+type output_Output<OUTPUT = any, PARTIAL = any> = Output<OUTPUT, PARTIAL>;
+declare const output_object: typeof object;
+declare const output_text: typeof text;
+declare namespace output {
+  export {
+    output_InferGenerateOutput as InferGenerateOutput,
+    output_InferStreamOutput as InferStreamOutput,
+    output_Output as Output,
+    output_object as object,
+    output_text as text,
+  };
+}
+
 type CallSettings = {
     /**
   Maximum number of tokens to generate.
@@ -806,55 +861,6 @@ type DownloadFunction = (options: Array<{
     mediaType: string | undefined;
 } | null>>;
 
-/**
-Create a type from an object with all keys and nested keys set to optional.
-The helper supports normal objects and schemas (which are resolved automatically).
-It always recurses into arrays.
-
-Adopted from [type-fest](https://github.com/sindresorhus/type-fest/tree/main) PartialDeep.
- */
-type DeepPartial<T> = T extends FlexibleSchema ? DeepPartialInternal<InferSchema<T>> : DeepPartialInternal<T>;
-type DeepPartialInternal<T> = T extends null | undefined | string | number | boolean | symbol | bigint | void | Date | RegExp | ((...arguments_: any[]) => unknown) | (new (...arguments_: any[]) => unknown) ? T : T extends Map<infer KeyType, infer ValueType> ? PartialMap<KeyType, ValueType> : T extends Set<infer ItemType> ? PartialSet<ItemType> : T extends ReadonlyMap<infer KeyType, infer ValueType> ? PartialReadonlyMap<KeyType, ValueType> : T extends ReadonlySet<infer ItemType> ? PartialReadonlySet<ItemType> : T extends object ? T extends ReadonlyArray<infer ItemType> ? ItemType[] extends T ? readonly ItemType[] extends T ? ReadonlyArray<DeepPartialInternal<ItemType | undefined>> : Array<DeepPartialInternal<ItemType | undefined>> : PartialObject<T> : PartialObject<T> : unknown;
-type PartialMap<KeyType, ValueType> = {} & Map<DeepPartialInternal<KeyType>, DeepPartialInternal<ValueType>>;
-type PartialSet<T> = {} & Set<DeepPartialInternal<T>>;
-type PartialReadonlyMap<KeyType, ValueType> = {} & ReadonlyMap<DeepPartialInternal<KeyType>, DeepPartialInternal<ValueType>>;
-type PartialReadonlySet<T> = {} & ReadonlySet<DeepPartialInternal<T>>;
-type PartialObject<ObjectType extends object> = {
-    [KeyType in keyof ObjectType]?: DeepPartialInternal<ObjectType[KeyType]>;
-};
-
-interface Output<OUTPUT, PARTIAL> {
-    readonly type: 'object' | 'text';
-    responseFormat: PromiseLike<LanguageModelV3CallOptions['responseFormat']>;
-    parsePartial(options: {
-        text: string;
-    }): Promise<{
-        partial: PARTIAL;
-    } | undefined>;
-    parseOutput(options: {
-        text: string;
-    }, context: {
-        response: LanguageModelResponseMetadata;
-        usage: LanguageModelUsage;
-        finishReason: FinishReason;
-    }): Promise<OUTPUT>;
-}
-declare const text: () => Output<string, string>;
-declare const object: <OUTPUT>({ schema: inputSchema, }: {
-    schema: FlexibleSchema<OUTPUT>;
-}) => Output<OUTPUT, DeepPartial<OUTPUT>>;
-
-type output_Output<OUTPUT, PARTIAL> = Output<OUTPUT, PARTIAL>;
-declare const output_object: typeof object;
-declare const output_text: typeof text;
-declare namespace output {
-  export {
-    output_Output as Output,
-    output_object as object,
-    output_text as text,
-  };
-}
-
 /**
 Function that you can use to provide different settings for a step.
 
@@ -2381,7 +2387,7 @@ type TextStreamPart<TOOLS extends ToolSet> = {
  * You can implement your own Agent by implementing the `Agent` interface,
  * or use the `ToolLoopAgent` class.
  */
-interface Agent<TOOLS extends ToolSet = {}, OUTPUT = never, OUTPUT_PARTIAL = never> {
+interface Agent<TOOLS extends ToolSet = {}, OUTPUT extends Output = never> {
     /**
      * The specification version of the agent interface. This will enable
      * us to evolve the agent interface and retain backwards compatibility.
@@ -2398,11 +2404,11 @@ interface Agent<TOOLS extends ToolSet = {}, OUTPUT = never, OUTPUT_PARTIAL = nev
     /**
      * Generates an output from the agent (non-streaming).
      */
-    generate(options: Prompt): PromiseLike<GenerateTextResult<TOOLS, OUTPUT>>;
+    generate(options: Prompt): PromiseLike<GenerateTextResult<TOOLS, InferGenerateOutput<OUTPUT>>>;
     /**
      * Streams an output from the agent (streaming).
      */
-    stream(options: Prompt): StreamTextResult<TOOLS, OUTPUT_PARTIAL>;
+    stream(options: Prompt): StreamTextResult<TOOLS, InferStreamOutput<OUTPUT>>;
 }
 
 /**
@@ -2431,7 +2437,7 @@ type ToolLoopAgentOnStepFinishCallback<TOOLS extends ToolSet = {}> = (stepResult
 /**
  * Configuration options for an agent.
  */
-type ToolLoopAgentSettings<TOOLS extends ToolSet = {}, OUTPUT = never, OUTPUT_PARTIAL = never> = CallSettings & {
+type ToolLoopAgentSettings<TOOLS extends ToolSet = {}, OUTPUT extends Output = never> = CallSettings & {
     /**
      * The id of the agent.
      */
@@ -2471,7 +2477,7 @@ type ToolLoopAgentSettings<TOOLS extends ToolSet = {}, OUTPUT = never, OUTPUT_PA
     /**
   Optional specification for parsing structured outputs from the LLM response.
      */
-    experimental_output?: Output<OUTPUT, OUTPUT_PARTIAL>;
+    experimental_output?: OUTPUT;
     /**
      * @deprecated Use `prepareStep` instead.
      */
@@ -2519,10 +2525,10 @@ type ToolLoopAgentSettings<TOOLS extends ToolSet = {}, OUTPUT = never, OUTPUT_PA
  * - A tool call needs approval, or
  * - A stop condition is met (default stop condition is stepCountIs(20))
  */
-declare class ToolLoopAgent<TOOLS extends ToolSet = {}, OUTPUT = never, OUTPUT_PARTIAL = never> implements Agent<TOOLS, OUTPUT, OUTPUT_PARTIAL> {
+declare class ToolLoopAgent<TOOLS extends ToolSet = {}, OUTPUT extends Output = never> implements Agent<TOOLS, OUTPUT> {
     readonly version = "agent-v1";
     private readonly settings;
-    constructor(settings: ToolLoopAgentSettings<TOOLS, OUTPUT, OUTPUT_PARTIAL>);
+    constructor(settings: ToolLoopAgentSettings<TOOLS, OUTPUT>);
     /**
      * The id of the agent.
      */
@@ -2534,17 +2540,17 @@ declare class ToolLoopAgent<TOOLS extends ToolSet = {}, OUTPUT = never, OUTPUT_P
     /**
      * Generates an output from the agent (non-streaming).
      */
-    generate(options: Prompt): Promise<GenerateTextResult<TOOLS, OUTPUT>>;
+    generate(options: Prompt): Promise<GenerateTextResult<TOOLS, InferGenerateOutput<OUTPUT>>>;
     /**
      * Streams an output from the agent (streaming).
      */
-    stream(options: Prompt): StreamTextResult<TOOLS, OUTPUT_PARTIAL>;
+    stream(options: Prompt): StreamTextResult<TOOLS, InferStreamOutput<OUTPUT>>;
 }
 
 /**
  * Infer the type of the tools of an agent.
  */
-type InferAgentTools<AGENT> = AGENT extends Agent<infer TOOLS, any, any> ? TOOLS : never;
+type InferAgentTools<AGENT> = AGENT extends Agent<infer TOOLS, never> ? TOOLS : never;
 
 /**
  * Infer the UI message type of an agent.
@@ -2559,8 +2565,8 @@ type InferAgentUIMessage<AGENT> = UIMessage<never, never, InferUITools<InferAgen
  *
  * @returns The response object.
  */
-declare function createAgentUIStreamResponse<TOOLS extends ToolSet = {}, OUTPUT = never, OUTPUT_PARTIAL = never>({ headers, status, statusText, consumeSseStream, ...options }: {
-    agent: Agent<TOOLS, OUTPUT, OUTPUT_PARTIAL>;
+declare function createAgentUIStreamResponse<TOOLS extends ToolSet = {}, OUTPUT extends Output = never>({ headers, status, statusText, consumeSseStream, ...options }: {
+    agent: Agent<TOOLS, OUTPUT>;
     messages: unknown[];
 } & UIMessageStreamResponseInit & UIMessageStreamOptions<UIMessage<never, never, InferUITools<TOOLS>>>): Promise<Response>;
 
@@ -3212,8 +3218,8 @@ declare const UI_MESSAGE_STREAM_HEADERS: {
  *
  * @returns The UI message stream.
  */
-declare function createAgentUIStream<TOOLS extends ToolSet = {}, OUTPUT = never, OUTPUT_PARTIAL = never>({ agent, messages, ...uiMessageStreamOptions }: {
-    agent: Agent<TOOLS, OUTPUT, OUTPUT_PARTIAL>;
+declare function createAgentUIStream<TOOLS extends ToolSet = {}, OUTPUT extends Output = never>({ agent, messages, ...uiMessageStreamOptions }: {
+    agent: Agent<TOOLS, OUTPUT>;
     messages: unknown[];
 } & UIMessageStreamOptions<UIMessage<never, never, InferUITools<TOOLS>>>): Promise<AsyncIterableStream<InferUIMessageChunk<UIMessage<never, never, InferUITools<TOOLS>>>>>;
 
@@ -3223,9 +3229,9 @@ declare function createAgentUIStream<TOOLS extends ToolSet = {}, OUTPUT = never,
  * @param agent - The agent to run.
  * @param messages - The input UI messages.
  */
-declare function pipeAgentUIStreamToResponse<TOOLS extends ToolSet = {}, OUTPUT = never, OUTPUT_PARTIAL = never>({ response, headers, status, statusText, consumeSseStream, ...options }: {
+declare function pipeAgentUIStreamToResponse<TOOLS extends ToolSet = {}, OUTPUT extends Output = never>({ response, headers, status, statusText, consumeSseStream, ...options }: {
     response: ServerResponse;
-    agent: Agent<TOOLS, OUTPUT, OUTPUT_PARTIAL>;
+    agent: Agent<TOOLS, OUTPUT>;
     messages: unknown[];
 } & UIMessageStreamResponseInit & UIMessageStreamOptions<UIMessage<never, never, InferUITools<TOOLS>>>): Promise<void>;
 

package/dist/index.js

@@ -873,7 +873,7 @@ function detectMediaType({
 var import_provider_utils2 = require("@ai-sdk/provider-utils");
 
 // src/version.ts
-var VERSION = true ? "6.0.0-beta.62" : "0.0.0-test";
+var VERSION = true ? "6.0.0-beta.63" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({ url }) => {