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

package/src/client/types.ts

@@ -19,12 +19,15 @@ import type {
   ToolSet,
 } from "ai";
 import type {
+  Auth,
   Expand,
   FunctionReference,
   GenericActionCtx,
   GenericDataModel,
   GenericMutationCtx,
   GenericQueryCtx,
+  PaginationOptions,
+  PaginationResult,
   StorageActionWriter,
   StorageReader,
   WithoutSystemFields,
@@ -351,6 +354,13 @@ export interface Thread<DefaultTools extends ToolSet> {
   updateMetadata: (
     patch: Partial<WithoutSystemFields<ThreadDoc>>
   ) => Promise<ThreadDoc>;
+  /**
+   * Search for threads by title.
+   */
+  searchThreadTitles(args: {
+    query: string;
+    paginationOpts?: PaginationOptions;
+  }): Promise<PaginationResult<ThreadDoc>>;
   /**
    * This behaves like {@link generateText} from the "ai" package except that
    * it add context based on the userId and threadId and saves the input and
@@ -495,6 +505,7 @@ export type RunActionCtx = {
   runAction: GenericActionCtx<GenericDataModel>["runAction"];
 };
 export type ActionCtx = RunActionCtx & {
+  auth: Auth;
   storage: StorageActionWriter;
 };
 export type QueryCtx = RunQueryCtx & {

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

@@ -1723,6 +1723,36 @@ export type Mounts = {
         splitCursor?: string | null;
       }
     >;
+    searchThreadTitles: FunctionReference<
+      "query",
+      "public",
+      {
+        paginationOpts?: {
+          cursor: string | null;
+          endCursor?: string | null;
+          id?: number;
+          maximumBytesRead?: number;
+          maximumRowsRead?: number;
+          numItems: number;
+        };
+        query: string;
+        userId?: string | null;
+      },
+      {
+        continueCursor: string;
+        isDone: boolean;
+        page: Array<{
+          _creationTime: number;
+          _id: string;
+          status: "active" | "archived";
+          summary?: string;
+          title?: string;
+          userId?: string;
+        }>;
+        pageStatus?: "SplitRecommended" | "SplitRequired" | null;
+        splitCursor?: string | null;
+      }
+    >;
     updateThread: FunctionReference<
       "mutation",
       "public",

package/src/component/schema.ts

@@ -27,7 +27,12 @@ export const schema = defineSchema({
     defaultSystemPrompt: v.optional(v.string()),
     parentThreadIds: v.optional(v.array(v.id("threads"))),
     order: /*DEPRECATED*/ v.optional(v.number()),
-  }).index("userId", ["userId"]),
+  })
+    .index("userId", ["userId"])
+    .searchIndex("title", {
+      searchField: "title",
+      filterFields: ["userId"],
+    }),
   messages: defineTable({
     id: v.optional(v.string()), // external id, e.g. from Vercel AI SDK
     userId: v.optional(v.string()), // useful for searching across threads

package/src/component/threads.ts

@@ -92,6 +92,29 @@ export const updateThread = mutation({
   returns: vThreadDoc,
 });
 
+export const searchThreadTitles = query({
+  args: {
+    query: v.string(),
+    userId: v.optional(v.union(v.string(), v.null())),
+    paginationOpts: v.optional(paginationOptsValidator),
+  },
+  handler: async (ctx, args) => {
+    const threads = await ctx.db
+      .query("threads")
+      .withSearchIndex("title", (q) =>
+        args.userId
+          ? q.search("title", args.query).eq("userId", args.userId ?? undefined)
+          : q.search("title", args.query)
+      )
+      .paginate(args.paginationOpts ?? { cursor: null, numItems: 100 });
+    return {
+      ...threads,
+      page: threads.page.map(publicThread),
+    };
+  },
+  returns: vPaginationResult(vThreadDoc),
+});
+
 // When we expose this, we need to also hide all the messages and steps
 // export const archiveThread = mutation({
 //   args: { threadId: v.id("threads") },