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

package/src/react/deltas.test.ts

@@ -10,7 +10,7 @@ import { omit } from "convex-helpers";
 function makeStreamMessage(
   streamId: string,
   order: number,
-  stepOrder: number
+  stepOrder: number,
 ): StreamMessage {
   return {
     streamId,
@@ -23,7 +23,7 @@ function makeDelta(
   streamId: string,
   start: number,
   end: number,
-  parts: TextStreamPart[]
+  parts: TextStreamPart[],
 ): StreamDelta {
   return {
     streamId,
@@ -44,7 +44,7 @@ describe("mergeDeltas", () => {
       "thread1",
       streamMessages,
       [],
-      deltas
+      deltas,
     );
     expect(messages).toHaveLength(1);
     expect(messages[0].text).toBe("Hello");
@@ -66,7 +66,7 @@ describe("mergeDeltas", () => {
       "thread1",
       streamMessages,
       [],
-      deltas
+      deltas,
     );
     expect(messages).toHaveLength(1);
     expect(messages[0].text).toBe("Hello World!");
@@ -99,7 +99,7 @@ describe("mergeDeltas", () => {
       "thread1",
       streamMessages,
       [],
-      deltas
+      deltas,
     );
     expect(messages).toHaveLength(2);
     expect(messages[0].message?.role).toBe("assistant");
@@ -134,7 +134,7 @@ describe("mergeDeltas", () => {
       "thread1",
       streamMessages,
       [],
-      deltas
+      deltas,
     );
     expect(messages).toHaveLength(0);
     expect(changed).toBe(false);
@@ -195,7 +195,7 @@ describe("mergeDeltas", () => {
     expect(messages).toHaveLength(1);
     if (Array.isArray(messages[0].message?.content)) {
       const reasoningParts = messages[0].message.content.filter(
-        (p) => p.type === "reasoning"
+        (p) => p.type === "reasoning",
       );
       expect(reasoningParts).toHaveLength(1);
       expect(reasoningParts[0].text).toBe("I'm thinking... Still thinking...");
@@ -216,7 +216,7 @@ describe("mergeDeltas", () => {
       "thread1",
       streamMessage,
       undefined,
-      deltas
+      deltas,
     );
     expect(result.messages).toHaveLength(1);
     expect(result.messages[0].text).toBe("Hello World!");
@@ -225,7 +225,7 @@ describe("mergeDeltas", () => {
       "thread1",
       streamMessage,
       result,
-      deltas
+      deltas,
     );
     expect(result.messages).toHaveLength(1);
     expect(result.messages[0].text).toBe("Hello World!");
@@ -238,7 +238,7 @@ describe("mergeDeltas", () => {
       "thread1",
       streamMessage,
       result,
-      moreDeltas
+      moreDeltas,
     );
     expect(changed).toBe(true);
     expect(result.messages).toHaveLength(1);
@@ -248,7 +248,7 @@ describe("mergeDeltas", () => {
       "thread1",
       streamMessage,
       result,
-      moreDeltas
+      moreDeltas,
     );
     expect(changed).toBe(false);
     expect(result.messages).toHaveLength(1);
@@ -280,27 +280,27 @@ describe("mergeDeltas", () => {
       "thread1",
       streamMessages,
       [],
-      deltas
+      deltas,
     );
     const [messages2, streams2, changed2] = mergeDeltas(
       "thread1",
       streamMessages,
       [],
-      deltas
+      deltas,
     );
     expect(messages1.map((m) => omit(m, ["_creationTime"]))).toEqual(
-      messages2.map((m) => omit(m, ["_creationTime"]))
+      messages2.map((m) => omit(m, ["_creationTime"])),
     );
     expect(
       streams1.map((s) => ({
         ...s,
         messages: s.messages.map((m) => omit(m, ["_creationTime"])),
-      }))
+      })),
     ).toEqual(
       streams2.map((s) => ({
         ...s,
         messages: s.messages.map((m) => omit(m, ["_creationTime"])),
-      }))
+      })),
     );
     expect(changed1).toBe(changed2);
     // Inputs should remain unchanged

package/src/react/deltas.ts

@@ -20,7 +20,7 @@ export function mergeDeltas(
     cursor: number;
     messages: MessageDoc[];
   }>,
-  allDeltas: StreamDelta[]
+  allDeltas: StreamDelta[],
 ): [
   MessageDoc[],
   Array<{ streamId: string; cursor: number; messages: MessageDoc[] }>,
@@ -35,16 +35,16 @@ export function mergeDeltas(
   let changed = false;
   for (const streamMessage of streamMessages) {
     const deltas = allDeltas.filter(
-      (d) => d.streamId === streamMessage.streamId
+      (d) => d.streamId === streamMessage.streamId,
     );
     const existing = existingStreams.find(
-      (s) => s.streamId === streamMessage.streamId
+      (s) => s.streamId === streamMessage.streamId,
     );
     const [newStream, messageChanged] = applyDeltasToStreamMessage(
       threadId,
       streamMessage,
       existing,
-      deltas
+      deltas,
     );
     newStreams.push(newStream);
     if (messageChanged) changed = true;
@@ -69,7 +69,7 @@ export function applyDeltasToStreamMessage(
   existing:
     | { streamId: string; cursor: number; messages: MessageDoc[] }
     | undefined,
-  deltas: StreamDelta[]
+  deltas: StreamDelta[],
 ): [{ streamId: string; cursor: number; messages: MessageDoc[] }, boolean] {
   let changed = false;
   let cursor = existing?.cursor ?? 0;
@@ -82,17 +82,17 @@ export function applyDeltasToStreamMessage(
     if (cursor !== delta.start) {
       if (cursor >= delta.end) {
         console.debug(
-          `Got duplicate delta for stream ${delta.streamId} at ${delta.start}`
+          `Got duplicate delta for stream ${delta.streamId} at ${delta.start}`,
         );
         continue;
       } else if (cursor < delta.start) {
         console.warn(
-          `Got delta for stream ${delta.streamId} that has a gap ${cursor} -> ${delta.start}`
+          `Got delta for stream ${delta.streamId} that has a gap ${cursor} -> ${delta.start}`,
         );
         continue;
       } else {
         throw new Error(
-          `Got unexpected delta for stream ${delta.streamId}: delta: ${delta.start} -> ${delta.end} existing cursor: ${cursor}`
+          `Got unexpected delta for stream ${delta.streamId}: delta: ${delta.start} -> ${delta.end} existing cursor: ${cursor}`,
         );
       }
     }
@@ -129,7 +129,7 @@ export function applyDeltasToStreamMessage(
       threadId,
       streamMessage,
       parts[0]!,
-      existingMessages.length
+      existingMessages.length,
     );
     parts = parts.slice(1);
     currentMessage = newMessage;
@@ -153,7 +153,7 @@ export function applyDeltasToStreamMessage(
         threadId,
         streamMessage,
         part,
-        newStream.messages.length
+        newStream.messages.length,
       );
       lastContent = getLastContent(currentMessage);
       newStream.messages.push(currentMessage);
@@ -161,9 +161,9 @@ export function applyDeltasToStreamMessage(
     }
     switch (part.type) {
       case "text-delta":
-        currentMessage.text += part.textDelta;
+        currentMessage.text = (currentMessage.text ?? "") + part.textDelta;
         if (lastContent?.type === "text") {
-          lastContent.text += part.textDelta;
+          lastContent.text = (lastContent.text ?? "") + part.textDelta;
         } else {
           contentToAdd = {
             type: "text",
@@ -189,7 +189,7 @@ export function applyDeltasToStreamMessage(
           if (typeof lastContent.args !== "string") {
             throw new Error("Expected args to be a string");
           }
-          lastContent.args += part.argsTextDelta;
+          lastContent.args = (lastContent.args ?? "") + part.argsTextDelta;
         }
         break;
       case "tool-call":
@@ -197,9 +197,10 @@ export function applyDeltasToStreamMessage(
         contentToAdd = part;
         break;
       case "reasoning":
-        currentMessage.reasoning += part.textDelta;
+        currentMessage.reasoning =
+          (currentMessage.reasoning ?? "") + part.textDelta;
         if (lastContent?.type === "reasoning") {
-          lastContent.text += part.textDelta;
+          lastContent.text = (lastContent.text ?? "") + part.textDelta;
         } else {
           contentToAdd = {
             type: "reasoning",
@@ -236,7 +237,7 @@ export function applyDeltasToStreamMessage(
 }
 
 function cloneMessageAndContent(
-  message: Message | undefined
+  message: Message | undefined,
 ): Message | undefined {
   return (
     message &&
@@ -257,7 +258,7 @@ function getLastContent(message: MessageDoc) {
 }
 
 function statusFromStreamStatus(
-  status: StreamMessage["status"]
+  status: StreamMessage["status"],
 ): MessageStatus {
   switch (status) {
     case "streaming":
@@ -275,7 +276,7 @@ export function createStreamingMessage(
   threadId: string,
   message: StreamMessage,
   part: TextStreamPart,
-  index: number
+  index: number,
 ): MessageDoc {
   const { streamId, ...rest } = message;
   const metadata: MessageDoc = {

package/src/react/index.ts

@@ -86,7 +86,7 @@ export function useThreadMessages<
     stream?: Query extends ThreadStreamQuery
       ? boolean
       : ErrorMessage<"To enable streaming, your query must take in streamArgs: vStreamArgs and return a streams object returned from agent.syncStreams. See docs.">;
-  }
+  },
 ): UsePaginatedQueryResult<
   ThreadMessagesResult<Query> & { streaming?: boolean }
 > {
@@ -94,7 +94,7 @@ export function useThreadMessages<
   const paginated = usePaginatedQuery(
     query,
     args as PaginatedQueryArgs<Query> | "skip",
-    { initialNumItems: options.initialNumItems }
+    { initialNumItems: options.initialNumItems },
   );
 
   // These are streaming messages that will not include full messages.
@@ -107,7 +107,7 @@ export function useThreadMessages<
       args === "skip" ||
       paginated.status === "LoadingFirstPage"
       ? "skip"
-      : { ...args, startOrder: paginated.results.at(-1)?.order }
+      : { ...args, startOrder: paginated.results.at(-1)?.order },
   );
 
   const merged = useMemo(() => {
@@ -122,14 +122,14 @@ export function useThreadMessages<
         .map((m) => ({ ...m, streaming: false }))
         .concat(streamListMessages)
         .sort((a, b) =>
-          a.order === b.order ? a.stepOrder - b.stepOrder : a.order - b.order
+          a.order === b.order ? a.stepOrder - b.stepOrder : a.order - b.order,
         )
         // They shouldn't overlap, but check for duplicates just in case.
         .filter(
           (m, i, arr) =>
             !arr[i - 1] ||
             m.order !== arr[i - 1].order ||
-            m.stepOrder !== arr[i - 1].stepOrder
+            m.stepOrder !== arr[i - 1].stepOrder,
         ),
     };
   }, [paginated, streamMessages]);
@@ -154,7 +154,7 @@ export function useStreamingThreadMessages<
   Query extends ThreadStreamQuery<any, any>,
 >(
   query: Query,
-  args: (ThreadMessagesArgs<Query> & { startOrder?: number }) | "skip"
+  args: (ThreadMessagesArgs<Query> & { startOrder?: number }) | "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
@@ -179,7 +179,7 @@ export function useStreamingThreadMessages<
             kind: "list",
             startOrder: startOrderRef.current,
           } as StreamArgs,
-        } as FunctionArgs<Query>)
+        } as FunctionArgs<Query>),
   ) as
     | { streams: Extract<SyncStreamsReturnValue, { kind: "list" }> }
     | undefined;
@@ -204,7 +204,7 @@ export function useStreamingThreadMessages<
           ...queryArgs,
           paginationOpts: { cursor: null, numItems: 0 },
           streamArgs: { kind: "deltas", cursors } as StreamArgs,
-        } as FunctionArgs<Query>)
+        } as FunctionArgs<Query>),
   ) as
     | { streams: Extract<SyncStreamsReturnValue, { kind: "deltas" }> }
     | undefined;
@@ -220,7 +220,7 @@ export function useStreamingThreadMessages<
       threadId,
       streamList.streams.messages,
       streams,
-      cursorQuery?.streams?.deltas ?? []
+      cursorQuery?.streams?.deltas ?? [],
     );
   }, [threadId, cursorQuery, streams, streamList]);
   // Now assemble the chunks into messages
@@ -248,7 +248,7 @@ export function useStreamingThreadMessages<
 export function useStreamingText(
   url: string,
   threadId: string | null,
-  token?: string
+  token?: string,
 ) {
   const [text, setText] = useState("");
   const [loading, setLoading] = useState(false);
@@ -295,7 +295,7 @@ export function useStreamingText(
         setLoading(false);
       }
     },
-    [threadId, token, url]
+    [threadId, token, url],
   );
   return [{ text, loading, error }, readStream] as const;
 }

package/src/react/optimisticallySendMessage.ts

@@ -4,10 +4,10 @@ import type { OptimisticLocalStore } from "convex/browser";
 import type { ThreadQuery } from "./types.js";
 
 export function optimisticallySendMessage(
-  query: ThreadQuery<unknown, MessageDoc>
+  query: ThreadQuery<unknown, MessageDoc>,
 ): (
   store: OptimisticLocalStore,
-  args: { threadId: string; prompt: string }
+  args: { threadId: string; prompt: string },
 ) => void {
   return (store, args) => {
     const queries = store.getAllQueries(query);

package/src/react/toUIMessages.test.ts

@@ -117,15 +117,15 @@ describe("toUIMessages", () => {
     expect(uiMessages).toHaveLength(2);
     expect(uiMessages[0].role).toBe("user");
     expect(uiMessages[0].parts.filter((p) => p.type === "text")).toHaveLength(
-      1
+      1,
     );
     expect(uiMessages[1].role).toBe("assistant");
     expect(
-      uiMessages[1].parts.filter((p) => p.type === "tool-invocation")
+      uiMessages[1].parts.filter((p) => p.type === "tool-invocation"),
     ).toHaveLength(1);
     expect(
       uiMessages[1].parts.filter((p) => p.type === "tool-invocation")[0]
-        .toolInvocation
+        .toolInvocation,
     ).toEqual({
       toolName: "myTool",
       toolCallId: "call1",
@@ -169,19 +169,19 @@ describe("toUIMessages", () => {
     expect(uiMessages[0].role).toBe("assistant");
     expect(uiMessages[0].content).toBe("Here's one idea. Here's another idea.");
     expect(
-      uiMessages[0].parts.filter((p) => p.type === "reasoning")
+      uiMessages[0].parts.filter((p) => p.type === "reasoning"),
     ).toHaveLength(1);
     expect(uiMessages[0].parts[0].type).toBe("reasoning");
     assert(uiMessages[0].parts[0].type === "reasoning");
     expect(uiMessages[0].parts[0].reasoning).toBe(
-      "I'm thinking...I'm thinking..."
+      "I'm thinking...I'm thinking...",
     );
 
     expect(uiMessages[0].parts.filter((p) => p.type === "text")).toHaveLength(
-      1
+      1,
     );
     expect(uiMessages[0].parts.filter((p) => p.type === "text")[0].text).toBe(
-      "Here's one idea. Here's another idea."
+      "Here's one idea. Here's another idea.",
     );
   });
 
@@ -226,11 +226,11 @@ describe("toUIMessages", () => {
     expect(uiMessages).toHaveLength(1);
     expect(uiMessages[0].role).toBe("assistant");
     expect(
-      uiMessages[0].parts.filter((p) => p.type === "tool-invocation")
+      uiMessages[0].parts.filter((p) => p.type === "tool-invocation"),
     ).toHaveLength(1);
     expect(
       uiMessages[0].parts.filter((p) => p.type === "tool-invocation")[0]
-        .toolInvocation
+        .toolInvocation,
     ).toEqual({
       toolName: "myTool",
       toolCallId: "call1",
@@ -277,7 +277,7 @@ describe("toUIMessages", () => {
     expect(uiMessages[0].role).toBe("assistant");
     // Should have a tool-invocation part
     expect(uiMessages[0].parts.some((p) => p.type === "tool-invocation")).toBe(
-      true
+      true,
     );
   });
 

package/src/react/toUIMessages.ts

@@ -13,7 +13,7 @@ export type UIMessage = AIUIMessage & {
 };
 
 export function toUIMessages(
-  messages: (MessageDoc & { streaming?: boolean })[]
+  messages: (MessageDoc & { streaming?: boolean })[],
 ): UIMessage[] {
   const uiMessages: UIMessage[] = [];
   let assistantMessage: UIMessage | undefined;
@@ -63,7 +63,7 @@ export function toUIMessages(
       if (coreMessage.role === "tool" && !assistantMessage) {
         console.warn(
           "Tool message without preceding assistant message.. skipping",
-          message
+          message,
         );
         continue;
       }
@@ -118,7 +118,7 @@ export function toUIMessages(
               toolInvocation: {
                 state: "call",
                 step: assistantMessage.parts.filter(
-                  (part) => part.type === "tool-invocation"
+                  (part) => part.type === "tool-invocation",
                 ).length,
                 toolCallId: contentPart.toolCallId,
                 toolName: contentPart.toolName,
@@ -130,7 +130,7 @@ export function toUIMessages(
             const call = assistantMessage.parts.find(
               (part) =>
                 part.type === "tool-invocation" &&
-                part.toolInvocation.toolCallId === contentPart.toolCallId
+                part.toolInvocation.toolCallId === contentPart.toolCallId,
             ) as ToolInvocationUIPart | undefined;
             const toolInvocation: ToolInvocationUIPart["toolInvocation"] = {
               state: "result",
@@ -141,7 +141,7 @@ export function toUIMessages(
               step:
                 call?.toolInvocation.step ??
                 assistantMessage.parts.filter(
-                  (part) => part.type === "tool-invocation"
+                  (part) => part.type === "tool-invocation",
                 ).length,
             };
             if (call) {
@@ -149,7 +149,7 @@ export function toUIMessages(
             } else {
               console.warn(
                 "Tool result without preceding tool call.. adding anyways",
-                contentPart
+                contentPart,
               );
               assistantMessage.parts.push({
                 type: "tool-invocation",

package/src/react/useSmoothText.ts

@@ -29,7 +29,7 @@ export function useSmoothText(
      * This will start streaming the first value it sees.
      */
     startStreaming?: boolean;
-  } = {}
+  } = {},
 ): [string, { cursor: number; isStreaming: boolean }] {
   const [visibleText, setVisibleText] = useState(startStreaming ? "" : text);
   const smoothState = useRef({
@@ -52,11 +52,11 @@ export function useSmoothText(
     // Smooth out the charsPerSec by averaging it with the previous value.
     smoothState.current.charsPerMs = Math.min(
       (2 * latestCharsPerMs + smoothState.current.charsPerMs) / 3,
-      smoothState.current.charsPerMs * 2
+      smoothState.current.charsPerMs * 2,
     );
     smoothState.current.tick = Math.max(
       smoothState.current.tick,
-      Date.now() - 2 * MS_PER_FRAME
+      Date.now() - 2 * MS_PER_FRAME,
     );
 
     function update() {
@@ -66,14 +66,14 @@ export function useSmoothText(
       const now = Date.now();
       const timeSinceLastUpdate = Math.min(
         MAX_TIME_JUMP_MS,
-        now - smoothState.current.tick
+        now - smoothState.current.tick,
       );
       const chars = Math.floor(
-        timeSinceLastUpdate * smoothState.current.charsPerMs
+        timeSinceLastUpdate * smoothState.current.charsPerMs,
       );
       smoothState.current.cursor = Math.min(
         smoothState.current.cursor + chars,
-        text.length
+        text.length,
       );
       smoothState.current.tick = now;
       setVisibleText(text.slice(0, smoothState.current.cursor));

package/src/validators.ts

@@ -11,7 +11,7 @@ import { vVectorDimension } from "./component/vector/tables.js";
 
 export const vProviderOptions = v.record(
   v.string(),
-  v.record(v.string(), v.any())
+  v.record(v.string(), v.any()),
 );
 const providerOptions = v.optional(vProviderOptions);
 export type ProviderOptions = Infer<typeof providerOptions>;
@@ -22,12 +22,12 @@ export type ProviderMetadata = Infer<typeof providerMetadata>;
 
 export const vThreadStatus = v.union(
   v.literal("active"),
-  v.literal("archived") // unused
+  v.literal("archived"), // unused
 );
 export const vMessageStatus = v.union(
   v.literal("pending"),
   v.literal("success"),
-  v.literal("failed")
+  v.literal("failed"),
 );
 export type MessageStatus = Infer<typeof vMessageStatus>;
 
@@ -35,7 +35,7 @@ export const vRole = v.union(
   v.literal("system"),
   v.literal("user"),
   v.literal("assistant"),
-  v.literal("tool")
+  v.literal("tool"),
 );
 
 export const vTextPart = v.object({
@@ -61,7 +61,7 @@ export const vFilePart = v.object({
 
 export const vUserContent = v.union(
   v.string(),
-  v.array(v.union(vTextPart, vImagePart, vFilePart))
+  v.array(v.union(vTextPart, vImagePart, vFilePart)),
 );
 
 export const vReasoningPart = v.object({
@@ -87,8 +87,8 @@ export const vReasoningDetails = v.array(
     v.object({
       type: v.literal("redacted"),
       data: v.string(),
-    })
-  )
+    }),
+  ),
 );
 
 export const vToolCallPart = v.object({
@@ -107,9 +107,9 @@ export const vAssistantContent = v.union(
       vFilePart,
       vReasoningPart,
       vRedactedReasoningPart,
-      vToolCallPart
-    )
-  )
+      vToolCallPart,
+    ),
+  ),
 );
 
 const vToolResultContent = v.array(
@@ -122,8 +122,8 @@ const vToolResultContent = v.array(
       type: v.literal("image"),
       data: v.string(),
       mimeType: v.optional(v.string()),
-    })
-  )
+    }),
+  ),
 );
 
 const vToolResultPart = v.object({
@@ -171,7 +171,7 @@ export const vMessage = v.union(
   vUserMessage,
   vAssistantMessage,
   vToolMessage,
-  vSystemMessage
+  vSystemMessage,
 );
 export type Message = Infer<typeof vMessage>;
 
@@ -221,7 +221,7 @@ export const vFinishReason = v.union(
   v.literal("tool-calls"),
   v.literal("error"),
   v.literal("other"),
-  v.literal("unknown")
+  v.literal("unknown"),
 );
 
 export const vUsage = v.object({
@@ -245,7 +245,7 @@ export const vLanguageModelV1CallWarning = v.union(
   v.object({
     type: v.literal("other"),
     message: v.string(),
-  })
+  }),
 );
 
 export const vMessageWithMetadataInternal = v.object({
@@ -307,7 +307,7 @@ export const vContextOptions = v.object({
 
 export const vStorageOptions = v.object({
   saveMessages: v.optional(
-    v.union(v.literal("all"), v.literal("none"), v.literal("promptAndOutput"))
+    v.union(v.literal("all"), v.literal("none"), v.literal("promptAndOutput")),
   ),
 });
 
@@ -352,8 +352,8 @@ export const vTextArgs = v.object({
       v.object({
         type: v.literal("tool"),
         toolName: v.string(),
-      })
-    )
+      }),
+    ),
   ),
   maxSteps: v.optional(v.number()),
   experimental_continueSteps: v.optional(v.boolean()),
@@ -382,8 +382,8 @@ export function vPaginationResult<
       v.union(
         v.literal("SplitRecommended"),
         v.literal("SplitRequired"),
-        v.null()
-      )
+        v.null(),
+      ),
     ),
   });
 }
@@ -413,7 +413,7 @@ export const vTextStreamPart = v.union(
     toolName: v.string(),
     argsTextDelta: v.string(),
   }),
-  vToolResultPart
+  vToolResultPart,
 );
 export type TextStreamPart = Infer<typeof vTextStreamPart>;
 
@@ -432,8 +432,8 @@ export const vStreamArgs = v.optional(
     v.object({
       kind: v.literal("deltas"),
       cursors: v.array(vStreamCursor),
-    })
-  )
+    }),
+  ),
 );
 export type StreamArgs = Infer<typeof vStreamArgs>;
 
@@ -442,7 +442,7 @@ export const vStreamMessage = v.object({
   status: v.union(
     v.literal("streaming"),
     v.literal("finished"),
-    v.literal("aborted")
+    v.literal("aborted"),
   ),
   order: v.number(),
   stepOrder: v.number(),