@columenlabs/lumen-ai 0.1.221 → 0.1.222

package/dist/assistant/engine.d.ts

@@ -49,6 +49,11 @@ export interface AllowanceOptions {
     approver?: (caller: Caller) => boolean;
     /** Hears each request and each decision: tell the approvers, then the person. */
     notify?: (n: AllowanceNotice) => void | Promise<void>;
+    /**
+     * Units a month for one person (`period: 'month'`): their plan's allowance or an admin's override.
+     * Nothing (or a throw) falls back to the admin's cap in Settings.
+     */
+    cap?: (id: string) => number | null | undefined | Promise<number | null | undefined>;
 }
 /** A request made, or decided. */
 export interface AllowanceNotice {

package/dist/assistant/engine.js

@@ -140,6 +140,7 @@ export class Assistant {
                         usd: unit.usd,
                         period: al.period ?? 'month',
                         starter: al.starter ?? DEFAULT_CAP / unit.usd,
+                        ...(al.cap ? { cap: al.cap } : {}),
                     },
                 }
                 : {}),
@@ -611,7 +612,7 @@ export class Assistant {
         const used = round2(budget.used / u.usd);
         const base = this.state.allowanceConfig?.period === 'once'
             ? (this.state.allowanceConfig?.starter ?? total)
-            : (await this.state.appCap()) / u.usd;
+            : (await this.state.capFor(caller.id)) / u.usd;
         const waiting = approver
             ? (await this.state.requests()).rows.filter((r) => r.status === 'open').length
             : undefined;

package/dist/assistant/state.d.ts

@@ -77,6 +77,8 @@ export interface AllowanceConfig {
     period: 'month' | 'once';
     /** Units each person starts with (once); a month's cap is the admin's cap in Settings. */
     starter: number;
+    /** Units a month for one person (their plan, an override); nothing falls back to the admin's cap. */
+    cap?: (id: string) => number | null | undefined | Promise<number | null | undefined>;
 }
 export interface AllowanceDoc {
     grants: {
@@ -165,6 +167,9 @@ export declare class State {
         total: number;
         used: number;
     }>;
+    /** USD a month this person may spend on the app's keys: their own cap when the app names one,
+     *  else the admin's. A cap that cannot be read falls back too, so an answer never waits on it. */
+    capFor(id: string): Promise<number>;
     /** USD a month when nobody set a cap. */
     get defaultCap(): number;
     month(): string;

package/dist/assistant/state.js

@@ -104,7 +104,14 @@ export class State {
         const granted = d.grants
             .filter((g) => monthIn(this.o.timeZone, new Date(g.at)) === month)
             .reduce((a, g) => a + g.amount, 0);
-        return { total: (await this.appCap()) + granted * al.usd, used };
+        return { total: (await this.capFor(id)) + granted * al.usd, used };
+    }
+    /** USD a month this person may spend on the app's keys: their own cap when the app names one,
+     *  else the admin's. A cap that cannot be read falls back too, so an answer never waits on it. */
+    async capFor(id) {
+        const al = this.o.allowance;
+        const own = al?.cap ? await Promise.resolve(al.cap(id)).catch(() => null) : null;
+        return typeof own === 'number' && own >= 0 ? own * al.usd : this.appCap();
     }
     /** USD a month when nobody set a cap. */
     get defaultCap() {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@columenlabs/lumen-ai",
-  "version": "0.1.221",
+  "version": "0.1.222",
   "description": "Lumen's assistant and agent parts: a chat assistant's server (Claude tiers, tools over MCP, the web, hand off, spend caps), serving an MCP server from a Worker, read-only tool sets and metered tool calls.",
   "license": "MIT",
   "type": "module",
@@ -52,8 +52,8 @@
     "@ai-sdk/openai": "4.0.108",
     "@anthropic-ai/sdk": "0.133.0",
     "@cloudflare/think": "0.23.0",
-    "@columenlabs/lumen-ops": "0.1.221",
-    "@columenlabs/lumen-ui-types": "0.1.221",
+    "@columenlabs/lumen-ops": "0.1.222",
+    "@columenlabs/lumen-ui-types": "0.1.222",
     "agents": "0.32.0",
     "ai": "7.0.162",
     "just-bash": "3.6.0",