@convex-dev/agent 0.0.12 → 0.0.14-alpha.0

package/README.md

@@ -289,8 +289,12 @@ const messages = await ctx.runQuery(
 ```
 
 ```ts
-const messages = await agent.saveMessages(ctx, { threadId, userId, messages });
+const messages = await agent.saveMessages(ctx, { threadId, userId, messages: [
+  { role: "user", content: "Hello, world!" },
+]});
 ```
+Note: you can also pass in message metadata if you want to save usage, etc.
+See the docstrings in [the implementation](./src/client/index.ts#L473).
 
 ```ts
 const messages = await agent.saveSteps(ctx, { threadId, userId, step });

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

@@ -6,7 +6,7 @@ import { z } from "zod";
 import { Mounts } from "../component/_generated/api.js";
 import { type VectorDimension } from "../component/vector/tables.js";
 import { type AIMessageWithoutId } from "../mapping.js";
-import { type CallSettings, type ProviderMetadata, type ProviderOptions, type SearchOptions } from "../validators.js";
+import { type CallSettings, MessageWithMetadata, type ProviderMetadata, type ProviderOptions, type SearchOptions, Usage } from "../validators.js";
 import type { OpaqueIds, RunActionCtx, RunMutationCtx, RunQueryCtx, UseApi } from "./types.js";
 import schema from "../component/schema.js";
 export type ThreadDoc = OpaqueIds<{
@@ -90,6 +90,14 @@ export type GenerationOutputMetadata = {
 type CoreMessageMaybeWithId = CoreMessage & {
     id?: string | undefined;
 };
+export type UsageHandler = (ctx: RunActionCtx, args: {
+    userId?: string;
+    threadId?: string;
+    usage: Usage;
+    providerMetadata?: ProviderMetadata;
+    model: string;
+    provider: string;
+}) => Promise<void>;
 export declare class Agent<AgentTools extends ToolSet> {
     component: UseApi<Mounts>;
     options: {
@@ -150,6 +158,10 @@ export declare class Agent<AgentTools extends ToolSet> {
          * This can be overridden at each generate/stream callsite.
          */
         maxRetries?: number;
+        /**
+         * The usage handler to use for this agent.
+         */
+        usageHandler?: UsageHandler;
     };
     constructor(component: UseApi<Mounts>, options: {
         /**
@@ -209,6 +221,10 @@ export declare class Agent<AgentTools extends ToolSet> {
          * This can be overridden at each generate/stream callsite.
          */
         maxRetries?: number;
+        /**
+         * The usage handler to use for this agent.
+         */
+        usageHandler?: UsageHandler;
     });
     /**
      * Start a new thread with the agent. This will have a fresh history, though if
@@ -320,12 +336,6 @@ export declare class Agent<AgentTools extends ToolSet> {
         dimension: VectorDimension;
         model: string;
     } | undefined>;
-    /**
-     * Explicitly save messages associated with the thread (& user if provided)
-     * @param ctx The ctx parameter to a mutation or action.
-     * @param args The messages and context to save
-     * @returns
-     */
     /**
      * Explicitly save messages associated with the thread (& user if provided)
      * @param ctx The ctx parameter to a mutation or action.
@@ -337,17 +347,16 @@ export declare class Agent<AgentTools extends ToolSet> {
         userId?: string;
         messages: CoreMessageMaybeWithId[];
         /**
-         * If false, it will "commit" the messages immediately.
-         * If true, it will mark them as pending until the final step has finished.
+         * 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;
-        /**
-         * The message that this is responding to.
-         */
         /**
          * The message that this is responding to.
          */
@@ -355,21 +364,13 @@ export declare class Agent<AgentTools extends ToolSet> {
         /**
          * Whether to mark all pending messages in the thread as failed.
          * This is used to recover from a failure via a retry that wipes the slate clean.
-         */
-        /**
-         * Whether to mark all pending messages in the thread as failed.
-         * This is used to recover from a failure via a retry that wipes the slate clean.
+         * Defaults to true.
          */
         failPendingSteps?: boolean;
     }): Promise<{
         lastMessageId: string;
         messageIds: string[];
     }>;
-    /**
-     * Explicitly save a "step" created by the AI SDK.
-     * @param ctx The ctx argument to a mutation or action.
-     * @param args What to save
-     */
     /**
      * Explicitly save a "step" created by the AI SDK.
      * @param ctx The ctx argument to a mutation or action.
@@ -385,15 +386,17 @@ export declare class Agent<AgentTools extends ToolSet> {
          * The step to save, possibly including multiple tool calls.
          */
         step: StepResult<TOOLS>;
+        /**
+         * The model used to generate the step.
+         * Defaults to the chat model for the Agent.
+         */
+        model?: string;
+        /**
+         * The provider of the model used to generate the step.
+         * Defaults to the chat provider for the Agent.
+         */
+        provider?: string;
     }): Promise<void>;
-    /**
-     * Commit or rollback a message that was pending.
-     * This is done automatically when saving messages by default.
-     * If creating pending messages, you can call this when the full "transaction" is done.
-     * @param ctx The ctx argument to your mutation or action.
-     * @param args What message to save. Generally the parent message sent into
-     *   the generateText call.
-     */
     /**
      * Commit or rollback a message that was pending.
      * This is done automatically when saving messages by default.
@@ -445,10 +448,11 @@ export declare class Agent<AgentTools extends ToolSet> {
         threadId?: string;
     }, args: TextArgs<AgentTools, TOOLS, Parameters<typeof streamText<TOOLS, OUTPUT, PARTIAL_OUTPUT>>[0]>): Promise<StreamTextResult<TOOLS, PARTIAL_OUTPUT> & GenerationOutputMetadata>;
     saveMessagesAndFetchContext<T extends {
+        id?: string;
         prompt?: string;
         messages?: CoreMessage[] | AIMessageWithoutId[];
         system?: string;
-    }>(ctx: RunActionCtx | RunMutationCtx, { userId, threadId, parentMessageId, system, ...args }: {
+    }>(ctx: RunActionCtx | RunMutationCtx, { id, userId, threadId, parentMessageId, system, ...args }: {
         userId: string | undefined;
         threadId: string | undefined;
         parentMessageId?: string;
@@ -486,7 +490,7 @@ export declare class Agent<AgentTools extends ToolSet> {
      * for the {@link ContextOptions} and {@link StorageOptions}.
      * @returns The result of the streamObject function.
      */
-    streamObject<T>(ctx: RunMutationCtx, { userId, threadId }: {
+    streamObject<T>(ctx: RunActionCtx, { userId, threadId }: {
         userId?: string;
         threadId?: string;
     }, args: OurStreamObjectArgs<T>): Promise<StreamObjectResult<DeepPartial<T>, T, never> & GenerationOutputMetadata>;
@@ -501,9 +505,36 @@ export declare class Agent<AgentTools extends ToolSet> {
         threadId: string;
         messageId: string;
         result: GenerateObjectResult<unknown>;
+        metadata?: Omit<MessageWithMetadata, "message">;
     }): Promise<void>;
     mergedContextOptions(opts: ContextOptions): ContextOptions;
     searchOptionsWithDefaults(contextOptions: ContextOptions, messages: CoreMessage[]): Promise<SearchOptions>;
+    /**
+     * Create a mutation that creates a thread so you can call it from a Workflow.
+     * e.g.
+     * ```ts
+     * // in convex/foo.ts
+     * export const createThread = weatherAgent.createThreadMutation();
+     *
+     * const workflow = new WorkflowManager(components.workflow);
+     * export const myWorkflow = workflow.define({
+     *   args: {},
+     *   handler: async (step) => {
+     *     const { threadId } = await step.runMutation(internal.foo.createThread);
+     *     // use the threadId to generate text, object, etc.
+     *   },
+     * });
+     * ```
+     * @returns A mutation that creates a thread.
+     */
+    createThreadMutation(): import("convex/server").RegisteredMutation<"internal", {
+        userId?: string | undefined;
+        title?: string | undefined;
+        summary?: string | undefined;
+        parentThreadIds?: string[] | undefined;
+    }, Promise<{
+        threadId: string;
+    }>>;
     /**
      * Create an action out of this agent so you can call it from workflows or other actions
      * without a wrapping function.

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/client/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAC1E,OAAO,KAAK,EACV,WAAW,EACX,WAAW,EACX,oBAAoB,EACpB,kBAAkB,EAClB,SAAS,EACT,kBAAkB,EAClB,MAAM,EACN,UAAU,EACV,kBAAkB,EAClB,gBAAgB,EAChB,iBAAiB,EACjB,IAAI,EACJ,UAAU,EACV,oBAAoB,EACpB,OAAO,EACR,MAAM,IAAI,CAAC;AACZ,OAAO,EACL,cAAc,EACd,YAAY,EAEZ,YAAY,EACZ,UAAU,EAEX,MAAM,IAAI,CAAC;AAGZ,OAAO,EAAE,KAAK,EAAK,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,MAAM,EAAE,MAAM,gCAAgC,CAAC;AACxD,OAAO,EAEL,KAAK,eAAe,EACrB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACL,KAAK,kBAAkB,EAOxB,MAAM,eAAe,CAAC;AAMvB,OAAO,EACL,KAAK,YAAY,EACjB,KAAK,gBAAgB,EACrB,KAAK,eAAe,EACpB,KAAK,aAAa,EAGnB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,KAAK,EACV,SAAS,EACT,YAAY,EACZ,cAAc,EACd,WAAW,EACX,MAAM,EACP,MAAM,YAAY,CAAC;AACpB,OAAO,MAAM,MAAM,wBAAwB,CAAC;AAE5C,MAAM,MAAM,SAAS,GAAG,SAAS,CAC/B;IAAE,GAAG,EAAE,MAAM,CAAC;IAAC,aAAa,EAAE,MAAM,CAAA;CAAE,GAAG,KAAK,CAC5C,OAAO,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CACvC,CACF,CAAC;AACF,MAAM,MAAM,UAAU,GAAG,SAAS,CAChC;IAAE,GAAG,EAAE,MAAM,CAAC;IAAC,aAAa,EAAE,MAAM,CAAA;CAAE,GAAG,KAAK,CAC5C,OAAO,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CACxC,CACF,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B;;;OAGG;IACH,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B;;;;OAIG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;OAEG;IACH,aAAa,CAAC,EAAE;QACd;;WAEG;QACH,KAAK,EAAE,MAAM,CAAC;QACd;;WAEG;QACH,UAAU,CAAC,EAAE,OAAO,CAAC;QACrB;;;WAGG;QACH,YAAY,CAAC,EAAE,OAAO,CAAC;QACvB;;;;;;WAMG;QACH,YAAY,CAAC,EAAE;YAAE,MAAM,EAAE,MAAM,CAAC;YAAC,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC;KAClD,CAAC;IACF;;;OAGG;IACH,kBAAkB,CAAC,EAAE,OAAO,CAAC;CAC9B,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B;;;;OAIG;IACH,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B,mFAAmF;IACnF,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B,2EAA2E;IAC3E,kBAAkB,CAAC,EAAE,OAAO,CAAC;CAC9B,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG;IAAE,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC;AAE9D,KAAK,sBAAsB,GAAG,WAAW,GAAG;IAAE,EAAE,CAAC,EAAE,MAAM,GAAG,SAAS,CAAA;CAAE,CAAC;AAExE,qBAAa,KAAK,CAAC,UAAU,SAAS,OAAO;IAElC,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC;IACzB,OAAO,EAAE;QACd;;;WAGG;QACH,IAAI,CAAC,EAAE,MAAM,CAAC;QACd;;;;;;WAMG;QACH,IAAI,EAAE,eAAe,CAAC;QACtB;;;;;;;;;WASG;QACH,aAAa,CAAC,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC;QACzC;;;WAGG;QACH,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB;;;;;WAKG;QACH,KAAK,CAAC,EAAE,UAAU,CAAC;QACnB;;;;WAIG;QACH,cAAc,CAAC,EAAE,cAAc,CAAC;QAChC;;;WAGG;QACH,cAAc,CAAC,EAAE,cAAc,CAAC;QAChC;;;WAGG;QACH,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB;;;WAGG;QACH,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB;gBA3DM,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC,EACzB,OAAO,EAAE;QACd;;;WAGG;QACH,IAAI,CAAC,EAAE,MAAM,CAAC;QACd;;;;;;WAMG;QACH,IAAI,EAAE,eAAe,CAAC;QACtB;;;;;;;;;WASG;QACH,aAAa,CAAC,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC;QACzC;;;WAGG;QACH,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB;;;;;WAKG;QACH,KAAK,CAAC,EAAE,UAAU,CAAC;QACnB;;;;WAIG;QACH,cAAc,CAAC,EAAE,cAAc,CAAC;QAChC;;;WAGG;QACH,cAAc,CAAC,EAAE,cAAc,CAAC;QAChC;;;WAGG;QACH,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB;;;WAGG;QACH,UAAU,CAAC,EAAE,MAAM,CAAC;KACrB;IAGH;;;;;;;;;OASG;IACG,YAAY,CAChB,GAAG,EAAE,YAAY,EACjB,IAAI,CAAC,EAAE;QACL;;;WAGG;QACH,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB;;;;WAIG;QACH,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;QAC3B;;WAEG;QACH,KAAK,CAAC,EAAE,MAAM,CAAC;QACf;;WAEG;QACH,OAAO,CAAC,EAAE,MAAM,CAAC;KAClB,GACA,OAAO,CAAC;QACT,QAAQ,EAAE,MAAM,CAAC;QACjB,MAAM,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC;KAC5B,CAAC;IACF;;;;;;;;OAQG;IACG,YAAY,CAChB,GAAG,EAAE,cAAc,EACnB,IAAI,CAAC,EAAE;QACL,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;QAC3B,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,OAAO,CAAC,EAAE,MAAM,CAAC;KAClB,GACA,OAAO,CAAC;QACT,QAAQ,EAAE,MAAM,CAAC;KAClB,CAAC;IAoCF;;;;;;;OAOG;IACH;;;;;;;OAOG;IACG,cAAc,CAClB,GAAG,EAAE,YAAY,EACjB,EACE,QAAQ,EACR,MAAM,GACP,EAAE;QACD;;WAEG;QACH;;WAEG;QACH,QAAQ,EAAE,MAAM,CAAC;QACjB;;;WAGG;QACH,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,GACA,OAAO,CAAC;QACT,MAAM,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC;KAC5B,CAAC;IAeF;;;;;;;OAOG;IACH;;;;;;;OAOG;IACG,oBAAoB,CACxB,GAAG,EAAE,WAAW,GAAG,YAAY,EAC/B,IAAI,EAAE;QACJ,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,QAAQ,EAAE,WAAW,EAAE,CAAC;QACxB,eAAe,CAAC,EAAE,MAAM,CAAC;KAC1B,GAAG,cAAc,GACjB,OAAO,CAAC,WAAW,EAAE,CAAC;IA+CnB,aAAa,CAAC,QAAQ,EAAE,WAAW,EAAE;iBAG1B,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,EAAE;mBACjB,eAAe;eACnB,MAAM;;IAkCrB;;;;;OAKG;IACH;;;;;OAKG;IACG,YAAY,CAChB,GAAG,EAAE,cAAc,EACnB,IAAI,EAAE;QACJ,QAAQ,EAAE,MAAM,CAAC;QACjB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,QAAQ,EAAE,sBAAsB,EAAE,CAAC;QACnC;;;WAGG;QACH;;;WAGG;QACH,OAAO,CAAC,EAAE,OAAO,CAAC;QAClB;;WAEG;QACH;;WAEG;QACH,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB;;;WAGG;QACH;;;WAGG;QACH,gBAAgB,CAAC,EAAE,OAAO,CAAC;KAC5B,GACA,OAAO,CAAC;QACT,aAAa,EAAE,MAAM,CAAC;QACtB,UAAU,EAAE,MAAM,EAAE,CAAC;KACtB,CAAC;IAkBF;;;;OAIG;IACH;;;;OAIG;IACG,QAAQ,CAAC,KAAK,SAAS,OAAO,EAClC,GAAG,EAAE,cAAc,EACnB,IAAI,EAAE;QACJ,QAAQ,EAAE,MAAM,CAAC;QACjB;;WAEG;QACH,SAAS,EAAE,MAAM,CAAC;QAClB;;WAEG;QACH,IAAI,EAAE,UAAU,CAAC,KAAK,CAAC,CAAC;KACzB,GACA,OAAO,CAAC,IAAI,CAAC;IAYhB;;;;;;;OAOG;IACH;;;;;;;OAOG;IACG,eAAe,CACnB,GAAG,EAAE,cAAc,EACnB,IAAI,EAAE;QACJ,QAAQ,EAAE,MAAM,CAAC;QACjB,SAAS,EAAE,MAAM,CAAC;QAClB,MAAM,EAAE;YAAE,IAAI,EAAE,OAAO,CAAC;YAAC,KAAK,EAAE,MAAM,CAAA;SAAE,GAAG;YAAE,IAAI,EAAE,SAAS,CAAA;SAAE,CAAC;KAChE,GACA,OAAO,CAAC,IAAI,CAAC;IAchB;;;;;;;;;;;OAWG;IACG,YAAY,CAChB,KAAK,SAAS,OAAO,EACrB,MAAM,GAAG,KAAK,EACd,cAAc,GAAG,KAAK,EAEtB,GAAG,EAAE,YAAY,EACjB,EACE,MAAM,EACN,QAAQ,GACT,EAAE;QACD,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB,EACD,IAAI,EAAE,QAAQ,CACZ,UAAU,EACV,KAAK,EACL,UAAU,CAAC,OAAO,YAAY,CAAC,KAAK,EAAE,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAClE,GACA,OAAO,CACR,kBAAkB,CAAC,KAAK,GAAG,UAAU,EAAE,MAAM,CAAC,GAAG,wBAAwB,CAC1E;IA6CD;;;;;;;;;;;OAWG;IACG,UAAU,CACd,KAAK,SAAS,OAAO,EACrB,MAAM,GAAG,KAAK,EACd,cAAc,GAAG,KAAK,EAEtB,GAAG,EAAE,YAAY,EACjB,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE;QAAE,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;KAAE,EAC5D,IAAI,EAAE,QAAQ,CACZ,UAAU,EACV,KAAK,EACL,UAAU,CAAC,OAAO,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAChE,GACA,OAAO,CACR,gBAAgB,CAAC,KAAK,EAAE,cAAc,CAAC,GAAG,wBAAwB,CACnE;IAkDK,2BAA2B,CAC/B,CAAC,SAAS;QACR,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,WAAW,EAAE,GAAG,kBAAkB,EAAE,CAAC;QAChD,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,EAED,GAAG,EAAE,YAAY,GAAG,cAAc,EAClC,EACE,MAAM,EACN,QAAQ,EACR,eAAe,EACf,MAAM,EACN,GAAG,IAAI,EACR,EAAE;QACD,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;QAC3B,QAAQ,EAAE,MAAM,GAAG,SAAS,CAAC;QAC7B,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,oBAAoB,CAAC,EAAE,OAAO,CAAC;QAC/B,oBAAoB,CAAC,EAAE,OAAO,CAAC;KAChC,GAAG,cAAc,GAChB,CAAC,GACF,OAAO,CAAC;QACT,IAAI,EAAE,CAAC,CAAC;QACR,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;KAC/B,CAAC;IAuCF;;;;;;;;;;;OAWG;IACG,cAAc,CAAC,CAAC,EACpB,GAAG,EAAE,YAAY,EACjB,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE;QAAE,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;KAAE,EAC5D,IAAI,EAAE,aAAa,CAAC,CAAC,CAAC,GACrB,OAAO,CAAC,oBAAoB,CAAC,CAAC,CAAC,GAAG,wBAAwB,CAAC;IAkC9D;;;;;;;;;;;OAWG;IACG,YAAY,CAAC,CAAC,EAClB,GAAG,EAAE,cAAc,EACnB,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE;QAAE,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;KAAE,EAC5D,IAAI,EAAE,mBAAmB,CAAC,CAAC,CAAC,GAC3B,OAAO,CACR,kBAAkB,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC,GAAG,wBAAwB,CACxE;IAgDD;;;;;;OAMG;IACG,UAAU,CACd,GAAG,EAAE,cAAc,EACnB,IAAI,EAAE;QACJ,QAAQ,EAAE,MAAM,CAAC;QACjB,SAAS,EAAE,MAAM,CAAC;QAClB,MAAM,EAAE,oBAAoB,CAAC,OAAO,CAAC,CAAC;KACvC,GACA,OAAO,CAAC,IAAI,CAAC;IAWhB,oBAAoB,CAAC,IAAI,EAAE,cAAc,GAAG,cAAc;IAcpD,yBAAyB,CAC7B,cAAc,EAAE,cAAc,EAC9B,QAAQ,EAAE,WAAW,EAAE,GACtB,OAAO,CAAC,aAAa,CAAC;IA+BzB;;;;;OAKG;IACH,YAAY,CAAC,IAAI,CAAC,EAAE;QAAE,cAAc,CAAC,EAAE,cAAc,CAAC;QAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;KAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;IAiB1E;;;;;;OAMG;IACH,cAAc,CAAC,CAAC,EAAE,IAAI,EAAE,aAAa,CAAC,CAAC,CAAC,GAAG;QAAE,QAAQ,CAAC,EAAE,MAAM,CAAA;KAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAoBjE;AAED,MAAM,MAAM,OAAO,GAAG,YAAY,GAAG;IACnC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AAIF,KAAK,cAAc,GAAG,CAAC,CAAC,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;AACjD,KAAK,eAAe,CAAC,UAAU,SAAS,cAAc,IAEpD,UAAU,SAAS,MAAM,CAAC,GAAG,CAAC,GAC1B,UAAU,CAAC,OAAO,CAAC,GACnB,UAAU,SAAS,CAAC,CAAC,UAAU,GAC7B,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,GACnB,KAAK,CAAC;AAEd;;;;;;GAMG;AACH,wBAAgB,UAAU,CAAC,UAAU,SAAS,cAAc,EAAE,MAAM,EAAE,CAAC,EAAE;IACvE;;;;SAIK;IACL,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;;;SAIK;IACL,IAAI,EAAE,UAAU,CAAC;IACjB;;;;;;SAMK;IACL,OAAO,EAAE,CACP,GAAG,EAAE,OAAO,EACZ,IAAI,EAAE,eAAe,CAAC,UAAU,CAAC,EACjC,OAAO,EAAE,oBAAoB,KAC1B,WAAW,CAAC,MAAM,CAAC,CAAC;IACzB,GAAG,CAAC,EAAE,OAAO,CAAC;CACf,GAAG,IAAI,CAAC,UAAU,EAAE,MAAM,CAAC,GAAG;IAC7B,OAAO,EAAE,CACP,IAAI,EAAE,eAAe,CAAC,UAAU,CAAC,EACjC,OAAO,EAAE,oBAAoB,KAC1B,WAAW,CAAC,MAAM,CAAC,CAAC;CAC1B,CAqBA;AAwBD,KAAK,QAAQ,CACX,UAAU,SAAS,OAAO,EAC1B,KAAK,SAAS,OAAO,EACrB,CAAC,SAAS;IACR,UAAU,CAAC,EAAE,UAAU,CAAC,KAAK,GAAG,UAAU,CAAC,CAAC;IAC5C,KAAK,CAAC,EAAE,KAAK,CAAC;IACd,KAAK,EAAE,eAAe,CAAC;CACxB,IACC,IAAI,CAAC,CAAC,EAAE,YAAY,GAAG,OAAO,GAAG,OAAO,CAAC,GAAG;IAC9C,KAAK,CAAC,EAAE,eAAe,CAAC;IACxB,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B,GAAG;IACF,KAAK,CAAC,EAAE,KAAK,CAAC;IACd,UAAU,CAAC,EAAE,UAAU,CAAC;SAAG,GAAG,IAAI,MAAM,KAAK,GAAG,MAAM,UAAU,GAAG,OAAO;KAAE,CAAC,CAAC;CAC/E,GAAG,cAAc,GAChB,cAAc,CAAC;AAEjB,KAAK,yBAAyB,GAAG,cAAc,GAC7C,cAAc,GACd,YAAY,GAAG;IACb,KAAK,CAAC,EAAE,eAAe,CAAC;IACxB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,WAAW,EAAE,CAAC;IACzB,uBAAuB,CAAC,EAAE,kBAAkB,CAAC;IAC7C,sBAAsB,CAAC,EAAE,iBAAiB,CAAC;IAC3C,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,6BAA6B,CAAC,EAAE,gBAAgB,CAAC;CAClD,CAAC;AAEJ,KAAK,2BAA2B,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,IAChE,yBAAyB,GAAG;IAC1B,MAAM,CAAC,EAAE,QAAQ,CAAC;IAClB,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,CAAC;IAChC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEJ,KAAK,0BAA0B,CAAC,CAAC,IAAI,yBAAyB,GAAG;IAC/D,MAAM,EAAE,OAAO,CAAC;IAChB,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,CAAC;IAChC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEF,KAAK,6BAA6B,CAAC,CAAC,SAAS,MAAM,IACjD,yBAAyB,GAAG;IAC1B,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;IACf,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,CAAC;CACjC,CAAC;AAEJ,KAAK,6BAA6B,GAAG,yBAAyB,GAAG;IAC/D,MAAM,CAAC,EAAE,SAAS,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf,CAAC;AAEF,KAAK,kBAAkB,CAAC,CAAC,IACvB,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC7B,2BAA2B,CAAC,CAAC,CAAC,GAC9B,CAAC,SAAS,KAAK,CAAC,OAAO,CAAC,GACtB,0BAA0B,CAAC,CAAC,CAAC,GAC7B,CAAC,SAAS,MAAM,GACd,6BAA6B,CAAC,CAAC,CAAC,GAChC,6BAA6B,CAAC;AAExC,KAAK,gBAAgB,CAAC,CAAC,IACrB,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC7B,2BAA2B,CAAC,CAAC,CAAC,GAC9B,CAAC,SAAS,KAAK,CAAC,OAAO,CAAC,GACtB,0BAA0B,CAAC,CAAC,CAAC,GAC7B,6BAA6B,CAAC;AAEtC,KAAK,aAAa,CAAC,CAAC,IAAI,kBAAkB,CAAC,CAAC,CAAC,GAC3C,IAAI,CAEF,UAAU,CAAC,OAAO,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,EACzC,yBAAyB,GAAG,aAAa,CAC1C,CAAC;AAEJ,KAAK,mBAAmB,CAAC,CAAC,IAAI,gBAAgB,CAAC,CAAC,CAAC,GAC/C,IAAI,CACF,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EACrC,SAAS,GAAG,UAAU,GAAG,aAAa,CACvC,CAAC;AAEJ,KAAK,oBAAoB,GAAG,wBAAwB,GAAG;IACrD,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF,UAAU,MAAM,CAAC,UAAU,SAAS,OAAO;IACzC;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;;;;;;;;OASG;IACH,YAAY,CAAC,KAAK,SAAS,OAAO,EAAE,MAAM,GAAG,KAAK,EAAE,cAAc,GAAG,KAAK,EACxE,IAAI,EAAE,QAAQ,CACZ,UAAU,EACV,KAAK,EACL,UAAU,CAAC,OAAO,YAAY,CAAC,KAAK,EAAE,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAClE,GACA,OAAO,CACR,kBAAkB,CAAC,KAAK,GAAG,UAAU,EAAE,MAAM,CAAC,GAAG,oBAAoB,CACtE,CAAC;IAEF;;;;;;;;;OASG;IACH,UAAU,CAAC,KAAK,SAAS,OAAO,EAAE,MAAM,GAAG,KAAK,EAAE,cAAc,GAAG,KAAK,EACtE,IAAI,EAAE,QAAQ,CACZ,UAAU,EACV,KAAK,EACL,UAAU,CAAC,OAAO,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAChE,GACA,OAAO,CACR,gBAAgB,CAAC,KAAK,GAAG,UAAU,EAAE,cAAc,CAAC,GAAG,oBAAoB,CAC5E,CAAC;IACF;;;;;;;;;OASG;IACH,cAAc,CAAC,CAAC,EACd,IAAI,EAAE,aAAa,CAAC,CAAC,CAAC,GACrB,OAAO,CAAC,oBAAoB,CAAC,CAAC,CAAC,GAAG,oBAAoB,CAAC,CAAC;IAC3D;;;;;;;;;OASG;IACH,cAAc,CACZ,IAAI,EAAE,6BAA6B,GAClC,OAAO,CAAC,oBAAoB,CAAC,SAAS,CAAC,GAAG,oBAAoB,CAAC,CAAC;IACnE;;;;;;;;;OASG;IACH,YAAY,CAAC,CAAC,EACZ,IAAI,EAAE,mBAAmB,CAAC,CAAC,CAAC,GAC3B,OAAO,CACR,kBAAkB,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC,GAAG,oBAAoB,CACpE,CAAC;CACH"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/client/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAC1E,OAAO,KAAK,EACV,WAAW,EACX,WAAW,EACX,oBAAoB,EACpB,kBAAkB,EAClB,SAAS,EACT,kBAAkB,EAClB,MAAM,EACN,UAAU,EACV,kBAAkB,EAClB,gBAAgB,EAChB,iBAAiB,EACjB,IAAI,EACJ,UAAU,EACV,oBAAoB,EACpB,OAAO,EACR,MAAM,IAAI,CAAC;AACZ,OAAO,EACL,cAAc,EACd,YAAY,EACZ,YAAY,EACZ,UAAU,EAEX,MAAM,IAAI,CAAC;AAGZ,OAAO,EAAE,KAAK,EAAK,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,MAAM,EAAE,MAAM,gCAAgC,CAAC;AACxD,OAAO,EAEL,KAAK,eAAe,EACrB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACL,KAAK,kBAAkB,EAOxB,MAAM,eAAe,CAAC;AAOvB,OAAO,EACL,KAAK,YAAY,EACjB,mBAAmB,EACnB,KAAK,gBAAgB,EACrB,KAAK,eAAe,EACpB,KAAK,aAAa,EAClB,KAAK,EAGN,MAAM,kBAAkB,CAAC;AAC1B,OAAO,KAAK,EACV,SAAS,EACT,YAAY,EACZ,cAAc,EACd,WAAW,EACX,MAAM,EACP,MAAM,YAAY,CAAC;AACpB,OAAO,MAAM,MAAM,wBAAwB,CAAC;AAE5C,MAAM,MAAM,SAAS,GAAG,SAAS,CAC/B;IAAE,GAAG,EAAE,MAAM,CAAC;IAAC,aAAa,EAAE,MAAM,CAAA;CAAE,GAAG,KAAK,CAC5C,OAAO,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CACvC,CACF,CAAC;AACF,MAAM,MAAM,UAAU,GAAG,SAAS,CAChC;IAAE,GAAG,EAAE,MAAM,CAAC;IAAC,aAAa,EAAE,MAAM,CAAA;CAAE,GAAG,KAAK,CAC5C,OAAO,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CACxC,CACF,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B;;;OAGG;IACH,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B;;;;OAIG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB;;OAEG;IACH,aAAa,CAAC,EAAE;QACd;;WAEG;QACH,KAAK,EAAE,MAAM,CAAC;QACd;;WAEG;QACH,UAAU,CAAC,EAAE,OAAO,CAAC;QACrB;;;WAGG;QACH,YAAY,CAAC,EAAE,OAAO,CAAC;QACvB;;;;;;WAMG;QACH,YAAY,CAAC,EAAE;YAAE,MAAM,EAAE,MAAM,CAAC;YAAC,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC;KAClD,CAAC;IACF;;;OAGG;IACH,kBAAkB,CAAC,EAAE,OAAO,CAAC;CAC9B,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B;;;;OAIG;IACH,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B,mFAAmF;IACnF,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B,2EAA2E;IAC3E,kBAAkB,CAAC,EAAE,OAAO,CAAC;CAC9B,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG;IAAE,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC;AAE9D,KAAK,sBAAsB,GAAG,WAAW,GAAG;IAAE,EAAE,CAAC,EAAE,MAAM,GAAG,SAAS,CAAA;CAAE,CAAC;AAExE,MAAM,MAAM,YAAY,GAAG,CACzB,GAAG,EAAE,YAAY,EACjB,IAAI,EAAE;IACJ,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,KAAK,CAAC;IAEb,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;IACpC,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC;CAClB,KACE,OAAO,CAAC,IAAI,CAAC,CAAC;AAEnB,qBAAa,KAAK,CAAC,UAAU,SAAS,OAAO;IAElC,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC;IACzB,OAAO,EAAE;QACd;;;WAGG;QACH,IAAI,CAAC,EAAE,MAAM,CAAC;QACd;;;;;;WAMG;QACH,IAAI,EAAE,eAAe,CAAC;QACtB;;;;;;;;;WASG;QACH,aAAa,CAAC,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC;QACzC;;;WAGG;QACH,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB;;;;;WAKG;QACH,KAAK,CAAC,EAAE,UAAU,CAAC;QACnB;;;;WAIG;QACH,cAAc,CAAC,EAAE,cAAc,CAAC;QAChC;;;WAGG;QACH,cAAc,CAAC,EAAE,cAAc,CAAC;QAChC;;;WAGG;QACH,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB;;;WAGG;QACH,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB;;WAEG;QACH,YAAY,CAAC,EAAE,YAAY,CAAC;KAC7B;gBA/DM,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC,EACzB,OAAO,EAAE;QACd;;;WAGG;QACH,IAAI,CAAC,EAAE,MAAM,CAAC;QACd;;;;;;WAMG;QACH,IAAI,EAAE,eAAe,CAAC;QACtB;;;;;;;;;WASG;QACH,aAAa,CAAC,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC;QACzC;;;WAGG;QACH,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB;;;;;WAKG;QACH,KAAK,CAAC,EAAE,UAAU,CAAC;QACnB;;;;WAIG;QACH,cAAc,CAAC,EAAE,cAAc,CAAC;QAChC;;;WAGG;QACH,cAAc,CAAC,EAAE,cAAc,CAAC;QAChC;;;WAGG;QACH,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB;;;WAGG;QACH,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB;;WAEG;QACH,YAAY,CAAC,EAAE,YAAY,CAAC;KAC7B;IAGH;;;;;;;;;OASG;IACG,YAAY,CAChB,GAAG,EAAE,YAAY,EACjB,IAAI,CAAC,EAAE;QACL;;;WAGG;QACH,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB;;;;WAIG;QACH,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;QAC3B;;WAEG;QACH,KAAK,CAAC,EAAE,MAAM,CAAC;QACf;;WAEG;QACH,OAAO,CAAC,EAAE,MAAM,CAAC;KAClB,GACA,OAAO,CAAC;QACT,QAAQ,EAAE,MAAM,CAAC;QACjB,MAAM,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC;KAC5B,CAAC;IACF;;;;;;;;OAQG;IACG,YAAY,CAChB,GAAG,EAAE,cAAc,EACnB,IAAI,CAAC,EAAE;QACL,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;QAC3B,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,OAAO,CAAC,EAAE,MAAM,CAAC;KAClB,GACA,OAAO,CAAC;QACT,QAAQ,EAAE,MAAM,CAAC;KAClB,CAAC;IAoCF;;;;;;;OAOG;IACH;;;;;;;OAOG;IACG,cAAc,CAClB,GAAG,EAAE,YAAY,EACjB,EACE,QAAQ,EACR,MAAM,GACP,EAAE;QACD;;WAEG;QACH;;WAEG;QACH,QAAQ,EAAE,MAAM,CAAC;QACjB;;;WAGG;QACH,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,GACA,OAAO,CAAC;QACT,MAAM,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC;KAC5B,CAAC;IAeF;;;;;;;OAOG;IACH;;;;;;;OAOG;IACG,oBAAoB,CACxB,GAAG,EAAE,WAAW,GAAG,YAAY,EAC/B,IAAI,EAAE;QACJ,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,QAAQ,EAAE,WAAW,EAAE,CAAC;QACxB,eAAe,CAAC,EAAE,MAAM,CAAC;KAC1B,GAAG,cAAc,GACjB,OAAO,CAAC,WAAW,EAAE,CAAC;IA+CnB,aAAa,CAAC,QAAQ,EAAE,WAAW,EAAE;iBAG1B,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,EAAE;mBACjB,eAAe;eACnB,MAAM;;IAmCrB;;;;;OAKG;IACG,YAAY,CAChB,GAAG,EAAE,cAAc,EACnB,IAAI,EAAE;QACJ,QAAQ,EAAE,MAAM,CAAC;QACjB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,QAAQ,EAAE,sBAAsB,EAAE,CAAC;QACnC;;;WAGG;QACH,QAAQ,CAAC,EAAE,IAAI,CAAC,mBAAmB,EAAE,SAAS,CAAC,EAAE,CAAC;QAClD;;;;WAIG;QACH,OAAO,CAAC,EAAE,OAAO,CAAC;QAClB;;WAEG;QACH,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB;;;;WAIG;QACH,gBAAgB,CAAC,EAAE,OAAO,CAAC;KAC5B,GACA,OAAO,CAAC;QACT,aAAa,EAAE,MAAM,CAAC;QACtB,UAAU,EAAE,MAAM,EAAE,CAAC;KACtB,CAAC;IA4BF;;;;OAIG;IACG,QAAQ,CAAC,KAAK,SAAS,OAAO,EAClC,GAAG,EAAE,cAAc,EACnB,IAAI,EAAE;QACJ,QAAQ,EAAE,MAAM,CAAC;QACjB;;WAEG;QACH,SAAS,EAAE,MAAM,CAAC;QAClB;;WAEG;QACH,IAAI,EAAE,UAAU,CAAC,KAAK,CAAC,CAAC;QACxB;;;WAGG;QACH,KAAK,CAAC,EAAE,MAAM,CAAC;QACf;;;WAGG;QACH,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB,GACA,OAAO,CAAC,IAAI,CAAC;IAwBhB;;;;;;;OAOG;IACG,eAAe,CACnB,GAAG,EAAE,cAAc,EACnB,IAAI,EAAE;QACJ,QAAQ,EAAE,MAAM,CAAC;QACjB,SAAS,EAAE,MAAM,CAAC;QAClB,MAAM,EAAE;YAAE,IAAI,EAAE,OAAO,CAAC;YAAC,KAAK,EAAE,MAAM,CAAA;SAAE,GAAG;YAAE,IAAI,EAAE,SAAS,CAAA;SAAE,CAAC;KAChE,GACA,OAAO,CAAC,IAAI,CAAC;IAchB;;;;;;;;;;;OAWG;IACG,YAAY,CAChB,KAAK,SAAS,OAAO,EACrB,MAAM,GAAG,KAAK,EACd,cAAc,GAAG,KAAK,EAEtB,GAAG,EAAE,YAAY,EACjB,EACE,MAAM,EACN,QAAQ,GACT,EAAE;QACD,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB,EACD,IAAI,EAAE,QAAQ,CACZ,UAAU,EACV,KAAK,EACL,UAAU,CAAC,OAAO,YAAY,CAAC,KAAK,EAAE,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAClE,GACA,OAAO,CACR,kBAAkB,CAAC,KAAK,GAAG,UAAU,EAAE,MAAM,CAAC,GAAG,wBAAwB,CAC1E;IAwDD;;;;;;;;;;;OAWG;IACG,UAAU,CACd,KAAK,SAAS,OAAO,EACrB,MAAM,GAAG,KAAK,EACd,cAAc,GAAG,KAAK,EAEtB,GAAG,EAAE,YAAY,EACjB,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE;QAAE,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;KAAE,EAC5D,IAAI,EAAE,QAAQ,CACZ,UAAU,EACV,KAAK,EACL,UAAU,CAAC,OAAO,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAChE,GACA,OAAO,CACR,gBAAgB,CAAC,KAAK,EAAE,cAAc,CAAC,GAAG,wBAAwB,CACnE;IA6DK,2BAA2B,CAC/B,CAAC,SAAS;QACR,EAAE,CAAC,EAAE,MAAM,CAAC;QACZ,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,WAAW,EAAE,GAAG,kBAAkB,EAAE,CAAC;QAChD,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,EAED,GAAG,EAAE,YAAY,GAAG,cAAc,EAClC,EACE,EAAE,EACF,MAAM,EACN,QAAQ,EACR,eAAe,EACf,MAAM,EACN,GAAG,IAAI,EACR,EAAE;QACD,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC;QAC3B,QAAQ,EAAE,MAAM,GAAG,SAAS,CAAC;QAC7B,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,oBAAoB,CAAC,EAAE,OAAO,CAAC;QAC/B,oBAAoB,CAAC,EAAE,OAAO,CAAC;KAChC,GAAG,cAAc,GAChB,CAAC,GACF,OAAO,CAAC;QACT,IAAI,EAAE,CAAC,CAAC;QACR,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;KAC/B,CAAC;IAyCF;;;;;;;;;;;OAWG;IACG,cAAc,CAAC,CAAC,EACpB,GAAG,EAAE,YAAY,EACjB,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE;QAAE,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;KAAE,EAC5D,IAAI,EAAE,aAAa,CAAC,CAAC,CAAC,GACrB,OAAO,CAAC,oBAAoB,CAAC,CAAC,CAAC,GAAG,wBAAwB,CAAC;IA4C9D;;;;;;;;;;;OAWG;IACG,YAAY,CAAC,CAAC,EAClB,GAAG,EAAE,YAAY,EACjB,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE;QAAE,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;KAAE,EAC5D,IAAI,EAAE,mBAAmB,CAAC,CAAC,CAAC,GAC3B,OAAO,CACR,kBAAkB,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC,GAAG,wBAAwB,CACxE;IA2DD;;;;;;OAMG;IACG,UAAU,CACd,GAAG,EAAE,cAAc,EACnB,IAAI,EAAE;QACJ,QAAQ,EAAE,MAAM,CAAC;QACjB,SAAS,EAAE,MAAM,CAAC;QAClB,MAAM,EAAE,oBAAoB,CAAC,OAAO,CAAC,CAAC;QACtC,QAAQ,CAAC,EAAE,IAAI,CAAC,mBAAmB,EAAE,SAAS,CAAC,CAAC;KACjD,GACA,OAAO,CAAC,IAAI,CAAC;IA8BhB,oBAAoB,CAAC,IAAI,EAAE,cAAc,GAAG,cAAc;IAcpD,yBAAyB,CAC7B,cAAc,EAAE,cAAc,EAC9B,QAAQ,EAAE,WAAW,EAAE,GACtB,OAAO,CAAC,aAAa,CAAC;IAgCzB;;;;;;;;;;;;;;;;;OAiBG;IACH,oBAAoB;;;;;;;;IAepB;;;;;OAKG;IACH,YAAY,CAAC,IAAI,CAAC,EAAE;QAAE,cAAc,CAAC,EAAE,cAAc,CAAC;QAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;KAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;IAiB1E;;;;;;OAMG;IACH,cAAc,CAAC,CAAC,EAAE,IAAI,EAAE,aAAa,CAAC,CAAC,CAAC,GAAG;QAAE,QAAQ,CAAC,EAAE,MAAM,CAAA;KAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAoBjE;AAED,MAAM,MAAM,OAAO,GAAG,YAAY,GAAG;IACnC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AAIF,KAAK,cAAc,GAAG,CAAC,CAAC,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;AACjD,KAAK,eAAe,CAAC,UAAU,SAAS,cAAc,IAEpD,UAAU,SAAS,MAAM,CAAC,GAAG,CAAC,GAC1B,UAAU,CAAC,OAAO,CAAC,GACnB,UAAU,SAAS,CAAC,CAAC,UAAU,GAC7B,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,GACnB,KAAK,CAAC;AAEd;;;;;;GAMG;AACH,wBAAgB,UAAU,CAAC,UAAU,SAAS,cAAc,EAAE,MAAM,EAAE,CAAC,EAAE;IACvE;;;;SAIK;IACL,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;;;SAIK;IACL,IAAI,EAAE,UAAU,CAAC;IACjB;;;;;;SAMK;IACL,OAAO,EAAE,CACP,GAAG,EAAE,OAAO,EACZ,IAAI,EAAE,eAAe,CAAC,UAAU,CAAC,EACjC,OAAO,EAAE,oBAAoB,KAC1B,WAAW,CAAC,MAAM,CAAC,CAAC;IACzB,GAAG,CAAC,EAAE,OAAO,CAAC;CACf,GAAG,IAAI,CAAC,UAAU,EAAE,MAAM,CAAC,GAAG;IAC7B,OAAO,EAAE,CACP,IAAI,EAAE,eAAe,CAAC,UAAU,CAAC,EACjC,OAAO,EAAE,oBAAoB,KAC1B,WAAW,CAAC,MAAM,CAAC,CAAC;CAC1B,CAqBA;AAwBD,KAAK,QAAQ,CACX,UAAU,SAAS,OAAO,EAC1B,KAAK,SAAS,OAAO,EACrB,CAAC,SAAS;IACR,UAAU,CAAC,EAAE,UAAU,CAAC,KAAK,GAAG,UAAU,CAAC,CAAC;IAC5C,KAAK,CAAC,EAAE,KAAK,CAAC;IACd,KAAK,EAAE,eAAe,CAAC;CACxB,IACC,IAAI,CAAC,CAAC,EAAE,YAAY,GAAG,OAAO,GAAG,OAAO,CAAC,GAAG;IAC9C,KAAK,CAAC,EAAE,eAAe,CAAC;IACxB,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B,GAAG;IACF,KAAK,CAAC,EAAE,KAAK,CAAC;IACd,UAAU,CAAC,EAAE,UAAU,CAAC;SAAG,GAAG,IAAI,MAAM,KAAK,GAAG,MAAM,UAAU,GAAG,OAAO;KAAE,CAAC,CAAC;CAC/E,GAAG,cAAc,GAChB,cAAc,CAAC;AAEjB,KAAK,yBAAyB,GAAG,cAAc,GAC7C,cAAc,GACd,YAAY,GAAG;IACb,KAAK,CAAC,EAAE,eAAe,CAAC;IACxB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,WAAW,EAAE,CAAC;IACzB,uBAAuB,CAAC,EAAE,kBAAkB,CAAC;IAC7C,sBAAsB,CAAC,EAAE,iBAAiB,CAAC;IAC3C,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,6BAA6B,CAAC,EAAE,gBAAgB,CAAC;CAClD,CAAC;AAEJ,KAAK,2BAA2B,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,IAChE,yBAAyB,GAAG;IAC1B,MAAM,CAAC,EAAE,QAAQ,CAAC;IAClB,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,CAAC;IAChC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEJ,KAAK,0BAA0B,CAAC,CAAC,IAAI,yBAAyB,GAAG;IAC/D,MAAM,EAAE,OAAO,CAAC;IAChB,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,CAAC;IAChC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B,CAAC;AAEF,KAAK,6BAA6B,CAAC,CAAC,SAAS,MAAM,IACjD,yBAAyB,GAAG;IAC1B,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;IACf,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,CAAC;CACjC,CAAC;AAEJ,KAAK,6BAA6B,GAAG,yBAAyB,GAAG;IAC/D,MAAM,CAAC,EAAE,SAAS,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf,CAAC;AAEF,KAAK,kBAAkB,CAAC,CAAC,IACvB,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC7B,2BAA2B,CAAC,CAAC,CAAC,GAC9B,CAAC,SAAS,KAAK,CAAC,OAAO,CAAC,GACtB,0BAA0B,CAAC,CAAC,CAAC,GAC7B,CAAC,SAAS,MAAM,GACd,6BAA6B,CAAC,CAAC,CAAC,GAChC,6BAA6B,CAAC;AAExC,KAAK,gBAAgB,CAAC,CAAC,IACrB,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC7B,2BAA2B,CAAC,CAAC,CAAC,GAC9B,CAAC,SAAS,KAAK,CAAC,OAAO,CAAC,GACtB,0BAA0B,CAAC,CAAC,CAAC,GAC7B,6BAA6B,CAAC;AAEtC,KAAK,aAAa,CAAC,CAAC,IAAI,kBAAkB,CAAC,CAAC,CAAC,GAC3C,IAAI,CAEF,UAAU,CAAC,OAAO,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,EACzC,yBAAyB,GAAG,aAAa,CAC1C,CAAC;AAEJ,KAAK,mBAAmB,CAAC,CAAC,IAAI,gBAAgB,CAAC,CAAC,CAAC,GAC/C,IAAI,CACF,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EACrC,SAAS,GAAG,UAAU,GAAG,aAAa,CACvC,CAAC;AAEJ,KAAK,oBAAoB,GAAG,wBAAwB,GAAG;IACrD,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF,UAAU,MAAM,CAAC,UAAU,SAAS,OAAO;IACzC;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IACjB;;;;;;;;;OASG;IACH,YAAY,CAAC,KAAK,SAAS,OAAO,EAAE,MAAM,GAAG,KAAK,EAAE,cAAc,GAAG,KAAK,EACxE,IAAI,EAAE,QAAQ,CACZ,UAAU,EACV,KAAK,EACL,UAAU,CAAC,OAAO,YAAY,CAAC,KAAK,EAAE,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAClE,GACA,OAAO,CACR,kBAAkB,CAAC,KAAK,GAAG,UAAU,EAAE,MAAM,CAAC,GAAG,oBAAoB,CACtE,CAAC;IAEF;;;;;;;;;OASG;IACH,UAAU,CAAC,KAAK,SAAS,OAAO,EAAE,MAAM,GAAG,KAAK,EAAE,cAAc,GAAG,KAAK,EACtE,IAAI,EAAE,QAAQ,CACZ,UAAU,EACV,KAAK,EACL,UAAU,CAAC,OAAO,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAChE,GACA,OAAO,CACR,gBAAgB,CAAC,KAAK,GAAG,UAAU,EAAE,cAAc,CAAC,GAAG,oBAAoB,CAC5E,CAAC;IACF;;;;;;;;;OASG;IACH,cAAc,CAAC,CAAC,EACd,IAAI,EAAE,aAAa,CAAC,CAAC,CAAC,GACrB,OAAO,CAAC,oBAAoB,CAAC,CAAC,CAAC,GAAG,oBAAoB,CAAC,CAAC;IAC3D;;;;;;;;;OASG;IACH,cAAc,CACZ,IAAI,EAAE,6BAA6B,GAClC,OAAO,CAAC,oBAAoB,CAAC,SAAS,CAAC,GAAG,oBAAoB,CAAC,CAAC;IACnE;;;;;;;;;OASG;IACH,YAAY,CAAC,CAAC,EACZ,IAAI,EAAE,mBAAmB,CAAC,CAAC,CAAC,GAC3B,OAAO,CACR,kBAAkB,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC,GAAG,oBAAoB,CACpE,CAAC;CACH"}

package/dist/commonjs/client/index.js

@@ -1,9 +1,10 @@
 import { generateObject, generateText, streamObject, streamText, tool, } from "ai";
 import { assert } from "convex-helpers";
-import { internalActionGeneric } from "convex/server";
+import { internalActionGeneric, internalMutationGeneric } from "convex/server";
+import { v } from "convex/values";
 import { validateVectorDimension, } from "../component/vector/tables.js";
-import { deserializeMessage, promptOrMessagesToCoreMessages, serializeMessageWithId, serializeNewMessagesInStep, serializeObjectResult, serializeStep, } from "../mapping.js";
-import { DEFAULT_MESSAGE_RANGE, DEFAULT_RECENT_MESSAGES, extractText, } from "../shared.js";
+import { deserializeMessage, promptOrMessagesToCoreMessages, serializeMessage, serializeNewMessagesInStep, serializeObjectResult, serializeStep, } from "../mapping.js";
+import { DEFAULT_MESSAGE_RANGE, DEFAULT_RECENT_MESSAGES, extractText, isTool, } from "../shared.js";
 import { vSafeObjectArgs, vTextArgs, } from "../validators.js";
 export class Agent {
     component;
@@ -119,7 +120,7 @@ export class Agent {
     async getEmbeddings(messages) {
         let embeddings;
         if (this.options.textEmbedding) {
-            const messageTexts = messages.map((m) => extractText(m));
+            const messageTexts = messages.map((m) => !isTool(m) && extractText(m));
             // Find the indexes of the messages that have text.
             const textIndexes = messageTexts
                 .map((t, i) => (t ? i : undefined))
@@ -131,6 +132,7 @@ export class Agent {
             const textEmbeddings = await this.options.textEmbedding.doEmbed({
                 values: messageTexts.filter((t) => !!t),
             });
+            // TODO: record usage of embeddings
             // Then assemble the embeddings into a single array with nulls for the messages without text.
             const embeddingsOrNull = Array(messages.length).fill(null);
             textIndexes.forEach((i, j) => {
@@ -148,12 +150,6 @@ export class Agent {
         }
         return embeddings;
     }
-    /**
-     * Explicitly save messages associated with the thread (& user if provided)
-     * @param ctx The ctx parameter to a mutation or action.
-     * @param args The messages and context to save
-     * @returns
-     */
     /**
      * Explicitly save messages associated with the thread (& user if provided)
      * @param ctx The ctx parameter to a mutation or action.
@@ -161,13 +157,20 @@ export class Agent {
      * @returns
      */
     async saveMessages(ctx, args) {
+        const embeddings = await this.getEmbeddings(args.messages);
         const result = await ctx.runMutation(this.component.messages.addMessages, {
             threadId: args.threadId,
             userId: args.userId,
             agentName: this.options.name,
-            model: this.options.chat.modelId,
-            messages: args.messages.map(serializeMessageWithId),
-            embeddings: await this.getEmbeddings(args.messages),
+            messages: args.messages.map((m, i) => ({
+                embedding: embeddings?.vectors[i] && {
+                    model: embeddings.model,
+                    dimension: embeddings.dimension,
+                    vector: embeddings.vectors[i],
+                },
+                ...args.metadata?.[i],
+                message: serializeMessage(m),
+            })),
             failPendingSteps: args.failPendingSteps ?? true,
             pending: args.pending ?? false,
             parentMessageId: args.parentMessageId,
@@ -177,11 +180,6 @@ export class Agent {
             messageIds: result.messages.map((m) => m._id),
         };
     }
-    /**
-     * Explicitly save a "step" created by the AI SDK.
-     * @param ctx The ctx argument to a mutation or action.
-     * @param args What to save
-     */
     /**
      * Explicitly save a "step" created by the AI SDK.
      * @param ctx The ctx argument to a mutation or action.
@@ -189,23 +187,27 @@ export class Agent {
      */
     async saveStep(ctx, args) {
         const step = serializeStep(args.step);
-        const messages = serializeNewMessagesInStep(args.step);
+        const messages = serializeNewMessagesInStep(args.step, {
+            provider: args.provider ?? this.options.chat.provider,
+            model: args.model ?? this.options.chat.modelId,
+        });
+        const embeddings = await this.getEmbeddings(messages.map((m) => m.message));
+        if (embeddings) {
+            const { model, dimension, vectors } = embeddings;
+            for (let i = 0; i < messages.length; i++) {
+                const vector = vectors[i];
+                if (vector) {
+                    messages[i].embedding = { model, dimension, vector };
+                }
+            }
+        }
         await ctx.runMutation(this.component.messages.addStep, {
             threadId: args.threadId,
             messageId: args.messageId,
             step: { step, messages },
             failPendingSteps: false,
-            embeddings: await this.getEmbeddings(messages.map((m) => m.message)),
         });
     }
-    /**
-     * Commit or rollback a message that was pending.
-     * This is done automatically when saving messages by default.
-     * If creating pending messages, you can call this when the full "transaction" is done.
-     * @param ctx The ctx argument to your mutation or action.
-     * @param args What message to save. Generally the parent message sent into
-     *   the generateText call.
-     */
     /**
      * Commit or rollback a message that was pending.
      * This is done automatically when saving messages by default.
@@ -246,13 +248,14 @@ export class Agent {
         const tools = wrapTools(toolCtx, this.options.tools, args.tools);
         const saveOutputMessages = args.saveOutputMessages ??
             this.options.storageOptions?.saveOutputMessages;
+        const model = aiArgs.model ?? this.options.chat;
         try {
             const result = (await generateText({
                 // Can be overridden
-                model: this.options.chat,
                 maxSteps: this.options.maxSteps,
                 maxRetries: this.options.maxRetries,
                 ...aiArgs,
+                model,
                 // eslint-disable-next-line @typescript-eslint/no-explicit-any
                 toolChoice: args.toolChoice,
                 tools,
@@ -264,6 +267,16 @@ export class Agent {
                             step,
                         });
                     }
+                    if (this.options.usageHandler && step.usage) {
+                        await this.options.usageHandler(ctx, {
+                            userId,
+                            threadId,
+                            model: model.modelId,
+                            provider: model.provider,
+                            usage: step.usage,
+                            providerMetadata: step.providerMetadata,
+                        });
+                    }
                     return args.onStepFinish?.(step);
                 },
             }));
@@ -299,12 +312,13 @@ export class Agent {
         const tools = wrapTools(toolCtx, this.options.tools, args.tools);
         const saveOutputMessages = args.saveOutputMessages ??
             this.options.storageOptions?.saveOutputMessages;
+        const model = aiArgs.model ?? this.options.chat;
         const result = streamText({
             // Can be overridden
-            model: this.options.chat,
             maxSteps: this.options.maxSteps,
             maxRetries: this.options.maxRetries,
             ...aiArgs,
+            model,
             // eslint-disable-next-line @typescript-eslint/no-explicit-any
             toolChoice: args.toolChoice,
             tools,
@@ -332,13 +346,23 @@ export class Agent {
                         step,
                     });
                 }
+                if (this.options.usageHandler && step.usage) {
+                    await this.options.usageHandler(ctx, {
+                        userId,
+                        threadId,
+                        model: model.modelId,
+                        provider: model.provider,
+                        usage: step.usage,
+                        providerMetadata: step.providerMetadata,
+                    });
+                }
                 return args.onStepFinish?.(step);
             },
         });
         result.messageId = messageId;
         return result;
     }
-    async saveMessagesAndFetchContext(ctx, { userId, threadId, parentMessageId, system, ...args }) {
+    async saveMessagesAndFetchContext(ctx, { id, userId, threadId, parentMessageId, system, ...args }) {
         const saveAny = args.saveAnyInputMessages ??
             this.options.storageOptions?.saveAnyInputMessages;
         const saveAll = args.saveAllInputMessages ??
@@ -353,10 +377,12 @@ export class Agent {
         });
         let messageId;
         if (threadId && saveAny !== false) {
+            const coreMessages = saveAll ? messages : messages.slice(-1);
             const saved = await this.saveMessages(ctx, {
                 threadId,
                 userId,
-                messages: saveAll ? messages : messages.slice(-1),
+                messages: coreMessages,
+                metadata: coreMessages.length === 1 ? [{ id }] : undefined,
                 pending: true,
                 // We should just fail if you pass in an ID for the message, fail those children
                 // failPendingSteps: true,
@@ -388,20 +414,31 @@ export class Agent {
      */
     async generateObject(ctx, { userId, threadId }, args) {
         const { args: aiArgs, messageId } = await this.saveMessagesAndFetchContext(ctx, { ...args, userId, threadId });
+        const model = aiArgs.model ?? this.options.chat;
         const saveOutputMessages = args.saveOutputMessages ??
             this.options.storageOptions?.saveOutputMessages;
         try {
             const result = (await generateObject({
                 // Can be overridden
-                model: this.options.chat,
                 maxRetries: this.options.maxRetries,
                 ...aiArgs,
+                model,
                 // eslint-disable-next-line @typescript-eslint/no-explicit-any
             }));
             if (threadId && messageId && saveOutputMessages !== false) {
                 await this.saveObject(ctx, { threadId, messageId, result });
             }
             result.messageId = messageId;
+            if (this.options.usageHandler && result.usage) {
+                await this.options.usageHandler(ctx, {
+                    userId,
+                    threadId,
+                    model: model.modelId,
+                    provider: model.provider,
+                    usage: result.usage,
+                    providerMetadata: result.providerMetadata,
+                });
+            }
             return result;
         }
         catch (error) {
@@ -429,14 +466,15 @@ export class Agent {
     async streamObject(ctx, { userId, threadId }, args) {
         // TODO: unify all this shared code between all the generate* and stream* functions
         const { args: aiArgs, messageId } = await this.saveMessagesAndFetchContext(ctx, { ...args, userId, threadId });
+        const model = aiArgs.model ?? this.options.chat;
         const saveOutputMessages = args.saveOutputMessages ??
             this.options.storageOptions?.saveOutputMessages;
         const stream = streamObject({
             // Can be overridden
-            model: this.options.chat,
             maxRetries: this.options.maxRetries,
             // eslint-disable-next-line @typescript-eslint/no-explicit-any
             ...aiArgs,
+            model,
             onError: async (error) => {
                 console.error("onError", error);
                 return args.onError?.(error);
@@ -460,6 +498,16 @@ export class Agent {
                         },
                     });
                 }
+                if (this.options.usageHandler && result.usage) {
+                    await this.options.usageHandler(ctx, {
+                        userId,
+                        threadId,
+                        model: model.modelId,
+                        provider: model.provider,
+                        usage: result.usage,
+                        providerMetadata: result.providerMetadata,
+                    });
+                }
                 // eslint-disable-next-line @typescript-eslint/no-explicit-any
                 return args.onFinish?.(result);
             },
@@ -475,13 +523,28 @@ export class Agent {
      * @param args The arguments to the saveObject function.
      */
     async saveObject(ctx, args) {
-        const step = serializeObjectResult(args.result);
+        const { step, messages: withoutEmbed } = serializeObjectResult(args.result, {
+            model: this.options.chat.modelId,
+            provider: this.options.chat.provider,
+        });
+        const embeddings = await this.getEmbeddings([withoutEmbed[0].message]);
+        const messages = embeddings?.vectors[0]
+            ? [
+                {
+                    ...withoutEmbed[0],
+                    embedding: {
+                        dimension: embeddings.dimension,
+                        model: embeddings.model,
+                        vector: embeddings.vectors[0],
+                    },
+                },
+            ]
+            : withoutEmbed;
         await ctx.runMutation(this.component.messages.addStep, {
             threadId: args.threadId,
             messageId: args.messageId,
             failPendingSteps: false,
-            embeddings: await this.getEmbeddings([step.messages[0].message]),
-            step,
+            step: { step, messages },
         });
     }
     mergedContextOptions(opts) {
@@ -516,10 +579,43 @@ export class Agent {
             search.vector = (await this.options.textEmbedding.doEmbed({
                 values: [text],
             })).embeddings[0];
+            // TODO: record usage of embeddings
             search.vectorModel = this.options.textEmbedding.modelId;
         }
         return search;
     }
+    /**
+     * Create a mutation that creates a thread so you can call it from a Workflow.
+     * e.g.
+     * ```ts
+     * // in convex/foo.ts
+     * export const createThread = weatherAgent.createThreadMutation();
+     *
+     * const workflow = new WorkflowManager(components.workflow);
+     * export const myWorkflow = workflow.define({
+     *   args: {},
+     *   handler: async (step) => {
+     *     const { threadId } = await step.runMutation(internal.foo.createThread);
+     *     // use the threadId to generate text, object, etc.
+     *   },
+     * });
+     * ```
+     * @returns A mutation that creates a thread.
+     */
+    createThreadMutation() {
+        return internalMutationGeneric({
+            args: {
+                userId: v.optional(v.string()),
+                parentThreadIds: v.optional(v.array(v.string())),
+                title: v.optional(v.string()),
+                summary: v.optional(v.string()),
+            },
+            handler: async (ctx, args) => {
+                const { threadId } = await this.createThread(ctx, args);
+                return { threadId };
+            },
+        });
+    }
     /**
      * Create an action out of this agent so you can call it from workflows or other actions
      * without a wrapping function.