@convex-dev/agent 0.6.0 → 0.6.1

package/dist/component/threads.js

@@ -0,0 +1,208 @@
+import { assert, omit, pick } from "convex-helpers";
+import { paginator } from "convex-helpers/server/pagination";
+import { partial } from "convex-helpers/validators";
+import { paginationOptsValidator } from "convex/server";
+import { vThreadDoc } from "../client/index.js";
+import { vPaginationResult } from "../validators.js";
+import { api, internal } from "./_generated/api.js";
+import { action, internalMutation, mutation, query, } from "./_generated/server.js";
+import { deleteMessage } from "./messages.js";
+import { schema, v } from "./schema.js";
+import { deleteStreamsPageForThreadId } from "./streams.js";
+function publicThreadOrNull(thread) {
+    if (thread === null) {
+        return null;
+    }
+    return publicThread(thread);
+}
+function publicThread(thread) {
+    return omit(thread, ["defaultSystemPrompt", "parentThreadIds", "order"]);
+}
+export const getThread = query({
+    args: { threadId: v.id("threads") },
+    handler: async (ctx, args) => {
+        return publicThreadOrNull(await ctx.db.get(args.threadId));
+    },
+    returns: v.union(vThreadDoc, v.null()),
+});
+export const listThreadsByUserId = query({
+    args: {
+        userId: v.optional(v.string()),
+        order: v.optional(v.union(v.literal("asc"), v.literal("desc"))),
+        paginationOpts: v.optional(paginationOptsValidator),
+    },
+    handler: async (ctx, args) => {
+        const threads = await paginator(ctx.db, schema)
+            .query("threads")
+            .withIndex("userId", (q) => q.eq("userId", args.userId))
+            .order(args.order ?? "desc")
+            .paginate(args.paginationOpts ?? { cursor: null, numItems: 100 });
+        return {
+            ...threads,
+            page: threads.page.map(publicThread),
+        };
+    },
+    returns: vPaginationResult(vThreadDoc),
+});
+const vThread = schema.tables.threads.validator;
+export const createThread = mutation({
+    args: omit(vThread.fields, ["order", "status"]),
+    handler: async (ctx, args) => {
+        const threadId = await ctx.db.insert("threads", {
+            ...args,
+            status: "active",
+        });
+        return publicThread((await ctx.db.get(threadId)));
+    },
+    returns: vThreadDoc,
+});
+export const threadFieldsSupportingPatch = [
+    "title",
+    "summary",
+    "status",
+    "userId",
+];
+export const updateThread = mutation({
+    args: {
+        threadId: v.id("threads"),
+        patch: v.object(partial(pick(vThread.fields, threadFieldsSupportingPatch))),
+    },
+    handler: async (ctx, args) => {
+        const thread = await ctx.db.get(args.threadId);
+        assert(thread, `Thread ${args.threadId} not found`);
+        await ctx.db.patch(args.threadId, args.patch);
+        return publicThread((await ctx.db.get(args.threadId)));
+    },
+    returns: vThreadDoc,
+});
+export const searchThreadTitles = query({
+    args: {
+        query: v.string(),
+        userId: v.optional(v.union(v.string(), v.null())),
+        limit: v.number(),
+    },
+    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))
+            .take(args.limit);
+        return threads.map(publicThread);
+    },
+    returns: v.array(vThreadDoc),
+});
+// When we expose this, we need to also hide all the messages and steps
+// export const archiveThread = mutation({
+//   args: { threadId: v.id("threads") },
+//   handler: async (ctx, args) => {
+//     const thread = await ctx.db.get(args.threadId);
+//     assert(thread, `Thread ${args.threadId} not found`);
+//     await ctx.db.patch(args.threadId, { status: "archived" });
+//     return publicThread((await ctx.db.get(args.threadId))!);
+//   },
+//   returns: vThreadDoc,
+// });
+// TODO: delete thread
+/**
+ * Use this to delete a thread and everything it contains.
+ * It will try to delete all pages synchronously.
+ * If it times out or fails, you'll have to run it again.
+ */
+export const deleteAllForThreadIdSync = action({
+    args: { threadId: v.id("threads"), limit: v.optional(v.number()) },
+    handler: async (ctx, args) => {
+        let cursor = undefined;
+        while (true) {
+            const result = await ctx.runMutation(internal.threads._deletePageForThreadId, { threadId: args.threadId, cursor, limit: args.limit });
+            if (result.isDone) {
+                break;
+            }
+            cursor = result.cursor;
+        }
+        await ctx.runAction(api.streams.deleteAllStreamsForThreadIdSync, {
+            threadId: args.threadId,
+        });
+    },
+    returns: v.null(),
+});
+const deleteThreadArgs = {
+    threadId: v.id("threads"),
+    cursor: v.optional(v.string()),
+    messagesDone: v.optional(v.boolean()),
+    streamsDone: v.optional(v.boolean()),
+    streamOrder: v.optional(v.number()),
+    deltaCursor: v.optional(v.string()),
+    limit: v.optional(v.number()),
+};
+const deleteThreadReturns = {
+    cursor: v.string(),
+    isDone: v.boolean(),
+};
+export const _deletePageForThreadId = internalMutation({
+    args: deleteThreadArgs,
+    handler: deletePageForThreadIdHandler,
+    returns: deleteThreadReturns,
+});
+/**
+ * Use this to delete a thread and everything it contains.
+ * It will continue deleting pages asynchronously.
+ */
+export const deleteAllForThreadIdAsync = mutation({
+    args: deleteThreadArgs,
+    handler: async (ctx, args) => {
+        let messagesResult = {
+            isDone: args.messagesDone ?? false,
+            cursor: args.cursor,
+        };
+        if (!args.messagesDone) {
+            messagesResult = await deletePageForThreadIdHandler(ctx, args);
+        }
+        let streamResult = {
+            isDone: args.streamsDone ?? false,
+            streamOrder: args.streamOrder,
+            deltaCursor: args.deltaCursor,
+        };
+        if (!args.streamsDone) {
+            streamResult = await deleteStreamsPageForThreadId(ctx, {
+                threadId: args.threadId,
+                streamOrder: args.streamOrder,
+                deltaCursor: args.deltaCursor,
+            });
+        }
+        const isDone = messagesResult.isDone && streamResult.isDone;
+        if (!isDone) {
+            await ctx.scheduler.runAfter(0, api.threads.deleteAllForThreadIdAsync, {
+                threadId: args.threadId,
+                cursor: messagesResult.cursor,
+                messagesDone: messagesResult.isDone,
+                streamsDone: streamResult.isDone,
+                streamOrder: streamResult.streamOrder,
+                deltaCursor: streamResult.deltaCursor,
+            });
+        }
+        return { isDone };
+    },
+    returns: v.object({ isDone: v.boolean() }),
+});
+async function deletePageForThreadIdHandler(ctx, args) {
+    const messages = await paginator(ctx.db, schema)
+        .query("messages")
+        .withIndex("threadId_status_tool_order_stepOrder", (q) => q.eq("threadId", args.threadId))
+        .paginate({
+        numItems: args.limit ?? 100,
+        cursor: args.cursor ?? null,
+    });
+    await Promise.all(messages.page.map((m) => deleteMessage(ctx, m)));
+    if (messages.isDone) {
+        const thread = await ctx.db.get(args.threadId);
+        if (thread) {
+            await ctx.db.delete(args.threadId);
+        }
+    }
+    return {
+        cursor: messages.continueCursor,
+        isDone: messages.isDone,
+    };
+}
+//# sourceMappingURL=threads.js.map

