@convex-dev/agent 0.0.1-alpha.1 → 0.0.1-alpha.3

package/dist/commonjs/client/index.d.ts

@@ -1,26 +1,52 @@
 import type { EmbeddingModelV1, LanguageModelV1 } from "@ai-sdk/provider";
-import type { CoreMessage, DeepPartial, GenerateObjectResult, GenerateTextResult, StepResult, StreamObjectResult, StreamTextResult, Tool, ToolChoice, ToolExecutionOptions, ToolSet, Message as UIMessage } from "ai";
+import type { CoreMessage, DeepPartial, GenerateObjectResult, GenerateTextResult, StepResult, StreamObjectResult, StreamTextResult, Tool, ToolChoice, ToolExecutionOptions, ToolSet } from "ai";
 import { generateObject, generateText, streamObject, streamText } from "ai";
-import type { ZodType } from "zod";
 import { api } from "../component/_generated/api";
-import { Message, MessageStatus, SearchOptions } from "../validators";
+import { SearchOptions } from "../validators";
 import { RunActionCtx, RunMutationCtx, RunQueryCtx, UseApi } from "./types";
-import { GenericActionCtx, GenericDataModel } from "convex/server";
+import { ConvexToZod } from "convex-helpers/server/zod";
 import { Infer, Validator } from "convex/values";
 export type ContextOptions = {
-    parentMessageId?: string;
-    includeToolMessages?: boolean;
+    /**
+     * Whether to include tool messages in the context.
+     */
+    includeToolCalls?: boolean;
+    /**
+     * How many recent messages to include. These are added after the search
+     * messages, and do not count against the search limit.
+     */
     recentMessages?: number;
+    /**
+     * Options for searching messages.
+     */
     searchOptions?: {
+        /**
+         * The maximum number of messages to fetch.
+         */
         limit: number;
+        /**
+         * Whether to use text search to find messages.
+         */
         textSearch?: boolean;
+        /**
+         * Whether to use vector search to find messages.
+         */
         vectorSearch?: boolean;
-        messageRange: {
+        /**
+         * Note, this is after the limit is applied.
+         * By default this will quadruple the number of messages fetched.
+         * (two before, and one after each message found in the search)
+         */
+        messageRange?: {
             before: number;
             after: number;
         };
     };
-    searchOtherChats?: boolean;
+    /**
+     * Whether to search across other threads for relevant messages.
+     * By default, only the current thread is searched.
+     */
+    searchOtherThreads?: boolean;
 };
 export type StorageOptions = {
     saveAllInputMessages?: boolean;
@@ -36,96 +62,107 @@ export declare class Agent<AgentTools extends ToolSet> {
     component: UseApi<typeof api>;
     options: {
         name?: string;
-        chat: LanguageModelV1;
+        thread: LanguageModelV1;
         textEmbedding?: EmbeddingModelV1<string>;
-        defaultSystemPrompt?: string;
+        instructions?: string;
         tools?: AgentTools;
+        contextOptions?: ContextOptions;
+        maxSteps?: number;
     };
     constructor(component: UseApi<typeof api>, options: {
         name?: string;
-        chat: LanguageModelV1;
+        thread: LanguageModelV1;
         textEmbedding?: EmbeddingModelV1<string>;
-        defaultSystemPrompt?: string;
+        instructions?: string;
         tools?: AgentTools;
+        contextOptions?: ContextOptions;
+        maxSteps?: number;
     });
     /**
-     * Start a new chat with the agent. This will have a fresh history, though if
-     * you pass in a userId you can have it search across other chats for relevant
+     * Start a new thread with the agent. This will have a fresh history, though if
+     * you pass in a userId you can have it search across other threads for relevant
      * messages as context for the LLM calls.
-     * @param ctx The context of the Convex function. From an action, you can chat
-     *   with the agent. From a mutation, you can start a chat and save the chatId
-     *   to pass to continueChat later.
-     * @param args The chat metadata.
-     * @returns The chatId of the new chat and the chat object.
+     * @param ctx The context of the Convex function. From an action, you can thread
+     *   with the agent. From a mutation, you can start a thread and save the threadId
+     *   to pass to continueThread later.
+     * @param args The thread metadata.
+     * @returns The threadId of the new thread and the thread object.
      */
-    startChat(ctx: RunActionCtx, args: {
+    createThread(ctx: RunActionCtx, args: {
         /**
-         * The userId to associate with the chat. If not provided, the chat will be
+         * The userId to associate with the thread. If not provided, the thread will be
          * anonymous.
          */
         userId?: string;
         /**
-         * The parent chatIds to merge with.
-         * If the chat is a continuation of one or many previous chats,
-         * you can pass in the chatIds of the parent chats to merge the histories.
+         * The parent threadIds to merge with.
+         * If the thread is a continuation of one or many previous threads,
+         * you can pass in the threadIds of the parent threads to merge the histories.
          */
-        parentChatIds?: string[];
+        parentThreadIds?: string[];
         /**
-         * The title of the chat. Not currently used.
+         * The title of the thread. Not currently used.
          */
         title?: string;
         /**
-         * The summary of the chat. Not currently used.
+         * The summary of the thread. Not currently used.
          */
         summary?: string;
     }): Promise<{
-        chatId: string;
-        chat: Chat<AgentTools>;
+        threadId: string;
+        thread: Thread<AgentTools>;
     }>;
     /**
-     * Start a new chat with the agent. This will have a fresh history, though if
-     * you pass in a userId you can have it search across other chats for relevant
+     * Start a new thread with the agent. This will have a fresh history, though if
+     * you pass in a userId you can have it search across other threads for relevant
      * messages as context for the LLM calls.
      * @param ctx The context of the Convex function. From a mutation, you can
-     * start a chat and save the chatId to pass to continueChat later.
-     * @param args The chat metadata.
-     * @returns The chatId of the new chat.
+     * start a thread and save the threadId to pass to continueThread later.
+     * @param args The thread metadata.
+     * @returns The threadId of the new thread.
      */
-    startChat(ctx: RunMutationCtx, args: {
+    createThread(ctx: RunMutationCtx, args: {
         userId?: string;
-        parentChatIds?: string[];
+        parentThreadIds?: string[];
         title?: string;
         summary?: string;
     }): Promise<{
-        chatId: string;
+        threadId: string;
     }>;
-    continueChat(ctx: RunActionCtx, { chatId, userId, }: {
-        chatId: string;
+    continueThread(ctx: RunActionCtx, { threadId, userId, }: {
+        threadId: string;
+        /**
+         * If supplied, the userId can be used to search across other threads for
+         * relevant messages from the same user as context for the LLM calls.
+         */
         userId?: string;
     }): Promise<{
-        chat: Chat<AgentTools>;
+        thread: Thread<AgentTools>;
     }>;
     fetchContextMessages(ctx: RunQueryCtx | RunActionCtx, args: {
         userId?: string;
-        chatId?: string;
+        threadId?: string;
         messages: CoreMessage[];
+        parentMessageId?: string;
     } & ContextOptions): Promise<CoreMessage[]>;
     saveMessages(ctx: RunMutationCtx, args: {
-        chatId: string;
+        threadId?: string;
+        userId?: string;
         messages: CoreMessageMaybeWithId[];
         pending?: boolean;
         parentMessageId?: string;
+        failPendingSteps?: boolean;
     }): Promise<{
         lastMessageId: string;
         messageIds: string[];
     }>;
     saveStep<TOOLS extends ToolSet>(ctx: RunMutationCtx, args: {
-        chatId: string;
+        threadId: string;
         messageId: string;
         step: StepResult<TOOLS>;
     }): Promise<void>;
     completeMessage<TOOLS extends ToolSet>(ctx: RunMutationCtx, args: {
-        chatId: string;
+        threadId: string;
         messageId: string;
         result: {
             kind: "error";
@@ -139,60 +176,466 @@ export declare class Agent<AgentTools extends ToolSet> {
     }): Promise<void>;
     /**
      * This behaves like {@link generateText} except that it add context based on
-     * the userId and chatId. It saves the input and resulting messages to the
-     * chat, if specified.
-     * however. To do that, use {@link continueChat} or {@link saveMessages}.
+     * the userId and threadId. It saves the input and resulting messages to the
+     * thread, if specified.
+     * however. To do that, use {@link continueThread} or {@link saveMessages}.
      * @param ctx The context of the agent.
      * @param args The arguments to the generateText function.
      * @returns The result of the generateText function.
      */
-    generateText<TOOLS extends ToolSet, OUTPUT = never, OUTPUT_PARTIAL = never>(ctx: RunActionCtx, { userId, chatId, }: {
+    generateText<TOOLS extends ToolSet, OUTPUT = never, OUTPUT_PARTIAL = never>(ctx: RunActionCtx, { userId, threadId, }: {
         userId?: string;
-        chatId: string;
+        threadId?: string;
     }, args: TextArgs<AgentTools, TOOLS, Parameters<typeof generateText<TOOLS, OUTPUT, OUTPUT_PARTIAL>>[0]>): Promise<GenerateTextResult<TOOLS & AgentTools, OUTPUT> & GenerationOutputMetadata>;
-    streamText<TOOLS extends ToolSet, OUTPUT = never, PARTIAL_OUTPUT = never>(ctx: RunActionCtx, { userId, chatId }: {
+    streamText<TOOLS extends ToolSet, OUTPUT = never, PARTIAL_OUTPUT = never>(ctx: RunActionCtx, { userId, threadId }: {
         userId?: string;
-        chatId: string;
-    }, args: Partial<Parameters<typeof streamText<TOOLS, OUTPUT, PARTIAL_OUTPUT>>[0]> & ContextOptions & StorageOptions): Promise<StreamTextResult<TOOLS, PARTIAL_OUTPUT> & GenerationOutputMetadata>;
-    generateObject<T>(ctx: RunActionCtx, { userId, chatId }: {
+        threadId?: string;
+    }, args: TextArgs<AgentTools, TOOLS, Parameters<typeof streamText<TOOLS, OUTPUT, PARTIAL_OUTPUT>>[0]>): Promise<StreamTextResult<TOOLS, PARTIAL_OUTPUT> & GenerationOutputMetadata>;
+    generateObject<T>(ctx: RunActionCtx, { userId, threadId }: {
         userId?: string;
-        chatId: string;
+        threadId?: string;
     }, args: Omit<Parameters<typeof generateObject>[0], "model"> & {
         model?: LanguageModelV1;
+    } & {
+        parentMessageId?: string;
     } & ContextOptions & StorageOptions): Promise<GenerateObjectResult<T> & GenerationOutputMetadata>;
-    streamObject<T>(ctx: RunMutationCtx, { userId, chatId }: {
+    streamObject<T>(ctx: RunMutationCtx, { userId, threadId }: {
         userId?: string;
-        chatId: string;
+        threadId?: string;
     }, args: Omit<Parameters<typeof streamObject<T>>[0], "model"> & {
         model?: LanguageModelV1;
+    } & {
+        parentMessageId?: string;
     } & ContextOptions & StorageOptions): Promise<StreamObjectResult<DeepPartial<T>, T, never> & GenerationOutputMetadata>;
-    searchOptionsWithDefaults(searchArgs: ContextOptions, messages: CoreMessage[]): Promise<SearchOptions>;
-    getChatMessages(ctx: RunQueryCtx, args: {
-        chatId: string;
-        limit?: number;
-        statuses?: MessageStatus[];
-        cursor?: string;
-        includeToolMessages?: boolean;
-        order?: "asc" | "desc";
-    }): Promise<{
-        messages: (Message & {
-            id: string;
-        })[];
-        continueCursor?: string;
-        isDone: boolean;
-    }>;
+    mergedContextOptions(opts: ContextOptions): ContextOptions;
+    searchOptionsWithDefaults(contextOptions: ContextOptions, messages: CoreMessage[]): Promise<SearchOptions>;
+    /**
+     *
+     */
+    asAction(spec: {
+        contextOptions?: ContextOptions;
+        maxSteps?: number;
+    }): import("convex/server").RegisteredAction<"internal", {
+        userId?: string | undefined;
+        threadId?: string | undefined;
+        createThread?: {
+            title?: string | undefined;
+            userId?: string | undefined;
+            summary?: string | undefined;
+            parentThreadIds?: string[] | undefined;
+        } | undefined;
+        maxRetries?: number | undefined;
+        contextOptions?: {
+            includeToolCalls?: boolean | undefined;
+            recentMessages?: number | undefined;
+            searchOptions?: {
+                messageRange?: {
+                    before: number;
+                    after: number;
+                } | undefined;
+                textSearch?: boolean | undefined;
+                vectorSearch?: boolean | undefined;
+                limit: number;
+            } | undefined;
+            searchOtherThreads?: boolean | undefined;
+        } | undefined;
+        storageOptions?: {
+            saveAllInputMessages?: boolean | undefined;
+            saveAllOutputMessages?: boolean | undefined;
+        } | undefined;
+        continueThread?: {
+            userId?: string | undefined;
+            threadId: string;
+        } | undefined;
+        generateText?: {
+            messages?: ({
+                providerOptions?: Record<string, any> | undefined;
+                experimental_providerMetadata?: Record<string, any> | undefined;
+                role: "user";
+                content: string | ({
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "text";
+                    text: string;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    mimeType?: string | undefined;
+                    type: "image";
+                    image: string | ArrayBuffer;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "file";
+                    mimeType: string;
+                    data: string | ArrayBuffer;
+                })[];
+            } | {
+                providerOptions?: Record<string, any> | undefined;
+                experimental_providerMetadata?: Record<string, any> | undefined;
+                role: "assistant";
+                content: string | ({
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "text";
+                    text: string;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "file";
+                    mimeType: string;
+                    data: string | ArrayBuffer;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "reasoning";
+                    text: string;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "redacted-reasoning";
+                    data: string;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "tool-call";
+                    toolCallId: string;
+                    toolName: string;
+                    args: any;
+                })[];
+            } | {
+                providerOptions?: Record<string, any> | undefined;
+                experimental_providerMetadata?: Record<string, any> | undefined;
+                role: "tool";
+                content: {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    args?: any;
+                    experimental_content?: ({
+                        type: "text";
+                        text: string;
+                    } | {
+                        mimeType?: string | undefined;
+                        type: "image";
+                        data: string;
+                    })[] | undefined;
+                    isError?: boolean | undefined;
+                    type: "tool-result";
+                    toolCallId: string;
+                    toolName: string;
+                    result: any;
+                }[];
+            } | {
+                providerOptions?: Record<string, any> | undefined;
+                experimental_providerMetadata?: Record<string, any> | undefined;
+                role: "system";
+                content: string;
+            })[] | undefined;
+            maxSteps?: number | undefined;
+            prompt?: string | undefined;
+        } | undefined;
+        streamText?: {
+            messages?: ({
+                providerOptions?: Record<string, any> | undefined;
+                experimental_providerMetadata?: Record<string, any> | undefined;
+                role: "user";
+                content: string | ({
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "text";
+                    text: string;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    mimeType?: string | undefined;
+                    type: "image";
+                    image: string | ArrayBuffer;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "file";
+                    mimeType: string;
+                    data: string | ArrayBuffer;
+                })[];
+            } | {
+                providerOptions?: Record<string, any> | undefined;
+                experimental_providerMetadata?: Record<string, any> | undefined;
+                role: "assistant";
+                content: string | ({
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "text";
+                    text: string;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "file";
+                    mimeType: string;
+                    data: string | ArrayBuffer;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "reasoning";
+                    text: string;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "redacted-reasoning";
+                    data: string;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "tool-call";
+                    toolCallId: string;
+                    toolName: string;
+                    args: any;
+                })[];
+            } | {
+                providerOptions?: Record<string, any> | undefined;
+                experimental_providerMetadata?: Record<string, any> | undefined;
+                role: "tool";
+                content: {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    args?: any;
+                    experimental_content?: ({
+                        type: "text";
+                        text: string;
+                    } | {
+                        mimeType?: string | undefined;
+                        type: "image";
+                        data: string;
+                    })[] | undefined;
+                    isError?: boolean | undefined;
+                    type: "tool-result";
+                    toolCallId: string;
+                    toolName: string;
+                    result: any;
+                }[];
+            } | {
+                providerOptions?: Record<string, any> | undefined;
+                experimental_providerMetadata?: Record<string, any> | undefined;
+                role: "system";
+                content: string;
+            })[] | undefined;
+            maxSteps?: number | undefined;
+            prompt?: string | undefined;
+        } | undefined;
+        generateObject?: {
+            messages?: ({
+                providerOptions?: Record<string, any> | undefined;
+                experimental_providerMetadata?: Record<string, any> | undefined;
+                role: "user";
+                content: string | ({
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "text";
+                    text: string;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    mimeType?: string | undefined;
+                    type: "image";
+                    image: string | ArrayBuffer;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "file";
+                    mimeType: string;
+                    data: string | ArrayBuffer;
+                })[];
+            } | {
+                providerOptions?: Record<string, any> | undefined;
+                experimental_providerMetadata?: Record<string, any> | undefined;
+                role: "assistant";
+                content: string | ({
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "text";
+                    text: string;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "file";
+                    mimeType: string;
+                    data: string | ArrayBuffer;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "reasoning";
+                    text: string;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "redacted-reasoning";
+                    data: string;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "tool-call";
+                    toolCallId: string;
+                    toolName: string;
+                    args: any;
+                })[];
+            } | {
+                providerOptions?: Record<string, any> | undefined;
+                experimental_providerMetadata?: Record<string, any> | undefined;
+                role: "tool";
+                content: {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    args?: any;
+                    experimental_content?: ({
+                        type: "text";
+                        text: string;
+                    } | {
+                        mimeType?: string | undefined;
+                        type: "image";
+                        data: string;
+                    })[] | undefined;
+                    isError?: boolean | undefined;
+                    type: "tool-result";
+                    toolCallId: string;
+                    toolName: string;
+                    result: any;
+                }[];
+            } | {
+                providerOptions?: Record<string, any> | undefined;
+                experimental_providerMetadata?: Record<string, any> | undefined;
+                role: "system";
+                content: string;
+            })[] | undefined;
+            prompt?: string | undefined;
+            output?: any;
+            mode?: "json" | undefined;
+        } | undefined;
+        streamObject?: {
+            messages?: ({
+                providerOptions?: Record<string, any> | undefined;
+                experimental_providerMetadata?: Record<string, any> | undefined;
+                role: "user";
+                content: string | ({
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "text";
+                    text: string;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    mimeType?: string | undefined;
+                    type: "image";
+                    image: string | ArrayBuffer;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "file";
+                    mimeType: string;
+                    data: string | ArrayBuffer;
+                })[];
+            } | {
+                providerOptions?: Record<string, any> | undefined;
+                experimental_providerMetadata?: Record<string, any> | undefined;
+                role: "assistant";
+                content: string | ({
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "text";
+                    text: string;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "file";
+                    mimeType: string;
+                    data: string | ArrayBuffer;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "reasoning";
+                    text: string;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "redacted-reasoning";
+                    data: string;
+                } | {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    type: "tool-call";
+                    toolCallId: string;
+                    toolName: string;
+                    args: any;
+                })[];
+            } | {
+                providerOptions?: Record<string, any> | undefined;
+                experimental_providerMetadata?: Record<string, any> | undefined;
+                role: "tool";
+                content: {
+                    providerOptions?: Record<string, any> | undefined;
+                    experimental_providerMetadata?: Record<string, any> | undefined;
+                    args?: any;
+                    experimental_content?: ({
+                        type: "text";
+                        text: string;
+                    } | {
+                        mimeType?: string | undefined;
+                        type: "image";
+                        data: string;
+                    })[] | undefined;
+                    isError?: boolean | undefined;
+                    type: "tool-result";
+                    toolCallId: string;
+                    toolName: string;
+                    result: any;
+                }[];
+            } | {
+                providerOptions?: Record<string, any> | undefined;
+                experimental_providerMetadata?: Record<string, any> | undefined;
+                role: "system";
+                content: string;
+            })[] | undefined;
+            prompt?: string | undefined;
+            output?: any;
+            mode?: "json" | undefined;
+            schema: any;
+        } | undefined;
+    }, Promise<any>>;
+    /**
+     * Create a tool that can call this agent.
+     * @param spec The specification for the arguments to this agent.
+     *   They will be encoded as JSON and passed to the agent.
+     * @returns The agent as a tool that can be passed to other agents.
+     */
+    asTool(spec: {
+        description: string;
+        args: Validator<unknown, "required", string>;
+        contextOptions?: ContextOptions;
+        maxSteps?: number;
+    }): Tool<ConvexToZod<Validator<unknown, "required", string>>, string>;
 }
-export declare function promptOrMessagesToCoreMessages(args: {
-    system?: string;
-    prompt?: string;
-    messages?: CoreMessage[] | Omit<UIMessage, "id">[];
-}): CoreMessage[];
+export type ToolCtx = RunActionCtx & {
+    userId?: string;
+    threadId?: string;
+    messageId?: string;
+};
+/**
+ * This is a wrapper around the ai.tool function that adds support for
+ * userId and threadId to the tool, if they're called within a thread from an agent.
+ * @param tool The AI tool. See https://sdk.vercel.ai/docs/ai-sdk-core/tools-and-tool-calling
+ * @returns The same tool, but with userId and threadId args support added.
+ */
+export declare function createTool<V extends Validator<any, any, any>, RESULT>(convexTool: {
+    args: V;
+    description?: string;
+    handler: (ctx: ToolCtx, args: Infer<V>, options: ToolExecutionOptions) => PromiseLike<RESULT>;
+    ctx?: ToolCtx;
+}): Tool<ConvexToZod<V>, RESULT>;
 type TextArgs<AgentTools extends ToolSet, TOOLS extends ToolSet, T extends {
     toolChoice?: ToolChoice<TOOLS & AgentTools>;
     tools?: TOOLS;
     model: LanguageModelV1;
 }> = Omit<T, "toolChoice" | "tools" | "model"> & {
     model?: LanguageModelV1;
+    parentMessageId?: string;
 } & {
     tools?: TOOLS;
     toolChoice?: ToolChoice<{
@@ -204,30 +647,11 @@ type ObjectArgs<T extends {
 }> = Omit<T, "model"> & {
     model?: LanguageModelV1;
 } & ContextOptions & StorageOptions;
-interface Chat<AgentTools extends ToolSet> {
+interface Thread<AgentTools extends ToolSet> {
     generateText<TOOLS extends ToolSet, OUTPUT = never, OUTPUT_PARTIAL = never>(args: TextArgs<AgentTools, TOOLS, Parameters<typeof generateText<TOOLS, OUTPUT, OUTPUT_PARTIAL>>[0]>): Promise<GenerateTextResult<TOOLS & AgentTools, OUTPUT> & GenerationOutputMetadata>;
     streamText<TOOLS extends ToolSet, OUTPUT = never, PARTIAL_OUTPUT = never>(args: TextArgs<AgentTools, TOOLS, Parameters<typeof streamText<TOOLS, OUTPUT, PARTIAL_OUTPUT>>[0]>): Promise<StreamTextResult<TOOLS & AgentTools, PARTIAL_OUTPUT> & GenerationOutputMetadata>;
     generateObject<T>(args: ObjectArgs<Parameters<typeof generateObject>[0]>): Promise<GenerateObjectResult<T> & GenerationOutputMetadata>;
     streamObject<T>(args: ObjectArgs<Parameters<typeof streamObject<T>>[0]>): Promise<StreamObjectResult<DeepPartial<T>, T, never> & GenerationOutputMetadata>;
 }
-/**
- * This is a wrapper around the ai.tool function that adds support for
- * userId and chatId to the tool, if they're called within a chat from an agent.
- * @param tool The AI tool. See https://sdk.vercel.ai/docs/ai-sdk-core/tools-and-tool-calling
- * @returns The same tool, but with userId and chatId args support added.
- */
-export declare function tool<V extends Validator<any, any, any>, RESULT>(convexTool: {
-    args: V;
-    description?: string;
-    handler: (ctx: GenericActionCtx<GenericDataModel> & {
-        userId?: string;
-        chatId?: string;
-    }, args: Infer<V>, options: ToolExecutionOptions) => PromiseLike<RESULT>;
-    ctx?: GenericActionCtx<GenericDataModel> & {
-        userId?: string;
-        chatId?: string;
-    };
-}): Tool<ZodType<Infer<V>>, RESULT>;
-export declare function wrapTools(actionCtx: RunActionCtx, chatId: string, userId?: string, ...toolSets: (ToolSet | undefined)[]): ToolSet;
 export {};
 //# sourceMappingURL=index.d.ts.map