@convex-dev/agent 0.0.16 → 0.0.17-alpha.0

package/dist/commonjs/client/playground.js

@@ -0,0 +1,200 @@
+import { paginationOptsValidator, queryGeneric, mutationGeneric, actionGeneric, } from "convex/server";
+import { vMessageDoc, vThreadDoc, paginationResultValidator, vMessage, vContextOptions, vStorageOptions, } from "./index";
+import { v } from "convex/values";
+// Playground API definition
+export function definePlaygroundAPI(component, { agents, userNameLookup, }) {
+    // Map agent name to instance
+    const agentMap = Object.fromEntries(agents.map((agent) => [agent.options.name, agent]));
+    async function validateApiKey(ctx, apiKey) {
+        await ctx.runQuery(component.apiKeys.validate, { apiKey });
+    }
+    const isApiKeyValid = queryGeneric({
+        args: {
+            apiKey: v.string(),
+        },
+        handler: async (ctx, args) => {
+            try {
+                await validateApiKey(ctx, args.apiKey);
+                return true;
+                // eslint-disable-next-line @typescript-eslint/no-unused-vars
+            }
+            catch (e) {
+                return false;
+            }
+        },
+        returns: v.boolean(),
+    });
+    // List all agents
+    const listAgents = queryGeneric({
+        args: {
+            apiKey: v.string(),
+        },
+        handler: async (ctx, args) => {
+            await validateApiKey(ctx, args.apiKey);
+            return Object.keys(agentMap);
+        },
+        returns: v.array(v.string()),
+    });
+    const listUsers = queryGeneric({
+        args: {
+            apiKey: v.string(),
+            paginationOpts: paginationOptsValidator,
+        },
+        handler: async (ctx, args) => {
+            await validateApiKey(ctx, args.apiKey);
+            const users = await ctx.runQuery(component.users.listUsersWithThreads, {
+                paginationOpts: args.paginationOpts,
+            });
+            return {
+                ...users,
+                page: await Promise.all(users.page.map(async (userId) => ({
+                    _id: userId,
+                    name: userNameLookup ? await userNameLookup(ctx, userId) : userId,
+                }))),
+            };
+        },
+        returns: paginationResultValidator(v.object({
+            _id: v.string(),
+            name: v.string(),
+        })),
+    });
+    // List threads for a user (query)
+    const listThreads = queryGeneric({
+        args: {
+            apiKey: v.string(),
+            userId: v.string(),
+            paginationOpts: paginationOptsValidator,
+        },
+        handler: async (ctx, args) => {
+            await validateApiKey(ctx, args.apiKey);
+            const results = await ctx.runQuery(component.threads.listThreadsByUserId, {
+                userId: args.userId,
+                paginationOpts: args.paginationOpts,
+                order: "desc",
+            });
+            return {
+                ...results,
+                page: await Promise.all(results.page.map(async (thread) => {
+                    const { page: [last], } = await ctx.runQuery(component.messages.listMessagesByThreadId, {
+                        threadId: thread._id,
+                        order: "desc",
+                        paginationOpts: {
+                            numItems: 1,
+                            cursor: null,
+                        },
+                    });
+                    return {
+                        ...thread,
+                        latestMessage: last?.text,
+                        lastMessageAt: last?._creationTime,
+                    };
+                })),
+            };
+        },
+        returns: paginationResultValidator(v.object({
+            ...vThreadDoc.fields,
+            latestMessage: v.optional(v.string()),
+            lastMessageAt: v.optional(v.number()),
+        })),
+    });
+    // List messages for a thread (query)
+    const listMessages = queryGeneric({
+        args: {
+            apiKey: v.string(),
+            threadId: v.string(),
+            paginationOpts: paginationOptsValidator,
+        },
+        handler: async (ctx, args) => {
+            await validateApiKey(ctx, args.apiKey);
+            return ctx.runQuery(component.messages.listMessagesByThreadId, {
+                threadId: args.threadId,
+                paginationOpts: args.paginationOpts,
+                order: "desc",
+                statuses: ["success", "failed", "pending"],
+            });
+        },
+        returns: paginationResultValidator(vMessageDoc),
+    });
+    // Create a thread (mutation)
+    const createThread = mutationGeneric({
+        args: {
+            apiKey: v.string(),
+            agentName: v.string(),
+            userId: v.string(),
+            title: v.optional(v.string()),
+            summary: v.optional(v.string()),
+        },
+        handler: async (ctx, args) => {
+            await validateApiKey(ctx, args.apiKey);
+            const agent = agentMap[args.agentName];
+            if (!agent)
+                throw new Error(`Unknown agent: ${args.agentName}`);
+            return agent.createThread(ctx, {
+                userId: args.userId,
+                title: args.title,
+                summary: args.summary,
+            });
+        },
+        returns: v.object({ threadId: v.string() }),
+    });
+    // Send a message (action)
+    const generateText = actionGeneric({
+        args: {
+            apiKey: v.string(),
+            agentName: v.string(),
+            userId: v.string(),
+            threadId: v.string(),
+            // Args passed through to generateText
+            prompt: v.optional(v.string()),
+            messages: v.optional(v.array(vMessage)),
+            contextOptions: v.optional(vContextOptions),
+            storageOptions: v.optional(vStorageOptions),
+        },
+        handler: async (ctx, args) => {
+            const { apiKey, agentName, userId, threadId, ...rest } = args;
+            await validateApiKey(ctx, apiKey);
+            const agent = agentMap[agentName];
+            if (!agent)
+                throw new Error(`Unknown agent: ${agentName}`);
+            const { thread } = await agent.continueThread(ctx, { threadId, userId });
+            const result = await thread.generateText(rest);
+            return result;
+        },
+    });
+    // Fetch prompt context (action)
+    const fetchPromptContext = actionGeneric({
+        args: {
+            apiKey: v.string(),
+            agentName: v.string(),
+            userId: v.optional(v.string()),
+            threadId: v.optional(v.string()),
+            messages: v.array(vMessage),
+            contextOptions: vContextOptions,
+            beforeMessageId: v.optional(v.string()),
+        },
+        handler: async (ctx, args) => {
+            await validateApiKey(ctx, args.apiKey);
+            const agent = agentMap[args.agentName];
+            if (!agent)
+                throw new Error(`Unknown agent: ${args.agentName}`);
+            return agent.fetchContextMessages(ctx, {
+                userId: args.userId,
+                threadId: args.threadId,
+                messages: args.messages,
+                contextOptions: args.contextOptions,
+                beforeMessageId: args.beforeMessageId,
+            });
+        },
+    });
+    return {
+        isApiKeyValid,
+        listUsers,
+        listThreads,
+        listMessages,
+        listAgents,
+        createThread,
+        generateText,
+        fetchPromptContext,
+    };
+}
+//# sourceMappingURL=playground.js.map

