@convex-dev/ai-budget 0.0.2-alpha.0 → 0.0.2-alpha.12

package/dist/component/_generated/component.d.ts

@@ -20,22 +20,28 @@ import type { FunctionReference } from "convex/server";
  */
 export type ComponentApi<Name extends string | undefined = string | undefined> = {
     lib: {
-        bumpAction: FunctionReference<"mutation", "internal", {
+        adjustBucket: FunctionReference<"mutation", "internal", {
+            deltaNanos: number;
+            dimension: string;
+            reason?: string;
+            tokens?: number;
+            value: string;
+        }, null, Name>;
+        bumpBucket: FunctionReference<"mutation", "internal", {
             dailyNanos?: number;
+            dimension: string;
             lifetimeNanos?: number;
-            name: string;
+            monthlyNanos?: number;
+            value: string;
         }, null, Name>;
         bumpGlobal: FunctionReference<"mutation", "internal", {
             dailyNanos?: number;
             lifetimeNanos?: number;
+            monthlyNanos?: number;
         }, null, Name>;
-        bumpUser: FunctionReference<"mutation", "internal", {
-            dailyNanos?: number;
-            lifetimeNanos?: number;
-            userId: string;
-        }, null, Name>;
-        deleteUser: FunctionReference<"mutation", "internal", {
-            userId: string;
+        deleteBucket: FunctionReference<"mutation", "internal", {
+            dimension: string;
+            value: string;
         }, {
             deletedThisBatch: number;
             done: boolean;
@@ -43,18 +49,26 @@ export type ComponentApi<Name extends string | undefined = string | undefined> =
         finishRequest: FunctionReference<"mutation", "internal", {
             cachedTokens?: number;
             completionTokens?: number;
+            costNanos?: number;
             error?: string;
             latencyMs?: number;
             promptTokens?: number;
             requestId: string;
             responseText?: string;
+            serverToolUses?: Record<string, number>;
         }, {
             costNanos: number;
         }, Name>;
+        getBucket: FunctionReference<"query", "internal", {
+            dimension: string;
+            value: string;
+        }, any, Name>;
         getGlobalStatus: FunctionReference<"query", "internal", {}, {
             dailySpendLimitNanos: number | null;
+            defaultWarnAtPct: number | null;
             enforcement: "hard" | "soft";
             lifetimeSpendLimitNanos: number | null;
+            retentionMs: number | null;
             spentTodayNanos: number;
             spentTotalNanos: number;
         }, Name>;
@@ -68,42 +82,51 @@ export type ComponentApi<Name extends string | undefined = string | undefined> =
         lineage: FunctionReference<"query", "internal", {
             requestId: string;
         }, any, Name>;
-        listActions: FunctionReference<"query", "internal", {}, any, Name>;
+        listAdjustments: FunctionReference<"query", "internal", {
+            dimension: string;
+            limit?: number;
+            value: string;
+        }, any, Name>;
+        listBuckets: FunctionReference<"query", "internal", {
+            dimension?: string;
+        }, any, Name>;
         listPrices: FunctionReference<"query", "internal", {}, any, Name>;
         listRequests: FunctionReference<"query", "internal", {
+            dimension?: string;
             limit?: number;
             userId?: string;
+            value?: string;
         }, any, Name>;
-        listUsers: FunctionReference<"query", "internal", {}, any, Name>;
-        setActionLimits: FunctionReference<"mutation", "internal", {
+        listServerToolPrices: FunctionReference<"query", "internal", {}, any, Name>;
+        setAlertDefaults: FunctionReference<"mutation", "internal", {
+            warnAtPct?: number;
+        }, null, Name>;
+        setBucketLimits: FunctionReference<"mutation", "internal", {
+            blocked?: boolean;
             dailySpendLimitNanos?: number;
             dailyTokenLimit?: number;
-            disabled?: boolean;
+            dimension: string;
             enforcement?: "hard" | "soft";
             lifetimeSpendLimitNanos?: number;
             lifetimeTokenLimit?: number;
-            name: string;
+            maxConcurrent?: number;
+            monthlySpendLimitNanos?: number;
+            monthlyTokenLimit?: number;
+            requestsPerMinute?: number;
+            value: string;
+            warnAtPct?: number;
         }, null, Name>;
         setGlobalLimits: FunctionReference<"mutation", "internal", {
             dailySpendLimitNanos?: number;
             enforcement?: "hard" | "soft";
             lifetimeSpendLimitNanos?: number;
         }, null, Name>;
-        setLimits: FunctionReference<"mutation", "internal", {
-            blocked?: boolean;
-            dailySpendLimitNanos?: number;
-            dailyTokenLimit?: number;
-            enforcement?: "hard" | "soft";
-            lifetimeSpendLimitNanos?: number;
-            lifetimeTokenLimit?: number;
-            requestsPerMinute?: number;
-            userId: string;
-        }, null, Name>;
         setModelPolicy: FunctionReference<"mutation", "internal", {
             mode: "open" | "allowlist" | "denylist";
             models: Array<string>;
         }, null, Name>;
         setPrice: FunctionReference<"mutation", "internal", {
+            cachedNanosPerMTok?: number;
             inputNanosPerMTok: number;
             model: string;
             outputNanosPerMTok: number;
@@ -111,6 +134,10 @@ export type ComponentApi<Name extends string | undefined = string | undefined> =
         setRetention: FunctionReference<"mutation", "internal", {
             retentionMs: number;
         }, null, Name>;
+        setServerToolPrice: FunctionReference<"mutation", "internal", {
+            nanosPerCall: number;
+            tool: string;
+        }, null, Name>;
         startRequest: FunctionReference<"mutation", "internal", {
             actionName?: string;
             messages: Array<{
@@ -119,9 +146,14 @@ export type ComponentApi<Name extends string | undefined = string | undefined> =
             }>;
             model: string;
             rerunOf?: string;
+            tags?: Array<{
+                dimension: string;
+                value: string;
+            }>;
             userId: string;
         }, {
             allowed: true;
+            notices: Array<string>;
             requestId: string;
             warnings: Array<string>;
         } | {
@@ -129,5 +161,11 @@ export type ComponentApi<Name extends string | undefined = string | undefined> =
             code: string;
             reason: string;
         }, Name>;
+        usageHistory: FunctionReference<"query", "internal", {
+            dimension: string;
+            limit?: number;
+            period: "day" | "month";
+            value: string;
+        }, any, Name>;
     };
 };

package/dist/component/lib.d.ts

@@ -1,5 +1,9 @@
 export declare const startRequest: import("convex/server").RegisteredMutation<"public", {
     actionName?: string | undefined;
+    tags?: {
+        dimension: string;
+        value: string;
+    }[] | undefined;
     rerunOf?: import("convex/values").GenericId<"requests"> | undefined;
     userId: string;
     model: string;
@@ -15,6 +19,7 @@ export declare const startRequest: import("convex/server").RegisteredMutation<"p
     allowed: true;
     requestId: import("convex/values").GenericId<"requests">;
     warnings: string[];
+    notices: string[];
 }>>;
 export declare const finishRequest: import("convex/server").RegisteredMutation<"public", {
     error?: string | undefined;
@@ -22,6 +27,8 @@ export declare const finishRequest: import("convex/server").RegisteredMutation<"
     promptTokens?: number | undefined;
     completionTokens?: number | undefined;
     cachedTokens?: number | undefined;
+    serverToolUses?: Record<string, number> | undefined;
+    costNanos?: number | undefined;
     latencyMs?: number | undefined;
     requestId: import("convex/values").GenericId<"requests">;
 }, Promise<{
@@ -45,6 +52,10 @@ export declare const lineage: import("convex/server").RegisteredQuery<"public",
         _id: import("convex/values").GenericId<"requests">;
         _creationTime: number;
         actionName?: string | undefined;
+        tags?: {
+            dimension: string;
+            value: string;
+        }[] | undefined;
         estimatedNanos?: number | undefined;
         estimatedTokens?: number | undefined;
         unpricedModel?: boolean | undefined;
@@ -55,6 +66,7 @@ export declare const lineage: import("convex/server").RegisteredQuery<"public",
         promptTokens?: number | undefined;
         completionTokens?: number | undefined;
         cachedTokens?: number | undefined;
+        serverToolUses?: Record<string, number> | undefined;
         costNanos?: number | undefined;
         latencyMs?: number | undefined;
         rerunOf?: import("convex/values").GenericId<"requests"> | undefined;
@@ -70,6 +82,10 @@ export declare const lineage: import("convex/server").RegisteredQuery<"public",
         _id: import("convex/values").GenericId<"requests">;
         _creationTime: number;
         actionName?: string | undefined;
+        tags?: {
+            dimension: string;
+            value: string;
+        }[] | undefined;
         estimatedNanos?: number | undefined;
         estimatedTokens?: number | undefined;
         unpricedModel?: boolean | undefined;
@@ -80,6 +96,7 @@ export declare const lineage: import("convex/server").RegisteredQuery<"public",
         promptTokens?: number | undefined;
         completionTokens?: number | undefined;
         cachedTokens?: number | undefined;
+        serverToolUses?: Record<string, number> | undefined;
         costNanos?: number | undefined;
         latencyMs?: number | undefined;
         rerunOf?: import("convex/values").GenericId<"requests"> | undefined;
@@ -98,6 +115,10 @@ export declare const getRequest: import("convex/server").RegisteredQuery<"public
     _id: import("convex/values").GenericId<"requests">;
     _creationTime: number;
     actionName?: string | undefined;
+    tags?: {
+        dimension: string;
+        value: string;
+    }[] | undefined;
     estimatedNanos?: number | undefined;
     estimatedTokens?: number | undefined;
     unpricedModel?: boolean | undefined;
@@ -108,6 +129,7 @@ export declare const getRequest: import("convex/server").RegisteredQuery<"public
     promptTokens?: number | undefined;
     completionTokens?: number | undefined;
     cachedTokens?: number | undefined;
+    serverToolUses?: Record<string, number> | undefined;
     costNanos?: number | undefined;
     latencyMs?: number | undefined;
     rerunOf?: import("convex/values").GenericId<"requests"> | undefined;
@@ -120,12 +142,18 @@ export declare const getRequest: import("convex/server").RegisteredQuery<"public
     status: "blocked" | "pending" | "success" | "error";
 } | null>>;
 export declare const listRequests: import("convex/server").RegisteredQuery<"public", {
+    dimension?: string | undefined;
+    value?: string | undefined;
     userId?: string | undefined;
     limit?: number | undefined;
 }, Promise<{
     _id: import("convex/values").GenericId<"requests">;
     _creationTime: number;
     actionName?: string | undefined;
+    tags?: {
+        dimension: string;
+        value: string;
+    }[] | undefined;
     estimatedNanos?: number | undefined;
     estimatedTokens?: number | undefined;
     unpricedModel?: boolean | undefined;
@@ -136,6 +164,7 @@ export declare const listRequests: import("convex/server").RegisteredQuery<"publ
     promptTokens?: number | undefined;
     completionTokens?: number | undefined;
     cachedTokens?: number | undefined;
+    serverToolUses?: Record<string, number> | undefined;
     costNanos?: number | undefined;
     latencyMs?: number | undefined;
     rerunOf?: import("convex/values").GenericId<"requests"> | undefined;
@@ -147,96 +176,155 @@ export declare const listRequests: import("convex/server").RegisteredQuery<"publ
     }[];
     status: "blocked" | "pending" | "success" | "error";
 }[]>>;
-export declare const listUsers: import("convex/server").RegisteredQuery<"public", {}, Promise<{
+export declare const listBuckets: import("convex/server").RegisteredQuery<"public", {
+    dimension?: string | undefined;
+}, Promise<{
     spendTodayNanos: number;
-    _id: import("convex/values").GenericId<"users">;
+    spendThisMonthNanos: number;
+    _id: import("convex/values").GenericId<"buckets">;
     _creationTime: number;
     requestsPerMinute?: number | undefined;
+    maxConcurrent?: number | undefined;
     dailySpendLimitNanos?: number | undefined;
+    monthlySpendLimitNanos?: number | undefined;
     lifetimeSpendLimitNanos?: number | undefined;
     dailyTokenLimit?: number | undefined;
+    monthlyTokenLimit?: number | undefined;
     lifetimeTokenLimit?: number | undefined;
     blocked?: boolean | undefined;
+    warnAtPct?: number | undefined;
     enforcement?: "hard" | "soft" | undefined;
     dailyBumpNanos?: number | undefined;
+    monthlyBumpNanos?: number | undefined;
     lifetimeBumpNanos?: number | undefined;
     bumpDayStamp?: string | undefined;
+    bumpMonthStamp?: string | undefined;
     tokensToday?: number | undefined;
+    monthStamp?: string | undefined;
+    tokensThisMonth?: number | undefined;
     reservedTodayNanos?: number | undefined;
+    reservedMonthNanos?: number | undefined;
     reservedTotalNanos?: number | undefined;
     reservedTodayTokens?: number | undefined;
+    reservedMonthTokens?: number | undefined;
     reservedTotalTokens?: number | undefined;
     pendingCount?: number | undefined;
-    userId: string;
+    dimension: string;
+    value: string;
     totalSpendNanos: number;
     totalRequests: number;
     totalTokens: number;
     dayStamp: string;
 }[]>>;
-export declare const setLimits: import("convex/server").RegisteredMutation<"public", {
-    requestsPerMinute?: number | undefined;
-    dailySpendLimitNanos?: number | undefined;
-    lifetimeSpendLimitNanos?: number | undefined;
-    dailyTokenLimit?: number | undefined;
-    lifetimeTokenLimit?: number | undefined;
-    blocked?: boolean | undefined;
-    enforcement?: "hard" | "soft" | undefined;
-    userId: string;
-}, Promise<null>>;
-export declare const deleteUser: import("convex/server").RegisteredMutation<"public", {
-    userId: string;
+export declare const getBucket: import("convex/server").RegisteredQuery<"public", {
+    dimension: string;
+    value: string;
 }, Promise<{
-    deletedThisBatch: number;
-    done: boolean;
-}>>;
-export declare const listActions: import("convex/server").RegisteredQuery<"public", {}, Promise<{
     spendTodayNanos: number;
-    _id: import("convex/values").GenericId<"actions">;
+    spendThisMonthNanos: number;
+    _id: import("convex/values").GenericId<"buckets">;
     _creationTime: number;
+    requestsPerMinute?: number | undefined;
+    maxConcurrent?: number | undefined;
     dailySpendLimitNanos?: number | undefined;
+    monthlySpendLimitNanos?: number | undefined;
     lifetimeSpendLimitNanos?: number | undefined;
     dailyTokenLimit?: number | undefined;
+    monthlyTokenLimit?: number | undefined;
     lifetimeTokenLimit?: number | undefined;
+    blocked?: boolean | undefined;
+    warnAtPct?: number | undefined;
     enforcement?: "hard" | "soft" | undefined;
     dailyBumpNanos?: number | undefined;
+    monthlyBumpNanos?: number | undefined;
     lifetimeBumpNanos?: number | undefined;
     bumpDayStamp?: string | undefined;
+    bumpMonthStamp?: string | undefined;
     tokensToday?: number | undefined;
+    monthStamp?: string | undefined;
+    tokensThisMonth?: number | undefined;
     reservedTodayNanos?: number | undefined;
+    reservedMonthNanos?: number | undefined;
     reservedTotalNanos?: number | undefined;
     reservedTodayTokens?: number | undefined;
+    reservedMonthTokens?: number | undefined;
     reservedTotalTokens?: number | undefined;
     pendingCount?: number | undefined;
-    disabled?: boolean | undefined;
+    dimension: string;
+    value: string;
     totalSpendNanos: number;
     totalRequests: number;
     totalTokens: number;
     dayStamp: string;
-    name: string;
-}[]>>;
-export declare const bumpUser: import("convex/server").RegisteredMutation<"public", {
-    dailyNanos?: number | undefined;
-    lifetimeNanos?: number | undefined;
-    userId: string;
-}, Promise<null>>;
-export declare const bumpAction: import("convex/server").RegisteredMutation<"public", {
-    dailyNanos?: number | undefined;
-    lifetimeNanos?: number | undefined;
-    name: string;
-}, Promise<null>>;
-export declare const bumpGlobal: import("convex/server").RegisteredMutation<"public", {
-    dailyNanos?: number | undefined;
-    lifetimeNanos?: number | undefined;
-}, Promise<null>>;
-export declare const setActionLimits: import("convex/server").RegisteredMutation<"public", {
+} | null>>;
+export declare const setBucketLimits: import("convex/server").RegisteredMutation<"public", {
+    requestsPerMinute?: number | undefined;
+    maxConcurrent?: number | undefined;
     dailySpendLimitNanos?: number | undefined;
+    monthlySpendLimitNanos?: number | undefined;
     lifetimeSpendLimitNanos?: number | undefined;
     dailyTokenLimit?: number | undefined;
+    monthlyTokenLimit?: number | undefined;
     lifetimeTokenLimit?: number | undefined;
+    blocked?: boolean | undefined;
+    warnAtPct?: number | undefined;
     enforcement?: "hard" | "soft" | undefined;
-    disabled?: boolean | undefined;
-    name: string;
+    dimension: string;
+    value: string;
+}, Promise<null>>;
+export declare const bumpBucket: import("convex/server").RegisteredMutation<"public", {
+    dailyNanos?: number | undefined;
+    monthlyNanos?: number | undefined;
+    lifetimeNanos?: number | undefined;
+    dimension: string;
+    value: string;
+}, Promise<null>>;
+export declare const adjustBucket: import("convex/server").RegisteredMutation<"public", {
+    tokens?: number | undefined;
+    reason?: string | undefined;
+    dimension: string;
+    value: string;
+    deltaNanos: number;
+}, Promise<null>>;
+export declare const listAdjustments: import("convex/server").RegisteredQuery<"public", {
+    limit?: number | undefined;
+    dimension: string;
+    value: string;
+}, Promise<{
+    _id: import("convex/values").GenericId<"adjustments">;
+    _creationTime: number;
+    tokens?: number | undefined;
+    reason?: string | undefined;
+    dimension: string;
+    value: string;
+    deltaNanos: number;
+}[]>>;
+export declare const usageHistory: import("convex/server").RegisteredQuery<"public", {
+    limit?: number | undefined;
+    dimension: string;
+    value: string;
+    period: "day" | "month";
+}, Promise<{
+    _id: import("convex/values").GenericId<"usage">;
+    _creationTime: number;
+    dimension: string;
+    value: string;
+    period: "day" | "month";
+    stamp: string;
+    spendNanos: number;
+    tokens: number;
+    requests: number;
+}[]>>;
+export declare const setAlertDefaults: import("convex/server").RegisteredMutation<"public", {
+    warnAtPct?: number | undefined;
 }, Promise<null>>;
+export declare const deleteBucket: import("convex/server").RegisteredMutation<"public", {
+    dimension: string;
+    value: string;
+}, Promise<{
+    deletedThisBatch: number;
+    done: boolean;
+}>>;
 export declare const getModelPolicy: import("convex/server").RegisteredQuery<"public", {}, Promise<{
     mode: "open" | "allowlist" | "denylist";
     models: string[];
@@ -247,17 +335,25 @@ export declare const getGlobalStatus: import("convex/server").RegisteredQuery<"p
     enforcement: "hard" | "soft";
     spentTodayNanos: number;
     spentTotalNanos: number;
+    retentionMs: number | null;
+    defaultWarnAtPct: number | null;
 }>>;
 export declare const setGlobalLimits: import("convex/server").RegisteredMutation<"public", {
     dailySpendLimitNanos?: number | undefined;
     lifetimeSpendLimitNanos?: number | undefined;
     enforcement?: "hard" | "soft" | undefined;
 }, Promise<null>>;
+export declare const bumpGlobal: import("convex/server").RegisteredMutation<"public", {
+    dailyNanos?: number | undefined;
+    monthlyNanos?: number | undefined;
+    lifetimeNanos?: number | undefined;
+}, Promise<null>>;
 export declare const setModelPolicy: import("convex/server").RegisteredMutation<"public", {
     models: string[];
     mode: "open" | "allowlist" | "denylist";
 }, Promise<null>>;
 export declare const setPrice: import("convex/server").RegisteredMutation<"public", {
+    cachedNanosPerMTok?: number | undefined;
     model: string;
     inputNanosPerMTok: number;
     outputNanosPerMTok: number;
@@ -265,5 +361,13 @@ export declare const setPrice: import("convex/server").RegisteredMutation<"publi
 export declare const listPrices: import("convex/server").RegisteredQuery<"public", {}, Promise<Record<string, {
     input: number;
     output: number;
+    cached?: number;
     overridden: boolean;
 }>>>;
+export declare const listServerToolPrices: import("convex/server").RegisteredQuery<"public", {}, Promise<{
+    [x: string]: number;
+}>>;
+export declare const setServerToolPrice: import("convex/server").RegisteredMutation<"public", {
+    tool: string;
+    nanosPerCall: number;
+}, Promise<null>>;