@convex-dev/agent 0.1.9-alpha.4 → 0.1.9-alpha.6

package/src/component/_generated/api.d.ts

@@ -1414,12 +1414,6 @@ export type Mounts = {
     >;
   };
   streams: {
-    abort: FunctionReference<
-      "mutation",
-      "public",
-      { reason: string; streamId: string },
-      null
-    >;
     addDelta: FunctionReference<
       "mutation",
       "public",
@@ -1575,7 +1569,7 @@ export type Mounts = {
     list: FunctionReference<
       "query",
       "public",
-      { startOrder?: number; threadId: string },
+      { threadId: string },
       Array<{
         agentName?: string;
         model?: string;

package/src/component/schema.ts

@@ -109,10 +109,6 @@ export const schema = defineSchema({
         kind: v.literal("finished"),
         endedAt: v.number(),
       }),
-      v.object({
-        kind: v.literal("aborted"),
-        reason: v.string(),
-      }),
       v.object({
         kind: v.literal("error"),
         error: v.string(),

package/src/component/streams.ts

@@ -6,7 +6,7 @@ import {
   vStreamMessage,
 } from "../validators.js";
 import { api, internal } from "./_generated/api.js";
-import type { Doc, Id } from "./_generated/dataModel.js";
+import type { Id } from "./_generated/dataModel.js";
 import {
   internalMutation,
   mutation,
@@ -102,17 +102,13 @@ export const create = mutation({
 export const list = query({
   args: {
     threadId: v.id("threads"),
-    startOrder: v.optional(v.number()),
   },
   returns: v.array(vStreamMessage),
   handler: async (ctx, args) => {
     return ctx.db
       .query("streamingMessages")
       .withIndex("threadId_state_order_stepOrder", (q) =>
-        q
-          .eq("threadId", args.threadId)
-          .eq("state.kind", "streaming")
-          .gte("order", args.startOrder ?? 0)
+        q.eq("threadId", args.threadId).eq("state.kind", "streaming")
       )
       .order("desc")
       .take(100)
@@ -133,42 +129,6 @@ export const list = query({
   },
 });
 
-export const abort = mutation({
-  args: {
-    streamId: v.id("streamingMessages"),
-    reason: v.string(),
-  },
-  returns: v.null(),
-  handler: async (ctx, args) => {
-    const stream = await ctx.db.get(args.streamId);
-    if (!stream) {
-      throw new Error(`Stream not found: ${args.streamId}`);
-    }
-    if (stream.state.kind !== "streaming") {
-      console.warn(
-        `Stream trying to abort but not currently streaming (${stream.state.kind}): ${args.streamId}`
-      );
-      return;
-    }
-    await cleanupTimeoutFn(ctx, stream);
-    await ctx.db.patch(args.streamId, {
-      state: { kind: "aborted", reason: args.reason },
-    });
-  },
-});
-
-async function cleanupTimeoutFn(
-  ctx: MutationCtx,
-  stream: Doc<"streamingMessages">
-) {
-  if (stream.state.kind === "streaming" && stream.state.timeoutFnId) {
-    const timeoutFn = await ctx.db.system.get(stream.state.timeoutFnId);
-    if (timeoutFn?.state.kind === "pending") {
-      await ctx.scheduler.cancel(stream.state.timeoutFnId);
-    }
-  }
-}
-
 export const finish = mutation({
   args: {
     streamId: v.id("streamingMessages"),
@@ -189,7 +149,12 @@ export const finish = mutation({
       );
       return;
     }
-    await cleanupTimeoutFn(ctx, stream);
+    if (stream.state.timeoutFnId) {
+      const timeoutFn = await ctx.db.system.get(stream.state.timeoutFnId);
+      if (timeoutFn?.state.kind === "pending") {
+        await ctx.scheduler.cancel(stream.state.timeoutFnId);
+      }
+    }
     await ctx.db.patch(args.streamId, {
       state: { kind: "finished", endedAt: Date.now() },
     });

package/src/react/deltas.ts

@@ -189,6 +189,7 @@ export function applyDeltasToStreamMessage(
         contentToAdd = part;
         break;
       case "reasoning":
+        currentMessage.reasoning += part.textDelta;
         if (lastContent?.type === "reasoning") {
           lastContent.text += part.textDelta;
         } else {

package/src/react/index.ts

@@ -7,7 +7,7 @@ import {
 } from "convex/react";
 import { usePaginatedQuery } from "convex-helpers/react";
 import type { FunctionArgs } from "convex/server";
-import { useMemo, useRef, useState } from "react";
+import { useMemo, useState } from "react";
 import type { MessageDoc } from "../client/index.js";
 import type { SyncStreamsReturnValue } from "../client/types.js";
 import type { StreamArgs } from "../validators.js";
@@ -103,11 +103,7 @@ export function useThreadMessages<
       ThreadMessagesArgs<Query>,
       ThreadMessagesResult<Query>
     >,
-    !options.stream ||
-      args === "skip" ||
-      paginated.status === "LoadingFirstPage"
-      ? "skip"
-      : { ...args, startOrder: paginated.results.at(0)?.order }
+    !options.stream ? "skip" : args
   );
 
   const merged = useMemo(() => {
@@ -154,7 +150,7 @@ export function useStreamingThreadMessages<
   Query extends ThreadStreamQuery<any, any>,
 >(
   query: Query,
-  args: (ThreadMessagesArgs<Query> & { startOrder?: number }) | "skip"
+  args: ThreadMessagesArgs<Query> | "skip"
 ): Array<ThreadMessagesResult<Query>> | undefined {
   // Invariant: streamMessages[streamId] is comprised of all deltas up to the
   // cursor. There can be multiple messages in the same stream, e.g. for tool
@@ -162,10 +158,6 @@ export function useStreamingThreadMessages<
   const [streams, setStreams] = useState<
     Array<{ streamId: string; cursor: number; messages: MessageDoc[] }>
   >([]);
-  const startOrderRef = useRef<number>(0);
-  if (args !== "skip" && !startOrderRef.current && args.startOrder) {
-    startOrderRef.current = args.startOrder;
-  }
   // Get all the active streams
   const streamList = useQuery(
     query,
@@ -174,10 +166,7 @@ export function useStreamingThreadMessages<
       : ({
           ...args,
           paginationOpts: { cursor: null, numItems: 0 },
-          streamArgs: {
-            kind: "list",
-            startOrder: startOrderRef.current,
-          } as StreamArgs,
+          streamArgs: { kind: "list" } as StreamArgs,
         } as FunctionArgs<Query>)
   ) as
     | { streams: Extract<SyncStreamsReturnValue, { kind: "list" }> }

package/src/validators.ts

@@ -460,7 +460,6 @@ export const vStreamArgs = v.optional(
   v.union(
     v.object({
       kind: v.literal("list"),
-      startOrder: v.optional(v.number()),
     }),
     v.object({
       kind: v.literal("deltas"),