package/dist/commonjs/client/playground.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"playground.js","sourceRoot":"","sources":["../../../src/client/playground.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,uBAAuB,EACvB,YAAY,EACZ,eAAe,EACf,aAAa,GAId,MAAM,eAAe,CAAC;AACvB,OAAO,EACL,WAAW,EACX,UAAU,EACV,yBAAyB,EACzB,QAAQ,EACR,eAAe,EACf,eAAe,GAGhB,MAAM,SAAS,CAAC;AAEjB,OAAO,EAAE,CAAC,EAAE,MAAM,eAAe,CAAC;AAMlC,4BAA4B;AAC5B,MAAM,UAAU,mBAAmB,CACjC,SAAyB,EACzB,EACE,MAAM,EACN,cAAc,GAOf;IAED,6BAA6B;IAC7B,MAAM,QAAQ,GAAmC,MAAM,CAAC,WAAW,CACjE,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CACnD,CAAC;IAEF,KAAK,UAAU,cAAc,CAAC,GAAgB,EAAE,MAAc;QAC5D,MAAM,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;IAC7D,CAAC;IAED,MAAM,aAAa,GAAG,YAAY,CAAC;QACjC,IAAI,EAAE;YACJ,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;SACnB;QACD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YAC3B,IAAI,CAAC;gBACH,MAAM,cAAc,CAAC,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;gBACvC,OAAO,IAAI,CAAC;gBACZ,6DAA6D;YAC/D,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,OAAO,KAAK,CAAC;YACf,CAAC;QACH,CAAC;QACD,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE;KACrB,CAAC,CAAC;IAEH,kBAAkB;IAClB,MAAM,UAAU,GAAG,YAAY,CAAC;QAC9B,IAAI,EAAE;YACJ,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;SACnB;QACD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YAC3B,MAAM,cAAc,CAAC,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YACvC,OAAO,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC;QACD,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;KAC7B,CAAC,CAAC;IAEH,MAAM,SAAS,GAAG,YAAY,CAAC;QAC7B,IAAI,EAAE;YACJ,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;YAClB,cAAc,EAAE,uBAAuB;SACxC;QACD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YAC3B,MAAM,cAAc,CAAC,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YACvC,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,KAAK,CAAC,oBAAoB,EAAE;gBACrE,cAAc,EAAE,IAAI,CAAC,cAAc;aACpC,CAAC,CAAC;YACH,OAAO;gBACL,GAAG,KAAK;gBACR,IAAI,EAAE,MAAM,OAAO,CAAC,GAAG,CACrB,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;oBAChC,GAAG,EAAE,MAAM;oBACX,IAAI,EAAE,cAAc,CAAC,CAAC,CAAC,MAAM,cAAc,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM;iBAClE,CAAC,CAAC,CACJ;aACF,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,yBAAyB,CAChC,CAAC,CAAC,MAAM,CAAC;YACP,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE;YACf,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;SACjB,CAAC,CACH;KACF,CAAC,CAAC;IAEH,kCAAkC;IAClC,MAAM,WAAW,GAAG,YAAY,CAAC;QAC/B,IAAI,EAAE;YACJ,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;YAClB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;YAClB,cAAc,EAAE,uBAAuB;SACxC;QACD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YAC3B,MAAM,cAAc,CAAC,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YACvC,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,QAAQ,CAChC,SAAS,CAAC,OAAO,CAAC,mBAAmB,EACrC;gBACE,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,cAAc,EAAE,IAAI,CAAC,cAAc;gBACnC,KAAK,EAAE,MAAM;aACd,CACF,CAAC;YACF,OAAO;gBACL,GAAG,OAAO;gBACV,IAAI,EAAE,MAAM,OAAO,CAAC,GAAG,CACrB,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE;oBAChC,MAAM,EACJ,IAAI,EAAE,CAAC,IAAI,CAAC,GACb,GAAG,MAAM,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,QAAQ,CAAC,sBAAsB,EAAE;wBAChE,QAAQ,EAAE,MAAM,CAAC,GAAG;wBACpB,KAAK,EAAE,MAAM;wBACb,cAAc,EAAE;4BACd,QAAQ,EAAE,CAAC;4BACX,MAAM,EAAE,IAAI;yBACb;qBACF,CAAC,CAAC;oBACH,OAAO;wBACL,GAAG,MAAM;wBACT,aAAa,EAAE,IAAI,EAAE,IAAI;wBACzB,aAAa,EAAE,IAAI,EAAE,aAAa;qBACnC,CAAC;gBACJ,CAAC,CAAC,CACH;aACF,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,yBAAyB,CAChC,CAAC,CAAC,MAAM,CAAC;YACP,GAAG,UAAU,CAAC,MAAM;YACpB,aAAa,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;YACrC,aAAa,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;SACtC,CAAC,CACH;KACF,CAAC,CAAC;IAEH,qCAAqC;IACrC,MAAM,YAAY,GAAG,YAAY,CAAC;QAChC,IAAI,EAAE;YACJ,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;YAClB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;YACpB,cAAc,EAAE,uBAAuB;SACxC;QACD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YAC3B,MAAM,cAAc,CAAC,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YACvC,OAAO,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,QAAQ,CAAC,sBAAsB,EAAE;gBAC7D,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,cAAc,EAAE,IAAI,CAAC,cAAc;gBACnC,KAAK,EAAE,MAAM;gBACb,QAAQ,EAAE,CAAC,SAAS,EAAE,QAAQ,EAAE,SAAS,CAAC;aAC3C,CAAC,CAAC;QACL,CAAC;QACD,OAAO,EAAE,yBAAyB,CAAC,WAAW,CAAC;KAChD,CAAC,CAAC;IAEH,6BAA6B;IAC7B,MAAM,YAAY,GAAG,eAAe,CAAC;QACnC,IAAI,EAAE;YACJ,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;YAClB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;YACrB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;YAClB,KAAK,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;YAC7B,OAAO,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;SAChC;QACD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YAC3B,MAAM,cAAc,CAAC,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YACvC,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACvC,IAAI,CAAC,KAAK;gBAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;YAChE,OAAO,KAAK,CAAC,YAAY,CAAC,GAAG,EAAE;gBAC7B,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,OAAO,EAAE,IAAI,CAAC,OAAO;aACtB,CAAC,CAAC;QACL,CAAC;QACD,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;KAC5C,CAAC,CAAC;IAEH,0BAA0B;IAC1B,MAAM,YAAY,GAAG,aAAa,CAAC;QACjC,IAAI,EAAE;YACJ,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;YAClB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;YACrB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;YAClB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;YACpB,sCAAsC;YACtC,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;YAC9B,QAAQ,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YACvC,cAAc,EAAE,CAAC,CAAC,QAAQ,CAAC,eAAe,CAAC;YAC3C,cAAc,EAAE,CAAC,CAAC,QAAQ,CAAC,eAAe,CAAC;SAC5C;QACD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YAC3B,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAI,EAAE,GAAG,IAAI,CAAC;YAC9D,MAAM,cAAc,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;YAClC,MAAM,KAAK,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC;YAClC,IAAI,CAAC,KAAK;gBAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,SAAS,EAAE,CAAC,CAAC;YAC3D,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,KAAK,CAAC,cAAc,CAAC,GAAG,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;YACzE,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YAC/C,OAAO,MAAM,CAAC;QAChB,CAAC;KACF,CAAC,CAAC;IAEH,gCAAgC;IAChC,MAAM,kBAAkB,GAAG,aAAa,CAAC;QACvC,IAAI,EAAE;YACJ,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;YAClB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;YACrB,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;YAC9B,QAAQ,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;YAChC,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC;YAC3B,cAAc,EAAE,eAAe;YAC/B,eAAe,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;SACxC;QACD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YAC3B,MAAM,cAAc,CAAC,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YACvC,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACvC,IAAI,CAAC,KAAK;gBAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;YAChE,OAAO,KAAK,CAAC,oBAAoB,CAAC,GAAG,EAAE;gBACrC,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,cAAc,EAAE,IAAI,CAAC,cAAc;gBACnC,eAAe,EAAE,IAAI,CAAC,eAAe;aACtC,CAAC,CAAC;QACL,CAAC;KACF,CAAC,CAAC;IAEH,OAAO;QACL,aAAa;QACb,SAAS;QACT,WAAW;QACX,YAAY;QACZ,UAAU;QACV,YAAY;QACZ,YAAY;QACZ,kBAAkB;KACnB,CAAC;AACJ,CAAC"}

