claude-token-saver 3.42.7 → 3.43.0

package/README.ko.md

@@ -559,10 +559,12 @@ v3.26.0부터 트랜스크립트의 모델 ID로 게이트웨이를 감지해
 
 LiteLLM 프록시로 Bedrock 등을 쓰면 Claude Code stdin 에 `rate_limits` 가 오지 않아 `✦ current`·`📅 weekly` 게이지가 아예 없습니다. 대신 LiteLLM 은 키별 `max_budget` 과 `spend` 를 관리하므로, 그 값을 가져와 같은 지점에 예산 게이지를 그립니다.
 
-- 감지 조건: `ANTHROPIC_BASE_URL` 이 공식 엔드포인트가 아니고, `ANTHROPIC_AUTH_TOKEN`(또는 `ANTHROPIC_API_KEY`)이 설정된 환경.
+- 감지 조건: `ANTHROPIC_BASE_URL` 이 공식 엔드포인트가 아닌 환경. 키는 `ANTHROPIC_AUTH_TOKEN`(또는 `ANTHROPIC_API_KEY`)에서 먼저 찾고, 없으면 `settings.json` 의 `apiKeyHelper` 를 실행해 얻습니다. 헬퍼 실행은 5분에 한 번 뜨는 갱신 프로세스 안에서만 하므로 통계선 렌더는 느려지지 않습니다 (v3.43.0).
+- `/key/info` 가 404 를 주어도 갱신을 멈추지 않습니다. Okta JWT 같은 커스텀 인증을 쓰는 배포에서는 404 가 정상이고, 예산은 `/user/info` 에 들어 있습니다. 두 엔드포인트가 모두 실패할 때만 실패로 봅니다 (v3.43.0).
+- 팀 멤버십이 아니라 internal user 한도로 게이지를 그리게 되면 `budget (user)` 로 출처를 함께 표시합니다. LiteLLM 이 차단 여부를 판정할 때 보는 값은 팀 멤버십 한도라서, 두 값이 수백 배 어긋나는 배포가 있습니다 (v3.43.0).
 - 조회는 LiteLLM 의 `GET /key/info` 와 `GET /user/info` 로 하고, 호출 키 자신의 정보만 받습니다. 예산 출처는 실무에서 가장 많이 쓰는 **팀 멤버십 예산**(team_memberships 의 spend·max_budget)을 먼저 보고, 없으면 키 자체의 max_budget, 그다음 internal user 예산 순으로 고릅니다. 렌더는 캐시 파일만 읽으며, 갱신은 5분에 한 번 분리된 백그라운드 프로세스가 수행합니다 (update-check 와 같은 구조라 statusline 이 네트워크를 기다리지 않습니다).
 - `max_budget` 이 없는 무제한 키는 게이지를 만들지 않습니다. 이 경우에도 `💵` 월 지출 세그먼트는 세션 로그 기반이라 그대로 표시됩니다.
-- 상태 확인: `sprag litellm-budget` (캐시 출력) · `litellm-budget --refresh` (즉시 조회).
+- 상태 확인: `sprag litellm-budget` (게이지·사용·잔여 금액 출력) · `--json` (캐시 원본) · `--refresh` (즉시 조회).
 
 세션 기본 모델이 sonnet 이면 sonnet 위임 규칙(T1)은 구조적으로 절감이 0입니다. 같은 급으로 내려보내 봐야 차액이 없기 때문이며 이는 정상 동작입니다. 다만 `route-scan rules` 가 이 경우를 "아직 위임 없음"과 같은 문구로 표시해 고장처럼 보였으므로, 이제 현재 기본 모델 기준으로 적용되지 않는다는 사실을 따로 적습니다.
 

package/README.md

@@ -1,3 +1,8 @@
+> **`claude-token-saver` is the old package name.**
+> Install `sprag-cli` instead — same tool, same binaries, same releases:
+> `npm i -g sprag-cli`
+> (switching needs a swap: `npm uninstall -g claude-token-saver && npm i -g sprag-cli`)
+
 <div align="center">
 
 <picture>
@@ -18,6 +23,8 @@
 ---
 
 <!-- ko:begin (generated from README.ko.md — edit that file, then run npm run sync:ko) -->