package/dist/component/threads.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"threads.js","sourceRoot":"","sources":["../../src/component/threads.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,SAAS,EAAE,MAAM,kCAAkC,CAAC;AAC7D,OAAO,EAAE,OAAO,EAAE,MAAM,2BAA2B,CAAC;AACpD,OAAO,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;AAExD,OAAO,EAAkB,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAChE,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,EAAE,GAAG,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAEpD,OAAO,EACL,MAAM,EACN,gBAAgB,EAChB,QAAQ,EAER,KAAK,GACN,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAC9C,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,4BAA4B,EAAE,MAAM,cAAc,CAAC;AAE5D,SAAS,kBAAkB,CAAC,MAA6B;IACvD,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;QACpB,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,YAAY,CAAC,MAAM,CAAC,CAAC;AAC9B,CAAC;AAED,SAAS,YAAY,CAAC,MAAsB;IAC1C,OAAO,IAAI,CAAC,MAAM,EAAE,CAAC,qBAAqB,EAAE,iBAAiB,EAAE,OAAO,CAAC,CAAC,CAAC;AAC3E,CAAC;AAED,MAAM,CAAC,MAAM,SAAS,GAAG,KAAK,CAAC;IAC7B,IAAI,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE;IACnC,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QAC3B,OAAO,kBAAkB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC7D,CAAC;IACD,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;CACvC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,mBAAmB,GAAG,KAAK,CAAC;IACvC,IAAI,EAAE;QACJ,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;QAC9B,KAAK,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;QAC/D,cAAc,EAAE,CAAC,CAAC,QAAQ,CAAC,uBAAuB,CAAC;KACpD;IACD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QAC3B,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,GAAG,CAAC,EAAE,EAAE,MAAM,CAAC;aAC5C,KAAK,CAAC,SAAS,CAAC;aAChB,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;aACvD,KAAK,CAAC,IAAI,CAAC,KAAK,IAAI,MAAM,CAAC;aAC3B,QAAQ,CAAC,IAAI,CAAC,cAAc,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAC;QACpE,OAAO;YACL,GAAG,OAAO;YACV,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC;SACrC,CAAC;IACJ,CAAC;IACD,OAAO,EAAE,iBAAiB,CAAC,UAAU,CAAC;CACvC,CAAC,CAAC;AAEH,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC;AAEhD,MAAM,CAAC,MAAM,YAAY,GAAG,QAAQ,CAAC;IACnC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IAC/C,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QAC3B,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,SAAS,EAAE;YAC9C,GAAG,IAAI;YACP,MAAM,EAAE,QAAQ;SACjB,CAAC,CAAC;QACH,OAAO,YAAY,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAE,CAAC,CAAC;IACrD,CAAC;IACD,OAAO,EAAE,UAAU;CACpB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,2BAA2B,GAAG;IACzC,OAAgB;IAChB,SAAkB;IAClB,QAAiB;IACjB,QAAiB;CAClB,CAAC;AAEF,MAAM,CAAC,MAAM,YAAY,GAAG,QAAQ,CAAC;IACnC,IAAI,EAAE;QACJ,QAAQ,EAAE,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC;QACzB,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,2BAA2B,CAAC,CAAC,CAAC;KAC5E;IACD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QAC3B,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,EAAE,UAAU,IAAI,CAAC,QAAQ,YAAY,CAAC,CAAC;QACpD,MAAM,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QAC9C,OAAO,YAAY,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAE,CAAC,CAAC;IAC1D,CAAC;IACD,OAAO,EAAE,UAAU;CACpB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,kBAAkB,GAAG,KAAK,CAAC;IACtC,IAAI,EAAE;QACJ,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;QACjB,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;QACjD,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;KAClB;IACD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QAC3B,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,EAAE;aACzB,KAAK,CAAC,SAAS,CAAC;aAChB,eAAe,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,CAC9B,IAAI,CAAC,MAAM;YACT,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,QAAQ,EAAE,IAAI,CAAC,MAAM,IAAI,SAAS,CAAC;YACtE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAClC;aACA,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpB,OAAO,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IACnC,CAAC;IACD,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC;CAC7B,CAAC,CAAC;AAEH,uEAAuE;AACvE,0CAA0C;AAC1C,yCAAyC;AACzC,oCAAoC;AACpC,sDAAsD;AACtD,2DAA2D;AAC3D,iEAAiE;AACjE,+DAA+D;AAC/D,OAAO;AACP,yBAAyB;AACzB,MAAM;AAEN,sBAAsB;AAEtB;;;;GAIG;AACH,MAAM,CAAC,MAAM,wBAAwB,GAAG,MAAM,CAAC;IAC7C,IAAI,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE;IAClE,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QAC3B,IAAI,MAAM,GAAuB,SAAS,CAAC;QAC3C,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,MAAM,GAAwB,MAAM,GAAG,CAAC,WAAW,CACvD,QAAQ,CAAC,OAAO,CAAC,sBAAsB,EACvC,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CACvD,CAAC;YACF,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBAClB,MAAM;YACR,CAAC;YACD,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QACzB,CAAC;QACD,MAAM,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,+BAA+B,EAAE;YAC/D,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACxB,CAAC,CAAC;IACL,CAAC;IACD,OAAO,EAAE,CAAC,CAAC,IAAI,EAAE;CAClB,CAAC,CAAC;AAEH,MAAM,gBAAgB,GAAG;IACvB,QAAQ,EAAE,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC;IACzB,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IAC9B,YAAY,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;IACrC,WAAW,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;IACpC,WAAW,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IACnC,WAAW,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IACnC,KAAK,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CAC9B,CAAC;AAEF,MAAM,mBAAmB,GAAG;IAC1B,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;IAClB,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE;CACpB,CAAC;AAGF,MAAM,CAAC,MAAM,sBAAsB,GAAG,gBAAgB,CAAC;IACrD,IAAI,EAAE,gBAAgB;IACtB,OAAO,EAAE,4BAA4B;IACrC,OAAO,EAAE,mBAAmB;CAC7B,CAAC,CAAC;AAEH;;;GAGG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,QAAQ,CAAC;IAChD,IAAI,EAAE,gBAAgB;IACtB,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QAC3B,IAAI,cAAc,GAAG;YACnB,MAAM,EAAE,IAAI,CAAC,YAAY,IAAI,KAAK;YAClC,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC;QACF,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,cAAc,GAAG,MAAM,4BAA4B,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QACjE,CAAC;QACD,IAAI,YAAY,GAAG;YACjB,MAAM,EAAE,IAAI,CAAC,WAAW,IAAI,KAAK;YACjC,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,WAAW,EAAE,IAAI,CAAC,WAAW;SAC9B,CAAC;QACF,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,YAAY,GAAG,MAAM,4BAA4B,CAAC,GAAG,EAAE;gBACrD,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,WAAW,EAAE,IAAI,CAAC,WAAW;aAC9B,CAAC,CAAC;QACL,CAAC;QACD,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,IAAI,YAAY,CAAC,MAAM,CAAC;QAC5D,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,OAAO,CAAC,yBAAyB,EAAE;gBACrE,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,MAAM,EAAE,cAAc,CAAC,MAAM;gBAC7B,YAAY,EAAE,cAAc,CAAC,MAAM;gBACnC,WAAW,EAAE,YAAY,CAAC,MAAM;gBAChC,WAAW,EAAE,YAAY,CAAC,WAAW;gBACrC,WAAW,EAAE,YAAY,CAAC,WAAW;aACtC,CAAC,CAAC;QACL,CAAC;QACD,OAAO,EAAE,MAAM,EAAE,CAAC;IACpB,CAAC;IACD,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC;CAC3C,CAAC,CAAC;AAEH,KAAK,UAAU,4BAA4B,CACzC,GAAgB,EAChB,IAAsB;IAEtB,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,GAAG,CAAC,EAAE,EAAE,MAAM,CAAC;SAC7C,KAAK,CAAC,UAAU,CAAC;SACjB,SAAS,CAAC,sCAAsC,EAAE,CAAC,CAAC,EAAE,EAAE,CACvD,CAAC,CAAC,EAAE,CAAC,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,CAChC;SACA,QAAQ,CAAC;QACR,QAAQ,EAAE,IAAI,CAAC,KAAK,IAAI,GAAG;QAC3B,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,IAAI;KAC5B,CAAC,CAAC;IACL,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,aAAa,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IACnE,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;QACpB,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/C,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IACD,OAAO;QACL,MAAM,EAAE,QAAQ,CAAC,cAAc;QAC/B,MAAM,EAAE,QAAQ,CAAC,MAAM;KACxB,CAAC;AACJ,CAAC"}

