@convex-dev/agent 0.1.18-alpha.1 → 0.5.0-alpha.1

package/dist/client/index.d.ts

@@ -1,25 +1,25 @@
-import type { EmbeddingModelV1, LanguageModelV1 } from "@ai-sdk/provider";
-import type { CoreMessage, DeepPartial, GenerateObjectResult, GenerateTextResult, StepResult, StreamObjectResult, StreamTextResult, ToolSet } from "ai";
+import type { LanguageModelV2, EmbeddingModelV2 } from "@ai-sdk/provider";
+import type { ModelMessage, DeepPartial, GenerateObjectResult, GenerateTextResult, StepResult, StreamObjectResult, StreamTextResult, ToolSet, StopCondition, Schema } from "ai";
 import { type GenericActionCtx, type GenericDataModel, type PaginationOptions, type PaginationResult } from "convex/server";
 import type { MessageDoc, ThreadDoc } from "../component/schema.js";
 import type { threadFieldsSupportingPatch } from "../component/threads.js";
 import { type VectorDimension } from "../component/vector/tables.js";
-import { type AIMessageWithoutId } from "../mapping.js";
 import { extractText, isTool } from "../shared.js";
-import { type MessageEmbeddings, type MessageStatus, type MessageWithMetadata, type ProviderMetadata, type StreamArgs, type Usage } from "../validators.js";
+import { type Message, type MessageEmbeddings, type MessageStatus, type MessageWithMetadata, type ProviderMetadata, type StreamArgs, type Usage } from "../validators.js";
 import { createTool } from "./createTool.js";
 import { listMessages } from "./listMessages.js";
-import { fetchContextMessages } from "./search.js";
-import { type StreamingOptions, syncStreams } from "./streaming.js";
+import { type StreamingOptions } from "./streaming.js";
 import type { ActionCtx, AgentComponent, ContextOptions, GenerationOutputMetadata, Options, OurObjectArgs, OurStreamObjectArgs, RawRequestResponseHandler, RunActionCtx, RunMutationCtx, RunQueryCtx, StorageOptions, StreamingTextArgs, SyncStreamsReturnValue, TextArgs, Thread, UsageHandler } from "./types.js";
+import type z from "zod";
+export { stepCountIs } from "ai";
 export { vMessageDoc, vThreadDoc } from "../component/schema.js";
-export { serializeDataOrUrl } from "../mapping.js";
+export { serializeDataOrUrl, deserializeMessage, serializeMessage, } from "../mapping.js";
 export { vAssistantMessage, vContextOptions, vMessage, vPaginationResult, vProviderMetadata, vStorageOptions, vStreamArgs, vSystemMessage, vToolMessage, vUsage, vUserMessage, } from "../validators.js";
 export type { ToolCtx } from "./createTool.js";
 export { getFile, storeFile } from "./files.js";
-export { filterOutOrphanedToolMessages } from "./search.js";
-export { abortStream, listStreams } from "./streaming.js";
-export { createTool, extractText, fetchContextMessages, isTool, listMessages, syncStreams, };
+export { filterOutOrphanedToolMessages, fetchContextMessages, } from "./search.js";
+export { abortStream, listStreams, syncStreams } from "./streaming.js";
+export { createTool, extractText, isTool, listMessages };
 export { definePlaygroundAPI, type PlaygroundAPI, type AgentsFn, } from "./definePlaygroundAPI.js";
 export type { AgentComponent, ContextOptions, MessageDoc, ProviderMetadata, RawRequestResponseHandler, StorageOptions, StreamArgs, SyncStreamsReturnValue, Thread, ThreadDoc, Usage, UsageHandler, };
 export declare class Agent<