package/dist/commonjs/component/apiKeys.d.ts

@@ -0,0 +1,11 @@
+export declare const issue: import("convex/server").RegisteredMutation<"public", {
+    name?: string | undefined;
+}, Promise<import("convex/values").GenericId<"apiKeys">>>;
+export declare const validate: import("convex/server").RegisteredQuery<"public", {
+    apiKey: import("convex/values").GenericId<"apiKeys">;
+}, Promise<boolean>>;
+export declare const destroy: import("convex/server").RegisteredMutation<"public", {
+    name?: string | undefined;
+    apiKey?: import("convex/values").GenericId<"apiKeys"> | undefined;
+}, Promise<"missing" | "deleted" | "name mismatch" | "must provide either apiKey or name">>;
+//# sourceMappingURL=apiKeys.d.ts.map

package/dist/commonjs/component/apiKeys.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"apiKeys.d.ts","sourceRoot":"","sources":["../../../src/component/apiKeys.ts"],"names":[],"mappings":"AAGA,eAAO,MAAM,KAAK;;yDAmBhB,CAAC;AAEH,eAAO,MAAM,QAAQ;;oBAYnB,CAAC;AAEH,eAAO,MAAM,OAAO;;;2FAmClB,CAAC"}

package/dist/commonjs/component/apiKeys.js

