opencode-visual-cache 1.3.0 → 1.5.0

package/dist/index.js

@@ -1,6 +1,7 @@
 import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "@opentui/solid/jsx-runtime";
 import { createMemo, createSignal, createEffect, onMount, onCleanup, Show, untrack } from "solid-js";
 import { PLUGIN_VERSION } from "./_version";
+import { balanceProviders, getBalanceProvider, maskKey, matchBalanceProvider } from "./balance-providers";
 // ── terminal-width helpers ────────────────────────────────────────
 // CJK characters occupy 2 terminal columns; padEnd/padStart count
 // string length (=1 per char), which breaks alignment with mixed text.
@@ -94,6 +95,14 @@ const ZH_T = {
     secDetail: "明细",
     secModel: "模型",
     secSkills: "已加载技能",
+    balTotal: "总余额:",
+    balNoKey: "未配置 {p} API Key",
+    balLoading: "查询中...",
+    balError: "查询失败",
+    balErr401: "API Key 无效",
+    balErr403: "余额查询被拒绝",
+    balErrEmpty: "未获取到余额数据",
+    balErrTimeout: "查询超时",
 };
 const EN_T = {
     title: "Token Cache",
@@ -125,6 +134,14 @@ const EN_T = {
     secDetail: "Detail",
     secModel: "Model",
     secSkills: "Loaded Skills",
+    balTotal: "Total:",
+    balNoKey: "{p} API Key not set",
+    balLoading: "Fetching...",
+    balError: "Fetch failed",
+    balErr401: "Invalid API Key",
+    balErr403: "Balance request rejected",
+    balErrEmpty: "No balance data",
+    balErrTimeout: "Request timed out",
 };
 // ── color helpers ────────────────────────────────────────────────
 /** Extract { r, g, b } (0–255) from a hex string or RGBA-like object. */
@@ -299,6 +316,46 @@ function estimateTokens(text) {
     const asciiPerToken = jsonLike ? 3.5 : codeLike ? 3.5 : 4;
     return Math.max(1, Math.ceil(ascii / asciiPerToken + cjk / 1.0));
 }
+const BALANCE_POLL_MS = 5 * 60 * 1000; // 5 minutes
+/**
+ * 将余额从来源币种换算为目标币种。
+ * DEFAULT_RATES 以 USD=1 为基准：先折算为 USD，再换算到目标币种。
+ */
+function convertBalance(target, targetRate, amount, from) {
+    if (from === target)
+        return amount;
+    const fromRate = DEFAULT_RATES[from] ?? 1;
+    const usd = from === "USD" ? amount : amount / fromRate;
+    return target === "USD" ? usd : usd * targetRate;
+}
+/**
+ * 从 OpenCode 已认证的 provider 读取 API key 作为余额查询的自动兜底。
+ * 匹配复用前缀逻辑：先精确匹配 id，再前缀匹配（如 moonshotai-cn → moonshot）。
+ * key 来源：auth.json（provider.key）或配置（provider.options.apiKey）。
+ * 仅当手动配置的 key 缺失时使用；读取失败或未匹配返回空串。
+ */
+function findOpencodeKey(api, provider) {
+    try {
+        const provs = api.state.provider;
+        // 大小写不敏感：精确匹配 id，否则前缀匹配（如 moonshotai-cn → moonshot）
+        const id = provider.id.toLowerCase();
+        const hit = provs.find((p) => p.id.toLowerCase() === id) ?? provs.find((p) => p.id.toLowerCase().startsWith(id));
+        if (!hit)
+            return "";
+        const k = typeof hit.key === "string" ? hit.key : "";
+        if (k)
+            return k;
+        return typeof hit.options?.apiKey === "string" ? hit.options.apiKey : "";
+    }
+    catch {
+        return "";
+    }
+}
+/** 货币符号：优先取 /cache-currency 内置映射，未知币种回退为代码。 */
+function balanceSymbol(currency) {
+    const sym = CURRENCIES[currency];
+    return sym ?? currency + " ";
+}
 const CURRENCIES = {
     USD: "$", CNY: "¥", EUR: "€", JPY: "JP¥", GBP: "£", KRW: "₩",
 };
@@ -325,7 +382,7 @@ function TokenCachePanel(props) {
     const [skillsOpen, setSkillsOpen] = createSignal(true);
     let boxEl;
     // ── shared signals (de-structured so internal code is unchanged) ──
-    const { currencySymbol, setCurrencySymbol, exchangeRate, setExchangeRate, langZH, setLangZH, sectionDetail, setSectionDetail, sectionModel, setSectionModel, sectionDist, setSectionDist, sectionSkills, setSectionSkills, borderVisible, setBorderVisible, } = props.signals;
+    const { currencySymbol, setCurrencySymbol, exchangeRate, setExchangeRate, langZH, setLangZH, sectionDetail, setSectionDetail, sectionModel, setSectionModel, sectionDist, setSectionDist, sectionSkills, setSectionSkills, sectionBalance, setSectionBalance, balanceRefresh, balanceProviderId, setBalanceProviderId, autoBalance, setAutoBalance, balanceCurrency, setBalanceCurrency, borderVisible, setBorderVisible, } = props.signals;
     // ── reactive translation (follows langZH signal) ──
     const t = createMemo(() => langZH() ? ZH_T : EN_T);
     // ── scan session messages reactively ──
@@ -352,6 +409,90 @@ function TokenCachePanel(props) {
         hasSkills: false,
     });
     const [refreshTick, setRefreshTick] = createSignal(0);
+    // ── balance state + polling ──────────────────────────────────
+    const [balanceState, setBalanceState] = createSignal({
+        status: "idle", data: null, lastFetch: 0,
+    });
+    // 请求序号：防止定时轮询与手动刷新并发时，慢的旧请求覆盖新结果
+    let balanceSeq = 0;
+    // 当前 provider 显示名
+    const providerName = createMemo(() => getBalanceProvider(balanceProviderId()).name);
+    const pollBalance = async () => {
+        const provider = getBalanceProvider(balanceProviderId());
+        // 手动配置的 key 优先；缺失时自动复用 OpenCode 已认证的 key（auth.json / config）
+        const key = props.api.kv.get(`${KV_PREFIX}.balance.${provider.id}.key`, "")
+            || findOpencodeKey(props.api, provider);
+        if (!key) {
+            setBalanceState({ status: "idle", data: null, lastFetch: 0, error: undefined, key: undefined });
+            return;
+        }
+        const now = Date.now();
+        const prev = balanceState();
+        // key 已更换（重新输入）→ 强制重新查询，绕过缓存
+        if (prev.status === "ok" && prev.key === key && now - prev.lastFetch < BALANCE_POLL_MS)
+            return; // cache still fresh
+        const seq = ++balanceSeq;
+        setBalanceState({ ...prev, status: "loading", error: undefined, key });
+        const controller = new AbortController();
+        let timedOut = false;
+        const timer = setTimeout(() => { timedOut = true; controller.abort(); }, 10_000);
+        try {
+            const data = await provider.fetchBalance(key, controller.signal);
+            clearTimeout(timer);
+            if (seq !== balanceSeq)
+                return; // 已被更新的请求取代，丢弃过期结果
+            setBalanceState({ status: "ok", data, lastFetch: Date.now(), error: undefined, key });
+        }
+        catch (err) {
+            clearTimeout(timer);
+            if (seq !== balanceSeq)
+                return;
+            const code = timedOut ? "TIMEOUT" : (err instanceof Error ? err.message : "");
+            // 失败时清空旧数据，避免显示过期余额
+            setBalanceState({ status: "error", data: null, lastFetch: 0, error: code, key });
+        }
+    };
+    // Re-fetch when the API key is (re)configured via /cache-balance-key.
+    // 注意：pollBalance 内部读写 balanceState 信号，若不做 untrack 包裹，
+    // effect 会追踪 balanceState 的变化并与 pollBalance 的 setBalanceState
+    // 形成无限循环（每次重跑都发起新的 fetch 请求）。
+    createEffect(() => {
+        void balanceRefresh();
+        untrack(() => { void pollBalance(); });
+    });
+    // 自动切换当前会话的 provider（前缀匹配）。手动切换会关闭此行为。
+    // 直接追踪 messages 取最后一条 assistant 消息的 providerID——
+    // 不依赖 session.model 的响应式更新（模型切换时该链路可能不触发重算）。
+    createEffect(() => {
+        if (!autoBalance())
+            return;
+        const sid = props.signals.overrideSessionId() ?? props.sessionId;
+        const msgs = props.api.state.session.messages(sid);
+        let pid = "";
+        for (let i = msgs.length - 1; i >= 0; i--) {
+            const m = msgs[i];
+            if (m.role === "assistant" && m.providerID) {
+                pid = m.providerID;
+                break;
+            }
+        }
+        // 会话尚无 assistant 消息（新会话 / 刚切换模型未对话 / 消息未加载）
+        // → 回退到会话级模型元数据，反映当前正在使用的 provider
+        if (!pid) {
+            try {
+                const session = props.api.state.session.get(sid);
+                pid = session?.model?.providerID ?? "";
+            }
+            catch { /* ignore */ }
+        }
+        if (!pid)
+            return;
+        const hit = matchBalanceProvider(pid);
+        if (hit && hit.id !== balanceProviderId()) {
+            setBalanceProviderId(hit.id);
+            props.signals.setBalanceRefresh(props.signals.balanceRefresh() + 1);
+        }
+    });
     // ── auto-clear override when the user navigates to a different main session ──
     let lastMainSid = props.sessionId;
     createEffect(() => {
@@ -602,10 +743,33 @@ function TokenCachePanel(props) {
                     setCurrencySymbol(sym);
                 if (typeof rate === "number" && rate > 0)
                     setExchangeRate(rate);
+                const balCur = props.api.kv.get(`${KV_PREFIX}.balance_currency`);
+                if (typeof balCur === "string")
+                    setBalanceCurrency(balCur);
+                // Restore balance provider (fall back to default when unknown)
+                const savedProvider = props.api.kv.get(`${KV_PREFIX}.balance.provider`);
+                if (typeof savedProvider === "string" && balanceProviders.some((p) => p.id === savedProvider)) {
+                    setBalanceProviderId(savedProvider);
+                }
+                // Restore auto-switch (default on)
+                const savedAuto = props.api.kv.get(`${KV_PREFIX}.balance.auto`);
+                if (typeof savedAuto === "boolean")
+                    setAutoBalance(savedAuto);
+                // Migrate legacy DeepSeek key (cache_panel.ds_key → cache_panel.balance.deepseek.key)
+                const legacyKey = props.api.kv.get(`${KV_PREFIX}.ds_key`, "");
+                if (legacyKey) {
+                    const dsKey = props.api.kv.get(`${KV_PREFIX}.balance.deepseek.key`, "");
+                    if (!dsKey)
+                        props.api.kv.set(`${KV_PREFIX}.balance.deepseek.key`, legacyKey);
+                    props.api.kv.set(`${KV_PREFIX}.ds_key`, "");
+                }
+                // 恢复的 provider 可能与默认值不同，强制重新查询
+                props.signals.setBalanceRefresh(props.signals.balanceRefresh() + 1);
                 setSectionDetail(Boolean(props.api.kv.get(`${KV_PREFIX}.section.detail`, true)));
                 setSectionModel(Boolean(props.api.kv.get(`${KV_PREFIX}.section.model`, true)));
                 setSectionDist(Boolean(props.api.kv.get(`${KV_PREFIX}.section.dist`, true)));
                 setSectionSkills(Boolean(props.api.kv.get(`${KV_PREFIX}.section.skills`, true)));
+                setSectionBalance(Boolean(props.api.kv.get(`${KV_PREFIX}.section.balance`, true)));
                 const bv = props.api.kv.get(`${KV_PREFIX}.border`, true);
                 setBorderVisible(bv !== false);
                 // Restore language preference
@@ -663,7 +827,8 @@ function TokenCachePanel(props) {
         const unsubMsg = props.api.event.on("message.updated", () => { bumpPartVersion(); setRefreshTick(v => v + 1); });
         const unsubSession = props.api.event.on("session.updated", () => { setRefreshTick(v => v + 1); });
         setRefreshTick(v => v + 1);
-        onCleanup(() => { clearTimeout(partTimer); unsubPart(); unsubMsg(); unsubSession(); });
+        const balanceTimer = setInterval(pollBalance, BALANCE_POLL_MS);
+        onCleanup(() => { clearTimeout(partTimer); clearInterval(balanceTimer); unsubPart(); unsubMsg(); unsubSession(); });
     });
     // ── colours ──
     // Pull from the current theme, auto-desaturate if too punchy,
@@ -732,7 +897,36 @@ function TokenCachePanel(props) {
                                                 const maxLabel = Math.max(4, panelWidth() - gutter() - rightW - 1);
                                                 const label = truncateVisual(sk.name, maxLabel);
                                                 return (_jsx("text", { fg: pal().muted, children: justify(label, fmt(sk.tokens), t().tok) }));
-                                            }) })] }) })] })] })] }));
+                                            }) })] }) }), _jsxs(Show, { when: sectionBalance(), children: [_jsx("text", { fg: pal().muted, children: sep() }), _jsx(Show, { when: balanceState().status === "idle", children: _jsxs("text", { fg: pal().muted, children: [_jsx("span", { style: { fg: pal().muted }, children: "> " }), _jsx("span", { children: t().balNoKey.replace("{p}", providerName()) })] }) }), _jsx(Show, { when: balanceState().status === "loading", children: _jsxs("text", { fg: pal().muted, children: [_jsx("span", { style: { fg: pal().muted }, children: "> " }), _jsx("span", { children: t().balLoading })] }) }), _jsx(Show, { when: balanceState().status === "error", children: _jsxs("text", { fg: pal().error, children: [_jsx("span", { style: { fg: pal().muted }, children: "> " }), _jsx("span", { children: (() => {
+                                                        const code = balanceState().error;
+                                                        if (code === "401")
+                                                            return t().balErr401;
+                                                        if (code === "403")
+                                                            return t().balErr403;
+                                                        if (code === "EMPTY")
+                                                            return t().balErrEmpty;
+                                                        if (code === "TIMEOUT")
+                                                            return t().balErrTimeout;
+                                                        return t().balError + (code ? ` (${code})` : "");
+                                                    })() })] }) }), _jsx(Show, { when: balanceState().status === "ok" && balanceState().data, children: (() => {
+                                            const list = balanceState().data;
+                                            const pref = balanceCurrency();
+                                            // 偏好币种是 DeepSeek 原生返回的（CNY/USD）→ 直接显示
+                                            const native = pref ? list.find(x => x.currency === pref) : undefined;
+                                            if (native) {
+                                                return (_jsx("text", { fg: pal().text, children: justify(t().balTotal, balanceSymbol(native.currency) + native.total) }));
+                                            }
+                                            // 非原生币种（EUR/JPY/GBP/KRW…）→ 取第一条余额按汇率换算
+                                            const base = list[0];
+                                            const baseAmt = parseFloat(base.total);
+                                            const converted = Number.isFinite(baseAmt)
+                                                ? convertBalance(pref || base.currency, exchangeRate(), baseAmt, base.currency)
+                                                : baseAmt;
+                                            const shown = pref && base.currency !== pref
+                                                ? converted.toLocaleString("en-US", { maximumFractionDigits: 2 })
+                                                : base.total;
+                                            return (_jsx("text", { fg: pal().text, children: justify(t().balTotal, balanceSymbol(pref || base.currency) + shown) }));
+                                        })() })] })] })] })] }));
 }
 // ---------------------------------------------------------------------------
 // Plugin entry