package/dist/component/users.d.ts

@@ -0,0 +1,52 @@
+export declare const listUsersWithThreads: import("convex/server").RegisteredQuery<"public", {
+    paginationOpts: {
+        id?: number;
+        endCursor?: string | null;
+        maximumRowsRead?: number;
+        maximumBytesRead?: number;
+        numItems: number;
+        cursor: string | null;
+    };
+}, Promise<{
+    page: string[];
+    isDone: boolean;
+    continueCursor: import("convex/server").Cursor;
+    splitCursor?: import("convex/server").Cursor | null;
+    pageStatus?: "SplitRecommended" | "SplitRequired" | null;
+}>>;
+export declare const deleteAllForUserId: import("convex/server").RegisteredAction<"public", {
+    userId: string;
+}, Promise<void>>;
+export declare const deleteAllForUserIdAsync: import("convex/server").RegisteredMutation<"public", {
+    userId: string;
+}, Promise<boolean>>;
+export declare const _deleteAllForUserIdAsync: import("convex/server").RegisteredMutation<"internal", {
+    deltaCursor?: string | undefined;
+    streamOrder?: number | undefined;
+    streamsInProgress?: boolean | undefined;
+    userId: string;
+    messagesCursor: string | null;
+    threadsCursor: string | null;
+    threadInProgress: import("convex/values").GenericId<"threads"> | null;
+}, Promise<boolean>>;
+export declare const _deletePageForUserId: import("convex/server").RegisteredMutation<"internal", {
+    deltaCursor?: string | undefined;
+    streamOrder?: number | undefined;
+    streamsInProgress?: boolean | undefined;
+    userId: string;
+    messagesCursor: string | null;
+    threadsCursor: string | null;
+    threadInProgress: import("convex/values").GenericId<"threads"> | null;
+}, Promise<{
+    deltaCursor?: string | undefined;
+    streamOrder?: number | undefined;
+    streamsInProgress?: boolean | undefined;
+    isDone: boolean;
+    messagesCursor: string | null;
+    threadsCursor: string;
+    threadInProgress: import("convex/values").GenericId<"threads"> | null;
+}>>;
+export declare const getThreadUserId: import("convex/server").RegisteredQuery<"internal", {
+    threadId: import("convex/values").GenericId<"threads">;
+}, Promise<string | null>>;
+//# sourceMappingURL=users.d.ts.map

