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

package/src/component/messages.ts

@@ -46,7 +46,7 @@ function publicMessage(message: Doc<"messages">): MessageDoc {
 
 export async function deleteMessage(
   ctx: MutationCtx,
-  messageDoc: Doc<"messages">
+  messageDoc: Doc<"messages">,
 ) {
   await ctx.db.delete(messageDoc._id);
   if (messageDoc.embeddingId) {
@@ -71,14 +71,14 @@ export const deleteByIds = mutation({
           return id;
         }
         return null;
-      })
+      }),
     );
     return deletedMessageIds.filter((id) => id !== null);
   },
 });
 
 export const messageStatuses = vMessageDoc.fields.status.members.map(
-  (m) => m.value
+  (m) => m.value,
 );
 
 export const deleteByOrder = mutation({
@@ -99,7 +99,7 @@ export const deleteByOrder = mutation({
       ctx,
       args.threadId,
       "asc",
-      args.startOrder
+      args.startOrder,
     )
       .narrow({
         lowerBound: args.startStepOrder
@@ -140,7 +140,7 @@ export const addMessages = mutation({
 });
 async function addMessagesHandler(
   ctx: MutationCtx,
-  args: ObjectType<typeof addMessagesArgs>
+  args: ObjectType<typeof addMessagesArgs>,
 ) {
   let userId = args.userId;
   const threadId = args.threadId;
@@ -163,15 +163,15 @@ async function addMessagesHandler(
     const pendingMessages = await ctx.db
       .query("messages")
       .withIndex("threadId_status_tool_order_stepOrder", (q) =>
-        q.eq("threadId", threadId).eq("status", "pending")
+        q.eq("threadId", threadId).eq("status", "pending"),
       )
       .collect();
     await Promise.all(
       pendingMessages
         .filter((m) => !parentMessage || m.order === parentMessage.order)
         .map((m) =>
-          ctx.db.patch(m._id, { status: "failed", error: "Restarting" })
-        )
+          ctx.db.patch(m._id, { status: "failed", error: "Restarting" }),
+        ),
     );
   }
   let order, stepOrder;
@@ -194,7 +194,7 @@ async function addMessagesHandler(
   if (embeddings) {
     assert(
       embeddings.vectors.length === messages.length,
-      "embeddings.vectors.length must match messages.length"
+      "embeddings.vectors.length must match messages.length",
     );
   }
   for (let i = 0; i < messages.length; i++) {
@@ -242,7 +242,7 @@ async function addMessagesHandler(
 export async function getMaxMessage(
   ctx: QueryCtx,
   threadId: Id<"threads">,
-  order?: number
+  order?: number,
 ) {
   return orderedMessagesStream(ctx, threadId, "desc", order).first();
 }
@@ -251,7 +251,7 @@ function orderedMessagesStream(
   ctx: QueryCtx,
   threadId: Id<"threads">,
   sortOrder: "asc" | "desc",
-  order?: number
+  order?: number,
 ) {
   return mergedStream(
     [true, false].flatMap((tool) =>
@@ -268,10 +268,10 @@ function orderedMessagesStream(
             }
             return qq;
           })
-          .order(sortOrder)
-      )
+          .order(sortOrder),
+      ),
     ),
-    ["order", "stepOrder"]
+    ["order", "stepOrder"],
   );
 }
 
@@ -288,7 +288,7 @@ export const rollbackMessage = mutation({
       ctx,
       message.threadId,
       "asc",
-      message.order
+      message.order,
     ).collect();
     for (const m of messages) {
       if (m.status === "pending") {
@@ -347,7 +347,7 @@ export const updateMessage = mutation({
 
 async function commitMessageHandler(
   ctx: MutationCtx,
-  { messageId }: { messageId: Id<"messages"> }
+  { messageId }: { messageId: Id<"messages"> },
 ) {
   const message = await ctx.db.get(messageId);
   assert(message, `Message ${messageId} not found`);
@@ -362,10 +362,10 @@ async function commitMessageHandler(
             .eq("threadId", message.threadId)
             .eq("status", "pending")
             .eq("tool", tool)
-            .eq("order", order)
-        )
+            .eq("order", order),
+        ),
     ),
-    ["order", "stepOrder"]
+    ["order", "stepOrder"],
   ).collect();
   for (const message of messages) {
     await ctx.db.patch(message._id, { status: "success" });
@@ -390,7 +390,7 @@ export const listMessagesByThreadId = query({
       (await ctx.db.get(args.upToAndIncludingMessageId));
     assert(
       !last || last.threadId === args.threadId,
-      "upToAndIncludingMessageId must be a message in the thread"
+      "upToAndIncludingMessageId must be a message in the thread",
     );
     const toolOptions = args.excludeToolMessages ? [false] : [true, false];
     const order = args.order ?? "desc";
@@ -411,9 +411,10 @@ export const listMessagesByThreadId = query({
           .order(order)
           .filterWith(
             // We allow all messages on the same order.
-            async (m) => !last || m.order < last.order || m.order === last.order
-          )
-      )
+            async (m) =>
+              !last || m.order < last.order || m.order === last.order,
+          ),
+      ),
     );
     const messages = await mergedStream(streams, [
       "order",
@@ -422,7 +423,7 @@ export const listMessagesByThreadId = query({
       args.paginationOpts ?? {
         numItems: DEFAULT_RECENT_MESSAGES,
         cursor: null,
-      }
+      },
     );
     return { ...messages, page: messages.page.map(publicMessage) };
   },
@@ -450,14 +451,14 @@ export const searchMessages = action({
     limit: v.number(),
     vectorScoreThreshold: v.optional(v.number()),
     messageRange: v.optional(
-      v.object({ before: v.number(), after: v.number() })
+      v.object({ before: v.number(), after: v.number() }),
     ),
   },
   returns: v.array(vMessageDoc),
   handler: async (ctx, args): Promise<MessageDoc[]> => {
     assert(
       args.searchAllMessagesForUserId || args.threadId,
-      "Specify userId or threadId"
+      "Specify userId or threadId",
     );
     const limit = args.limit;
     let textSearchMessages: MessageDoc[] | undefined;
@@ -496,20 +497,20 @@ export const searchMessages = action({
             1 / ((textEmbeddingIds?.indexOf(v._id) ?? Infinity) + k),
         }))
         .sort((a, b) => b.score - a.score);
-      const vectorIds = vectorScores.slice(0, limit).map((v) => v.id);
+      const embeddingIds = vectorScores.slice(0, limit).map((v) => v.id);
       const messages: MessageDoc[] = await ctx.runQuery(
         internal.messages._fetchSearchMessages,
         {
           searchAllMessagesForUserId: args.searchAllMessagesForUserId,
           threadId: args.threadId,
-          vectorIds,
+          embeddingIds,
           textSearchMessages: textSearchMessages?.filter(
-            (m) => !vectorIds.includes(m.embeddingId! as VectorTableId)
+            (m) => !embeddingIds.includes(m.embeddingId! as VectorTableId),
           ),
           messageRange: args.messageRange ?? DEFAULT_MESSAGE_RANGE,
           beforeMessageId: args.beforeMessageId,
           limit,
-        }
+        },
       );
       return messages;
     }
@@ -520,7 +521,7 @@ export const searchMessages = action({
 export const _fetchSearchMessages = internalQuery({
   args: {
     threadId: v.optional(v.id("threads")),
-    vectorIds: v.array(vVectorId),
+    embeddingIds: v.array(vVectorId),
     searchAllMessagesForUserId: v.optional(v.string()),
     textSearchMessages: v.optional(v.array(vMessageDoc)),
     messageRange: v.object({ before: v.number(), after: v.number() }),
@@ -534,28 +535,28 @@ export const _fetchSearchMessages = internalQuery({
     const { searchAllMessagesForUserId, threadId } = args;
     assert(
       searchAllMessagesForUserId || threadId,
-      "Specify searchAllMessagesForUserId or threadId to search"
+      "Specify searchAllMessagesForUserId or threadId to search",
     );
     let messages: MessageDoc[] = (
       await Promise.all(
-        args.vectorIds.map((embeddingId) =>
+        args.embeddingIds.map((embeddingId) =>
           ctx.db
             .query("messages")
             .withIndex("embeddingId_threadId", (q) =>
               searchAllMessagesForUserId
                 ? q.eq("embeddingId", embeddingId)
-                : q.eq("embeddingId", embeddingId).eq("threadId", threadId!)
+                : q.eq("embeddingId", embeddingId).eq("threadId", threadId!),
             )
             .filter((q) =>
               q.and(
                 q.eq(q.field("status"), "success"),
                 searchAllMessagesForUserId
                   ? q.eq(q.field("userId"), searchAllMessagesForUserId)
-                  : q.eq(q.field("threadId"), threadId)
-              )
+                  : q.eq(q.field("threadId"), threadId),
+              ),
             )
-            .first()
-        )
+            .first(),
+        ),
       )
     )
       .filter(
@@ -566,7 +567,7 @@ export const _fetchSearchMessages = internalQuery({
           (!beforeMessage ||
             m.order < beforeMessage.order ||
             (m.order === beforeMessage.order &&
-              m.stepOrder < beforeMessage.stepOrder))
+              m.stepOrder < beforeMessage.stepOrder)),
       )
       .map(publicMessage);
     messages.push(...(args.textSearchMessages ?? []));
@@ -614,7 +615,7 @@ export const _fetchSearchMessages = internalQuery({
               .eq("status", "success")
               .eq("tool", false)
               .gte("order", earliest)
-              .lte("order", latest)
+              .lte("order", latest),
           )
           .collect();
         if (!ranges[searchId]) {
@@ -645,7 +646,7 @@ export const textSearch = query({
   handler: async (ctx, args) => {
     assert(
       args.searchAllMessagesForUserId || args.threadId,
-      "Specify userId or threadId"
+      "Specify userId or threadId",
     );
     const beforeMessage =
       args.beforeMessageId && (await ctx.db.get(args.beforeMessageId));
@@ -657,7 +658,7 @@ export const textSearch = query({
           ? q
               .search("text", args.text)
               .eq("userId", args.searchAllMessagesForUserId)
-          : q.search("text", args.text).eq("threadId", args.threadId!)
+          : q.search("text", args.text).eq("threadId", args.threadId!),
       )
       // Just in case tool messages slip through
       .filter((q) => {
@@ -674,7 +675,7 @@ export const textSearch = query({
           !beforeMessage ||
           m.order < beforeMessage.order ||
           (m.order === beforeMessage.order &&
-            m.stepOrder < beforeMessage.stepOrder)
+            m.stepOrder < beforeMessage.stepOrder),
       )
       .map(publicMessage);
   },

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({