@convex-dev/agent 0.3.1 → 0.3.2

package/src/component/messages.ts

@@ -102,12 +102,12 @@ export const deleteByOrder = mutation({
     lastOrder?: number;
     lastStepOrder?: number;
   }> => {
-    const messages = await orderedMessagesStream(
-      ctx,
-      args.threadId,
-      "asc",
-      args.startOrder,
-    )
+    const messages = await orderedMessagesStream(ctx, {
+      threadId: args.threadId,
+      sortOrder: "asc",
+      startOrder: args.startOrder,
+      startOrderBound: "gte",
+    })
       .narrow({
         lowerBound: args.startStepOrder
           ? [args.startOrder, args.startStepOrder]
@@ -312,14 +312,22 @@ export async function getMaxMessage(
   threadId: Id<"threads">,
   order?: number,
 ) {
-  return orderedMessagesStream(ctx, threadId, "desc", order).first();
+  return orderedMessagesStream(ctx, {
+    threadId,
+    sortOrder: "desc",
+    startOrder: order,
+    startOrderBound: "eq",
+  }).first();
 }
 
 function orderedMessagesStream(
   ctx: QueryCtx,
-  threadId: Id<"threads">,
-  sortOrder: "asc" | "desc",
-  order?: number,
+  args: {
+    threadId: Id<"threads">;
+    sortOrder: "asc" | "desc";
+    startOrder?: number;
+    startOrderBound?: "gte" | "eq";
+  },
 ) {
   return mergedStream(
     [true, false].flatMap((tool) =>
@@ -328,15 +336,19 @@ function orderedMessagesStream(
           .query("messages")
           .withIndex("threadId_status_tool_order_stepOrder", (q) => {
             const qq = q
-              .eq("threadId", threadId)
+              .eq("threadId", args.threadId)
               .eq("status", status)
               .eq("tool", tool);
-            if (order !== undefined) {
-              return qq.eq("order", order);
+            if (args.startOrder !== undefined) {
+              if (args.startOrderBound === "gte") {
+                return qq.gte("order", args.startOrder);
+              } else {
+                return qq.eq("order", args.startOrder);
+              }
             }
             return qq;
           })
-          .order(sortOrder),
+          .order(args.sortOrder),
       ),
     ),
     ["order", "stepOrder"],