package/dist/component/users.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"users.d.ts","sourceRoot":"","sources":["../../src/component/users.ts"],"names":[],"mappings":"AAqBA,eAAO,MAAM,oBAAoB;;;;;;;;;;;;;;;GAgB/B,CAAC;AAEH,eAAO,MAAM,kBAAkB;;iBAiC7B,CAAC;AAEH,eAAO,MAAM,uBAAuB;;oBAiBlC,CAAC;AAuBH,eAAO,MAAM,wBAAwB;;;;;;;;oBAInC,CAAC;AAqBH,eAAO,MAAM,oBAAoB;;;;;;;;;;;;;;;;GAI/B,CAAC;AA6GH,eAAO,MAAM,eAAe;;0BAS1B,CAAC"}

package/dist/component/users.js

@@ -0,0 +1,229 @@
+import { paginator } from "convex-helpers/server/pagination";
+import { stream } from "convex-helpers/server/stream";
+import { nullable } from "convex-helpers/validators";
+import { paginationOptsValidator } from "convex/server";
+import { vPaginationResult } from "../validators.js";
+import { internal } from "./_generated/api.js";
+import { action, internalMutation, internalQuery, mutation, query, } from "./_generated/server.js";
+import { deleteMessage } from "./messages.js";
+import { deleteStreamsPageForThreadId } from "./streams.js";
+import { schema, v } from "./schema.js";
+// Note: it only searches for users with threads
+export const listUsersWithThreads = query({
+    args: {
+        paginationOpts: paginationOptsValidator,
+    },
+    handler: async (ctx, args) => {
+        const results = await stream(ctx.db, schema)
+            .query("threads")
+            .withIndex("userId", (q) => q.gt("userId", ""))
+            .distinct(["userId"])
+            .paginate(args.paginationOpts);
+        return {
+            ...results,
+            page: results.page.map((t) => t.userId).filter((t) => !!t),
+        };
+    },
+    returns: vPaginationResult(v.string()),
+});
+export const deleteAllForUserId = action({
+    args: { userId: v.string() },
+    handler: async (ctx, args) => {
+        let threadsCursor = null;
+        let threadInProgress = null;
+        let messagesCursor = null;
+        let streamsInProgress = false;
+        let streamOrder = undefined;
+        let deltaCursor = undefined;
+        let isDone = false;
+        while (!isDone) {
+            const result = await ctx.runMutation(internal.users._deletePageForUserId, {
+                userId: args.userId,
+                messagesCursor,
+                threadInProgress,
+                threadsCursor,
+                streamsInProgress,
+                streamOrder,
+                deltaCursor,
+            });
+            messagesCursor = result.messagesCursor;
+            threadInProgress = result.threadInProgress;
+            threadsCursor = result.threadsCursor;
+            streamsInProgress = result.streamsInProgress ?? false;
+            streamOrder = result.streamOrder;
+            deltaCursor = result.deltaCursor;
+            isDone = result.isDone;
+        }
+    },
+    returns: v.null(),
+});
+export const deleteAllForUserIdAsync = mutation({
+    args: {
+        userId: v.string(),
+    },
+    handler: async (ctx, args) => {
+        const isDone = await deleteAllForUserIdAsyncHandler(ctx, {
+            userId: args.userId,
+            messagesCursor: null,
+            threadsCursor: null,
+            threadInProgress: null,
+            streamsInProgress: false,
+            streamOrder: undefined,
+            deltaCursor: undefined,
+        });
+        return isDone;
+    },
+    returns: v.boolean(),
+});
+const deleteAllArgs = {
+    userId: v.string(),
+    messagesCursor: nullable(v.string()),
+    threadsCursor: nullable(v.string()),
+    threadInProgress: nullable(v.id("threads")),
+    streamsInProgress: v.optional(v.boolean()),
+    streamOrder: v.optional(v.number()),
+    deltaCursor: v.optional(v.string()),
+};
+const deleteAllReturns = {
+    threadsCursor: v.string(),
+    threadInProgress: nullable(v.id("threads")),
+    messagesCursor: nullable(v.string()),
+    streamsInProgress: v.optional(v.boolean()),
+    streamOrder: v.optional(v.number()),
+    deltaCursor: v.optional(v.string()),
+    isDone: v.boolean(),
+};
+export const _deleteAllForUserIdAsync = internalMutation({
+    args: deleteAllArgs,
+    handler: deleteAllForUserIdAsyncHandler,
+    returns: v.boolean(),
+});
+async function deleteAllForUserIdAsyncHandler(ctx, args) {
+    const result = await deletePageForUserId(ctx, args);
+    if (!result.isDone) {
+        await ctx.scheduler.runAfter(0, internal.users._deleteAllForUserIdAsync, {
+            userId: args.userId,
+            messagesCursor: result.messagesCursor,
+            threadsCursor: result.threadsCursor,
+            threadInProgress: result.threadInProgress,
+            streamsInProgress: result.streamsInProgress ?? false,
+            streamOrder: result.streamOrder,
+            deltaCursor: result.deltaCursor,
+        });
+    }
+    return result.isDone;
+}
+export const _deletePageForUserId = internalMutation({
+    args: deleteAllArgs,
+    handler: deletePageForUserId,
+    returns: deleteAllReturns,
+});
+async function deletePageForUserId(ctx, args) {
+    let threadInProgress = args.threadInProgress;
+    let threadsCursor = args.threadsCursor;
+    let messagesCursor = args.messagesCursor;
+    let streamsInProgress = args.streamsInProgress ?? false;
+    let streamOrder = args.streamOrder;
+    let deltaCursor = args.deltaCursor;
+    // Phase 1: Get a thread to work on if we don't have one
+    if (!threadsCursor || !threadInProgress) {
+        const threads = await paginator(ctx.db, schema)
+            .query("threads")
+            .withIndex("userId", (q) => q.eq("userId", args.userId))
+            .order("desc")
+            .paginate({
+            numItems: 1,
+            cursor: args.threadsCursor ?? null,
+        });
+        threadsCursor = threads.continueCursor;
+        if (threads.page.length > 0) {
+            threadInProgress = threads.page[0]._id;
+            messagesCursor = null;
+            streamsInProgress = false;
+            streamOrder = undefined;
+            deltaCursor = undefined;
+        }
+        else {
+            return {
+                isDone: true,
+                threadsCursor,
+                threadInProgress,
+                messagesCursor,
+                streamsInProgress,
+                streamOrder,
+                deltaCursor,
+            };
+        }
+    }
+    // Phase 2: Delete messages for the current thread
+    if (!streamsInProgress) {
+        const messages = await paginator(ctx.db, schema)
+            .query("messages")
+            .withIndex("threadId_status_tool_order_stepOrder", (q) => q.eq("threadId", threadInProgress))
+            .order("desc")
+            .paginate({
+            numItems: 100,
+            cursor: args.messagesCursor,
+        });
+        await Promise.all(messages.page.map((m) => deleteMessage(ctx, m)));
+        if (messages.isDone) {
+            // Messages are done, move to streams deletion phase
+            streamsInProgress = true;
+            messagesCursor = null;
+            streamOrder = undefined;
+            deltaCursor = undefined;
+        }
+        else {
+            messagesCursor = messages.continueCursor;
+        }
+        return {
+            messagesCursor,
+            threadsCursor,
+            threadInProgress,
+            streamsInProgress,
+            streamOrder,
+            deltaCursor,
+            isDone: false,
+        };
+    }
+    // Phase 3: Delete streams for the current thread
+    const streamResult = await deleteStreamsPageForThreadId(ctx, {
+        threadId: threadInProgress,
+        streamOrder,
+        deltaCursor,
+    });
+    if (streamResult.isDone) {
+        // Streams are done, delete the thread and reset for next thread
+        await ctx.db.delete(threadInProgress);
+        threadInProgress = null;
+        messagesCursor = null;
+        streamsInProgress = false;
+        streamOrder = undefined;
+        deltaCursor = undefined;
+    }
+    else {
+        // Continue with streams deletion
+        streamOrder = streamResult.streamOrder;
+        deltaCursor = streamResult.deltaCursor;
+    }
+    return {
+        messagesCursor,
+        threadsCursor,
+        threadInProgress,
+        streamsInProgress,
+        streamOrder,
+        deltaCursor,
+        isDone: false,
+    };
+}
+export const getThreadUserId = internalQuery({
+    args: {
+        threadId: v.id("threads"),
+    },
+    returns: v.union(v.string(), v.null()),
+    handler: async (ctx, args) => {
+        const thread = await ctx.db.get(args.threadId);
+        return thread?.userId ?? null;
+    },
+});
+//# sourceMappingURL=users.js.map