@@ -0,0 +1,69 @@
+import { v } from "convex/values";
+import { mutation, query } from "./_generated/server";
+export const issue = mutation({
+    args: {
+        name: v.optional(v.string()),
+    },
+    handler: async (ctx, args) => {
+        if (args.name) {
+            const existingApiKey = await ctx.db
+                .query("apiKeys")
+                .withIndex("name", (q) => q.eq("name", args.name))
+                .first();
+            if (existingApiKey) {
+                console.warn(`API key ${args.name} already exists, deleting...`);
+                await ctx.db.delete(existingApiKey._id);
+            }
+        }
+        const apiKey = await ctx.db.insert("apiKeys", args);
+        return apiKey;
+    },
+    returns: v.id("apiKeys"),
+});
+export const validate = query({
+    args: {
+        apiKey: v.id("apiKeys"),
+    },
+    handler: async (ctx, args) => {
+        const apiKey = await ctx.db.get(args.apiKey);
+        if (!apiKey) {
+            throw new Error("Invalid API key");
+        }
+        return true;
+    },
+    returns: v.boolean(),
+});
+export const destroy = mutation({
+    args: v.object({
+        apiKey: v.optional(v.id("apiKeys")),
+        name: v.optional(v.string()),
+    }),
+    handler: async (ctx, args) => {
+        if (args.apiKey) {
+            const apiKey = await ctx.db.get(args.apiKey);
+            if (!apiKey) {
+                return "missing";
+            }
+            if (apiKey.name !== args.name) {
+                return "name mismatch";
+            }
+            await ctx.db.delete(args.apiKey);
+        }
+        else if (args.name) {
+            const apiKey = await ctx.db
+                .query("apiKeys")
+                .withIndex("name", (q) => q.eq("name", args.name))
+                .first();
+            if (!apiKey) {
+                return "missing";
+            }
+            await ctx.db.delete(apiKey._id);
+        }
+        else {
+            return "must provide either apiKey or name";
+        }
+        return "deleted";
+    },
+    returns: v.union(v.literal("missing"), v.literal("deleted"), v.literal("name mismatch"), v.literal("must provide either apiKey or name")),
+});
+//# sourceMappingURL=apiKeys.js.map

