@convex-dev/agent 0.1.18 → 0.2.0

package/dist/client/index.d.ts

@@ -1,27 +1,78 @@
-import type { EmbeddingModelV1, LanguageModelV1 } from "@ai-sdk/provider";
-import type { CoreMessage, DeepPartial, GenerateObjectResult, GenerateTextResult, StepResult, StreamObjectResult, StreamTextResult, ToolSet } from "ai";
-import { type GenericActionCtx, type GenericDataModel, type PaginationOptions, type PaginationResult } from "convex/server";
+import type { FlexibleSchema, IdGenerator, InferSchema } from "@ai-sdk/provider-utils";
+import type { CallSettings, EmbeddingModel, GenerateObjectResult, GenerateTextResult, LanguageModel, ModelMessage, StepResult, StopCondition, StreamTextResult, ToolSet } from "ai";
+import { generateObject, streamObject } from "ai";
+import { 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 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 { ActionCtx, AgentComponent, ContextOptions, GenerationOutputMetadata, Options, OurObjectArgs, OurStreamObjectArgs, RawRequestResponseHandler, RunActionCtx, RunMutationCtx, RunQueryCtx, StorageOptions, StreamingTextArgs, SyncStreamsReturnValue, TextArgs, Thread, UsageHandler } from "./types.js";
+import { type SaveMessageArgs, type SaveMessagesArgs } from "./messages.js";
+import { type StreamingOptions } from "./streaming.js";
+import type { ActionCtx, AgentComponent, ContextOptions, DefaultObjectSchema, GenerationOutputMetadata, MaybeCustomCtx, GenerateObjectArgs, ObjectMode, ObjectSchema, Options, RawRequestResponseHandler, RunActionCtx, RunMutationCtx, RunQueryCtx, StorageOptions, StreamingTextArgs, StreamObjectArgs, SyncStreamsReturnValue, TextArgs, Thread, UsageHandler } from "./types.js";
+import type { DataModel } from "../component/_generated/dataModel.js";
+export { stepCountIs } from "ai";
 export { vMessageDoc, vThreadDoc } from "../component/schema.js";
-export { serializeDataOrUrl } from "../mapping.js";
+export { deserializeMessage, serializeDataOrUrl, 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 { definePlaygroundAPI, type AgentsFn, type PlaygroundAPI, } from "./definePlaygroundAPI.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 { definePlaygroundAPI, type PlaygroundAPI, type AgentsFn, } from "./definePlaygroundAPI.js";
+export { listMessages, saveMessage, saveMessages, type SaveMessageArgs, type SaveMessagesArgs, } from "./messages.js";
+export { fetchContextMessages, filterOutOrphanedToolMessages, } from "./search.js";
+export { abortStream, listStreams, syncStreams } from "./streaming.js";
+export { createThread, getThreadMetadata } from "./threads.js";
+export { createTool, extractText, isTool };
 export type { AgentComponent, ContextOptions, MessageDoc, ProviderMetadata, RawRequestResponseHandler, StorageOptions, StreamArgs, SyncStreamsReturnValue, Thread, ThreadDoc, Usage, UsageHandler, };
+export type Config = {
+    /**
+     * The LLM model to use for generating / streaming text and objects.
+     * e.g.
+     * import { openai } from "@ai-sdk/openai"
+     * const myAgent = new Agent(components.agent, {
+     *   languageModel: openai.chat("gpt-4o-mini"),
+     */
+    languageModel?: LanguageModel;
+    /**
+     * The model to use for text embeddings. Optional.
+     * If specified, it will use this for generating vector embeddings
+     * of chats, and can opt-in to doing vector search for automatic context
+     * on generateText, etc.
+     * e.g.
+     * import { openai } from "@ai-sdk/openai"
+     * const myAgent = new Agent(components.agent, {
+     *   ...
+     *   textEmbeddingModel: openai.embedding("text-embedding-3-small")
+     */
+    textEmbeddingModel?: EmbeddingModel<string>;
+    /**
+     * Options to determine what messages are included as context in message
+     * generation. To disable any messages automatically being added, pass:
+     * { recentMessages: 0 }
+     */
+    contextOptions?: ContextOptions;
+    /**
+     * Determines whether messages are automatically stored when passed as
+     * arguments or generated.
+     */
+    storageOptions?: StorageOptions;
+    /**
+     * The default settings to use for the LLM calls.
+     * This can be overridden at each generate/stream callsite on a per-field
+     * basis. To clear a default setting, you'll need to pass `undefined`.
+     */
+    callSettings?: CallSettings;
+    /**
+     * The usage handler to use for this agent.
+     */
+    usageHandler?: UsageHandler;
+    /**
+     * Called for each LLM request/response, so you can do things like
+     * log the raw request body or response headers to a table, or logs.
+     */
+    rawRequestResponseHandler?: RawRequestResponseHandler;
+};
 export declare class Agent<
 /**
  * You can require that all `ctx` args to generateText & streamText
@@ -44,7 +95,7 @@ export declare class Agent<
  */
 CustomCtx extends object = object, AgentTools extends ToolSet = any> {
     component: AgentComponent;
-    options: {
+    options: Config & {
         /**
          * The name for the agent. This will be attributed on each message
          * created by this agent.
@@ -55,20 +106,9 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
          * e.g.
          * import { openai } from "@ai-sdk/openai"
          * const myAgent = new Agent(components.agent, {
-         *   chat: openai.chat("gpt-4o-mini"),
-         */
-        chat: LanguageModelV1;
-        /**
-         * The model to use for text embeddings. Optional.
-         * If specified, it will use this for generating vector embeddings
-         * of chats, and can opt-in to doing vector search for automatic context
-         * on generateText, etc.
-         * e.g.
-         * import { openai } from "@ai-sdk/openai"
-         * const myAgent = new Agent(components.agent, {
-         *   textEmbedding: openai.embedding("text-embedding-3-small")
+         *   languageModel: openai.chat("gpt-4o-mini"),
          */
-        textEmbedding?: EmbeddingModelV1<string>;
+        languageModel: LanguageModel;
         /**
          * The default system prompt to put in each request.
          * Override per-prompt by passing the "system" parameter.
@@ -81,38 +121,17 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
          * (import { createTool } from "@convex-dev/agent")
          */
         tools?: AgentTools;
-        /**
-         * Options to determine what messages are included as context in message
-         * generation. To disable any messages automatically being added, pass:
-         * { recentMessages: 0 }
-         */
-        contextOptions?: ContextOptions;
-        /**
-         * Determines whether messages are automatically stored when passed as
-         * arguments or generated.
-         */
-        storageOptions?: StorageOptions;
         /**
          * When generating or streaming text with tools available, this
-         * determines the default max number of iterations.
-         */
-        maxSteps?: number;
-        /**
-         * The maximum number of calls to make to an LLM in case it fails.
-         * This can be overridden at each generate/stream callsite.
-         */
-        maxRetries?: number;
-        /**
-         * The usage handler to use for this agent.
+         * determines when to stop. Defaults to stepCountIs(1).
          */
-        usageHandler?: UsageHandler;
+        stopWhen?: StopCondition<AgentTools> | Array<StopCondition<AgentTools>>;
         /**
-         * Called for each LLM request/response, so you can do things like
-         * log the raw request body or response headers to a table, or logs.
+         * @deprecated Use `languageEmbeddingModel` instead.
          */
-        rawRequestResponseHandler?: RawRequestResponseHandler;
+        chat?: LanguageModel;
     };
-    constructor(component: AgentComponent, options: {
+    constructor(component: AgentComponent, options: Config & {
         /**
          * The name for the agent. This will be attributed on each message
          * created by this agent.
@@ -123,20 +142,9 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
          * e.g.
          * import { openai } from "@ai-sdk/openai"
          * const myAgent = new Agent(components.agent, {
-         *   chat: openai.chat("gpt-4o-mini"),
+         *   languageModel: openai.chat("gpt-4o-mini"),
          */
-        chat: LanguageModelV1;
-        /**
-         * The model to use for text embeddings. Optional.
-         * If specified, it will use this for generating vector embeddings
-         * of chats, and can opt-in to doing vector search for automatic context
-         * on generateText, etc.
-         * e.g.
-         * import { openai } from "@ai-sdk/openai"
-         * const myAgent = new Agent(components.agent, {
-         *   textEmbedding: openai.embedding("text-embedding-3-small")
-         */
-        textEmbedding?: EmbeddingModelV1<string>;
+        languageModel: LanguageModel;
         /**
          * The default system prompt to put in each request.
          * Override per-prompt by passing the "system" parameter.
@@ -149,36 +157,15 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
          * (import { createTool } from "@convex-dev/agent")
          */
         tools?: AgentTools;
-        /**
-         * Options to determine what messages are included as context in message
-         * generation. To disable any messages automatically being added, pass:
-         * { recentMessages: 0 }
-         */
-        contextOptions?: ContextOptions;
-        /**
-         * Determines whether messages are automatically stored when passed as
-         * arguments or generated.
-         */
-        storageOptions?: StorageOptions;
         /**
          * When generating or streaming text with tools available, this
-         * determines the default max number of iterations.
-         */
-        maxSteps?: number;
-        /**
-         * The maximum number of calls to make to an LLM in case it fails.
-         * This can be overridden at each generate/stream callsite.
-         */
-        maxRetries?: number;
-        /**
-         * The usage handler to use for this agent.
+         * determines when to stop. Defaults to stepCountIs(1).
          */
-        usageHandler?: UsageHandler;
+        stopWhen?: StopCondition<AgentTools> | Array<StopCondition<AgentTools>>;
         /**
-         * Called for each LLM request/response, so you can do things like
-         * log the raw request body or response headers to a table, or logs.
+         * @deprecated Use `languageEmbeddingModel` instead.
          */
-        rawRequestResponseHandler?: RawRequestResponseHandler;
+        chat?: LanguageModel;
     });
     /**
      * Start a new thread with the agent. This will have a fresh history, though if
@@ -190,7 +177,7 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
      * @param args The thread metadata.
      * @returns The threadId of the new thread and the thread object.
      */
-    createThread<ThreadTools extends ToolSet | undefined = undefined>(ctx: RunActionCtx & CustomCtx, args?: {
+    createThread(ctx: RunActionCtx & CustomCtx, args?: {
         /**
          * The userId to associate with the thread. If not provided, the thread will be
          * anonymous.
@@ -204,19 +191,9 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
          * The summary of the thread. Not currently used for anything.
          */
         summary?: string;
-        /**
-         * The usage handler to use for this thread. Overrides any handler
-         * set in the agent constructor.
-         */
-        usageHandler?: UsageHandler;
-        /**
-         * The tools to use for this thread.
-         * Overrides any tools passed in the agent constructor.
-         */
-        tools?: ThreadTools;
     }): Promise<{
         threadId: string;
-        thread: Thread<ThreadTools extends undefined ? AgentTools : ThreadTools>;
+        thread: Thread<AgentTools>;
     }>;
     /**
      * Start a new thread with the agent. This will have a fresh history, though if
@@ -227,7 +204,7 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
      * @param args The thread metadata.
      * @returns The threadId of the new thread.
      */
-    createThread<ThreadTools extends ToolSet | undefined = undefined>(ctx: RunMutationCtx, args?: {
+    createThread(ctx: RunMutationCtx, args?: {
         /**
          * The userId to associate with the thread. If not provided, the thread will be
          * anonymous.
@@ -241,16 +218,6 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
          * The summary of the thread. Not currently used for anything.
          */
         summary?: string;
-        /**
-         * The usage handler to use for this thread. Overrides any handler
-         * set in the agent constructor.
-         */
-        usageHandler?: UsageHandler;
-        /**
-         * The tools to use for this thread.
-         * Overrides any tools passed in the agent constructor.
-         */
-        tools?: ThreadTools;
     }): Promise<{
         threadId: string;
     }>;
@@ -262,7 +229,7 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
      * @param { threadId, userId }: the thread and user to associate the messages with.
      * @returns Functions bound to the userId and threadId on a `{thread}` object.
      */
-    continueThread<ThreadTools extends ToolSet | undefined = undefined>(ctx: ActionCtx & CustomCtx, args: {
+    continueThread(ctx: ActionCtx & CustomCtx, args: {
         /**
          * The associated thread created by {@link createThread}
          */
@@ -272,29 +239,87 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
          * relevant messages from the same user as context for the LLM calls.
          */
         userId?: string | null;
+    }): Promise<{
+        thread: Thread<AgentTools>;
+    }>;
+    start<TOOLS extends ToolSet | undefined, T>(ctx: ActionCtx & CustomCtx, 
+    /**
+     * These are the arguments you'll pass to the LLM call such as
+     * `generateText` or `streamText`. This function will look up the context
+     * and provide functions to save the steps, abort the generation, and more.
+     * The type of the arguments returned infers from the type of the arguments
+     * you pass here.
+     */
+    args: T & {
+        /**
+         * If provided, this message will be used as the "prompt" for the LLM call,
+         * instead of the prompt or messages.
+         * This is useful if you want to first save a user message, then use it as
+         * the prompt for the LLM call in another call.
+         */
+        promptMessageId?: string;
+        /**
+         * The model to use for the LLM calls. This will override the model specified
+         * in the Agent constructor.
+         */
+        model?: LanguageModel;
         /**
-         * The usage handler to use for this thread. Overrides any handler
-         * set in the agent constructor.
+         * The tools to use for the tool calls. This will override tools specified
+         * in the Agent constructor or createThread / continueThread.
          */
-        usageHandler?: UsageHandler;
+        tools?: TOOLS;
         /**
-         * The tools to use for this thread.
-         * Overrides any tools passed in the agent constructor.
+         * The single prompt message to use for the LLM call. This will be the
+         * last message in the context. If it's a string, it will be a user role.
          */
-        tools?: ThreadTools;
+        prompt?: string | (ModelMessage | Message)[];
+        /**
+         * If provided alongside prompt, the ordering will be:
+         * 1. system prompt
+         * 2. search context
+         * 3. recent messages
+         * 4. these messages
+         * 5. prompt messages, including those already on the same `order` as
+         *   the promptMessageId message, if provided.
+         */
+        messages?: (ModelMessage | Message)[];
+        /**
+         * This will be the first message in the context, and overrides the
+         * agent's instructions.
+         */
+        system?: string;
+        /**
+         * The abort signal to be passed to the LLM call. If triggered, it will
+         * mark the pending message as failed. If the generation is asynchronously
+         * aborted, it will trigger this signal when detected.
+         */
+        abortSignal?: AbortSignal;
+        _internal?: {
+            generateId?: IdGenerator;
+        };
+    }, options?: Options & {
+        userId?: string | null;
+        threadId?: string;
     }): Promise<{
-        thread: Thread<ThreadTools extends undefined ? AgentTools : ThreadTools>;
+        args: T & {
+            system?: string;
+            model: LanguageModel;
+            messages: ModelMessage[];
+            tools?: TOOLS extends undefined ? AgentTools : TOOLS;
+        } & CallSettings;
+        order: number;
+        stepOrder: number;
+        userId: string | undefined;
+        promptMessageId: string | undefined;
+        updateModel: (model: LanguageModel | undefined) => void;
+        save: <TOOLS extends ToolSet>(toSave: {
+            step: StepResult<TOOLS>;
+        } | {
+            object: GenerateObjectResult<unknown>;
+        }, createPendingMessage?: boolean) => Promise<void>;
+        fail: (reason: string) => Promise<void>;
+        getSavedMessages: () => MessageDoc[];
     }>;
-    /**
-     * Search for threads by title, paginated.
-     * @param ctx The context passed from the query/mutation/action.
-     * @returns The threads matching the search, paginated.
-     */
-    searchThreadTitles(ctx: RunQueryCtx, { userId, query, limit, }: {
-        userId?: string | undefined;
-        query: string;
-        limit?: number;
-    }): Promise<ThreadDoc[]>;
     /**
      * This behaves like {@link generateText} from the "ai" package except that
      * it add context based on the userId and threadId and saves the input and
@@ -303,21 +328,14 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
      * to a thread (and optionally userId).
      * @param ctx The context passed from the action function calling this.
      * @param { userId, threadId }: The user and thread to associate the message with
-     * @param args The arguments to the generateText function, along with extra controls
+     * @param generateTextArgs The arguments to the generateText function, along with extra controls
      * for the {@link ContextOptions} and {@link StorageOptions}.
      * @returns The result of the generateText function.
      */
-    generateText<TOOLS extends ToolSet | undefined = undefined, OUTPUT = never, OUTPUT_PARTIAL = never>(ctx: ActionCtx & CustomCtx, { userId: argsUserId, threadId, tools: threadTools, ...usageHandler }: {
+    generateText<TOOLS extends ToolSet | undefined = undefined, OUTPUT = never, OUTPUT_PARTIAL = never>(ctx: ActionCtx & CustomCtx, threadOpts: {
         userId?: string | null;
         threadId?: string;
-        /**
-         * The usage handler to use for this thread. Overrides any handler
-         * set in the agent constructor.
-         */
-        usageHandler?: UsageHandler;
-        /** Note: to get better type inference, pass tools in the next arg */
-        tools?: ToolSet;
-    }, args: TextArgs<AgentTools, TOOLS, OUTPUT, OUTPUT_PARTIAL>, options?: Options): Promise<GenerateTextResult<TOOLS extends undefined ? AgentTools : TOOLS, OUTPUT> & GenerationOutputMetadata>;
+    }, generateTextArgs: TextArgs<AgentTools, TOOLS, OUTPUT, OUTPUT_PARTIAL>, options?: Options): Promise<GenerateTextResult<TOOLS extends undefined ? AgentTools : TOOLS, OUTPUT> & GenerationOutputMetadata>;
     /**
      * This behaves like {@link streamText} from the "ai" package except that
      * it add context based on the userId and threadId and saves the input and
@@ -325,18 +343,14 @@ 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).
      */
-    streamText<TOOLS extends ToolSet | undefined = undefined, OUTPUT = never, PARTIAL_OUTPUT = never>(ctx: ActionCtx & CustomCtx, { userId: argsUserId, threadId, 
-    /** Note: to get better type inference, pass tools in the next arg */
-    tools: threadTools, ...usageHandler }: {
+    streamText<TOOLS extends ToolSet | undefined = undefined, OUTPUT = never, PARTIAL_OUTPUT = never>(ctx: ActionCtx & CustomCtx, threadOpts: {
         userId?: string | null;
         threadId?: string;
-        usageHandler?: UsageHandler;
-        tools?: ToolSet;
     }, 
     /**
      * The arguments to the streamText function, similar to the ai `streamText` function.
      */
-    args: StreamingTextArgs<AgentTools, TOOLS, OUTPUT, PARTIAL_OUTPUT>, 
+    streamTextArgs: StreamingTextArgs<AgentTools, TOOLS, OUTPUT, PARTIAL_OUTPUT>, 
     /**
      * The {@link ContextOptions} and {@link StorageOptions}
      * options to use for fetching contextual messages and saving input/output messages.
@@ -361,20 +375,19 @@ 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).
      */
-    generateObject<T>(ctx: ActionCtx, { userId: argsUserId, threadId, ...usageHandler }: {
+    generateObject<SCHEMA extends ObjectSchema = DefaultObjectSchema, OUTPUT extends ObjectMode = InferSchema<SCHEMA> extends string ? "enum" : "object", RESULT = OUTPUT extends "array" ? Array<InferSchema<SCHEMA>> : InferSchema<SCHEMA>>(ctx: ActionCtx & CustomCtx, threadOpts: {
         userId?: string | null;
         threadId?: string;
-        usageHandler?: UsageHandler;
     }, 
     /**
      * The arguments to the generateObject function, similar to the ai.generateObject function.
      */
-    args: OurObjectArgs<T>, 
+    generateObjectArgs: GenerateObjectArgs<SCHEMA, OUTPUT, RESULT>, 
     /**
      * The {@link ContextOptions} and {@link StorageOptions}
      * options to use for fetching contextual messages and saving input/output messages.
      */
-    options?: Options): Promise<GenerateObjectResult<T> & GenerationOutputMetadata>;
+    options?: Options): Promise<GenerateObjectResult<RESULT> & GenerationOutputMetadata>;
     /**
      * This behaves like `streamObject` from the "ai" package except that
      * it add context based on the userId and threadId and saves the input and
@@ -382,20 +395,32 @@ 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<SCHEMA extends ObjectSchema = DefaultObjectSchema, OUTPUT extends ObjectMode = InferSchema<SCHEMA> extends string ? "enum" : "object", RESULT = OUTPUT extends "array" ? Array<InferSchema<SCHEMA>> : InferSchema<SCHEMA>>(ctx: ActionCtx & CustomCtx, threadOpts: {
         userId?: string | null;
         threadId?: string;
-        usageHandler?: UsageHandler;
     }, 
     /**
      * The arguments to the streamObject function, similar to the ai `streamObject` function.
      */
-    args: OurStreamObjectArgs<T>, 
+    streamObjectArgs: StreamObjectArgs<SCHEMA, OUTPUT, RESULT> & {
+        /**
+         * If provided, this message will be used as the "prompt" for the LLM call,
+         * instead of the prompt or messages.
+         * This is useful if you want to first save a user message, then use it as
+         * the prompt for the LLM call in another call.
+         */
+        promptMessageId?: string;
+        /**
+         * The model to use for the LLM calls. This will override the model specified
+         * in the Agent constructor.
+         */
+        model?: LanguageModel;
+    }, 
     /**
      * The {@link ContextOptions} and {@link StorageOptions}
      * options to use for fetching contextual messages and saving input/output messages.
      */
-    options?: Options): Promise<StreamObjectResult<DeepPartial<T>, T, never> & GenerationOutputMetadata>;
+    options?: Options): Promise<ReturnType<typeof streamObject<SCHEMA, OUTPUT, RESULT>> & GenerationOutputMetadata>;
     /**
      * Save a message to the thread.
      * @param ctx A ctx object from a mutation or action.
@@ -458,6 +483,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 +492,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 +504,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 +514,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,25 +527,28 @@ 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;
                 id: string;
                 sourceType: "url";
                 url: string;
-            }[] | undefined;
-            reasoningDetails?: ({
-                signature?: string | undefined;
-                type: "text";
-                text: string;
             } | {
-                type: "redacted";
-                data: string;
+                filename?: string | undefined;
+                providerMetadata?: Record<string, Record<string, any>> | undefined;
+                id: string;
+                title: string;
+                type: "source";
+                sourceType: "document";
+                mediaType: string;
             })[] | undefined;
             warnings?: ({
                 details?: string | undefined;
@@ -532,6 +563,20 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
                 message: string;
             })[] | undefined;
             finishReason?: "length" | "error" | "other" | "stop" | "content-filter" | "tool-calls" | "unknown" | 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;
+            } | {
+                type: "redacted";
+                data: string;
+            })[] | undefined;
             _id: string;
             _creationTime: number;
             status: "pending" | "success" | "failed";
@@ -558,7 +603,6 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
          */
         skipEmbeddings?: boolean;
     }): Promise<{
-        lastMessageId: string;
         messages: MessageDoc[];
     }>;
     /**
@@ -601,7 +645,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 +682,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;
@@ -682,7 +726,6 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
         provider?: string;
     }): Promise<{
         messages: MessageDoc[];
-        pending?: MessageDoc;
     }>;
     /**
      * Manually save the result of a generateObject call to the thread.
@@ -699,7 +742,9 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
         provider: string | undefined;
         result: GenerateObjectResult<unknown>;
         metadata?: Omit<MessageWithMetadata, "message">;
-    }): Promise<void>;
+    }): Promise<{
+        messages: MessageDoc[];
+    }>;
     /**
      * Commit or rollback a message that was pending.
      * This is done automatically when saving messages by default.
@@ -708,14 +753,13 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
      * @param args What message to save. Generally the parent message sent into
      *   the generateText call.
      */
-    completeMessage(ctx: RunMutationCtx, args: {
-        threadId: string;
+    finalizeMessage(ctx: RunMutationCtx, args: {
         messageId: string;
         result: {
-            kind: "error";
+            status: "failed";
             error: string;
         } | {
-            kind: "success";
+            status: "success";
         };
     }): Promise<void>;
     /**
@@ -728,9 +772,7 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
         messageId: string;
         patch: {
             /** The message to replace the existing message. */
-            message: CoreMessage & {
-                id?: string;
-            };
+            message: ModelMessage | Message;
             /** The status to set on the message. */
             status: "success" | "error";
             /** The error message to set on the message. */
@@ -837,24 +879,26 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
         pageSize?: number;
     }): 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;
-        maxRetries?: number;
+        pendingMessageId?: string;
+        model?: LanguageModel;
     }>(ctx: RunActionCtx, args: T, { userId: argsUserId, threadId, contextOptions, storageOptions, }: {
-        userId: string | undefined;
+        userId: string | null | undefined;
         threadId: string | undefined;
     } & Options): Promise<{
-        args: T & {
-            model: LanguageModelV1;
-        };
+        args: Extract<T, {
+            model: LanguageModel;
+            messages: ModelMessage[];
+        }> & CallSettings;
         userId: string | undefined;
-        messageId: string | undefined;
+        promptMessageId: string | undefined;
+        pendingMessageId: string | undefined;
         order: number | undefined;
         stepOrder: number | undefined;
+        savedMessages: MessageDoc[] | undefined;
     }>;
     doEmbed(ctx: RunActionCtx, options: {
         userId: string | undefined;
@@ -865,20 +909,6 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
     }): Promise<{
         embeddings: number[][];
     }>;
-    /**
-     * Process messages to inline file and image URLs that point to localhost
-     * by converting them to base64. This solves the problem of LLMs not being
-     * able to access localhost URLs.
-     */
-    private _inlineMessagesFiles;
-    /**
-     * Check if a URL points to localhost
-     */
-    private _isLocalhostUrl;
-    /**
-     * Download a file from a URL
-     */
-    private _downloadFile;
     /**
      * WORKFLOW UTILITIES
      */
@@ -911,25 +941,9 @@ 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.
-         */
-        maxSteps?: number;
-        /**
-         * The {@link ContextOptions} to use for fetching contextual messages and
-         * saving input/output messages.
-         * Defaults to the {@link Agent.contextOptions} option.
-         */
-        contextOptions?: ContextOptions;
-        /**
-         * The {@link StorageOptions} to use for saving input/output messages.
-         * Defaults to the {@link Agent.storageOptions} option.
-         */
-        storageOptions?: StorageOptions;
+    asTextAction<DataModel extends GenericDataModel>(spec: MaybeCustomCtx<CustomCtx, DataModel, AgentTools> & {
         /**
          * Whether to stream the text.
          * If false, it will generate the text in a single call. (default)
@@ -938,30 +952,12 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
          * defaults if you pass true.
          */
         stream?: boolean | StreamingOptions;
-    } & (CustomCtx extends Record<string, unknown> ? {
         /**
-         * If you have a custom ctx that you use with the Agent
-         * (e.g. new Agent<{ orgId: string }>(...))
-         * you need to provide this function to add any extra fields.
-         * e.g.
-         * ```ts
-         * const myAgent = new Agent<{ orgId: string }>(...);
-         * const myAction = myAgent.asTextAction({
-         *   customCtx: (ctx: ActionCtx, target, llmArgs) => {
-         *     const orgId = await lookupOrgId(ctx, target.threadId);
-         *     return { orgId };
-         *   },
-         * });
-         * ```
-         * Then, in your tools, you can
+         * When to stop generating text.
+         * Defaults to the {@link Agent["options"].stopWhen} option.
          */
-        customCtx: (ctx: GenericActionCtx<DataModel>, target: {
-            userId?: string | undefined;
-            threadId?: string | undefined;
-        }, llmArgs: TextArgs<AgentTools>) => CustomCtx;
-    } : {
-        customCtx?: never;
-    })): import("convex/server").RegisteredAction<"internal", {
+        stopWhen?: StopCondition<AgentTools> | Array<StopCondition<AgentTools>>;
+    } & Options, overrides?: CallSettings): import("convex/server").RegisteredAction<"internal", {
         userId?: string | undefined;
         threadId?: string | undefined;
         providerOptions?: Record<string, Record<string, any>> | undefined;
@@ -1001,6 +997,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;
             } | {
@@ -1009,6 +1006,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;
@@ -1020,6 +1018,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;
@@ -1028,7 +1028,6 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
                     type: "image";
                     data: string;
                 })[] | undefined;
-                isError?: boolean | undefined;
                 type: "tool-result";
                 toolCallId: string;
                 toolName: string;
@@ -1039,15 +1038,6 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
             role: "system";
             content: string;
         })[] | undefined;
-        headers?: Record<string, string> | undefined;
-        maxTokens?: number | undefined;
-        temperature?: number | undefined;
-        topP?: number | undefined;
-        topK?: number | undefined;
-        presencePenalty?: number | undefined;
-        frequencyPenalty?: number | undefined;
-        seed?: number | undefined;
-        maxRetries?: number | undefined;
         stream?: boolean | undefined;
         toolChoice?: "required" | "none" | "auto" | {
             type: "tool";
@@ -1075,32 +1065,41 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
         storageOptions?: {
             saveMessages?: "all" | "none" | "promptAndOutput" | undefined;
         } | undefined;
+        callSettings?: {
+            headers?: Record<string, string> | undefined;
+            maxOutputTokens?: number | undefined;
+            temperature?: number | undefined;
+            topP?: number | undefined;
+            topK?: number | undefined;
+            presencePenalty?: number | undefined;
+            frequencyPenalty?: number | undefined;
+            stopSequences?: string[] | undefined;
+            seed?: number | undefined;
+            maxRetries?: number | undefined;
+        } | undefined;
     }, Promise<{
         text: string;
-        messageId: string | undefined;
+        promptMessageId: 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>;
+        savedMessageIds: string[];
     } | {
         text: string;
-        messageId: string | undefined;
+        promptMessageId: 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;
+        savedMessageIds: string[];
     }>>;
     /**
      * 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?: {
-        contextOptions?: ContextOptions;
-        storageOptions?: StorageOptions;
-    }): import("convex/server").RegisteredAction<"internal", {
+    asObjectAction<T>(objectArgs: Omit<Parameters<typeof generateObject<FlexibleSchema<T>>>[0], "model">, options?: Options & MaybeCustomCtx<CustomCtx, DataModel, AgentTools>): import("convex/server").RegisteredAction<"internal", {
         userId?: string | undefined;
         threadId?: string | undefined;
         providerOptions?: Record<string, Record<string, any>> | undefined;
@@ -1140,6 +1139,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;
             } | {
@@ -1148,6 +1148,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;
@@ -1159,6 +1160,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;
@@ -1167,7 +1170,6 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
                     type: "image";
                     data: string;
                 })[] | undefined;
-                isError?: boolean | undefined;
                 type: "tool-result";
                 toolCallId: string;
                 toolName: string;
@@ -1178,15 +1180,6 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
             role: "system";
             content: string;
         })[] | undefined;
-        headers?: Record<string, string> | undefined;
-        maxTokens?: number | undefined;
-        temperature?: number | undefined;
-        topP?: number | undefined;
-        topK?: number | undefined;
-        presencePenalty?: number | undefined;
-        frequencyPenalty?: number | undefined;
-        seed?: number | undefined;
-        maxRetries?: number | undefined;
         prompt?: string | undefined;
         promptMessageId?: string | undefined;
         contextOptions?: {
@@ -1207,12 +1200,25 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
         storageOptions?: {
             saveMessages?: "all" | "none" | "promptAndOutput" | undefined;
         } | undefined;
+        callSettings?: {
+            headers?: Record<string, string> | undefined;
+            maxOutputTokens?: number | undefined;
+            temperature?: number | undefined;
+            topP?: number | undefined;
+            topK?: number | undefined;
+            presencePenalty?: number | undefined;
+            frequencyPenalty?: number | undefined;
+            stopSequences?: string[] | undefined;
+            seed?: number | undefined;
+            maxRetries?: number | undefined;
+        } | undefined;
     }, Promise<{
         object: T;
-        messageId: string | undefined;
+        promptMessageId: 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;
+        savedMessageIds: string[];
     }>>;
     /**
      * Save messages to the thread.
@@ -1236,12 +1242,15 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
      */
     asSaveMessagesMutation(): import("convex/server").RegisteredMutation<"internal", {
         userId?: string | undefined;
-        pending?: boolean | undefined;
         promptMessageId?: string | undefined;
+        embeddings?: {
+            model: string;
+            vectors: (number[] | null)[];
+        } | undefined;
         failPendingSteps?: boolean | undefined;
         threadId: string;
         messages: {
-            id?: string | undefined;
+            status?: "pending" | "success" | "failed" | undefined;
             fileIds?: string[] | undefined;
             error?: string | undefined;
             model?: string | undefined;
@@ -1249,25 +1258,28 @@ 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;
                 id: string;
                 sourceType: "url";
                 url: string;
-            }[] | undefined;
-            reasoningDetails?: ({
-                signature?: string | undefined;
-                type: "text";
-                text: string;
             } | {
-                type: "redacted";
-                data: string;
+                filename?: string | undefined;
+                providerMetadata?: Record<string, Record<string, any>> | undefined;
+                id: string;
+                title: string;
+                type: "source";
+                sourceType: "document";
+                mediaType: string;
             })[] | undefined;
             warnings?: ({
                 details?: string | undefined;
@@ -1282,6 +1294,20 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
                 message: string;
             })[] | undefined;
             finishReason?: "length" | "error" | "other" | "stop" | "content-filter" | "tool-calls" | "unknown" | 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;
+            } | {
+                type: "redacted";
+                data: string;
+            })[] | undefined;
             message: {
                 providerOptions?: Record<string, Record<string, any>> | undefined;
                 role: "user";
@@ -1317,6 +1343,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;
                 } | {
@@ -1325,6 +1352,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;
@@ -1336,6 +1364,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;
@@ -1344,7 +1374,6 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
                         type: "image";
                         data: string;
                     })[] | undefined;
-                    isError?: boolean | undefined;
                     type: "tool-result";
                     toolCallId: string;
                     toolName: string;
@@ -1358,497 +1387,11 @@ CustomCtx extends object = object, AgentTools extends ToolSet = any> {
         }[];
     }, Promise<{
         lastMessageId: string;
-        messageIds: string[];
+        messages: {
+            _id: string;
+            order: number;
+            stepOrder: number;
+        }[];
     }>>;
 }
-type CoreMessageMaybeWithId = CoreMessage & {
-    id?: string | undefined;
-};
-/**
- * Create a thread to store messages with an Agent.
- * @param ctx The context from a mutation or action.
- * @param component The Agent component, usually `components.agent`.
- * @param args The associated thread metadata.
- * @returns The id of the created thread.
- */
-export declare function createThread(ctx: RunMutationCtx, component: AgentComponent, args?: {
-    userId?: string | null;
-    title?: string;
-    summary?: string;
-}): Promise<string>;
-/**
- * Get the metadata for a thread.
- * @param ctx A ctx object from a query, mutation, or action.
- * @param args.threadId The thread to get the metadata for.
- * @returns The metadata for the thread.
- */
-export declare function getThreadMetadata(ctx: RunQueryCtx, component: AgentComponent, args: {
-    threadId: string;
-}): Promise<ThreadDoc>;
-type SaveMessagesArgs = {
-    threadId: string;
-    userId?: string | null;
-    /**
-     * The message that these messages are in response to. They will be
-     * the same "order" as this message, at increasing stepOrder(s).
-     */
-    promptMessageId?: string;
-    /**
-     * The messages to save.
-     */
-    messages: CoreMessageMaybeWithId[];
-    /**
-     * Metadata to save with the messages. Each element corresponds to the
-     * message at the same index.
-     */
-    metadata?: Omit<MessageWithMetadata, "message">[];
-    /**
-     * If false, it will "commit" the messages immediately.
-     * If true, it will mark them as pending until the final step has finished.
-     * Defaults to false.
-     */
-    pending?: boolean;
-    /**
-     * If true, it will fail any pending steps.
-     * Defaults to false.
-     */
-    failPendingSteps?: boolean;
-    /**
-     * The embeddings to save with the messages.
-     */
-    embeddings?: Omit<MessageEmbeddings, "dimension">;
-};
-/**
- * Explicitly save messages associated with the thread (& user if provided)
- */
-export declare function saveMessages(ctx: RunMutationCtx, component: AgentComponent, args: SaveMessagesArgs & {
-    /**
-     * The agent name to associate with the messages.
-     */
-    agentName?: string;
-}): Promise<{
-    lastMessageId: string;
-    messages: {
-        _creationTime: number;
-        _id: string;
-        agentName?: string | undefined;
-        embeddingId?: string | undefined;
-        error?: string | undefined;
-        fileIds?: string[] | undefined;
-        finishReason?: "stop" | "length" | "content-filter" | "tool-calls" | "error" | "other" | "unknown" | undefined;
-        id?: string | undefined;
-        message?: {
-            content: string | ({
-                providerOptions?: {
-                    [x: string]: {
-                        [x: string]: any;
-                    };
-                } | undefined;
-                text: string;
-                type: "text";
-            } | {
-                image: string | ArrayBuffer;
-                mimeType?: string | undefined;
-                providerOptions?: {
-                    [x: string]: {
-                        [x: string]: any;
-                    };
-                } | undefined;
-                type: "image";
-            } | {
-                data: string | ArrayBuffer;
-                filename?: string | undefined;
-                mimeType: string;
-                providerOptions?: {
-                    [x: string]: {
-                        [x: string]: any;
-                    };
-                } | undefined;
-                type: "file";
-            })[];
-            providerOptions?: {
-                [x: string]: {
-                    [x: string]: any;
-                };
-            } | undefined;
-            role: "user";
-        } | {
-            content: string | ({
-                providerOptions?: {
-                    [x: string]: {
-                        [x: string]: any;
-                    };
-                } | undefined;
-                text: string;
-                type: "text";
-            } | {
-                data: string | ArrayBuffer;
-                filename?: string | undefined;
-                mimeType: string;
-                providerOptions?: {
-                    [x: string]: {
-                        [x: string]: any;
-                    };
-                } | undefined;
-                type: "file";
-            } | {
-                providerOptions?: {
-                    [x: string]: {
-                        [x: string]: any;
-                    };
-                } | undefined;
-                signature?: string | undefined;
-                text: string;
-                type: "reasoning";
-            } | {
-                data: string;
-                providerOptions?: {
-                    [x: string]: {
-                        [x: string]: any;
-                    };
-                } | undefined;
-                type: "redacted-reasoning";
-            } | {
-                args: any;
-                providerOptions?: {
-                    [x: string]: {
-                        [x: string]: any;
-                    };
-                } | undefined;
-                toolCallId: string;
-                toolName: string;
-                type: "tool-call";
-            })[];
-            providerOptions?: {
-                [x: string]: {
-                    [x: string]: any;
-                };
-            } | undefined;
-            role: "assistant";
-        } | {
-            content: {
-                args?: any;
-                experimental_content?: ({
-                    text: string;
-                    type: "text";
-                } | {
-                    data: string;
-                    mimeType?: string | undefined;
-                    type: "image";
-                })[] | undefined;
-                isError?: boolean | undefined;
-                providerOptions?: {
-                    [x: string]: {
-                        [x: string]: any;
-                    };
-                } | undefined;
-                result: any;
-                toolCallId: string;
-                toolName: string;
-                type: "tool-result";
-            }[];
-            providerOptions?: {
-                [x: string]: {
-                    [x: string]: any;
-                };
-            } | undefined;
-            role: "tool";
-        } | {
-            content: string;
-            providerOptions?: {
-                [x: string]: {
-                    [x: string]: any;
-                };
-            } | undefined;
-            role: "system";
-        } | undefined;
-        model?: string | undefined;
-        order: number;
-        provider?: string | undefined;
-        providerMetadata?: {
-            [x: string]: {
-                [x: string]: any;
-            };
-        } | undefined;
-        providerOptions?: {
-            [x: string]: {
-                [x: string]: any;
-            };
-        } | undefined;
-        reasoning?: string | undefined;
-        reasoningDetails?: ({
-            signature?: string | undefined;
-            text: string;
-            type: "text";
-        } | {
-            data: string;
-            type: "redacted";
-        })[] | undefined;
-        sources?: {
-            id: string;
-            providerOptions?: {
-                [x: string]: {
-                    [x: string]: any;
-                };
-            } | undefined;
-            sourceType: "url";
-            title?: string | undefined;
-            url: string;
-        }[] | undefined;
-        status: "pending" | "success" | "failed";
-        stepOrder: number;
-        text?: string | undefined;
-        threadId: string;
-        tool: boolean;
-        usage?: {
-            completionTokens: number;
-            promptTokens: number;
-            totalTokens: number;
-        } | undefined;
-        userId?: string | undefined;
-        warnings?: ({
-            details?: string | undefined;
-            setting: string;
-            type: "unsupported-setting";
-        } | {
-            details?: string | undefined;
-            tool: any;
-            type: "unsupported-tool";
-        } | {
-            message: string;
-            type: "other";
-        })[] | undefined;
-    }[];
-}>;
-type SaveMessageArgs = {
-    threadId: string;
-    userId?: string | null;
-    /**
-     * Metadata to save with the messages. Each element corresponds to the
-     * message at the same index.
-     */
-    metadata?: Omit<MessageWithMetadata, "message">;
-    /**
-     * The embedding to save with the message.
-     */
-    embedding?: {
-        vector: number[];
-        model: string;
-    };
-} & ({
-    prompt?: undefined;
-    /**
-     * The message to save.
-     */
-    message: CoreMessage;
-} | {
-    prompt: string;
-    message?: undefined;
-});
-/**
- * Save a message to the thread.
- * @param ctx A ctx object from a mutation or action.
- * @param args The message and what to associate it with (user / thread)
- * You can pass extra metadata alongside the message, e.g. associated fileIds.
- * @returns The messageId of the saved message.
- */
-export declare function saveMessage(ctx: RunMutationCtx, component: AgentComponent, args: SaveMessageArgs & {
-    /**
-     * The agent name to associate with the message.
-     */
-    agentName?: string;
-}): Promise<{
-    messageId: string;
-    message: {
-        _creationTime: number;
-        _id: string;
-        agentName?: string | undefined;
-        embeddingId?: string | undefined;
-        error?: string | undefined;
-        fileIds?: string[] | undefined;
-        finishReason?: "stop" | "length" | "content-filter" | "tool-calls" | "error" | "other" | "unknown" | undefined;
-        id?: string | undefined;
-        message?: {
-            content: string | ({
-                providerOptions?: {
-                    [x: string]: {
-                        [x: string]: any;
-                    };
-                } | undefined;
-                text: string;
-                type: "text";
-            } | {
-                image: string | ArrayBuffer;
-                mimeType?: string | undefined;
-                providerOptions?: {
-                    [x: string]: {
-                        [x: string]: any;
-                    };
-                } | undefined;
-                type: "image";
-            } | {
-                data: string | ArrayBuffer;
-                filename?: string | undefined;
-                mimeType: string;
-                providerOptions?: {
-                    [x: string]: {
-                        [x: string]: any;
-                    };
-                } | undefined;
-                type: "file";
-            })[];
-            providerOptions?: {
-                [x: string]: {
-                    [x: string]: any;
-                };
-            } | undefined;
-            role: "user";
-        } | {
-            content: string | ({
-                providerOptions?: {
-                    [x: string]: {
-                        [x: string]: any;
-                    };
-                } | undefined;
-                text: string;
-                type: "text";
-            } | {
-                data: string | ArrayBuffer;
-                filename?: string | undefined;
-                mimeType: string;
-                providerOptions?: {
-                    [x: string]: {
-                        [x: string]: any;
-                    };
-                } | undefined;
-                type: "file";
-            } | {
-                providerOptions?: {
-                    [x: string]: {
-                        [x: string]: any;
-                    };
-                } | undefined;
-                signature?: string | undefined;
-                text: string;
-                type: "reasoning";
-            } | {
-                data: string;
-                providerOptions?: {
-                    [x: string]: {
-                        [x: string]: any;
-                    };
-                } | undefined;
-                type: "redacted-reasoning";
-            } | {
-                args: any;
-                providerOptions?: {
-                    [x: string]: {
-                        [x: string]: any;
-                    };
-                } | undefined;
-                toolCallId: string;
-                toolName: string;
-                type: "tool-call";
-            })[];
-            providerOptions?: {
-                [x: string]: {
-                    [x: string]: any;
-                };
-            } | undefined;
-            role: "assistant";
-        } | {
-            content: {
-                args?: any;
-                experimental_content?: ({
-                    text: string;
-                    type: "text";
-                } | {
-                    data: string;
-                    mimeType?: string | undefined;
-                    type: "image";
-                })[] | undefined;
-                isError?: boolean | undefined;
-                providerOptions?: {
-                    [x: string]: {
-                        [x: string]: any;
-                    };
-                } | undefined;
-                result: any;
-                toolCallId: string;
-                toolName: string;
-                type: "tool-result";
-            }[];
-            providerOptions?: {
-                [x: string]: {
-                    [x: string]: any;
-                };
-            } | undefined;
-            role: "tool";
-        } | {
-            content: string;
-            providerOptions?: {
-                [x: string]: {
-                    [x: string]: any;
-                };
-            } | undefined;
-            role: "system";
-        } | undefined;
-        model?: string | undefined;
-        order: number;
-        provider?: string | undefined;
-        providerMetadata?: {
-            [x: string]: {
-                [x: string]: any;
-            };
-        } | undefined;
-        providerOptions?: {
-            [x: string]: {
-                [x: string]: any;
-            };
-        } | undefined;
-        reasoning?: string | undefined;
-        reasoningDetails?: ({
-            signature?: string | undefined;
-            text: string;
-            type: "text";
-        } | {
-            data: string;
-            type: "redacted";
-        })[] | undefined;
-        sources?: {
-            id: string;
-            providerOptions?: {
-                [x: string]: {
-                    [x: string]: any;
-                };
-            } | undefined;
-            sourceType: "url";
-            title?: string | undefined;
-            url: string;
-        }[] | undefined;
-        status: "pending" | "success" | "failed";
-        stepOrder: number;
-        text?: string | undefined;
-        threadId: string;
-        tool: boolean;
-        usage?: {
-            completionTokens: number;
-            promptTokens: number;
-            totalTokens: number;
-        } | undefined;
-        userId?: string | undefined;
-        warnings?: ({
-            details?: string | undefined;
-            setting: string;
-            type: "unsupported-setting";
-        } | {
-            details?: string | undefined;
-            tool: any;
-            type: "unsupported-tool";
-        } | {
-            message: string;
-            type: "other";
-        })[] | undefined;
-    };
-}>;
 //# sourceMappingURL=index.d.ts.map