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

package/src/client/listMessages.ts

@@ -22,7 +22,7 @@ export async function listMessages(
     paginationOpts: PaginationOptions;
     excludeToolMessages?: boolean;
     statuses?: MessageStatus[];
-  }
+  },
 ): Promise<PaginationResult<MessageDoc>> {
   if (args.paginationOpts.numItems === 0) {
     return {

package/src/client/search.ts

@@ -16,9 +16,8 @@ import {
 const DEFAULT_VECTOR_SCORE_THRESHOLD = 0.0;
 
 export type GetEmbedding = (text: string) => Promise<{
-  vector: number[];
-  vectorModel: string;
-  vectorScoreThreshold?: number;
+  embedding: number[];
+  embeddingModel: string;
 }>;
 
 /**
@@ -44,7 +43,7 @@ export async function fetchContextMessages(
     upToAndIncludingMessageId?: string;
     contextOptions: ContextOptions;
     getEmbedding?: GetEmbedding;
-  }
+  },
 ): Promise<MessageDoc[]> {
   assert(args.userId || args.threadId, "Specify userId or threadId");
   const opts = args.contextOptions;
@@ -67,17 +66,17 @@ export async function fetchContextMessages(
         upToAndIncludingMessageId: args.upToAndIncludingMessageId,
         order: "desc",
         statuses: ["success"],
-      }
+      },
     );
     included = new Set(page.map((m) => m._id));
     contextMessages.push(
       // Reverse since we fetched in descending order
-      ...page.reverse()
+      ...page.reverse(),
     );
   }
   if (opts.searchOptions?.textSearch || opts.searchOptions?.vectorSearch) {
     const targetMessage = contextMessages.find(
-      (m) => m._id === args.upToAndIncludingMessageId
+      (m) => m._id === args.upToAndIncludingMessageId,
     )?.message;
     const messagesToSearch = targetMessage ? [targetMessage] : args.messages;
     if (!("runAction" in ctx)) {
@@ -89,11 +88,11 @@ export async function fetchContextMessages(
     assert(text, `No text to search in message ${JSON.stringify(lastMessage)}`);
     assert(
       !args.contextOptions?.searchOptions?.vectorSearch || "runAction" in ctx,
-      "You must do vector search from an action"
+      "You must do vector search from an action",
     );
     if (opts.searchOptions?.vectorSearch && !args.getEmbedding) {
       throw new Error(
-        "You must provide an embedding and embeddingModel to use vector search"
+        "You must provide an embedding and embeddingModel to use vector search",
       );
     }
     const embeddingFields = opts.searchOptions?.vectorSearch
@@ -122,20 +121,21 @@ export async function fetchContextMessages(
         vectorScoreThreshold:
           opts.searchOptions?.vectorScoreThreshold ??
           DEFAULT_VECTOR_SCORE_THRESHOLD,
-        ...embeddingFields,
-      }
+        embedding: embeddingFields?.embedding,
+        embeddingModel: embeddingFields?.embeddingModel,
+      },
     );
     // TODO: track what messages we used for context
     contextMessages.unshift(
-      ...searchMessages.filter((m) => !included?.has(m._id))
+      ...searchMessages.filter((m) => !included?.has(m._id)),
     );
   }
   // Ensure we don't include tool messages without a corresponding tool call
   return filterOutOrphanedToolMessages(
     contextMessages.sort((a, b) =>
       // Sort the raw MessageDocs by order and stepOrder
-      a.order === b.order ? a.stepOrder - b.stepOrder : a.order - b.order
-    )
+      a.order === b.order ? a.stepOrder - b.stepOrder : a.order - b.order,
+    ),
   );
 }
 

package/src/client/streaming.ts

@@ -38,7 +38,7 @@ export async function syncStreams(
     streamArgs: StreamArgs | undefined;
     // By default, only streaming messages are included.
     includeStatuses?: ("streaming" | "finished" | "aborted")[];
-  }
+  },
 ): Promise<SyncStreamsReturnValue | undefined> {
   if (!args.streamArgs) return undefined;
   if (args.streamArgs.kind === "list") {
@@ -66,7 +66,7 @@ export async function abortStream(
   component: AgentComponent,
   args: {
     reason: string;
-  } & ({ streamId: string } | { threadId: string; order: number })
+  } & ({ streamId: string } | { threadId: string; order: number }),
 ): Promise<boolean> {
   if ("streamId" in args) {
     return await ctx.runMutation(component.streams.abort, {
@@ -102,7 +102,7 @@ export async function listStreams(
     threadId: string;
     startOrder?: number;
     includeStatuses?: ("streaming" | "finished" | "aborted")[];
-  }
+  },
 ): Promise<StreamMessage[]> {
   return ctx.runQuery(component.streams.list, {
     threadId,
@@ -137,7 +137,7 @@ export function mergeTransforms<TOOLS extends ToolSet>(
   existing:
     | StreamTextTransform<TOOLS>
     | Array<StreamTextTransform<TOOLS>>
-    | undefined
+    | undefined,
 ) {
   if (!options) {
     return existing;
@@ -180,7 +180,7 @@ export class DeltaStreamer {
       order: number | undefined;
       stepOrder: number | undefined;
       abortSignal: AbortSignal | undefined;
-    }
+    },
   ) {
     this.options =
       typeof options === "boolean"
@@ -216,7 +216,7 @@ export class DeltaStreamer {
           ...omit(this.metadata, ["abortSignal"]),
           order: this.#nextOrder,
           stepOrder: this.#nextStepOrder,
-        }
+        },
       );
     }
     this.#nextParts.push(...parts);
@@ -237,7 +237,7 @@ export class DeltaStreamer {
     try {
       const success = await this.ctx.runMutation(
         this.component.streams.addDelta,
-        delta
+        delta,
       );
       if (!success) {
         this.abortController.abort();

package/src/client/types.ts

@@ -22,13 +22,14 @@ import type {
   Auth,
   Expand,
   FunctionReference,
-  GenericActionCtx,
-  GenericDataModel,
-  GenericMutationCtx,
-  GenericQueryCtx,
   StorageActionWriter,
   StorageReader,
   WithoutSystemFields,
+  FunctionArgs,
+  FunctionReturnType,
+  GenericActionCtx,
+  GenericDataModel,
+  OptionalRestArgs,
 } from "convex/server";
 import type { GenericId } from "convex/values";
 import type { Schema } from "zod";
@@ -127,7 +128,7 @@ export type UsageHandler = (
     providerMetadata: ProviderMetadata | undefined;
     model: string;
     provider: string;
-  }
+  },
 ) => void | Promise<void>;
 
 export type RawRequestResponseHandler = (
@@ -138,7 +139,7 @@ export type RawRequestResponseHandler = (
     agentName: string | undefined;
     request: LanguageModelRequestMetadata;
     response: LanguageModelResponseMetadata;
-  }
+  },
 ) => void | Promise<void>;
 
 export type AgentComponent = UseApi<Mounts>;
@@ -335,7 +336,7 @@ export interface Thread<DefaultTools extends ToolSet> {
    * Update the metadata for the thread.
    */
   updateMetadata: (
-    patch: Partial<WithoutSystemFields<ThreadDoc>>
+    patch: Partial<WithoutSystemFields<ThreadDoc>>,
   ) => Promise<ThreadDoc>;
   /**
    * This behaves like {@link generateText} from the "ai" package except that
@@ -358,7 +359,7 @@ export interface Thread<DefaultTools extends ToolSet> {
       OUTPUT,
       OUTPUT_PARTIAL
     >,
-    options?: Options
+    options?: Options,
   ): Promise<
     GenerateTextResult<TOOLS extends undefined ? DefaultTools : TOOLS, OUTPUT> &
       ThreadOutputMetadata
@@ -397,7 +398,7 @@ export interface Thread<DefaultTools extends ToolSet> {
        * iterating over the text, streaming it over HTTP, etc.
        */
       saveStreamDeltas?: boolean | StreamingOptions;
-    }
+    },
   ): Promise<
     StreamTextResult<
       TOOLS extends undefined ? DefaultTools : TOOLS,
@@ -417,7 +418,7 @@ export interface Thread<DefaultTools extends ToolSet> {
    */
   generateObject<T>(
     args: OurObjectArgs<T>,
-    options?: Options
+    options?: Options,
   ): Promise<GenerateObjectResult<T> & ThreadOutputMetadata>;
   /**
    * This behaves like {@link generateObject} from the "ai" package except that
@@ -431,7 +432,7 @@ export interface Thread<DefaultTools extends ToolSet> {
    */
   generateObject(
     args: GenerateObjectNoSchemaOptions,
-    options?: Options
+    options?: Options,
   ): Promise<GenerateObjectResult<JSONValue> & ThreadOutputMetadata>;
   /**
    * This behaves like {@link streamObject} from the "ai" package except that
@@ -445,7 +446,7 @@ export interface Thread<DefaultTools extends ToolSet> {
    */
   streamObject<T>(
     args: OurStreamObjectArgs<T>,
-    options?: Options
+    options?: Options,
   ): Promise<
     StreamObjectResult<DeepPartial<T>, T, never> & ThreadOutputMetadata
   >;
@@ -469,17 +470,24 @@ export type SyncStreamsReturnValue =
 
 /* Type utils follow */
 export type RunQueryCtx = {
-  runQuery: GenericQueryCtx<GenericDataModel>["runQuery"];
+  runQuery: <Query extends FunctionReference<"query", "internal">>(
+    query: Query,
+    args: FunctionArgs<Query>,
+  ) => Promise<FunctionReturnType<Query>>;
 };
-export type RunMutationCtx = {
-  runQuery: GenericMutationCtx<GenericDataModel>["runQuery"];
-  runMutation: GenericMutationCtx<GenericDataModel>["runMutation"];
+export type RunMutationCtx = RunQueryCtx & {
+  runMutation: <Mutation extends FunctionReference<"mutation", "internal">>(
+    mutation: Mutation,
+    args: FunctionArgs<Mutation>,
+  ) => Promise<FunctionReturnType<Mutation>>;
 };
-export type RunActionCtx = {
-  runQuery: GenericActionCtx<GenericDataModel>["runQuery"];
-  runMutation: GenericActionCtx<GenericDataModel>["runMutation"];
-  runAction: GenericActionCtx<GenericDataModel>["runAction"];
+export type RunActionCtx = RunMutationCtx & {
+  runAction<Action extends FunctionReference<"action", "internal">>(
+    action: Action,
+    args: FunctionArgs<Action>,
+  ): Promise<FunctionReturnType<Action>>;
 };
+export type UserActionCtx = GenericActionCtx<GenericDataModel>;
 export type ActionCtx = RunActionCtx & {
   auth: Auth;
   storage: StorageActionWriter;
@@ -496,9 +504,7 @@ export type OpaqueIds<T> =
       : T extends ArrayBuffer
         ? ArrayBuffer
         : T extends object
-          ? {
-              [K in keyof T]: OpaqueIds<T[K]>;
-            }
+          ? { [K in keyof T]: OpaqueIds<T[K]> }
           : T;
 
 export type UseApi<API> = Expand<{

package/src/component/apiKeys.ts

@@ -69,6 +69,6 @@ export const destroy = mutation({
     v.literal("missing"),
     v.literal("deleted"),
     v.literal("name mismatch"),
-    v.literal("must provide either apiKey or name")
+    v.literal("must provide either apiKey or name"),
   ),
 });

package/src/component/files.test.ts

@@ -22,7 +22,7 @@ describe("files", () => {
         hash,
         filename,
         mimeType: "text/plain",
-      }
+      },
     );
     expect(fileId).toBeTruthy();
     expect(returnedStorageId).toBe(storageId);

package/src/component/files.ts

@@ -23,7 +23,7 @@ export const addFile = mutation({
 
 export async function addFileHandler(
   ctx: MutationCtx,
-  args: Infer<typeof addFileArgs>
+  args: Infer<typeof addFileArgs>,
 ) {
   const existingFile = await ctx.db
     .query("files")
@@ -93,14 +93,14 @@ export const useExistingFile = mutation({
     v.object({
       fileId: v.id("files"),
       storageId: v.string(),
-    })
+    }),
   ),
 });
 
 export async function changeRefcount(
   ctx: MutationCtx,
   prev: Id<"files">[],
-  next: Id<"files">[]
+  next: Id<"files">[],
 ) {
   const prevSet = new Set(prev);
   const nextSet = new Set(next);
@@ -140,7 +140,7 @@ export const copyFile = mutation({
 
 export async function copyFileHandler(
   ctx: MutationCtx,
-  args: { fileId: Id<"files"> }
+  args: { fileId: Id<"files"> },
 ) {
   const file = await ctx.db.get(args.fileId);
   if (!file) {
@@ -194,14 +194,14 @@ export const deleteFiles = mutation({
         if (file.refcount && file.refcount > 0) {
           if (!args.force) {
             console.error(
-              `File ${fileId} has refcount ${file.refcount} > 0, skipping...`
+              `File ${fileId} has refcount ${file.refcount} > 0, skipping...`,
             );
             return null;
           }
         }
         await ctx.db.delete(fileId);
         return fileId;
-      })
+      }),
     );
     return deletedFileIds.filter((fileId) => fileId !== null);
   },

package/src/component/messages.test.ts

@@ -280,7 +280,7 @@ describe("agent", () => {
         patch: {
           message: { role: "user", content: "test" },
         },
-      })
+      }),
     ).rejects.toThrow();
   });
 
@@ -311,7 +311,7 @@ describe("agent", () => {
       {
         threadId: thread._id as Id<"threads">,
         order: "asc",
-      }
+      },
     );
     expect(remainingMessages.page).toHaveLength(1);
     expect(remainingMessages.page[0]._id).toBe(messageIds[1]);
@@ -346,7 +346,7 @@ describe("agent", () => {
       {
         threadId: thread._id as Id<"threads">,
         order: "asc",
-      }
+      },
     );
     expect(remainingMessages.page).toHaveLength(0);
   });
@@ -398,7 +398,7 @@ describe("agent", () => {
       {
         threadId: thread._id as Id<"threads">,
         order: "asc",
-      }
+      },
     );
 
     expect(remainingMessages.page).toHaveLength(3); // Should have messages from order 1 and 2
@@ -441,7 +441,7 @@ describe("agent", () => {
       {
         threadId: thread._id as Id<"threads">,
         order: "asc",
-      }
+      },
     );
 
     expect(remainingMessages.page).toHaveLength(2);
@@ -504,7 +504,7 @@ describe("agent", () => {
       {
         threadId: thread._id as Id<"threads">,
         order: "asc",
-      }
+      },
     );
     expect(remainingMessages.page).toHaveLength(1);
   });