package/dist/commonjs/component/apiKeys.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"apiKeys.js","sourceRoot":"","sources":["../../../src/component/apiKeys.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,eAAe,CAAC;AAClC,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AAEtD,MAAM,CAAC,MAAM,KAAK,GAAG,QAAQ,CAAC;IAC5B,IAAI,EAAE;QACJ,IAAI,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;KAC7B;IACD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QAC3B,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACd,MAAM,cAAc,GAAG,MAAM,GAAG,CAAC,EAAE;iBAChC,KAAK,CAAC,SAAS,CAAC;iBAChB,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;iBACjD,KAAK,EAAE,CAAC;YACX,IAAI,cAAc,EAAE,CAAC;gBACnB,OAAO,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,IAAI,8BAA8B,CAAC,CAAC;gBACjE,MAAM,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QACpD,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC;CACzB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,QAAQ,GAAG,KAAK,CAAC;IAC5B,IAAI,EAAE;QACJ,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC;KACxB;IACD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QAC3B,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;QACrC,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE;CACrB,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,OAAO,GAAG,QAAQ,CAAC;IAC9B,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;QACb,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC;QACnC,IAAI,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;KAC7B,CAAC;IACF,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QAC3B,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC7C,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,IAAI,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI,EAAE,CAAC;gBAC9B,OAAO,eAAe,CAAC;YACzB,CAAC;YACD,MAAM,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnC,CAAC;aAAM,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACrB,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,EAAE;iBACxB,KAAK,CAAC,SAAS,CAAC;iBAChB,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;iBACjD,KAAK,EAAE,CAAC;YACX,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,MAAM,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAClC,CAAC;aAAM,CAAC;YACN,OAAO,oCAAoC,CAAC;QAC9C,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,EAAE,CAAC,CAAC,KAAK,CACd,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,EACpB,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,EACpB,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC,EAC1B,CAAC,CAAC,OAAO,CAAC,oCAAoC,CAAC,CAChD;CACF,CAAC,CAAC"}

