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

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[]));

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);
@@ -237,7 +237,7 @@ export function applyDeltasToStreamMessage(
 }
 
 function cloneMessageAndContent(
-  message: Message | undefined
+  message: Message | undefined,
 ): Message | undefined {
   return (
     message &&
@@ -258,7 +258,7 @@ function getLastContent(message: MessageDoc) {
 }
 
 function statusFromStreamStatus(
-  status: StreamMessage["status"]
+  status: StreamMessage["status"],
 ): MessageStatus {
   switch (status) {
     case "streaming":
@@ -276,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);