package/dist/component/users.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"users.js","sourceRoot":"","sources":["../../src/component/users.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,kCAAkC,CAAC;AAC7D,OAAO,EAAE,MAAM,EAAE,MAAM,8BAA8B,CAAC;AACtD,OAAO,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AACrD,OAAO,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;AAExD,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAE/C,OAAO,EACL,MAAM,EACN,gBAAgB,EAChB,aAAa,EACb,QAAQ,EAER,KAAK,GACN,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAC9C,OAAO,EAAE,4BAA4B,EAAE,MAAM,cAAc,CAAC;AAC5D,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,aAAa,CAAC;AAExC,gDAAgD;AAChD,MAAM,CAAC,MAAM,oBAAoB,GAAG,KAAK,CAAC;IACxC,IAAI,EAAE;QACJ,cAAc,EAAE,uBAAuB;KACxC;IACD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QAC3B,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,MAAM,CAAC;aACzC,KAAK,CAAC,SAAS,CAAC;aAChB,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;aAC9C,QAAQ,CAAC,CAAC,QAAQ,CAAC,CAAC;aACpB,QAAQ,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACjC,OAAO;YACL,GAAG,OAAO;YACV,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAe,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;SACxE,CAAC;IACJ,CAAC;IACD,OAAO,EAAE,iBAAiB,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CACvC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,kBAAkB,GAAG,MAAM,CAAC;IACvC,IAAI,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE;IAC5B,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QAC3B,IAAI,aAAa,GAAG,IAAI,CAAC;QACzB,IAAI,gBAAgB,GAAG,IAAI,CAAC;QAC5B,IAAI,cAAc,GAAG,IAAI,CAAC;QAC1B,IAAI,iBAAiB,GAAG,KAAK,CAAC;QAC9B,IAAI,WAAW,GAAG,SAAS,CAAC;QAC5B,IAAI,WAAW,GAAG,SAAS,CAAC;QAC5B,IAAI,MAAM,GAAG,KAAK,CAAC;QACnB,OAAO,CAAC,MAAM,EAAE,CAAC;YACf,MAAM,MAAM,GAAqB,MAAM,GAAG,CAAC,WAAW,CACpD,QAAQ,CAAC,KAAK,CAAC,oBAAoB,EACnC;gBACE,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,cAAc;gBACd,gBAAgB;gBAChB,aAAa;gBACb,iBAAiB;gBACjB,WAAW;gBACX,WAAW;aACZ,CACF,CAAC;YACF,cAAc,GAAG,MAAM,CAAC,cAAc,CAAC;YACvC,gBAAgB,GAAG,MAAM,CAAC,gBAAgB,CAAC;YAC3C,aAAa,GAAG,MAAM,CAAC,aAAa,CAAC;YACrC,iBAAiB,GAAG,MAAM,CAAC,iBAAiB,IAAI,KAAK,CAAC;YACtD,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;YACjC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;YACjC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QACzB,CAAC;IACH,CAAC;IACD,OAAO,EAAE,CAAC,CAAC,IAAI,EAAE;CAClB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,uBAAuB,GAAG,QAAQ,CAAC;IAC9C,IAAI,EAAE;QACJ,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;KACnB;IACD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QAC3B,MAAM,MAAM,GAAG,MAAM,8BAA8B,CAAC,GAAG,EAAE;YACvD,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,cAAc,EAAE,IAAI;YACpB,aAAa,EAAE,IAAI;YACnB,gBAAgB,EAAE,IAAI;YACtB,iBAAiB,EAAE,KAAK;YACxB,WAAW,EAAE,SAAS;YACtB,WAAW,EAAE,SAAS;SACvB,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE;CACrB,CAAC,CAAC;AAEH,MAAM,aAAa,GAAG;IACpB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;IAClB,cAAc,EAAE,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IACpC,aAAa,EAAE,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IACnC,gBAAgB,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC;IAC3C,iBAAiB,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;IAC1C,WAAW,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IACnC,WAAW,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CACpC,CAAC;AAEF,MAAM,gBAAgB,GAAG;IACvB,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE;IACzB,gBAAgB,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC;IAC3C,cAAc,EAAE,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IACpC,iBAAiB,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;IAC1C,WAAW,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IACnC,WAAW,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;IACnC,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE;CACpB,CAAC;AAGF,MAAM,CAAC,MAAM,wBAAwB,GAAG,gBAAgB,CAAC;IACvD,IAAI,EAAE,aAAa;IACnB,OAAO,EAAE,8BAA8B;IACvC,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE;CACrB,CAAC,CAAC;AAEH,KAAK,UAAU,8BAA8B,CAC3C,GAAgB,EAChB,IAAmB;IAEnB,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;IACpD,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QACnB,MAAM,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,EAAE,QAAQ,CAAC,KAAK,CAAC,wBAAwB,EAAE;YACvE,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,cAAc,EAAE,MAAM,CAAC,cAAc;YACrC,aAAa,EAAE,MAAM,CAAC,aAAa;YACnC,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;YACzC,iBAAiB,EAAE,MAAM,CAAC,iBAAiB,IAAI,KAAK;YACpD,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,WAAW,EAAE,MAAM,CAAC,WAAW;SAChC,CAAC,CAAC;IACL,CAAC;IACD,OAAO,MAAM,CAAC,MAAM,CAAC;AACvB,CAAC;AAED,MAAM,CAAC,MAAM,oBAAoB,GAAG,gBAAgB,CAAC;IACnD,IAAI,EAAE,aAAa;IACnB,OAAO,EAAE,mBAAmB;IAC5B,OAAO,EAAE,gBAAgB;CAC1B,CAAC,CAAC;AACH,KAAK,UAAU,mBAAmB,CAChC,GAAgB,EAChB,IAAmB;IAEnB,IAAI,gBAAgB,GAAyB,IAAI,CAAC,gBAAgB,CAAC;IACnE,IAAI,aAAa,GAAkB,IAAI,CAAC,aAAa,CAAC;IACtD,IAAI,cAAc,GAAkB,IAAI,CAAC,cAAc,CAAC;IACxD,IAAI,iBAAiB,GAAY,IAAI,CAAC,iBAAiB,IAAI,KAAK,CAAC;IACjE,IAAI,WAAW,GAAuB,IAAI,CAAC,WAAW,CAAC;IACvD,IAAI,WAAW,GAAuB,IAAI,CAAC,WAAW,CAAC;IAEvD,wDAAwD;IACxD,IAAI,CAAC,aAAa,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxC,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,GAAG,CAAC,EAAE,EAAE,MAAM,CAAC;aAC5C,KAAK,CAAC,SAAS,CAAC;aAChB,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;aACvD,KAAK,CAAC,MAAM,CAAC;aACb,QAAQ,CAAC;YACR,QAAQ,EAAE,CAAC;YACX,MAAM,EAAE,IAAI,CAAC,aAAa,IAAI,IAAI;SACnC,CAAC,CAAC;QACL,aAAa,GAAG,OAAO,CAAC,cAAc,CAAC;QACvC,IAAI,OAAO,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;YACvC,cAAc,GAAG,IAAI,CAAC;YACtB,iBAAiB,GAAG,KAAK,CAAC;YAC1B,WAAW,GAAG,SAAS,CAAC;YACxB,WAAW,GAAG,SAAS,CAAC;QAC1B,CAAC;aAAM,CAAC;YACN,OAAO;gBACL,MAAM,EAAE,IAAI;gBACZ,aAAa;gBACb,gBAAgB;gBAChB,cAAc;gBACd,iBAAiB;gBACjB,WAAW;gBACX,WAAW;aACZ,CAAC;QACJ,CAAC;IACH,CAAC;IAED,kDAAkD;IAClD,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACvB,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,GAAG,CAAC,EAAE,EAAE,MAAM,CAAC;aAC7C,KAAK,CAAC,UAAU,CAAC;aACjB,SAAS,CAAC,sCAAsC,EAAE,CAAC,CAAC,EAAE,EAAE,CACvD,CAAC,CAAC,EAAE,CAAC,UAAU,EAAE,gBAAiB,CAAC,CACpC;aACA,KAAK,CAAC,MAAM,CAAC;aACb,QAAQ,CAAC;YACR,QAAQ,EAAE,GAAG;YACb,MAAM,EAAE,IAAI,CAAC,cAAc;SAC5B,CAAC,CAAC;QACL,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,aAAa,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAEnE,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;YACpB,oDAAoD;YACpD,iBAAiB,GAAG,IAAI,CAAC;YACzB,cAAc,GAAG,IAAI,CAAC;YACtB,WAAW,GAAG,SAAS,CAAC;YACxB,WAAW,GAAG,SAAS,CAAC;QAC1B,CAAC;aAAM,CAAC;YACN,cAAc,GAAG,QAAQ,CAAC,cAAc,CAAC;QAC3C,CAAC;QAED,OAAO;YACL,cAAc;YACd,aAAa;YACb,gBAAgB;YAChB,iBAAiB;YACjB,WAAW;YACX,WAAW;YACX,MAAM,EAAE,KAAK;SACd,CAAC;IACJ,CAAC;IAED,iDAAiD;IACjD,MAAM,YAAY,GAAG,MAAM,4BAA4B,CAAC,GAAG,EAAE;QAC3D,QAAQ,EAAE,gBAAiB;QAC3B,WAAW;QACX,WAAW;KACZ,CAAC,CAAC;IAEH,IAAI,YAAY,CAAC,MAAM,EAAE,CAAC;QACxB,gEAAgE;QAChE,MAAM,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;QACtC,gBAAgB,GAAG,IAAI,CAAC;QACxB,cAAc,GAAG,IAAI,CAAC;QACtB,iBAAiB,GAAG,KAAK,CAAC;QAC1B,WAAW,GAAG,SAAS,CAAC;QACxB,WAAW,GAAG,SAAS,CAAC;IAC1B,CAAC;SAAM,CAAC;QACN,iCAAiC;QACjC,WAAW,GAAG,YAAY,CAAC,WAAW,CAAC;QACvC,WAAW,GAAG,YAAY,CAAC,WAAW,CAAC;IACzC,CAAC;IAED,OAAO;QACL,cAAc;QACd,aAAa;QACb,gBAAgB;QAChB,iBAAiB;QACjB,WAAW;QACX,WAAW;QACX,MAAM,EAAE,KAAK;KACd,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,MAAM,eAAe,GAAG,aAAa,CAAC;IAC3C,IAAI,EAAE;QACJ,QAAQ,EAAE,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC;KAC1B;IACD,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IACtC,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QAC3B,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/C,OAAO,MAAM,EAAE,MAAM,IAAI,IAAI,CAAC;IAChC,CAAC;CACF,CAAC,CAAC"}