@@ -764,6 +958,11 @@ const tui = async (api) => {
     const [sectionModel, setSectionModel] = createSignal(true);
     const [sectionDist, setSectionDist] = createSignal(true);
     const [sectionSkills, setSectionSkills] = createSignal(true);
+    const [sectionBalance, setSectionBalance] = createSignal(true);
+    const [balanceRefresh, setBalanceRefresh] = createSignal(0);
+    const [balanceProviderId, setBalanceProviderId] = createSignal("deepseek");
+    const [autoBalance, setAutoBalance] = createSignal(true);
+    const [balanceCurrency, setBalanceCurrency] = createSignal("");
     const [borderVisible, setBorderVisible] = createSignal(true);
     const [langZH, setLangZH] = createSignal(LANG_ZH);
     const [overrideSessionId, setOverrideSessionId] = createSignal(undefined);
@@ -775,12 +974,57 @@ const tui = async (api) => {
         sectionModel, setSectionModel,
         sectionDist, setSectionDist,
         sectionSkills, setSectionSkills,
+        sectionBalance, setSectionBalance,
+        balanceRefresh, setBalanceRefresh,
+        balanceProviderId, setBalanceProviderId,
+        autoBalance, setAutoBalance,
+        balanceCurrency, setBalanceCurrency,
         borderVisible, setBorderVisible,
         overrideSessionId, setOverrideSessionId,
     };
     api.slots.register(createSidebarSlot(api, signals));
     // ── slash commands for runtime config ──
     const KV_PREFIX = "cache_panel";
+    /** 菜单中 provider 选项标题：标注 key 来源（手动配置 / OpenCode 自动复用 / 未配置）。 */
+    const providerOptionTitle = (p, current) => {
+        const zh = langZH();
+        const hasManual = !!api.kv.get(`${KV_PREFIX}.balance.${p.id}.key`, "");
+        const hasAuto = !hasManual && !!findOpencodeKey(api, p);
+        const mark = hasManual
+            ? (zh ? "（用户 key）" : " (user key)")
+            : hasAuto
+                ? (zh ? "（OpenCode）" : " (OpenCode)")
+                : (zh ? "（未配置）" : " (not set)");
+        return p.name + mark + (current && p.id === current ? " *" : "");
+    };
+    /** 弹出指定 provider 的 API Key 输入框（脱敏预填；空清除 / 含 * 保留原 key / 新 key 实时刷新）。 */
+    const promptBalanceKey = (dialog, provider) => {
+        const zh = langZH();
+        const current = api.kv.get(`${KV_PREFIX}.balance.${provider.id}.key`, "");
+        const masked = maskKey(current);
+        dialog?.replace(() => (_jsx(api.ui.DialogPrompt, { title: provider.name, description: () => _jsx("text", { children: zh ? `输入 ${provider.name} API Key 以显示账户余额（留空清除）` : `Enter your ${provider.name} API key to show account balance (leave empty to clear)` }), placeholder: provider.keyPlaceholder ?? "sk-...", value: masked, onConfirm: (val) => {
+                const input = val.trim();
+                let key;
+                if (input === "") {
+                    key = "";
+                }
+                else if (input.includes("*")) {
+                    key = current;
+                }
+                else {
+                    key = input;
+                }
+                api.kv.set(`${KV_PREFIX}.balance.${provider.id}.key`, key);
+                setBalanceRefresh(v => v + 1);
+                if (key) {
+                    api.ui.toast({ message: zh ? "API Key 已保存，正在查询余额..." : "API Key saved, fetching balance..." });
+                }
+                else {
+                    api.ui.toast({ message: zh ? "API Key 已清除" : "API Key cleared" });
+                }
+                dialog?.clear();
+            }, onCancel: () => dialog?.clear() })));
+    };
     api.command?.register(() => [
         {
             title: "Cache: Set Currency",
@@ -796,6 +1040,9 @@ const tui = async (api) => {
                         const defRate = DEFAULT_RATES[opt.value] ?? 1;
                         api.kv.set(`${KV_PREFIX}.currency`, sym);
                         api.kv.set(`${KV_PREFIX}.rate`, defRate);
+                        // 同步余额显示币种偏好：CNY/USD 原生直显，其余币种按汇率换算
+                        api.kv.set(`${KV_PREFIX}.balance_currency`, opt.value);
+                        signals.setBalanceCurrency(opt.value);
                         signals.setCurrencySymbol(sym);
                         signals.setExchangeRate(defRate);
                         api.ui.toast({ message: `Currency: ${opt.value} (${sym}), rate: ${defRate}` });
@@ -830,12 +1077,14 @@ const tui = async (api) => {
                 const modelOn = Boolean(api.kv.get(`${KV_PREFIX}.section.model`, true));
                 const distOn = Boolean(api.kv.get(`${KV_PREFIX}.section.dist`, true));
                 const skillsOn = Boolean(api.kv.get(`${KV_PREFIX}.section.skills`, true));
+                const balanceOn = Boolean(api.kv.get(`${KV_PREFIX}.section.balance`, true));
                 const borderOn = Boolean(api.kv.get(`${KV_PREFIX}.border`, true));
                 dialog?.replace(() => (_jsx(api.ui.DialogSelect, { title: "Toggle Section", options: [
                         { title: `Token Detail    [${detailOn ? "ON" : "OFF"}]`, value: "detail" },
                         { title: `Model & Pricing [${modelOn ? "ON" : "OFF"}]`, value: "model" },
                         { title: `Token Dist.     [${distOn ? "ON" : "OFF"}]`, value: "dist" },
                         { title: `Loaded Skills   [${skillsOn ? "ON" : "OFF"}]`, value: "skills" },
+                        { title: `Balance         [${balanceOn ? "ON" : "OFF"}]`, value: "balance" },
                         { title: `Panel Border    [${borderOn ? "ON" : "OFF"}]`, value: "border" },
                     ], onSelect: (opt) => {
                         if (opt.value === "border") {
@@ -856,6 +1105,8 @@ const tui = async (api) => {
                                 signals.setSectionDist(!cur);
                             if (opt.value === "skills")
                                 signals.setSectionSkills(!cur);
+                            if (opt.value === "balance")
+                                signals.setSectionBalance(!cur);
                             api.ui.toast({ message: `${opt.value} section ${!cur ? "shown" : "hidden"}` });
                         }
                         dialog?.clear();
@@ -874,9 +1125,10 @@ const tui = async (api) => {
                 const model = Boolean(api.kv.get(`${KV_PREFIX}.section.model`, true));
                 const dist = Boolean(api.kv.get(`${KV_PREFIX}.section.dist`, true));
                 const skills = Boolean(api.kv.get(`${KV_PREFIX}.section.skills`, true));
+                const balance = Boolean(api.kv.get(`${KV_PREFIX}.section.balance`, true));
                 api.ui.toast({
                     title: "Cache Panel Config",
-                    message: `Currency: ${sym}  |  Rate: ${rate}  |  Detail: ${detail ? "ON" : "OFF"}  |  Model: ${model ? "ON" : "OFF"}  |  Dist: ${dist ? "ON" : "OFF"}  |  Skills: ${skills ? "ON" : "OFF"}`,
+                    message: `Currency: ${sym}  |  Rate: ${rate}  |  Detail: ${detail ? "ON" : "OFF"}  |  Model: ${model ? "ON" : "OFF"}  |  Dist: ${dist ? "ON" : "OFF"}  |  Skills: ${skills ? "ON" : "OFF"}  |  Balance: ${balance ? "ON" : "OFF"}`,
                     duration: 8000,
                 });
                 dialog?.clear();
@@ -901,6 +1153,82 @@ const tui = async (api) => {
                     } })));
             },
         },
+        {
+            title: "Cache: Switch Balance Provider",
+            value: "cache.balance",
+            description: "切换余额提供商 / 自动切换当前会话提供商 | Switch balance provider / auto-switch session provider",
+            slash: { name: "cache-balance" },
+            onSelect: (dialog) => {
+                const zh = langZH();
+                const current = signals.balanceProviderId();
+                const auto = signals.autoBalance();
+                const autoLabel = auto
+                    ? (zh ? "自动切换提供商 [开]" : "Auto-switch provider [ON]")
+                    : (zh ? "自动切换提供商 [关]" : "Auto-switch provider [OFF]");
+                dialog?.replace(() => (_jsx(api.ui.DialogSelect, { title: zh ? "余额提供商 / 自动切换" : "Balance Provider / Auto-switch", options: [
+                        {
+                            title: autoLabel,
+                            value: "__auto__",
+                        },
+                        ...balanceProviders.map((p) => ({
+                            title: providerOptionTitle(p, current),
+                            value: p.id,
+                        })),
+                    ], onSelect: (opt) => {
+                        if (opt.value === "__auto__") {
+                            const next = !auto;
+                            api.kv.set(`${KV_PREFIX}.balance.auto`, next);
+                            signals.setAutoBalance(next);
+                            api.ui.toast({ message: zh ? `自动切换余额提供商: ${next ? "开" : "关"}` : `Auto-switch balance provider: ${next ? "ON" : "OFF"}` });
+                            dialog?.clear();
+                        }
+                        else {
+                            const provider = getBalanceProvider(opt.value);
+                            // 手动切换会关闭自动切换
+                            api.kv.set(`${KV_PREFIX}.balance.provider`, provider.id);
+                            api.kv.set(`${KV_PREFIX}.balance.auto`, false);
+                            signals.setBalanceProviderId(provider.id);
+                            signals.setAutoBalance(false);
+                            // 切换后立即按新 provider 刷新显示（无 key 时显示 idle，避免残留上一 provider 余额）
+                            signals.setBalanceRefresh(signals.balanceRefresh() + 1);
+                            const hasKey = !!api.kv.get(`${KV_PREFIX}.balance.${provider.id}.key`, "");
+                            if (!hasKey) {
+                                // 未配置 key → 进入设置流程（对话框保持打开等待输入）
+                                promptBalanceKey(dialog, provider);
+                            }
+                            else {
+                                api.ui.toast({ message: zh ? `余额提供商: ${provider.name}（自动切换已关闭）` : `Balance provider: ${provider.name} (auto-switch off)` });
+                                dialog?.clear();
+                            }
+                        }
+                    } })));
+            },
+        },
+        {
+            title: "Cache: Set Balance API Key",
+            value: "cache.balance.key",
+            description: "Select a provider and set its API key for balance display",
+            slash: { name: "cache-balance-key" },
+            onSelect: (dialog) => {
+                const zh = langZH();
+                // 步骤 1：选择 provider
+                dialog?.replace(() => (_jsx(api.ui.DialogSelect, { title: zh ? "选择余额提供商" : "Select Balance Provider", options: balanceProviders.map((p) => ({
+                        title: providerOptionTitle(p),
+                        value: p.id,
+                    })), onSelect: (opt) => {
+                        const provider = getBalanceProvider(opt.value);
+                        // 手动指定 provider 会关闭自动切换
+                        api.kv.set(`${KV_PREFIX}.balance.provider`, provider.id);
+                        api.kv.set(`${KV_PREFIX}.balance.auto`, false);
+                        signals.setBalanceProviderId(provider.id);
+                        signals.setAutoBalance(false);
+                        // 切换后立即刷新显示（防止取消输入时残留上一 provider 的余额）
+                        signals.setBalanceRefresh(signals.balanceRefresh() + 1);
+                        // 步骤 2：输入 key
+                        promptBalanceKey(dialog, provider);
+                    } })));
+            },
+        },
         {
             title: "Cache: Debug Skills Detection",
             value: "cache.debug-skills",