@convex-dev/agent 0.1.16-alpha.0 → 0.1.16-alpha.2

package/src/component/schema.ts

@@ -118,7 +118,7 @@ export const schema = defineSchema({
       v.object({
         kind: v.literal("aborted"),
         reason: v.string(),
-      })
+      }),
     ),
   })
     // There should only be one per "order" index
@@ -197,5 +197,4 @@ export const vMessageDoc = v.object({
 });
 export type MessageDoc = Infer<typeof vMessageDoc>;
 
-
 export default schema;

package/src/component/streams.ts

@@ -51,7 +51,7 @@ export const listDeltas = query({
       v.object({
         streamId: v.id("streamingMessages"),
         cursor: v.number(),
-      })
+      }),
     ),
   },
   returns: v.array(vStreamDelta),
@@ -62,16 +62,16 @@ export const listDeltas = query({
       const streamDeltas = await ctx.db
         .query("streamDeltas")
         .withIndex("streamId_start_end", (q) =>
-          q.eq("streamId", cursor.streamId).gte("start", cursor.cursor)
+          q.eq("streamId", cursor.streamId).gte("start", cursor.cursor),
         )
         .take(
-          Math.min(MAX_DELTAS_PER_STREAM, MAX_DELTAS_PER_REQUEST - totalDeltas)
+          Math.min(MAX_DELTAS_PER_STREAM, MAX_DELTAS_PER_REQUEST - totalDeltas),
         );
       totalDeltas += streamDeltas.length;
       deltas.push(
         ...streamDeltas.map((d) =>
-          pick(d, ["streamId", "start", "end", "parts"])
-        )
+          pick(d, ["streamId", "start", "end", "parts"]),
+        ),
       );
       if (totalDeltas >= MAX_DELTAS_PER_REQUEST) {
         break;
@@ -96,7 +96,7 @@ export const create = mutation({
     const timeoutFnId = await ctx.scheduler.runAfter(
       TIMEOUT_INTERVAL,
       internal.streams.timeoutStream,
-      { streamId }
+      { streamId },
     );
     await ctx.db.patch(streamId, { state: { ...state, timeoutFnId } });
     return streamId;
@@ -112,9 +112,9 @@ export const list = query({
         v.union(
           v.literal("streaming"),
           v.literal("finished"),
-          v.literal("aborted")
-        )
-      )
+          v.literal("aborted"),
+        ),
+      ),
     ),
   },
   returns: v.array(vStreamMessage),
@@ -128,11 +128,11 @@ export const list = query({
             q
               .eq("threadId", args.threadId)
               .eq("state.kind", status)
-              .gte("order", args.startOrder ?? 0)
+              .gte("order", args.startOrder ?? 0),
           )
-          .order("desc")
+          .order("desc"),
       ),
-      ["order", "stepOrder"]
+      ["order", "stepOrder"],
     ).take(100);
 
     return messages.map((m) => ({
@@ -165,7 +165,7 @@ export const abortByOrder = mutation({
         q
           .eq("threadId", args.threadId)
           .eq("state.kind", "streaming")
-          .eq("order", args.order)
+          .eq("order", args.order),
       )
       .take(100);
     for (const stream of streams) {
@@ -189,7 +189,7 @@ export const abort = mutation({
 
 async function abortById(
   ctx: MutationCtx,
-  args: { streamId: Id<"streamingMessages">; reason: string }
+  args: { streamId: Id<"streamingMessages">; reason: string },
 ) {
   const stream = await ctx.db.get(args.streamId);
   if (!stream) {
@@ -197,7 +197,7 @@ async function abortById(
   }
   if (stream.state.kind !== "streaming") {
     console.warn(
-      `Stream trying to abort but not currently streaming (${stream.state.kind}): ${args.streamId}`
+      `Stream trying to abort but not currently streaming (${stream.state.kind}): ${args.streamId}`,
     );
     return false;
   }
@@ -210,7 +210,7 @@ async function abortById(
 
 async function cleanupTimeoutFn(
   ctx: MutationCtx,
-  stream: Doc<"streamingMessages">
+  stream: Doc<"streamingMessages">,
 ) {
   if (stream.state.kind === "streaming" && stream.state.timeoutFnId) {
     const timeoutFn = await ctx.db.system.get(stream.state.timeoutFnId);
@@ -236,7 +236,7 @@ export const finish = mutation({
     }
     if (stream.state.kind !== "streaming") {
       console.warn(
-        `Stream trying to finish but not currently streaming: ${args.streamId}`
+        `Stream trying to finish but not currently streaming: ${args.streamId}`,
       );
       return;
     }
@@ -244,7 +244,7 @@ export const finish = mutation({
     const cleanupFnId = await ctx.scheduler.runAfter(
       DELETE_STREAM_DELAY,
       api.streams.deleteStreamAsync,
-      { streamId: args.streamId }
+      { streamId: args.streamId },
     );
     await ctx.db.patch(args.streamId, {
       state: { kind: "finished", endedAt: Date.now(), cleanupFnId },
@@ -254,7 +254,7 @@ export const finish = mutation({
 
 async function heartbeatStream(
   ctx: MutationCtx,
-  args: { streamId: Id<"streamingMessages"> }
+  args: { streamId: Id<"streamingMessages"> },
 ) {
   const stream = await ctx.db.get(args.streamId);
   if (!stream) {
@@ -283,7 +283,7 @@ async function heartbeatStream(
   const timeoutFnId = await ctx.scheduler.runAfter(
     TIMEOUT_INTERVAL,
     internal.streams.timeoutStream,
-    { streamId: args.streamId }
+    { streamId: args.streamId },
   );
   await ctx.db.patch(args.streamId, {
     state: {
@@ -314,7 +314,7 @@ export const timeoutStream = internalMutation({
 
 async function deletePageForStreamId(
   ctx: MutationCtx,
-  args: { streamId: Id<"streamingMessages">; cursor?: string }
+  args: { streamId: Id<"streamingMessages">; cursor?: string },
 ) {
   const deltas = await paginator(ctx.db, schema)
     .query("streamDeltas")
@@ -339,7 +339,7 @@ async function deletePageForStreamId(
 
 export async function deleteStreamsPageForThreadId(
   ctx: MutationCtx,
-  args: { threadId: Id<"threads">; streamOrder?: number; deltaCursor?: string }
+  args: { threadId: Id<"threads">; streamOrder?: number; deltaCursor?: string },
 ) {
   const allStreamMessages =
     schema.tables.streamingMessages.validator.fields.state.members
@@ -351,8 +351,8 @@ export async function deleteStreamsPageForThreadId(
             q
               .eq("threadId", args.threadId)
               .eq("state.kind", stateKind)
-              .gte("order", args.streamOrder ?? 0)
-          )
+              .gte("order", args.streamOrder ?? 0),
+          ),
       );
   let deltaCursor = args.deltaCursor;
   const streamMessage = await mergedStream(allStreamMessages, [
@@ -417,7 +417,7 @@ export const deleteAllStreamsForThreadIdAsync = mutation({
           threadId: args.threadId,
           streamOrder: result.streamOrder,
           deltaCursor: result.deltaCursor,
-        }
+        },
       );
     } else {
       await ctx.db.delete(args.threadId);
@@ -460,7 +460,7 @@ export const deleteAllStreamsForThreadIdSync = action({
   handler: async (ctx, args) => {
     let result = await ctx.runMutation(
       internal.streams.deleteStreamsPageForThreadIdMutation,
-      args
+      args,
     );
     while (!result.isDone) {
       result = await ctx.runMutation(
@@ -469,7 +469,7 @@ export const deleteAllStreamsForThreadIdSync = action({
           ...args,
           streamOrder: result.streamOrder,
           deltaCursor: result.deltaCursor,
-        }
+        },
       );
     }
   },

package/src/component/threads.test.ts

@@ -225,7 +225,7 @@ describe("threads", () => {
       t.mutation(api.threads.updateThread, {
         threadId: thread._id as Id<"threads">,
         patch: { title: "New Title" },
-      })
+      }),
     ).rejects.toThrow();
   });
 
@@ -368,7 +368,7 @@ describe("threads", () => {
         {
           threadId: thread._id as Id<"threads">,
           cursor: currentResult.cursor,
-        }
+        },
       );
       iterations++;
     }
@@ -471,7 +471,7 @@ describe("threads", () => {
     await expect(
       t.action(api.threads.deleteAllForThreadIdSync, {
         threadId: thread._id as Id<"threads">,
-      })
+      }),
     ).resolves.not.toThrow();
 
     // Thread should be deleted
@@ -499,7 +499,7 @@ describe("threads", () => {
     await expect(
       t.action(api.threads.deleteAllForThreadIdSync, {
         threadId: thread._id as Id<"threads">,
-      })
+      }),
     ).resolves.not.toThrow();
   });
 

package/src/component/threads.ts

@@ -104,7 +104,7 @@ export const searchThreadTitles = query({
       .withSearchIndex("title", (q) =>
         args.userId
           ? q.search("title", args.query).eq("userId", args.userId ?? undefined)
-          : q.search("title", args.query)
+          : q.search("title", args.query),
       )
       .take(args.limit);
     return threads.map(publicThread);
@@ -138,7 +138,7 @@ export const deleteAllForThreadIdSync = action({
     while (true) {
       const result: DeleteThreadReturns = await ctx.runMutation(
         internal.threads._deletePageForThreadId,
-        { threadId: args.threadId, cursor, limit: args.limit }
+        { threadId: args.threadId, cursor, limit: args.limit },
       );
       if (result.isDone) {
         break;
@@ -218,12 +218,12 @@ export const deleteAllForThreadIdAsync = mutation({
 
 async function deletePageForThreadIdHandler(
   ctx: MutationCtx,
-  args: DeleteThreadArgs
+  args: DeleteThreadArgs,
 ): Promise<DeleteThreadReturns> {
   const messages = await paginator(ctx.db, schema)
     .query("messages")
     .withIndex("threadId_status_tool_order_stepOrder", (q) =>
-      q.eq("threadId", args.threadId)
+      q.eq("threadId", args.threadId),
     )
     .paginate({
       numItems: args.limit ?? 100,

package/src/component/users.test.ts

@@ -60,7 +60,7 @@ describe("users", () => {
     // Should not have duplicate users
     console.log(firstPage.page, secondPage.page);
     expect(
-      firstPage.page.every((user) => !secondPage.page.includes(user))
+      firstPage.page.every((user) => !secondPage.page.includes(user)),
     ).toBe(true);
   });
 
@@ -470,7 +470,7 @@ describe("users", () => {
 
     // Try to delete for a user that doesn't exist
     await expect(
-      t.action(api.users.deleteAllForUserId, { userId: "nonexistentUser" })
+      t.action(api.users.deleteAllForUserId, { userId: "nonexistentUser" }),
     ).resolves.not.toThrow();
 
     // Should complete successfully without errors

package/src/component/users.ts

@@ -59,7 +59,7 @@ export const deleteAllForUserId = action({
           streamsInProgress,
           streamOrder,
           deltaCursor,
-        }
+        },
       );
       messagesCursor = result.messagesCursor;
       threadInProgress = result.threadInProgress;
@@ -121,7 +121,7 @@ export const _deleteAllForUserIdAsync = internalMutation({
 
 async function deleteAllForUserIdAsyncHandler(
   ctx: MutationCtx,
-  args: DeleteAllArgs
+  args: DeleteAllArgs,
 ): Promise<boolean> {
   const result = await deletePageForUserId(ctx, args);
   if (!result.isDone) {
@@ -145,7 +145,7 @@ export const _deletePageForUserId = internalMutation({
 });
 async function deletePageForUserId(
   ctx: MutationCtx,
-  args: DeleteAllArgs
+  args: DeleteAllArgs,
 ): Promise<DeleteAllReturns> {
   let threadInProgress: Id<"threads"> | null = args.threadInProgress;
   let threadsCursor: string | null = args.threadsCursor;
@@ -189,7 +189,7 @@ async function deletePageForUserId(
     const messages = await paginator(ctx.db, schema)
       .query("messages")
       .withIndex("threadId_status_tool_order_stepOrder", (q) =>
-        q.eq("threadId", threadInProgress!)
+        q.eq("threadId", threadInProgress!),
       )
       .order("desc")
       .paginate({

package/src/component/vector/index.ts

@@ -40,7 +40,7 @@ export const paginate = query({
         args.table
           ? // eslint-disable-next-line @typescript-eslint/no-explicit-any
             (q.eq("model", args.targetModel) as any).eq("table", args.table)
-          : q.eq("model", args.targetModel)
+          : q.eq("model", args.targetModel),
       )
       .paginate({
         cursor: args.cursor ?? null,
@@ -77,10 +77,10 @@ export const deleteBatchForThread = mutation({
             q
               .eq("model", args.model)
               .eq("table", table)
-              .eq("threadId", args.threadId)
-          )
+              .eq("threadId", args.threadId),
+          ),
       ),
-      ["threadId"]
+      ["threadId"],
     ).paginate({
       cursor: args.cursor ?? null,
       numItems: args.limit,
@@ -101,7 +101,7 @@ export const insertBatch = mutation({
       v.object({
         ...vEmbeddingsWithoutDenormalizedFields.fields,
         messageId: v.optional(v.id("messages")),
-      })
+      }),
     ),
   },
   returns: v.array(vVectorId),
@@ -113,7 +113,7 @@ export const insertBatch = mutation({
           await ctx.db.patch(messageId, { embeddingId });
         }
         return embeddingId;
-      })
+      }),
     );
   },
 });
@@ -121,7 +121,7 @@ export const insertBatch = mutation({
 export async function insertVector(
   ctx: MutationCtx,
   dimension: VectorDimension,
-  v: EmbeddingsWithoutDenormalizedFields
+  v: EmbeddingsWithoutDenormalizedFields,
 ) {
   return ctx.db.insert(getVectorTableName(dimension), {
     ...v,
@@ -143,7 +143,7 @@ export function searchVectors(
     threadId?: Id<"threads">;
     searchAllMessagesForUserId?: string;
     limit?: number;
-  }
+  },
 ) {
   const tableName = getVectorTableName(args.dimension);
   return ctx.vectorSearch(tableName, "vector", {
@@ -176,7 +176,7 @@ export const updateBatch = mutation({
         // deleting from one table and inserting into another.
         // However this requires updating all the messages that reference
         // the vector.
-      })
+      }),
     ),
   },
   returns: v.null(),
@@ -186,8 +186,8 @@ export const updateBatch = mutation({
         ctx.db.patch(embedding.id, {
           model: embedding.model,
           vector: embedding.vector,
-        })
-      )
+        }),
+      ),
     );
   },
 });

package/src/component/vector/tables.ts

@@ -31,7 +31,7 @@ const embeddings = {
 };
 
 export const vEmbeddingsWithoutDenormalizedFields = v.object(
-  omit(embeddings, ["model_table_userId", "model_table_threadId"])
+  omit(embeddings, ["model_table_userId", "model_table_threadId"]),
 );
 export type EmbeddingsWithoutDenormalizedFields = Infer<
   typeof vEmbeddingsWithoutDenormalizedFields
@@ -71,17 +71,17 @@ export const VectorDimensions = [
   128, 256, 512, 768, 1024, 1408, 1536, 2048, 3072, 4096,
 ] as const;
 export function validateVectorDimension(
-  dimension: number
+  dimension: number,
 ): asserts dimension is VectorDimension {
   if (!VectorDimensions.includes(dimension as VectorDimension)) {
     throw new Error(
-      `Unsupported vector dimension${dimension}. Supported: ${VectorDimensions.join(", ")}`
+      `Unsupported vector dimension${dimension}. Supported: ${VectorDimensions.join(", ")}`,
     );
   }
 }
 export type VectorDimension = (typeof VectorDimensions)[number];
 export const VectorTableNames = VectorDimensions.map(
-  (d) => `embeddings_${d}`
+  (d) => `embeddings_${d}`,
 ) as `embeddings_${(typeof VectorDimensions)[number]}`[];
 export type VectorTableName = (typeof VectorTableNames)[number];
 export type VectorTableId = GenericId<(typeof VectorTableNames)[number]>;
@@ -89,7 +89,7 @@ export type VectorTableId = GenericId<(typeof VectorTableNames)[number]>;
 export const vVectorDimension = literals(...VectorDimensions);
 export const vVectorTableName = literals(...VectorTableNames);
 export const vVectorId = v.union(
-  ...VectorTableNames.map((name) => v.id(name))
+  ...VectorTableNames.map((name) => v.id(name)),
 ) as VUnion<
   GenericId<(typeof VectorTableNames)[number]>,
   VId<(typeof VectorTableNames)[number]>[]
@@ -117,7 +117,7 @@ const tables: {
   VectorDimensions.map((dimensions) => [
     `embeddings_${dimensions}`,
     table(dimensions),
-  ])
+  ]),
 ) as Record<
   `embeddings_${(typeof VectorDimensions)[number]}`,
   VectorTable<(typeof VectorDimensions)[number]>

package/src/mapping.test.ts

@@ -89,7 +89,7 @@ describe("mapping", () => {
     const bigArr = new Uint8Array(1024 * 65).fill(1);
     const ab = bigArr.buffer.slice(
       bigArr.byteOffset,
-      bigArr.byteOffset + bigArr.byteLength
+      bigArr.byteOffset + bigArr.byteLength,
     );
     let called = false;
     const ctx = {
@@ -117,7 +117,7 @@ describe("mapping", () => {
     const { content: ser, fileIds } = await serializeContent(
       ctx,
       component,
-      content
+      content,
     );
     expect(called).toBe(true);
     expect(fileIds).toEqual(["file-123"]);
@@ -125,7 +125,7 @@ describe("mapping", () => {
     const serArr = ser as SerializedContent;
     expect(typeof (serArr as { data: unknown }[])[0].data).toBe("string");
     expect((serArr as { data: unknown }[])[0].data as string).toMatch(
-      /^https?:\/\//
+      /^https?:\/\//,
     );
   });
 
@@ -133,7 +133,7 @@ describe("mapping", () => {
     const arr = new Uint8Array([1, 2, 3, 4, 5]);
     const ab = arr.buffer.slice(
       arr.byteOffset,
-      arr.byteOffset + arr.byteLength
+      arr.byteOffset + arr.byteLength,
     );
     const ctx = {
       runAction: async () => undefined,

package/src/mapping.ts

@@ -20,7 +20,6 @@ import type { ActionCtx, AgentComponent } from "./client/types.js";
 import type { RunMutationCtx } from "./client/types.js";
 import { MAX_FILE_SIZE, storeFile } from "./client/files.js";
 
-
 export type AIMessageWithoutId = Omit<AIMessage, "id">;
 
 export type SerializeUrlsAndUint8Arrays<T> = T extends URL
@@ -42,13 +41,13 @@ export type SerializedMessage = SerializeUrlsAndUint8Arrays<CoreMessage>;
 export async function serializeMessage(
   ctx: ActionCtx | RunMutationCtx,
   component: AgentComponent,
-  messageWithId: CoreMessage & { id?: string }
+  messageWithId: CoreMessage & { id?: string },
 ): Promise<{ message: SerializedMessage; fileIds?: string[] }> {
   const { id: _, experimental_providerMetadata, ...message } = messageWithId;
   const { content, fileIds } = await serializeContent(
     ctx,
     component,
-    message.content
+    message.content,
   );
   return {
     message: {
@@ -72,7 +71,7 @@ export async function serializeNewMessagesInStep<TOOLS extends ToolSet>(
   ctx: ActionCtx,
   component: AgentComponent,
   step: StepResult<TOOLS>,
-  metadata: { model: string; provider: string }
+  metadata: { model: string; provider: string },
 ): Promise<MessageWithMetadata[]> {
   // If there are tool results, there's another message with the tool results
   // ref: https://github.com/vercel/ai/blob/main/packages/ai/core/generate-text/to-response-messages.ts
@@ -98,7 +97,7 @@ export async function serializeNewMessagesInStep<TOOLS extends ToolSet>(
       const { message, fileIds } = await serializeMessage(
         ctx,
         component,
-        messageWithId
+        messageWithId,
       );
       return {
         message,
@@ -109,7 +108,7 @@ export async function serializeNewMessagesInStep<TOOLS extends ToolSet>(
         text: step.text,
         fileIds,
       };
-    })
+    }),
   );
   return messages;
 }
@@ -118,7 +117,7 @@ export async function serializeObjectResult(
   ctx: ActionCtx,
   component: AgentComponent,
   result: GenerateObjectResult<unknown>,
-  metadata: { model: string; provider: string }
+  metadata: { model: string; provider: string },
 ): Promise<{ messages: MessageWithMetadata[] }> {
   const text = JSON.stringify(result.object);
 
@@ -147,7 +146,7 @@ export async function serializeObjectResult(
 export async function serializeContent(
   ctx: ActionCtx | RunMutationCtx,
   component: AgentComponent,
-  content: Content
+  content: Content,
 ): Promise<{ content: SerializedContent; fileIds?: string[] }> {
   if (typeof content === "string") {
     return { content };
@@ -166,7 +165,9 @@ export async function serializeContent(
             const { file } = await storeFile(
               ctx,
               component,
-              new Blob([image], { type: part.mimeType || guessMimeType(image) })
+              new Blob([image], {
+                type: part.mimeType || guessMimeType(image),
+              }),
             );
             image = file.url;
             fileIds.push(file.fileId);
@@ -179,7 +180,7 @@ export async function serializeContent(
             const { file } = await storeFile(
               ctx,
               component,
-              new Blob([data], { type: part.mimeType })
+              new Blob([data], { type: part.mimeType }),
             );
             data = file.url;
             fileIds.push(file.fileId);
@@ -192,7 +193,7 @@ export async function serializeContent(
         default:
           return part;
       }
-    })
+    }),
   );
   return {
     content: serialized as SerializedContent,
@@ -289,7 +290,7 @@ export function guessMimeType(buf: ArrayBuffer | string): string {
  * @returns The serialized data as an ArrayBuffer or the URL as a string.
  */
 export function serializeDataOrUrl(
-  dataOrUrl: DataContent | URL
+  dataOrUrl: DataContent | URL,
 ): ArrayBuffer | string {
   if (typeof dataOrUrl === "string") {
     return dataOrUrl;
@@ -302,12 +303,12 @@ export function serializeDataOrUrl(
   }
   return dataOrUrl.buffer.slice(
     dataOrUrl.byteOffset,
-    dataOrUrl.byteOffset + dataOrUrl.byteLength
+    dataOrUrl.byteOffset + dataOrUrl.byteLength,
   ) as ArrayBuffer;
 }
 
 export function deserializeUrl(
-  urlOrString: string | ArrayBuffer
+  urlOrString: string | ArrayBuffer,
 ): URL | DataContent {
   if (typeof urlOrString === "string") {
     if (
@@ -323,7 +324,7 @@ export function deserializeUrl(
 
 export function toUIFilePart(part: ImagePart | FilePart): FileUIPart {
   const dataOrUrl = serializeDataOrUrl(
-    part.type === "image" ? part.image : part.data
+    part.type === "image" ? part.image : part.data,
   );
 
   return {
@@ -346,7 +347,7 @@ export function promptOrMessagesToCoreMessages(args: {
   const messages: CoreMessage[] = [];
   assert(
     args.prompt || args.messages || args.promptMessageId,
-    "messages or prompt or promptMessageId is required"
+    "messages or prompt or promptMessageId is required",
   );
   if (args.messages) {
     if (
@@ -357,7 +358,7 @@ export function promptOrMessagesToCoreMessages(args: {
           (m.role === "data" || // UI-only role
             "toolInvocations" in m || // UI-specific field
             "parts" in m || // UI-specific field
-            "experimental_attachments" in m)
+            "experimental_attachments" in m),
       )
     ) {
       messages.push(...convertToCoreMessages(args.messages as AIMessage[]));