+**🇰🇷 한국어 문서: [README.ko.md](https://github.com/rootstudioyaml/sprag/blob/main/README.ko.md)** (아래 토글로도 펼칠 수 있습니다.)
+
 <details>
 <summary><strong>🇰🇷 한국어로 보기 (클릭하면 한국어 전문이 펼쳐집니다)</strong></summary>
 
@@ -563,10 +570,12 @@ v3.26.0부터 트랜스크립트의 모델 ID로 게이트웨이를 감지해
 
 LiteLLM 프록시로 Bedrock 등을 쓰면 Claude Code stdin 에 `rate_limits` 가 오지 않아 `✦ current`·`📅 weekly` 게이지가 아예 없습니다. 대신 LiteLLM 은 키별 `max_budget` 과 `spend` 를 관리하므로, 그 값을 가져와 같은 지점에 예산 게이지를 그립니다.
 
-- 감지 조건: `ANTHROPIC_BASE_URL` 이 공식 엔드포인트가 아니고, `ANTHROPIC_AUTH_TOKEN`(또는 `ANTHROPIC_API_KEY`)이 설정된 환경.
+- 감지 조건: `ANTHROPIC_BASE_URL` 이 공식 엔드포인트가 아닌 환경. 키는 `ANTHROPIC_AUTH_TOKEN`(또는 `ANTHROPIC_API_KEY`)에서 먼저 찾고, 없으면 `settings.json` 의 `apiKeyHelper` 를 실행해 얻습니다. 헬퍼 실행은 5분에 한 번 뜨는 갱신 프로세스 안에서만 하므로 통계선 렌더는 느려지지 않습니다 (v3.43.0).
+- `/key/info` 가 404 를 주어도 갱신을 멈추지 않습니다. Okta JWT 같은 커스텀 인증을 쓰는 배포에서는 404 가 정상이고, 예산은 `/user/info` 에 들어 있습니다. 두 엔드포인트가 모두 실패할 때만 실패로 봅니다 (v3.43.0).
+- 팀 멤버십이 아니라 internal user 한도로 게이지를 그리게 되면 `budget (user)` 로 출처를 함께 표시합니다. LiteLLM 이 차단 여부를 판정할 때 보는 값은 팀 멤버십 한도라서, 두 값이 수백 배 어긋나는 배포가 있습니다 (v3.43.0).
 - 조회는 LiteLLM 의 `GET /key/info` 와 `GET /user/info` 로 하고, 호출 키 자신의 정보만 받습니다. 예산 출처는 실무에서 가장 많이 쓰는 **팀 멤버십 예산**(team_memberships 의 spend·max_budget)을 먼저 보고, 없으면 키 자체의 max_budget, 그다음 internal user 예산 순으로 고릅니다. 렌더는 캐시 파일만 읽으며, 갱신은 5분에 한 번 분리된 백그라운드 프로세스가 수행합니다 (update-check 와 같은 구조라 statusline 이 네트워크를 기다리지 않습니다).
 - `max_budget` 이 없는 무제한 키는 게이지를 만들지 않습니다. 이 경우에도 `💵` 월 지출 세그먼트는 세션 로그 기반이라 그대로 표시됩니다.
-- 상태 확인: `sprag litellm-budget` (캐시 출력) · `litellm-budget --refresh` (즉시 조회).
+- 상태 확인: `sprag litellm-budget` (게이지·사용·잔여 금액 출력) · `--json` (캐시 원본) · `--refresh` (즉시 조회).
 
 세션 기본 모델이 sonnet 이면 sonnet 위임 규칙(T1)은 구조적으로 절감이 0입니다. 같은 급으로 내려보내 봐야 차액이 없기 때문이며 이는 정상 동작입니다. 다만 `route-scan rules` 가 이 경우를 "아직 위임 없음"과 같은 문구로 표시해 고장처럼 보였으므로, 이제 현재 기본 모델 기준으로 적용되지 않는다는 사실을 따로 적습니다.
 
@@ -1312,11 +1321,13 @@ If the detection is wrong, pin it with `sprag mode ttl=5m` (or `ttl=1h`). An exp
 
 Behind a LiteLLM proxy (Bedrock and friends), Claude Code's stdin never carries `rate_limits`, so the `✦ current` / `📅 weekly` gauges simply do not exist. LiteLLM does track per-key budgets, so the statusline draws a budget gauge in their place.
 
-- Detection: `ANTHROPIC_BASE_URL` points somewhere other than the official endpoint, and `ANTHROPIC_AUTH_TOKEN` (or `ANTHROPIC_API_KEY`) is set.
+- Detection: `ANTHROPIC_BASE_URL` points somewhere other than the official endpoint. The key comes from `ANTHROPIC_AUTH_TOKEN` / `ANTHROPIC_API_KEY`, or — when neither is set — from running the `apiKeyHelper` script in `settings.json`. The helper only runs inside the 5-minute refresh child, never on the render path (v3.43.0).
+- A 404 from `/key/info` no longer aborts the refresh: custom-auth deployments (Okta JWT and friends) always 404 there, and the budget lives in `/user/info`. Only a failure of both endpoints counts as a failure (v3.43.0).
+- When the gauge falls back to the internal-user budget it is labeled `budget (user)`. LiteLLM enforces the team-membership limit, which can differ from the user limit by orders of magnitude (v3.43.0).
 - The proxy is asked via `GET /key/info` and `GET /user/info` — only the calling key's own data. Renders read a cache file; a detached background process refreshes it every 5 minutes (same shape as the update check), so the statusline never waits on the network.
 - Budget source priority follows real-world usage: the **team-membership budget** (`team_memberships[].spend` + its linked budget table row) first, then the key's own `max_budget`, then the internal-user budget. Verified against a Dockerized LiteLLM, including memberships whose budget diverges from the team max into a separate budget-table row.
 - Unlimited keys (no `max_budget`) get no gauge. The `💵` monthly-spend segment still shows, since it comes from session logs.
-- Inspect with `sprag litellm-budget` (cached) or `litellm-budget --refresh` (query now).
+- Inspect with `sprag litellm-budget` (gauge with used/remaining amounts), `--json` (raw cache), or `--refresh` (query now).
 
 One related non-bug: if your session model is already sonnet, a sonnet-delegation (T1) rule can never save anything, because there is no price gap to capture. That is correct, but `route-scan rules` displayed it identically to "no delegations yet", so it now says outright that the rule does not apply at the current default model.
 

package/bin/cli.js

@@ -292,10 +292,12 @@ async function main() {
   }
 
   // Subcommand: litellm-budget · LiteLLM 게이트웨이 키의 max_budget/spend 조회.
-  //   sprag litellm-budget            # 캐시된 예산 상태 출력
+  //   sprag litellm-budget            # 캐시된 예산을 게이지·금액으로 출력
+  //   sprag litellm-budget --json     # 캐시 원본 JSON 출력
   //   sprag litellm-budget --refresh  # 지금 프록시에 물어봄 (detached 자식이 사용)
   if (args[0] === 'litellm-budget') {
-    const { gatewayEnv, readBudgetState, refreshBudgetState } = await import('../src/litellm-budget.js');
+    const { gatewayBase, readBudgetState, refreshBudgetState, formatBudgetReport } =
+      await import('../src/litellm-budget.js');
     const quiet = hasFlag('--quiet');
     if (hasFlag('--refresh')) {
       try {
@@ -308,12 +310,22 @@ async function main() {
       }
       return;
     }
-    const gw = gatewayEnv();
-    if (!gw) {
-      console.log('게이트웨이가 감지되지 않았습니다 (ANTHROPIC_BASE_URL + 키 필요).');
+    // 읽기 경로는 캐시만 보므로 키를 요구하지 않습니다 (apiKeyHelper 환경 배려).
+    const base = gatewayBase();
+    if (!base) {
+      console.log('게이트웨이가 감지되지 않았습니다 (ANTHROPIC_BASE_URL 필요).');
       return;
     }
-    console.log(JSON.stringify(readBudgetState(), null, 2));
+    const state = readBudgetState();
+    if (hasFlag('--json')) {
+      console.log(JSON.stringify(state, null, 2));
+      return;
+    }
+    if (state.base !== base || state.maxBudget === undefined) {
+      console.log('예산 캐시가 비어 있습니다. `sprag litellm-budget --refresh` 로 먼저 조회하십시오.');
+      return;
+    }
+    for (const line of formatBudgetReport(state)) console.log(line);
     return;
   }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "claude-token-saver",
-  "version": "3.42.7",
+  "version": "3.43.0",
   "description": "Legacy name of sprag-cli (Sprag) - same tool, kept publishing so existing installs stay current. Prefer: npm i -g sprag-cli",
   "type": "module",
   "bin": {

package/src/formatters/statusline.js

@@ -50,7 +50,7 @@ const MAGENTA = fg(167, 139, 250, '\x1b[35m');
 const GRAY = fg(100, 116, 139, '\x1b[90m');
 const BOLD = '\x1b[1m';
 
-function formatMoney(usd) {
+export function formatMoney(usd) {
   // Boundaries sit at 999.5/999500 so a value that would round up into the
   // next band's width ($999.9 → "$1000") jumps to that band's unit instead —
   // otherwise "$1000" renders visually larger than "$1.0K".
@@ -81,7 +81,7 @@ function formatPct(v) {
  * for ~4% effective precision around the fill edge. Stable monospace width
  * across all terminal fonts that ship Block Elements (U+2580–U+259F).
  */
-function gaugeBar(pct) {
+export function gaugeBar(pct) {
   const cells = 6;
   const clamped = Math.max(0, Math.min(100, pct));
   const filled = (clamped / 100) * cells; // e.g. 4.32 cells filled
@@ -640,15 +640,19 @@ export function formatReport(data, { color = true, verbose = false, timer = true
       Number.isFinite(info.maxBudget) && info.maxBudget > 0
         ? ` ${formatMoney(Number(info.spend) || 0)}/${formatMoney(info.maxBudget)}`
         : '';
+    // LiteLLM 의 internal user 한도는 차단 판정에 쓰이는 값이 아니라서, 팀
+    // 멤버십 한도와 수백 배 어긋나는 배포가 있습니다. 그 한도로 그린 게이지는
+    // 출처를 밝혀 두어야 사용자가 정확한 값으로 오해하지 않습니다.
+    const src = info.source === 'user' ? ' (user)' : '';
     if (isIcon) {
-      const labelPart = labels.usageLabel ? `${labels.usageLabel} ` : '';
+      const labelPart = labels.usageLabel ? `${labels.usageLabel}${src} ` : '';
       const bar = gaugeBar(pct);
       return `${c(tone)}${labels.icon} ${labelPart}${bar} ${pct}%${money}${tail}${c(RESET)}`;
     }
     if (verbose) {
-      return `${c(tone)}${labels.short} cap ${pct}% used${money}${tail}${c(RESET)}`;
+      return `${c(tone)}${labels.short}${src} cap ${pct}% used${money}${tail}${c(RESET)}`;
     }
-    return `${c(tone)}${labels.short} cap ${pct}%${money}${tail}${c(RESET)}`;
+    return `${c(tone)}${labels.short}${src} cap ${pct}%${money}${tail}${c(RESET)}`;
   }
   // Color tone: green <70%, yellow 70-89%, red 90+% (a 90+% window only
   // renders here when a *different* window won the cap-warn chip slot).

package/src/litellm-budget.js

@@ -18,9 +18,12 @@
  */
 
 import { readFileSync, writeFileSync, existsSync, mkdirSync } from 'node:fs';
-import { spawn } from 'node:child_process';
+import { spawn, execSync } from 'node:child_process';
 import { join } from 'node:path';
+import { homedir } from 'node:os';
 import { userDataDir } from './paths.js';
+import { gaugeBar, formatMoney } from './formatters/statusline.js';
+import { formatResetClock } from './format-time.js';
 import { cliEntryPath } from './update-check.js';
 import { debug } from './debug.js';
 
@@ -28,22 +31,79 @@ import { debug } from './debug.js';
 // 통계선 렌더(수 초 간격)가 프록시를 두들기지 않습니다.
 const CHECK_INTERVAL_MS = 5 * 60 * 1000;
 const FETCH_TIMEOUT_MS = 5000;
+const HELPER_TIMEOUT_MS = 10000;
 
 export function budgetStatePath() {
   return join(userDataDir(), 'litellm-budget.json');
 }
 
-/** 게이트웨이 환경 정보. 감지되지 않으면 null. */
-export function gatewayEnv(env = process.env) {
+/**
+ * 게이트웨이 주소만 판정합니다. 캐시만 읽는 경로는 키가 필요하지 않으므로,
+ * 키 확보와 주소 판정을 분리해 둡니다. apiKeyHelper 로만 인증하는 환경에는
+ * 환경변수에 토큰이 없어서, 키를 함께 요구하면 렌더 경로가 통째로 탈락합니다.
+ */
+export function gatewayBase(env = process.env) {
   const base = (env.ANTHROPIC_BASE_URL || '').trim().replace(/\/+$/, '');
   if (!base) return null;
   // 공식 엔드포인트를 그대로 가리키면 게이트웨이가 아닙니다.
   if (/^https?:\/\/api\.anthropic\.com/i.test(base)) return null;
-  const key = (env.ANTHROPIC_AUTH_TOKEN || env.ANTHROPIC_API_KEY || '').trim();
+  return base;
+}
+
+/** 게이트웨이 주소와 키를 함께 돌려줍니다. 둘 중 하나라도 없으면 null. */
+export function gatewayEnv(env = process.env) {
+  const base = gatewayBase(env);
+  if (!base) return null;
+  const key = resolveKey(env);
   if (!key) return null;
   return { base, key };
 }
 
+/**
+ * Claude Code 와 같은 방식으로 apiKeyHelper 를 실행해 토큰을 얻습니다.
+ * 환경변수에 토큰을 두지 않고 헬퍼 스크립트로 매번 발급받는 구성이 공식 인증
+ * 방식 가운데 하나이며, 그 구성에서는 statusline 자식 프로세스의 process.env
+ * 안에 토큰이 존재하지 않습니다.
+ *
+ * 헬퍼 실행은 비용이 있으므로 갱신 경로(5분에 한 번 뜨는 detached 자식)에서만
+ * 부릅니다. 수 초 간격으로 도는 렌더 경로에서 부르면 통계선이 그만큼 느려집니다.
+ */
+export function keyFromApiKeyHelper(cwd = process.cwd()) {
+  const candidates = [
+    join(cwd, '.claude', 'settings.local.json'),
+    join(cwd, '.claude', 'settings.json'),
+    join(homedir(), '.claude', 'settings.json'),
+  ];
+  for (const p of candidates) {
+    let helper;
+    try {
+      helper = JSON.parse(readFileSync(p, 'utf8'))?.apiKeyHelper;
+    } catch {
+      continue;
+    }
+    if (typeof helper !== 'string' || !helper.trim()) continue;
+    try {
+      const out = execSync(helper, {
+        encoding: 'utf8',
+        timeout: HELPER_TIMEOUT_MS,
+        stdio: ['ignore', 'pipe', 'ignore'],
+      });
+      // 진행 로그를 함께 출력하는 헬퍼가 있으므로 마지막 비어 있지 않은 줄을 취합니다.
+      const token = out.trim().split(/\r?\n/).filter(Boolean).pop();
+      if (token) return token.trim();
+    } catch (e) {
+      debug('litellm-budget:helper', e);
+    }
+  }
+  return null;
+}
+
+/** 환경변수 토큰을 먼저 보고, 없으면 apiKeyHelper 로 내려갑니다. */
+export function resolveKey(env = process.env) {
+  const direct = (env.ANTHROPIC_AUTH_TOKEN || env.ANTHROPIC_API_KEY || '').trim();
+  return direct || keyFromApiKeyHelper();
+}
+
 export function readBudgetState() {
   try {
     const s = JSON.parse(readFileSync(budgetStatePath(), 'utf8'));
@@ -67,11 +127,11 @@ function writeBudgetState(next) {
  *            spend:number, maxBudget:number}|null}
  */
 export function budgetWindow(env = process.env) {
-  const gw = gatewayEnv(env);
-  if (!gw) return null;
+  const base = gatewayBase(env);
+  if (!base) return null;
   const s = readBudgetState();
   // 다른 프록시의 캐시를 재사용하지 않도록 base 단위로 격리합니다.
-  if (s.base !== gw.base) return null;
+  if (s.base !== base) return null;
   const max = Number(s.maxBudget);
   const spend = Number(s.spend);
   if (!Number.isFinite(max) || max <= 0 || !Number.isFinite(spend)) return null;
@@ -84,19 +144,66 @@ export function budgetWindow(env = process.env) {
     resetsAt: resetMs ? Math.round(resetMs / 1000) : null,
     spend,
     maxBudget: max,
+    // 렌더러가 출처를 함께 보여 줄 수 있도록 싣습니다. internal user 한도는
+    // LiteLLM 이 차단 판정에 쓰는 값이 아니라서 그대로 보여 주면 오해를 부릅니다.
+    source: typeof s.source === 'string' ? s.source : null,
   };
 }
 
+/**
+ * `sprag litellm-budget` 이 출력하는 사람용 보고서입니다. 통계선의 cap 게이지와
+ * 같은 모양(비주얼 바 + 백분율 + 금액)을 쓰되, 통계선에서는 폭이 부족해 생략하는
+ * 잔여 금액과 예산 출처, 캐시 신선도를 함께 적습니다.
+ *
+ * @param {object} state - readBudgetState() 가 돌려준 캐시 상태
+ * @param {Date} [now]
+ * @returns {string[]} 출력할 줄들
+ */
+export function formatBudgetReport(state, now = new Date()) {
+  const max = Number(state?.maxBudget);
+  const spend = Number(state?.spend);
+  if (!Number.isFinite(max) || max <= 0) {
+    const spent = Number.isFinite(spend) ? formatMoney(spend) : '알 수 없음';
+    return [`예산 한도가 설정되지 않은 키입니다 (무제한). 누적 지출: ${spent}`];
+  }
+  const used = Number.isFinite(spend) ? Math.max(0, spend) : 0;
+  const left = Math.max(0, max - used);
+  const pct = Math.min(100, (used / max) * 100);
+  const lines = [
+    `🔑 budget ${gaugeBar(pct)} ${Math.round(pct)}% ${formatMoney(used)}/${formatMoney(max)}`,
+    `   사용 ${formatMoney(used)} · 잔여 ${formatMoney(left)} (${(100 - pct).toFixed(1)}%)`,
+  ];
+  const resetMs = Number(state?.budgetResetAt);
+  if (Number.isFinite(resetMs) && resetMs > 0) {
+    const clock = formatResetClock(Math.round(resetMs / 1000), now);
+    if (clock) lines.push(`   리셋 ${clock}`);
+  }
+  const labels = { team: '팀 멤버십 예산', key: '키 예산', user: '사용자 예산' };
+  if (state?.source) lines.push(`   출처 ${labels[state.source] || state.source}`);
+  if (state?.source === 'user') {
+    // LiteLLM 은 팀 멤버십 한도로 차단 여부를 판정하므로, 사용자 한도만 잡힌
+    // 상태에서는 실제 한도와 크게 어긋날 수 있습니다.
+    lines.push('   주의: 사용자 한도는 차단 판정 기준과 다를 수 있습니다.');
+  }
+  const checkedAt = Number(state?.checkedAt);
+  if (Number.isFinite(checkedAt) && checkedAt > 0) {
+    const ageMin = Math.max(0, Math.round((now.getTime() - checkedAt) / 60000));
+    lines.push(`   조회 ${ageMin}분 전`);
+  }
+  return lines;
+}
+
 /** 캐시가 오래됐으면 detached 자식으로 갱신을 예약합니다. 즉시 반환. */
 export function maybeSpawnBudgetCheck(env = process.env) {
-  const gw = gatewayEnv(env);
-  if (!gw) return false;
+  // 키 확보는 자식이 담당합니다. 렌더 경로에서 헬퍼를 돌리면 통계선이 느려집니다.
+  const base = gatewayBase(env);
+  if (!base) return false;
   const s = readBudgetState();
   const age = Date.now() - (Number(s.checkedAt) || 0);
-  if (s.base === gw.base && age < CHECK_INTERVAL_MS) return false;
+  if (s.base === base && age < CHECK_INTERVAL_MS) return false;
   // 오프라인/오류 시 렌더마다 자식을 다시 띄우지 않도록 시도 시각을 먼저 기록합니다.
   try {
-    writeBudgetState({ ...s, base: gw.base, checkedAt: Date.now() });
+    writeBudgetState({ ...s, base, checkedAt: Date.now() });
   } catch (e) {
     debug('litellm-budget:stamp', e);
     return false;
@@ -135,7 +242,10 @@ function parseResetMs(v) {
  *            budgetResetAt:number|null}|null}
  */
 export function pickBudgetSource(keyInfo, userInfo) {
-  const userId = keyInfo?.user_id ?? null;
+  // /key/info 가 404 인 게이트웨이(커스텀 인증)에서는 keyInfo 가 비므로,
+  // /user/info 가 함께 주는 내 user_id 를 폴백으로 씁니다. 이 폴백이 없으면
+  // 팀 전체 멤버십을 돌려주는 배포에서 남의 예산을 집을 수 있습니다.
+  const userId = keyInfo?.user_id ?? userInfo?.user_info?.user_id ?? null;
   const teamId = keyInfo?.team_id ?? null;
   // ① 팀 멤버십 예산: 키의 team_id 에 해당하는 팀에서 내 user_id 의 멤버십을 찾는다.
   const teams = Array.isArray(userInfo?.teams) ? userInfo.teams : [];
@@ -184,34 +294,48 @@ export function pickBudgetSource(keyInfo, userInfo) {
  * /key/info(키·소속 식별)와 /user/info(팀 멤버십 예산)를 함께 조회합니다.
  */
 export async function refreshBudgetState(env = process.env, fetchImpl = fetch) {
-  const gw = gatewayEnv(env);
-  if (!gw) return null;
+  const base = gatewayBase(env);
+  if (!base) return null;
+  const key = resolveKey(env);
+  if (!key) return null;
   const controller = new AbortController();
   const timer = setTimeout(() => controller.abort(), FETCH_TIMEOUT_MS);
   const headers = {
     accept: 'application/json',
-    authorization: `Bearer ${gw.key}`,
+    authorization: `Bearer ${key}`,
   };
   try {
-    const keyRes = await fetchImpl(`${gw.base}/key/info`, { signal: controller.signal, headers });
-    if (!keyRes.ok) throw new Error(`LiteLLM responded ${keyRes.status}`);
-    const keyBody = await keyRes.json();
-    const keyInfo = keyBody && typeof keyBody.info === 'object' ? keyBody.info : null;
-    if (!keyInfo) throw new Error('key/info response carried no info object');
-    // /user/info 는 팀 멤버십 예산 전용이라 실패해도 키 예산 폴백으로 진행한다.
+    // /key/info 는 커스텀 인증(JWT)을 쓰는 배포에서 404 가 정상입니다. 발급된
+    // 토큰에 해당하는 행이 LiteLLM_VerificationToken 에 없기 때문입니다. 예산
+    // 자체는 /user/info 에 있으므로, 여기서 중단하면 안 됩니다.
+    let keyInfo = null;
+    try {
+      const keyRes = await fetchImpl(`${base}/key/info`, { signal: controller.signal, headers });
+      if (keyRes.ok) {
+        const keyBody = await keyRes.json();
+        if (keyBody && typeof keyBody.info === 'object') keyInfo = keyBody.info;
+      } else {
+        debug('litellm-budget:key-info', new Error(`LiteLLM responded ${keyRes.status}`));
+      }
+    } catch (e) {
+      debug('litellm-budget:key-info', e);
+    }
     let userInfo = null;
     try {
-      const userRes = await fetchImpl(`${gw.base}/user/info`, { signal: controller.signal, headers });
+      const userRes = await fetchImpl(`${base}/user/info`, { signal: controller.signal, headers });
       if (userRes.ok) userInfo = await userRes.json();
+      else debug('litellm-budget:user-info', new Error(`LiteLLM responded ${userRes.status}`));
     } catch (e) {
       debug('litellm-budget:user-info', e);
     }
+    // 둘 다 받지 못했을 때만 실패로 봅니다.
+    if (!keyInfo && !userInfo) throw new Error('neither key/info nor user/info returned data');
     const picked = pickBudgetSource(keyInfo, userInfo);
     const next = {
-      base: gw.base,
+      base,
       checkedAt: Date.now(),
       source: picked ? picked.source : null,
-      spend: picked ? picked.spend : (numOrNull(keyInfo.spend) ?? 0),
+      spend: picked ? picked.spend : (numOrNull(keyInfo?.spend) ?? 0),
       maxBudget: picked ? picked.maxBudget : null,
       budgetResetAt: picked ? picked.budgetResetAt : null,
     };