@odla-ai/cli 0.46.4 → 0.46.5

package/dist/bin.cjs

@@ -1725,7 +1725,7 @@ var init_surface = __esm({
           audit: {},
           credential: { set: {} }
         },
-        spend: { show: {}, reset: {} }
+        spend: { show: {}, set: {}, reset: {} }
       },
       app: {
         archive: {},
@@ -1820,11 +1820,15 @@ var init_surface = __esm({
 });
 
 // src/admin-spend.ts
-async function call(ctx, method, scope) {
+async function call(ctx, method, scope, payload) {
   const url = `${ctx.platformUrl.replace(/\/$/, "")}/registry/platform/spend?scope=${encodeURIComponent(scope)}`;
   const response2 = await ctx.doFetch(url, {
     method,
-    headers: { authorization: `Bearer ${ctx.token}` }
+    headers: {
+      authorization: `Bearer ${ctx.token}`,
+      ...payload === void 0 ? {} : { "content-type": "application/json" }
+    },
+    ...payload === void 0 ? {} : { body: JSON.stringify(payload) }
   });
   const body = await response2.json().catch(() => ({}));
   if (!response2.ok) {
@@ -1867,14 +1871,36 @@ async function spendReset(ctx, scope) {
 async function adminSpend(parsed, ctx) {
   const action2 = parsed.positionals[2];
   const scope = parsed.positionals[3] ?? stringOpt(parsed.options.scope);
-  if (action2 !== "show" && action2 !== "reset") rejectWord(["admin", "spend"], action2);
+  if (action2 !== "show" && action2 !== "reset" && action2 !== "set") rejectWord(["admin", "spend"], action2);
   if (!scope) {
     throw new Error(
       `"admin spend ${action2}" needs a scope, e.g. odla-ai admin spend ${action2} app:my-app:<incarnation>`
     );
   }
+  if (action2 === "set") return adminSpendSet(ctx, parsed, scope);
   return action2 === "show" ? spendShow(ctx, scope) : spendReset(ctx, scope);
 }
+async function adminSpendSet(ctx, parsed, scope) {
+  const daily = stringOpt(parsed.options.daily);
+  const none = parsed.options.none === true;
+  if (none === (daily !== void 0)) {
+    throw new Error("admin spend set needs exactly one of --daily <usd> or --none");
+  }
+  const dailyCapUsd = none ? null : Number(daily);
+  if (dailyCapUsd !== null && (!Number.isFinite(dailyCapUsd) || dailyCapUsd < 0)) {
+    throw new Error("--daily takes a non-negative dollar amount, for example --daily 25");
+  }
+  const result = await call(ctx, "PUT", scope, { dailyCapUsd });
+  if (ctx.json) {
+    ctx.out.log(JSON.stringify(result, null, 2));
+    return;
+  }
+  ctx.out.log(`${result.scope}: ${result.message}`);
+  ctx.out.log(`  spent today ${money(result.spentUsd)}`);
+  if (result.alreadyOverCap) {
+    ctx.out.log("  today's spend already exceeds this cap \u2014 the next call will latch it.");
+  }
+}
 var money;
 var init_admin_spend = __esm({
   "src/admin-spend.ts"() {
@@ -2723,7 +2749,7 @@ async function adminCommand(parsed, deps = {}) {
     rejectWord(["admin", "ai", "credential"], parsed.positionals[3]);
   }
   if (area === "spend") {
-    assertArgs(parsed, JSON_OPTIONS, 4);
+    assertArgs(parsed, [...JSON_OPTIONS, "daily", "none", "scope"], 4);
     const context2 = await resolveOperatorContext(parsed, { allowMissingConfig: true });
     const out = deps.stdout ?? console;
     const doFetch = deps.fetch ?? fetch;
@@ -12139,6 +12165,7 @@ Usage:
   odla-ai admin ai usage [--context <name>] [--app-id <id>] [--env <env>] [--run-id <id>] [--limit <1-500>] [--json]
   odla-ai admin ai audit [--context <name>] [--limit <1-200>] [--json]
   odla-ai admin spend show <app:<id>:<incarnation>> [--context <name>] [--json]
+  odla-ai admin spend set <app:<id>:<incarnation>> (--daily <usd>|--none) [--context <name>] [--json]
   odla-ai admin spend reset <app:<id>:<incarnation>> [--context <name>] [--json]
   odla-ai security github connect [--repo owner/name] [--env dev]             [continue in Studio; human session required]
   odla-ai security github disconnect --source <id> [--env dev]          [continue in Studio; human session required]