@@ -57,7 +57,7 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
          * const myAgent = new Agent(components.agent, {
          *   chat: openai.chat("gpt-4o-mini"),
          */
-        chat: LanguageModelV1;
+        chat: LanguageModelV2;
         /**
          * The model to use for text embeddings. Optional.
          * If specified, it will use this for generating vector embeddings
@@ -68,7 +68,7 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
          * const myAgent = new Agent(components.agent, {
          *   textEmbedding: openai.embedding("text-embedding-3-small")
          */
-        textEmbedding?: EmbeddingModelV1<string>;
+        textEmbedding?: EmbeddingModelV2<string>;
         /**
          * The default system prompt to put in each request.
          * Override per-prompt by passing the "system" parameter.
@@ -94,9 +94,9 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
         storageOptions?: StorageOptions;
         /**
          * When generating or streaming text with tools available, this
-         * determines the default max number of iterations.
+         * determines when to stop. Defaults to stepCountIs(1).
          */
-        maxSteps?: number;
+        stopWhen?: StopCondition<AgentTools> | Array<StopCondition<AgentTools>>;
         /**
          * The maximum number of calls to make to an LLM in case it fails.
          * This can be overridden at each generate/stream callsite.
@@ -125,7 +125,7 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
          * const myAgent = new Agent(components.agent, {
          *   chat: openai.chat("gpt-4o-mini"),
          */
-        chat: LanguageModelV1;
+        chat: LanguageModelV2;
         /**
          * The model to use for text embeddings. Optional.
          * If specified, it will use this for generating vector embeddings
@@ -136,7 +136,7 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
          * const myAgent = new Agent(components.agent, {
          *   textEmbedding: openai.embedding("text-embedding-3-small")
          */
-        textEmbedding?: EmbeddingModelV1<string>;
+        textEmbedding?: EmbeddingModelV2<string>;
         /**
          * The default system prompt to put in each request.
          * Override per-prompt by passing the "system" parameter.
@@ -162,9 +162,9 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
         storageOptions?: StorageOptions;
         /**
          * When generating or streaming text with tools available, this
-         * determines the default max number of iterations.
+         * determines when to stop. Defaults to stepCountIs(1).
          */
-        maxSteps?: number;
+        stopWhen?: StopCondition<AgentTools> | Array<StopCondition<AgentTools>>;
         /**
          * The maximum number of calls to make to an LLM in case it fails.
          * This can be overridden at each generate/stream callsite.
@@ -382,7 +382,7 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
      * Use {@link continueThread} to get a version of this function already scoped
      * to a thread (and optionally userId).
      */
-    streamObject<T>(ctx: ActionCtx, { userId: argsUserId, threadId, usageHandler, }: {
+    streamObject<T extends z.Schema | Schema>(ctx: ActionCtx, { userId: argsUserId, threadId, usageHandler, }: {
         userId?: string | null;
         threadId?: string;
         usageHandler?: UsageHandler;
@@ -458,6 +458,7 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
                 } | {
                     providerOptions?: Record<string, Record<string, any>> | undefined;
                     signature?: string | undefined;
+                    state?: "streaming" | "done" | undefined;
                     type: "reasoning";
                     text: string;
                 } | {
@@ -466,6 +467,7 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
                     data: string;
                 } | {
                     providerOptions?: Record<string, Record<string, any>> | undefined;
+                    providerExecuted?: boolean | undefined;
                     type: "tool-call";
                     toolCallId: string;
                     toolName: string;
@@ -477,6 +479,8 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
                 content: {
                     providerOptions?: Record<string, Record<string, any>> | undefined;
                     args?: any;
+                    providerExecuted?: boolean | undefined;
+                    isError?: boolean | undefined;
                     experimental_content?: ({
                         type: "text";
                         text: string;
@@ -485,7 +489,6 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
                         type: "image";
                         data: string;
                     })[] | undefined;
-                    isError?: boolean | undefined;
                     type: "tool-result";
                     toolCallId: string;
                     toolName: string;
@@ -499,19 +502,36 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
             text?: string | undefined;
             reasoning?: string | undefined;
             usage?: {
+                reasoningTokens?: number | undefined;
+                cachedInputTokens?: number | undefined;
                 promptTokens: number;
                 completionTokens: number;
                 totalTokens: number;
             } | undefined;
             providerMetadata?: Record<string, Record<string, any>> | undefined;
-            sources?: {
+            sources?: ({
                 title?: string | undefined;
+                type?: "source" | undefined;
                 providerOptions?: Record<string, Record<string, any>> | undefined;
+                url?: string | undefined;
                 id: string;
                 sourceType: "url";
-                url: string;
-            }[] | undefined;
+            } | {
+                filename?: string | undefined;
+                providerMetadata?: Record<string, Record<string, any>> | undefined;
+                id: string;
+                title: string;
+                type: "source";
+                sourceType: "document";
+                mediaType: string;
+            })[] | undefined;
             reasoningDetails?: ({
+                providerOptions?: Record<string, Record<string, any>> | undefined;
+                signature?: string | undefined;
+                state?: "streaming" | "done" | undefined;
+                type: "reasoning";
+                text: string;
+            } | {
                 signature?: string | undefined;
                 type: "text";
                 text: string;
@@ -601,7 +621,7 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
     fetchContextMessages(ctx: RunQueryCtx | RunActionCtx, args: {
         userId: string | undefined;
         threadId: string | undefined;
-        messages: CoreMessage[];
+        messages: (ModelMessage | Message)[];
         /**
          * If provided, it will search for messages up to and including this message.
          * Note: if this is far in the past, text and vector search results may be more
@@ -638,7 +658,7 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
     generateEmbeddings(ctx: RunActionCtx, { userId, threadId, }: {
         userId: string | undefined;
         threadId: string | undefined;
-    }, messages: CoreMessage[]): Promise<{
+    }, messages: (ModelMessage | Message)[]): Promise<{
         vectors: (number[] | null)[];
         dimension: VectorDimension;
         model: string;
@@ -728,9 +748,9 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
         messageId: string;
         patch: {
             /** The message to replace the existing message. */
-            message: CoreMessage & {
+            message: (ModelMessage & {
                 id?: string;
-            };
+            }) | Message;
             /** The status to set on the message. */
             status: "success" | "error";
             /** The error message to set on the message. */
@@ -809,7 +829,11 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
         startStepOrder?: number;
         endOrder: number;
         endStepOrder?: number;
-    }): Promise<void>;
+    }): Promise<{
+        isDone: boolean;
+        lastOrder?: number;
+        lastStepOrder?: number;
+    }>;
     /**
      * Delete a thread and all its messages and streams asynchronously (in batches)
      * This uses a mutation to that processes one page and recursively queues the
@@ -834,18 +858,18 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
     }): Promise<void>;
     _saveMessagesAndFetchContext<T extends {
         id?: string;
-        prompt?: string;
-        messages?: CoreMessage[] | AIMessageWithoutId[];
+        prompt?: string | (ModelMessage | Message)[];
+        messages?: (ModelMessage | Message)[];
         system?: string;
         promptMessageId?: string;
-        model?: LanguageModelV1;
+        model?: LanguageModelV2;
         maxRetries?: number;
     }>(ctx: RunActionCtx, args: T, { userId: argsUserId, threadId, contextOptions, storageOptions, }: {
         userId: string | undefined;
         threadId: string | undefined;
     } & Options): Promise<{
         args: T & {
-            model: LanguageModelV1;
+            model: LanguageModelV2;
         };
         userId: string | undefined;
         messageId: string | undefined;
@@ -907,14 +931,14 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
      * Create an action out of this agent so you can call it from workflows or other actions
      * without a wrapping function.
      * @param spec Configuration for the agent acting as an action, including
-     *   {@link ContextOptions}, {@link StorageOptions}, and maxSteps.
+     *   {@link ContextOptions}, {@link StorageOptions}, and {@link stopWhen}.
      */
     asTextAction<DataModel extends GenericDataModel>(spec?: {
         /**
-         * The maximum number of steps to take in this action.
-         * Defaults to the {@link Agent.maxSteps} option.
+         * When to stop generating text.
+         * Defaults to the {@link Agent["options"].stopWhen} option.
          */
-        maxSteps?: number;
+        stopWhen?: StopCondition<AgentTools> | Array<StopCondition<AgentTools>>;
         /**
          * The {@link ContextOptions} to use for fetching contextual messages and
          * saving input/output messages.
@@ -997,6 +1021,7 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
             } | {
                 providerOptions?: Record<string, Record<string, any>> | undefined;
                 signature?: string | undefined;
+                state?: "streaming" | "done" | undefined;
                 type: "reasoning";
                 text: string;
             } | {
@@ -1005,6 +1030,7 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
                 data: string;
             } | {
                 providerOptions?: Record<string, Record<string, any>> | undefined;
+                providerExecuted?: boolean | undefined;
                 type: "tool-call";
                 toolCallId: string;
                 toolName: string;
@@ -1016,6 +1042,8 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
             content: {
                 providerOptions?: Record<string, Record<string, any>> | undefined;
                 args?: any;
+                providerExecuted?: boolean | undefined;
+                isError?: boolean | undefined;
                 experimental_content?: ({
                     type: "text";
                     text: string;
@@ -1024,7 +1052,6 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
                     type: "image";
                     data: string;
                 })[] | undefined;
-                isError?: boolean | undefined;
                 type: "tool-result";
                 toolCallId: string;
                 toolName: string;
@@ -1075,25 +1102,23 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
         text: string;
         messageId: string | undefined;
         order: number | undefined;
-        finishReason: import("@ai-sdk/provider").LanguageModelV1FinishReason;
-        warnings: Promise<import("@ai-sdk/provider").LanguageModelV1CallWarning[] | undefined>;
+        finishReason: import("@ai-sdk/provider").LanguageModelV2FinishReason;
+        warnings: Promise<import("@ai-sdk/provider").LanguageModelV2CallWarning[] | undefined>;
     } | {
         text: string;
         messageId: string | undefined;
         order: number | undefined;
-        finishReason: import("@ai-sdk/provider").LanguageModelV1FinishReason;
-        warnings: import("@ai-sdk/provider").LanguageModelV1CallWarning[] | undefined;
+        finishReason: import("@ai-sdk/provider").LanguageModelV2FinishReason;
+        warnings: import("@ai-sdk/provider").LanguageModelV2CallWarning[] | undefined;
     }>>;
     /**
      * Create an action that generates an object out of this agent so you can call
      * it from workflows or other actions without a wrapping function.
      * @param spec Configuration for the agent acting as an action, including
      * the normal parameters to {@link generateObject}, plus {@link ContextOptions}
-     * and maxSteps.
+     * and stopWhen.
      */
-    asObjectAction<T>(spec: OurObjectArgs<T> & {
-        maxSteps?: number;
-    }, options?: {
+    asObjectAction<T>(spec: OurObjectArgs<T>, options?: {
         contextOptions?: ContextOptions;
         storageOptions?: StorageOptions;
     }): import("convex/server").RegisteredAction<"internal", {
@@ -1136,6 +1161,7 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
             } | {
                 providerOptions?: Record<string, Record<string, any>> | undefined;
                 signature?: string | undefined;
+                state?: "streaming" | "done" | undefined;
                 type: "reasoning";
                 text: string;
             } | {
@@ -1144,6 +1170,7 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
                 data: string;
             } | {
                 providerOptions?: Record<string, Record<string, any>> | undefined;
+                providerExecuted?: boolean | undefined;
                 type: "tool-call";
                 toolCallId: string;
                 toolName: string;
@@ -1155,6 +1182,8 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
             content: {
                 providerOptions?: Record<string, Record<string, any>> | undefined;
                 args?: any;
+                providerExecuted?: boolean | undefined;
+                isError?: boolean | undefined;
                 experimental_content?: ({
                     type: "text";
                     text: string;
@@ -1163,7 +1192,6 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
                     type: "image";
                     data: string;
                 })[] | undefined;
-                isError?: boolean | undefined;
                 type: "tool-result";
                 toolCallId: string;
                 toolName: string;
@@ -1207,8 +1235,8 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
         object: T;
         messageId: string | undefined;
         order: number | undefined;
-        finishReason: import("@ai-sdk/provider").LanguageModelV1FinishReason;
-        warnings: import("@ai-sdk/provider").LanguageModelV1CallWarning[] | undefined;
+        finishReason: import("@ai-sdk/provider").LanguageModelV2FinishReason;
+        warnings: import("@ai-sdk/provider").LanguageModelV2CallWarning[] | undefined;
     }>>;
     /**
      * Save messages to the thread.
@@ -1245,19 +1273,36 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
             text?: string | undefined;
             reasoning?: string | undefined;
             usage?: {
+                reasoningTokens?: number | undefined;
+                cachedInputTokens?: number | undefined;
                 promptTokens: number;
                 completionTokens: number;
                 totalTokens: number;
             } | undefined;
             providerMetadata?: Record<string, Record<string, any>> | undefined;
-            sources?: {
+            sources?: ({
                 title?: string | undefined;
+                type?: "source" | undefined;
                 providerOptions?: Record<string, Record<string, any>> | undefined;
+                url?: string | undefined;
                 id: string;
                 sourceType: "url";
-                url: string;
-            }[] | undefined;
+            } | {
+                filename?: string | undefined;
+                providerMetadata?: Record<string, Record<string, any>> | undefined;
+                id: string;
+                title: string;
+                type: "source";
+                sourceType: "document";
+                mediaType: string;
+            })[] | undefined;
             reasoningDetails?: ({
+                providerOptions?: Record<string, Record<string, any>> | undefined;
+                signature?: string | undefined;
+                state?: "streaming" | "done" | undefined;
+                type: "reasoning";
+                text: string;
+            } | {
                 signature?: string | undefined;
                 type: "text";
                 text: string;
@@ -1313,6 +1358,7 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
                 } | {
                     providerOptions?: Record<string, Record<string, any>> | undefined;
                     signature?: string | undefined;
+                    state?: "streaming" | "done" | undefined;
                     type: "reasoning";
                     text: string;
                 } | {
@@ -1321,6 +1367,7 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
                     data: string;
                 } | {
                     providerOptions?: Record<string, Record<string, any>> | undefined;
+                    providerExecuted?: boolean | undefined;
                     type: "tool-call";
                     toolCallId: string;
                     toolName: string;
@@ -1332,6 +1379,8 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
                 content: {
                     providerOptions?: Record<string, Record<string, any>> | undefined;
                     args?: any;
+                    providerExecuted?: boolean | undefined;
+                    isError?: boolean | undefined;
                     experimental_content?: ({
                         type: "text";
                         text: string;
@@ -1340,7 +1389,6 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
                         type: "image";
                         data: string;
                     })[] | undefined;
-                    isError?: boolean | undefined;
                     type: "tool-result";
                     toolCallId: string;
                     toolName: string;
@@ -1357,9 +1405,6 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
         messageIds: string[];
     }>>;
 }
-type CoreMessageMaybeWithId = CoreMessage & {
-    id?: string | undefined;
-};
 /**
  * Create a thread to store messages with an Agent.
  * @param ctx The context from a mutation or action.
@@ -1392,7 +1437,9 @@ type SaveMessagesArgs = {
     /**
      * The messages to save.
      */
-    messages: CoreMessageMaybeWithId[];
+    messages: ((ModelMessage & {
+        id?: string | undefined;
+    }) | Message)[];
     /**
      * Metadata to save with the messages. Each element corresponds to the
      * message at the same index.
@@ -1494,6 +1541,7 @@ export declare function saveMessages(ctx: RunMutationCtx, component: AgentCompon
                     };
                 } | undefined;
                 signature?: string | undefined;
+                state?: "streaming" | "done" | undefined;
                 text: string;
                 type: "reasoning";
             } | {
@@ -1506,6 +1554,7 @@ export declare function saveMessages(ctx: RunMutationCtx, component: AgentCompon
                 type: "redacted-reasoning";
             } | {
                 args: any;
+                providerExecuted?: boolean | undefined;
                 providerOptions?: {
                     [x: string]: {
                         [x: string]: any;
@@ -1533,6 +1582,7 @@ export declare function saveMessages(ctx: RunMutationCtx, component: AgentCompon
                     type: "image";
                 })[] | undefined;
                 isError?: boolean | undefined;
+                providerExecuted?: boolean | undefined;
                 providerOptions?: {
                     [x: string]: {
                         [x: string]: any;
@@ -1573,6 +1623,16 @@ export declare function saveMessages(ctx: RunMutationCtx, component: AgentCompon
         } | undefined;
         reasoning?: string | undefined;
         reasoningDetails?: ({
+            providerOptions?: {
+                [x: string]: {
+                    [x: string]: any;
+                };
+            } | undefined;
+            signature?: string | undefined;
+            state?: "streaming" | "done" | undefined;
+            text: string;
+            type: "reasoning";
+        } | {
             signature?: string | undefined;
             text: string;
             type: "text";
@@ -1580,7 +1640,7 @@ export declare function saveMessages(ctx: RunMutationCtx, component: AgentCompon
             data: string;
             type: "redacted";
         })[] | undefined;
-        sources?: {
+        sources?: ({
             id: string;
             providerOptions?: {
                 [x: string]: {
@@ -1589,16 +1649,31 @@ export declare function saveMessages(ctx: RunMutationCtx, component: AgentCompon
             } | undefined;
             sourceType: "url";
             title?: string | undefined;
-            url: string;
-        }[] | undefined;
+            type?: "source" | undefined;
+            url?: string | undefined;
+        } | {
+            filename?: string | undefined;
+            id: string;
+            mediaType: string;
+            providerMetadata?: {
+                [x: string]: {
+                    [x: string]: any;
+                };
+            } | undefined;
+            sourceType: "document";
+            title: string;
+            type: "source";
+        })[] | undefined;
         status: "pending" | "success" | "failed";
         stepOrder: number;
         text?: string | undefined;
         threadId: string;
         tool: boolean;
         usage?: {
+            cachedInputTokens?: number | undefined;
             completionTokens: number;
             promptTokens: number;
+            reasoningTokens?: number | undefined;
             totalTokens: number;
         } | undefined;
         userId?: string | undefined;
@@ -1636,7 +1711,7 @@ type SaveMessageArgs = {
     /**
      * The message to save.
      */
-    message: CoreMessage;
+    message: ModelMessage | Message;
 } | {
     prompt: string;
     message?: undefined;
@@ -1725,6 +1800,7 @@ export declare function saveMessage(ctx: RunMutationCtx, component: AgentCompone
                     };
                 } | undefined;
                 signature?: string | undefined;
+                state?: "streaming" | "done" | undefined;
                 text: string;
                 type: "reasoning";
             } | {
@@ -1737,6 +1813,7 @@ export declare function saveMessage(ctx: RunMutationCtx, component: AgentCompone
                 type: "redacted-reasoning";
             } | {
                 args: any;
+                providerExecuted?: boolean | undefined;
                 providerOptions?: {
                     [x: string]: {
                         [x: string]: any;
@@ -1764,6 +1841,7 @@ export declare function saveMessage(ctx: RunMutationCtx, component: AgentCompone
                     type: "image";
                 })[] | undefined;
                 isError?: boolean | undefined;
+                providerExecuted?: boolean | undefined;
                 providerOptions?: {
                     [x: string]: {
                         [x: string]: any;
@@ -1804,6 +1882,16 @@ export declare function saveMessage(ctx: RunMutationCtx, component: AgentCompone
         } | undefined;
         reasoning?: string | undefined;
         reasoningDetails?: ({
+            providerOptions?: {
+                [x: string]: {
+                    [x: string]: any;
+                };
+            } | undefined;
+            signature?: string | undefined;
+            state?: "streaming" | "done" | undefined;
+            text: string;
+            type: "reasoning";
+        } | {
             signature?: string | undefined;
             text: string;
             type: "text";
@@ -1811,7 +1899,7 @@ export declare function saveMessage(ctx: RunMutationCtx, component: AgentCompone
             data: string;
             type: "redacted";
         })[] | undefined;
-        sources?: {
+        sources?: ({
             id: string;
             providerOptions?: {
                 [x: string]: {
@@ -1820,16 +1908,31 @@ export declare function saveMessage(ctx: RunMutationCtx, component: AgentCompone
             } | undefined;
             sourceType: "url";
             title?: string | undefined;
-            url: string;
-        }[] | undefined;
+            type?: "source" | undefined;
+            url?: string | undefined;
+        } | {
+            filename?: string | undefined;
+            id: string;
+            mediaType: string;
+            providerMetadata?: {
+                [x: string]: {
+                    [x: string]: any;
+                };
+            } | undefined;
+            sourceType: "document";
+            title: string;
+            type: "source";
+        })[] | undefined;
         status: "pending" | "success" | "failed";
         stepOrder: number;
         text?: string | undefined;
         threadId: string;
         tool: boolean;
         usage?: {
+            cachedInputTokens?: number | undefined;
             completionTokens: number;
             promptTokens: number;
+            reasoningTokens?: number | undefined;
             totalTokens: number;
         } | undefined;
         userId?: string | undefined;