package/dist/commonjs/component/files.d.ts

@@ -0,0 +1,31 @@
+import { MutationCtx } from "./_generated/server.js";
+import { Id } from "./_generated/dataModel.js";
+export declare const addFile: import("convex/server").RegisteredMutation<"public", {
+    storageId: string;
+    hash: string;
+}, Promise<import("convex/values").GenericId<"files">>>;
+export declare function addFileHandler(ctx: MutationCtx, args: {
+    storageId: string;
+    hash: string;
+}): Promise<import("convex/values").GenericId<"files">>;
+export declare const resuseFile: import("convex/server").RegisteredMutation<"public", {
+    fileId: Id<"files">;
+}, Promise<void>>;
+export declare function reuseFileHandler(ctx: MutationCtx, args: {
+    fileId: Id<"files">;
+}): Promise<void>;
+export declare const getFilesToDelete: import("convex/server").RegisteredQuery<"public", {
+    limit?: number | undefined;
+    cursor?: string | undefined;
+}, Promise<{
+    files: {
+        _id: import("convex/values").GenericId<"files">;
+        _creationTime: number;
+        storageId: string;
+        hash: string;
+        refcount: number;
+    }[];
+    continueCursor: string;
+    isDone: boolean;
+}>>;
+//# sourceMappingURL=files.d.ts.map

package/dist/commonjs/component/files.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"files.d.ts","sourceRoot":"","sources":["../../../src/component/files.ts"],"names":[],"mappings":"AACA,OAAO,EAAY,WAAW,EAAS,MAAM,wBAAwB,CAAC;AAEtE,OAAO,EAAE,EAAE,EAAE,MAAM,2BAA2B,CAAC;AAE/C,eAAO,MAAM,OAAO;eAWC,MAAM;UAAQ,MAAM;uDAJvC,CAAC;AAEH,wBAAsB,cAAc,CAClC,GAAG,EAAE,WAAW,EAChB,IAAI,EAAE;IAAE,SAAS,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,uDAQ1C;AAED,eAAO,MAAM,UAAU;YAUL,EAAE,CAAC,OAAO,CAAC;iBAJ3B,CAAC;AAEH,wBAAsB,gBAAgB,CACpC,GAAG,EAAE,WAAW,EAChB,IAAI,EAAE;IAAE,MAAM,EAAE,EAAE,CAAC,OAAO,CAAC,CAAA;CAAE,iBAS9B;AAED,eAAO,MAAM,gBAAgB;;;;;;;;;;;;;GAwB3B,CAAC"}

package/dist/commonjs/component/files.js