package/dist/component/vector/index.d.ts

@@ -0,0 +1,61 @@
+import type { Id } from "../_generated/dataModel.js";
+import { type ActionCtx, type MutationCtx } from "../_generated/server.js";
+import { type EmbeddingsWithoutDenormalizedFields, type VectorDimension } from "./tables.js";
+export declare const paginate: import("convex/server").RegisteredQuery<"public", {
+    table?: string | undefined;
+    cursor?: string | undefined;
+    limit: number;
+    vectorDimension: 128 | 256 | 512 | 768 | 1024 | 1408 | 1536 | 2048 | 3072 | 4096;
+    targetModel: string;
+}, Promise<{
+    ids: (import("convex/values").GenericId<"embeddings_128"> | import("convex/values").GenericId<"embeddings_256"> | import("convex/values").GenericId<"embeddings_512"> | import("convex/values").GenericId<"embeddings_768"> | import("convex/values").GenericId<"embeddings_1024"> | import("convex/values").GenericId<"embeddings_1408"> | import("convex/values").GenericId<"embeddings_1536"> | import("convex/values").GenericId<"embeddings_2048"> | import("convex/values").GenericId<"embeddings_3072"> | import("convex/values").GenericId<"embeddings_4096">)[];
+    isDone: boolean;
+    continueCursor: string;
+}>>;
+export declare const deleteBatchForThread: import("convex/server").RegisteredMutation<"public", {
+    cursor?: string | undefined;
+    threadId: string;
+    model: string;
+    limit: number;
+    vectorDimension: 128 | 256 | 512 | 768 | 1024 | 1408 | 1536 | 2048 | 3072 | 4096;
+}, Promise<{
+    isDone: boolean;
+    continueCursor: string;
+}>>;
+export declare const insertBatch: import("convex/server").RegisteredMutation<"public", {
+    vectors: {
+        userId?: string | undefined;
+        threadId?: string | undefined;
+        messageId?: import("convex/values").GenericId<"messages"> | undefined;
+        model: string;
+        table: string;
+        vector: number[];
+    }[];
+    vectorDimension: 128 | 256 | 512 | 768 | 1024 | 1408 | 1536 | 2048 | 3072 | 4096;
+}, Promise<(string & {
+    __tableName: "embeddings_128" | "embeddings_256" | "embeddings_512" | "embeddings_768" | "embeddings_1024" | "embeddings_1408" | "embeddings_1536" | "embeddings_2048" | "embeddings_3072" | "embeddings_4096";
+})[]>>;
+export declare function insertVector(ctx: MutationCtx, dimension: VectorDimension, v: EmbeddingsWithoutDenormalizedFields): Promise<import("convex/values").GenericId<"embeddings_128" | "embeddings_256" | "embeddings_512" | "embeddings_768" | "embeddings_1024" | "embeddings_1408" | "embeddings_1536" | "embeddings_2048" | "embeddings_3072" | "embeddings_4096">>;
+export declare function searchVectors(ctx: ActionCtx, vector: number[], args: {
+    dimension: VectorDimension;
+    model: string;
+    table: string;
+    userId?: string;
+    threadId?: Id<"threads">;
+    searchAllMessagesForUserId?: string;
+    limit?: number;
+}): Promise<{
+    _id: import("convex/values").GenericId<"embeddings_128" | "embeddings_256" | "embeddings_512" | "embeddings_768" | "embeddings_1024" | "embeddings_1408" | "embeddings_1536" | "embeddings_2048" | "embeddings_3072" | "embeddings_4096">;
+    _score: number;
+}[]>;
+export declare const updateBatch: import("convex/server").RegisteredMutation<"public", {
+    vectors: {
+        id: import("convex/values").GenericId<"embeddings_128" | "embeddings_256" | "embeddings_512" | "embeddings_768" | "embeddings_1024" | "embeddings_1408" | "embeddings_1536" | "embeddings_2048" | "embeddings_3072" | "embeddings_4096">;
+        model: string;
+        vector: number[];
+    }[];
+}, Promise<void>>;
+export declare const deleteBatch: import("convex/server").RegisteredMutation<"public", {
+    ids: import("convex/values").GenericId<"embeddings_128" | "embeddings_256" | "embeddings_512" | "embeddings_768" | "embeddings_1024" | "embeddings_1408" | "embeddings_1536" | "embeddings_2048" | "embeddings_3072" | "embeddings_4096">[];
+}, Promise<void>>;
+//# sourceMappingURL=index.d.ts.map