@@ -0,0 +1,61 @@
+import { paginator } from "convex-helpers/server/pagination";
+import { mutation, query } from "./_generated/server.js";
+import { schema, v } from "./schema.js";
+export const addFile = mutation({
+    args: {
+        storageId: v.string(),
+        hash: v.string(),
+    },
+    handler: addFileHandler,
+    returns: v.id("files"),
+});
+export async function addFileHandler(ctx, args) {
+    const fileId = await ctx.db.insert("files", {
+        storageId: args.storageId,
+        hash: args.hash,
+        refcount: 1,
+    });
+    return fileId;
+}
+export const resuseFile = mutation({
+    args: {
+        fileId: v.id("files"),
+    },
+    handler: reuseFileHandler,
+    returns: v.null(),
+});
+export async function reuseFileHandler(ctx, args) {
+    const file = await ctx.db.get(args.fileId);
+    if (!file) {
+        throw new Error("File not found");
+    }
+    await ctx.db.patch(args.fileId, {
+        refcount: file.refcount + 1,
+    });
+}
+export const getFilesToDelete = query({
+    args: {
+        cursor: v.optional(v.string()),
+        limit: v.optional(v.number()),
+    },
+    handler: async (ctx, args) => {
+        const files = await paginator(ctx.db, schema)
+            .query("files")
+            .withIndex("refcount", (q) => q.eq("refcount", 0))
+            .paginate({
+            numItems: args.limit ?? 100,
+            cursor: args.cursor ?? null,
+        });
+        return {
+            files: files.page,
+            continueCursor: files.continueCursor,
+            isDone: files.isDone,
+        };
+    },
+    returns: v.object({
+        files: v.array(v.doc("files")),
+        continueCursor: v.string(),
+        isDone: v.boolean(),
+    }),
+});
+//# sourceMappingURL=files.js.map

package/dist/commonjs/component/files.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"files.js","sourceRoot":"","sources":["../../../src/component/files.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,kCAAkC,CAAC;AAC7D,OAAO,EAAE,QAAQ,EAAe,KAAK,EAAE,MAAM,wBAAwB,CAAC;AACtE,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,aAAa,CAAC;AAGxC,MAAM,CAAC,MAAM,OAAO,GAAG,QAAQ,CAAC;IAC9B,IAAI,EAAE;QACJ,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;QACrB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;KACjB;IACD,OAAO,EAAE,cAAc;IACvB,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC;CACvB,CAAC,CAAC;AAEH,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,GAAgB,EAChB,IAAyC;IAEzC,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,OAAO,EAAE;QAC1C,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,QAAQ,EAAE,CAAC;KACZ,CAAC,CAAC;IACH,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,CAAC,MAAM,UAAU,GAAG,QAAQ,CAAC;IACjC,IAAI,EAAE;QACJ,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC;KACtB;IACD,OAAO,EAAE,gBAAgB;IACzB,OAAO,EAAE,CAAC,CAAC,IAAI,EAAE;CAClB,CAAC,CAAC;AAEH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,GAAgB,EAChB,IAA6B;IAE7B,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC3C,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,MAAM,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC;IACpC,CAAC;IACD,MAAM,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE;QAC9B,QAAQ,EAAE,IAAI,CAAC,QAAQ,GAAG,CAAC;KAC5B,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,MAAM,gBAAgB,GAAG,KAAK,CAAC;IACpC,IAAI,EAAE;QACJ,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;QAC9B,KAAK,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;KAC9B;IACD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QAC3B,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,GAAG,CAAC,EAAE,EAAE,MAAM,CAAC;aAC1C,KAAK,CAAC,OAAO,CAAC;aACd,SAAS,CAAC,UAAU,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;aACjD,QAAQ,CAAC;YACR,QAAQ,EAAE,IAAI,CAAC,KAAK,IAAI,GAAG;YAC3B,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,IAAI;SAC5B,CAAC,CAAC;QACL,OAAO;YACL,KAAK,EAAE,KAAK,CAAC,IAAI;YACjB,cAAc,EAAE,KAAK,CAAC,cAAc;YACpC,MAAM,EAAE,KAAK,CAAC,MAAM;SACrB,CAAC;IACJ,CAAC;IACD,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC;QAChB,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC9B,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE;QAC1B,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE;KACpB,CAAC;CACH,CAAC,CAAC"}