package/dist/component/vector/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/component/vector/index.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,EAAE,EAAE,MAAM,4BAA4B,CAAC;AACrD,OAAO,EACL,KAAK,SAAS,EAEd,KAAK,WAAW,EAEjB,MAAM,yBAAyB,CAAC;AAEjC,OAAO,EACL,KAAK,mCAAmC,EAExC,KAAK,eAAe,EAIrB,MAAM,aAAa,CAAC;AAErB,eAAO,MAAM,QAAQ;;;;;;;;;;GAiCnB,CAAC;AAEH,eAAO,MAAM,oBAAoB;;;;;;;;;GAqC/B,CAAC;AAEH,eAAO,MAAM,WAAW;;;;;;;;;;;;MAsBtB,CAAC;AAEH,wBAAsB,YAAY,CAChC,GAAG,EAAE,WAAW,EAChB,SAAS,EAAE,eAAe,EAC1B,CAAC,EAAE,mCAAmC,iPASvC;AAED,wBAAgB,aAAa,CAC3B,GAAG,EAAE,SAAS,EACd,MAAM,EAAE,MAAM,EAAE,EAChB,IAAI,EAAE;IACJ,SAAS,EAAE,eAAe,CAAC;IAC3B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,EAAE,CAAC,SAAS,CAAC,CAAC;IACzB,0BAA0B,CAAC,EAAE,MAAM,CAAC;IACpC,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;;;KAoBF;AAED,eAAO,MAAM,WAAW;;;;;;iBAyBtB,CAAC;AAEH,eAAO,MAAM,WAAW;;iBAQtB,CAAC"}