opencode-visual-cache 1.2.16 → 1.4.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-visual-cache",
-  "version": "1.2.16",
+  "version": "1.4.0",
   "description": "OpenCode TUI plugin displaying real-time token cache hit rate in the sidebar",
   "type": "module",
   "types": "dist/index.d.ts",

package/src/_version.ts

@@ -1,2 +1,2 @@
 // auto-generated
-export const PLUGIN_VERSION="1.2.16";
+export const PLUGIN_VERSION="1.4.0";

package/src/index.tsx

@@ -113,6 +113,14 @@ const ZH_T = {
   secDetail:  "明细",
   secModel:   "模型",
   secSkills:  "已加载技能",
+  balTotal:   "总余额:",
+  balNoKey:   "未配置 API Key",
+  balLoading: "查询中...",
+  balError:   "查询失败",
+  balErr401:  "API Key 无效",
+  balErr403:  "余额查询被拒绝",
+  balErrEmpty:"未获取到余额数据",
+  balErrTimeout: "查询超时",
 } as const
 
 const EN_T = {
@@ -145,6 +153,14 @@ const EN_T = {
   secDetail:  "Detail",
   secModel:   "Model",
   secSkills:  "Loaded Skills",
+  balTotal:   "Total:",
+  balNoKey:   "No API Key set",
+  balLoading: "Fetching...",
+  balError:   "Fetch failed",
+  balErr401:  "Invalid API Key",
+  balErr403:  "Balance request rejected",
+  balErrEmpty:"No balance data",
+  balErrTimeout: "Request timed out",
 } as const
 
 // ── color helpers ────────────────────────────────────────────────
@@ -223,6 +239,16 @@ function desaturateTo(raw: unknown, maxSat: number, fallback: string): string {
   return "#" + [nr, ng, nb].map((v) => Math.max(0, Math.min(255, v)).toString(16).padStart(2, "0")).join("")
 }
 
+/** Darken a hex colour by multiplying each channel by `factor` (0–1). */
+function dimColor(hex: string, factor = 0.5): string {
+  const c = rgb(hex)
+  if (!c) return hex
+  const r = Math.round(c.r * factor)
+  const g = Math.round(c.g * factor)
+  const b = Math.round(c.b * factor)
+  return "#" + [r, g, b].map((v) => Math.max(0, Math.min(255, v)).toString(16).padStart(2, "0")).join("")
+}
+
 // Morandi fallbacks — used when a theme colour cannot be resolved
 const FALLBACK = {
   primary: "#8B9DAF",
@@ -321,6 +347,64 @@ interface TokenDist {
   stepCost: number
 }
 
+// ---------------------------------------------------------------------------
+// DeepSeek Balance
+// ---------------------------------------------------------------------------
+
+interface DeepSeekBalance {
+  currency: string
+  total: string
+}
+
+interface BalanceState {
+  status: "idle" | "loading" | "ok" | "error"
+  data: DeepSeekBalance[] | null
+  lastFetch: number
+  error?: string
+  key?: string           // 上次成功/尝试查询所用的 key，用于检测 key 是否更换
+}
+
+const BALANCE_POLL_MS = 5 * 60 * 1000 // 5 minutes
+
+async function fetchDeepSeekBalance(apiKey: string, signal?: AbortSignal): Promise<DeepSeekBalance[]> {
+  const res = await fetch("https://api.deepseek.com/user/balance", {
+    headers: { Authorization: `Bearer ${apiKey}`, Accept: "application/json" },
+    signal,
+  })
+  if (!res.ok) {
+    if (res.status === 401) throw new Error("401")
+    if (res.status === 402 || res.status === 403) throw new Error("403")
+    throw new Error(String(res.status))
+  }
+  const json = await res.json() as {
+    is_available?: boolean
+    balance_infos?: { currency: string; total_balance: string; granted_balance: string; topped_up_balance: string }[]
+  }
+  const infos = json.balance_infos ?? []
+  if (infos.length === 0) throw new Error("EMPTY")
+  return infos.map((info) => ({
+    currency: info.currency ?? "CNY",
+    total: info.total_balance ?? "0",
+  }))
+}
+
+/**
+ * 将余额从来源币种换算为目标币种。
+ * DEFAULT_RATES 以 USD=1 为基准：先折算为 USD，再换算到目标币种。
+ */
+function convertBalance(target: string, targetRate: number, amount: number, from: string): number {
+  if (from === target) return amount
+  const fromRate = DEFAULT_RATES[from] ?? 1
+  const usd = from === "USD" ? amount : amount / fromRate
+  return target === "USD" ? usd : usd * targetRate
+}
+
+/** 货币符号：优先取 /cache-currency 内置映射，未知币种回退为代码。 */
+function balanceSymbol(currency: string): string {
+  const sym = CURRENCIES[currency]
+  return sym ?? currency + " "
+}
+
 // ---------------------------------------------------------------------------
 // Sidebar component
 // ---------------------------------------------------------------------------
@@ -343,8 +427,19 @@ interface PanelSignals {
   setSectionDist: (v: boolean) => void
   sectionSkills: () => boolean
   setSectionSkills: (v: boolean) => void
+  sectionBalance: () => boolean
+  setSectionBalance: (v: boolean) => void
+  /** Increment to force a DeepSeek balance re-fetch. */
+  balanceRefresh: () => number
+  setBalanceRefresh: (v: number) => void
+  /** Preferred currency code for balance display (CNY / USD / …). Empty = first entry. */
+  balanceCurrency: () => string
+  setBalanceCurrency: (v: string) => void
   borderVisible: () => boolean
   setBorderVisible: (v: boolean) => void
+  /** When set, the panel renders stats for this session instead of the main one. */
+  overrideSessionId: () => string | undefined
+  setOverrideSessionId: (v: string | undefined) => void
 }
 
 const CURRENCIES: Record<string, string> = {
@@ -391,6 +486,9 @@ function TokenCachePanel(props: {
     sectionModel, setSectionModel,
     sectionDist, setSectionDist,
     sectionSkills, setSectionSkills,
+    sectionBalance, setSectionBalance,
+    balanceRefresh,
+    balanceCurrency, setBalanceCurrency,
     borderVisible, setBorderVisible,
   } = props.signals
 
@@ -424,8 +522,63 @@ function TokenCachePanel(props: {
   })
   const [refreshTick, setRefreshTick] = createSignal(0)
 
+  // ── balance state + polling ──────────────────────────────────
+  const [balanceState, setBalanceState] = createSignal<BalanceState>({
+    status: "idle", data: null, lastFetch: 0,
+  })
+  // 请求序号：防止定时轮询与手动刷新并发时，慢的旧请求覆盖新结果
+  let balanceSeq = 0
+
+  const pollBalance = async () => {
+    const key = props.api.kv.get<string>(`${KV_PREFIX}.ds_key`, "")
+    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 fetchDeepSeekBalance(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() })
+  })
+
+  // ── auto-clear override when the user navigates to a different main session ──
+  let lastMainSid = props.sessionId
   createEffect(() => {
     const sid = props.sessionId
+    if (sid !== lastMainSid) {
+      lastMainSid = sid
+      if (props.signals.overrideSessionId()) {
+        props.signals.setOverrideSessionId(undefined)
+        props.api.kv.set(`${KV_PREFIX}.session`, "")
+      }
+    }
+  })
+
+  createEffect(() => {
+    const sid = props.signals.overrideSessionId() ?? props.sessionId
     void refreshTick()
     void partVersion()
 
@@ -608,10 +761,13 @@ function TokenCachePanel(props: {
         const rate = props.api.kv.get<number>(`${KV_PREFIX}.rate`)
         if (typeof sym === "string") setCurrencySymbol(sym)
         if (typeof rate === "number" && rate > 0) setExchangeRate(rate)
+        const balCur = props.api.kv.get<string>(`${KV_PREFIX}.balance_currency`)
+        if (typeof balCur === "string") setBalanceCurrency(balCur)
         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<boolean>(`${KV_PREFIX}.border`, true)
         setBorderVisible(bv !== false)
         // Restore language preference
@@ -666,7 +822,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 ──
@@ -751,7 +908,7 @@ function TokenCachePanel(props: {
         <span style={{ fg: pal().primary }}>
             <b>{t().title}</b>
             <Show when={open()}>
-              <span style={{ fg: pal().muted }}> (v{PLUGIN_VERSION})</span>
+              <span style={{ fg: dimColor(pal().muted, 0.75) }}> v{PLUGIN_VERSION}</span>
             </Show>
           </span>
         <Show when={!open() && data().hasData}>
@@ -774,6 +931,18 @@ function TokenCachePanel(props: {
       </text>
 
       <Show when={open()}>
+        <Show when={props.signals.overrideSessionId()}>
+          {(() => {
+            const prefix = "  \u21b3 " + (langZH() ? "\u5B50\u4EE3\u7406: " : "Sub: ")
+            const maxSidW = Math.max(6, panelWidth() - visualWidth(prefix))
+            return (
+              <text>
+                <span style={{ fg: pal().muted }}>{prefix}</span>
+                <span style={{ fg: pal().text }}>{truncateVisual(props.signals.overrideSessionId()!, maxSidW)}</span>
+              </text>
+            )
+          })()}
+        </Show>
         <Show when={data().hasData} fallback={
           <>
             <text fg={pal().muted}>{sep()}</text>
@@ -939,6 +1108,65 @@ function TokenCachePanel(props: {
             </Show>
           </Show>
           </Show>
+
+          {/* ── DeepSeek balance (single line) ── */}
+          <Show when={sectionBalance()}>
+            <text fg={pal().muted}>{sep()}</text>
+            <Show when={balanceState().status === "idle"}>
+              <text fg={pal().muted}>
+                <span style={{ fg: pal().muted }}>{"> "}</span>
+                <span>{t().balNoKey}</span>
+              </text>
+            </Show>
+            <Show when={balanceState().status === "loading"}>
+              <text fg={pal().muted}>
+                <span style={{ fg: pal().muted }}>{"> "}</span>
+                <span>{t().balLoading}</span>
+              </text>
+            </Show>
+            <Show when={balanceState().status === "error"}>
+              <text fg={pal().error}>
+                <span style={{ fg: pal().muted }}>{"> "}</span>
+                <span>{(() => {
+                  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})` : "")
+                })()}</span>
+              </text>
+            </Show>
+            <Show when={balanceState().status === "ok" && balanceState().data}>
+              {(() => {
+                const list = balanceState().data!
+                const pref = balanceCurrency()
+                // 偏好币种是 DeepSeek 原生返回的（CNY/USD）→ 直接显示
+                const native = pref ? list.find(x => x.currency === pref) : undefined
+                if (native) {
+                  return (
+                    <text fg={pal().text}>
+                      {justify(t().balTotal, balanceSymbol(native.currency) + native.total)}
+                    </text>
+                  )
+                }
+                // 非原生币种（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 (
+                  <text fg={pal().text}>
+                    {justify(t().balTotal, balanceSymbol(pref || base.currency) + shown)}
+                  </text>
+                )
+              })()}
+            </Show>
+          </Show>
         </Show>
       </Show>
     </box>
@@ -950,10 +1178,19 @@ function TokenCachePanel(props: {
 // ---------------------------------------------------------------------------
 
 function createSidebarSlot(api: TuiPluginApi, signals: PanelSignals): TuiSlotPlugin {
+  let lastSlotSid = ""
   return {
     order: 55,
     slots: {
       sidebar_content(ctx: TuiSlotContext, input: { session_id: string }): JSX.Element {
+        // ── auto-clear override when the user navigates to a different main session ──
+        if (input.session_id !== lastSlotSid) {
+          lastSlotSid = input.session_id
+          if (signals.overrideSessionId()) {
+            signals.setOverrideSessionId(undefined)
+            api.kv.set("cache_panel.session", "")
+          }
+        }
         return (
           <TokenCachePanel
             theme={ctx.theme.current}
@@ -975,8 +1212,12 @@ const tui: TuiPlugin = async (api: TuiPluginApi) => {
   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 [balanceCurrency, setBalanceCurrency] = createSignal("")
   const [borderVisible, setBorderVisible] = createSignal(true)
   const [langZH, setLangZH] = createSignal(LANG_ZH)
+  const [overrideSessionId, setOverrideSessionId] = createSignal<string | undefined>(undefined)
 
   const signals: PanelSignals = {
     currencySymbol, setCurrencySymbol,
@@ -986,7 +1227,11 @@ const tui: TuiPlugin = async (api: TuiPluginApi) => {
     sectionModel, setSectionModel,
     sectionDist, setSectionDist,
     sectionSkills, setSectionSkills,
+    sectionBalance, setSectionBalance,
+    balanceRefresh, setBalanceRefresh,
+    balanceCurrency, setBalanceCurrency,
     borderVisible, setBorderVisible,
+    overrideSessionId, setOverrideSessionId,
   }
 
   api.slots.register(createSidebarSlot(api, signals))
@@ -1012,6 +1257,9 @@ const tui: TuiPlugin = async (api: TuiPluginApi) => {
               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}` })
@@ -1056,6 +1304,7 @@ const tui: TuiPlugin = async (api: TuiPluginApi) => {
         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(() => (
           <api.ui.DialogSelect
@@ -1065,6 +1314,7 @@ const tui: TuiPlugin = async (api: TuiPluginApi) => {
               { 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: `DS Balance      [${balanceOn ? "ON" : "OFF"}]`, value: "balance" },
               { title: `Panel Border    [${borderOn ? "ON" : "OFF"}]`,  value: "border" },
             ]}
             onSelect={(opt) => {
@@ -1081,6 +1331,7 @@ const tui: TuiPlugin = async (api: TuiPluginApi) => {
                 if (opt.value === "model")  signals.setSectionModel(!cur)
                 if (opt.value === "dist")   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()
@@ -1101,9 +1352,10 @@ const tui: TuiPlugin = async (api: TuiPluginApi) => {
         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()
@@ -1134,6 +1386,54 @@ const tui: TuiPlugin = async (api: TuiPluginApi) => {
         ))
       },
     },
+    {
+      title: "Cache: Set DeepSeek API Key",
+      value: "cache.balance.key",
+      description: "Set or update the DeepSeek API key for balance display",
+      slash: { name: "cache-balance-key" },
+      onSelect: (dialog) => {
+        const zh = langZH()
+        const current = api.kv.get<string>(`${KV_PREFIX}.ds_key`, "")
+        // 已保存的 key 以脱敏形式预填：保留 "sk-" 前缀 + 头 5 尾 5 字符，中间用 * 填充
+        const maskKey = (k: string): string => {
+          if (!k) return ""
+          const prefix = k.startsWith("sk-") ? "sk-" : ""
+          const body = prefix ? k.slice(3) : k
+          if (body.length <= 10) return prefix + body.slice(0, 5) + "*".repeat(Math.max(3, body.length - 5))
+          return prefix + body.slice(0, 5) + "*".repeat(Math.max(3, body.length - 10)) + body.slice(-5)
+        }
+        const masked = maskKey(current)
+        dialog?.replace(() => (
+          <api.ui.DialogPrompt
+            title={zh ? "DeepSeek API Key" : "DeepSeek API Key"}
+            description={() => <text>{zh ? "输入 DeepSeek API Key 以显示账户余额（留空清除）" : "Enter your DeepSeek API key to show account balance (leave empty to clear)"}</text>}
+            placeholder="sk-..."
+            value={masked}
+            onConfirm={(val) => {
+              const input = val.trim()
+              // 空 → 清除；含 * （脱敏占位符残留）→ 视为未修改，保留原 key；否则为新 key
+              let key: string
+              if (input === "") {
+                key = ""
+              } else if (input.includes("*")) {
+                key = current
+              } else {
+                key = input
+              }
+              api.kv.set(`${KV_PREFIX}.ds_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()}
+          />
+        ))
+      },
+    },
     {
       title: "Cache: Debug Skills Detection",
       value: "cache.debug-skills",
@@ -1174,6 +1474,120 @@ const tui: TuiPlugin = async (api: TuiPluginApi) => {
         })
       },
     },
+    {
+      title: "Cache: Sub-Agent Stats",
+      value: "cache.session",
+      description: "View token cache statistics for a sub-agent by session ID",
+      slash: { name: "cache-session" },
+      onSelect: (dialog) => {
+        // ── 扫描当前主 session 的子代理 session ID 列表 ──
+        const rt = api.route.current
+        const parentSid = rt.name === "session" && rt.params ? String(rt.params.sessionID) : ""
+        const SUBAGENT_TOOLS = new Set(["task", "delegate", "call_omo_agent"])
+
+        interface ChildEntry { title: string; value: string; description: string }
+        const children: ChildEntry[] = []
+        if (parentSid) {
+          try {
+            const msgs = api.state.session.messages(parentSid)
+            for (const msg of msgs) {
+              if (msg.role !== "assistant") continue
+              let parts: readonly Part[] = []
+              try { parts = api.state.part(msg.id) } catch {}
+              for (const p of parts) {
+                if (p.type !== "tool") continue
+                const tool = String((p as ToolPart).tool ?? "")
+                if (!SUBAGENT_TOOLS.has(tool)) continue
+                const st = (p as any).state as Record<string, unknown> | undefined
+                const stMeta = st?.metadata as Record<string, unknown> | undefined
+                const subSid = stMeta?.session_id ?? stMeta?.sessionId
+                if (!subSid) continue
+                const sidStr = String(subSid)
+                const input = st?.input as Record<string, unknown> | undefined
+                const agent = String((p as any).subagent_type ?? input?.subagent_type ?? input?.category ?? tool)
+                const prompt = String(input?.prompt ?? "")
+                const desc = input?.description ? String(input.description) : ""
+                const title = desc || prompt.replace(/\n/g, " ").replace(/\s+/g, " ").trim().slice(0, 40) || agent
+                children.push({ title, value: sidStr, description: `${agent} · ${sidStr.slice(0, 24)}…` })
+              }
+            }
+          } catch {}
+        }
+
+        // 去重
+        const seen = new Set<string>()
+        const unique = children.filter(c => { if (seen.has(c.value)) return false; seen.add(c.value); return true })
+
+        if (unique.length > 0) {
+          // ── 有子代理 → DialogSelect 列表选择 ──
+          const zh = langZH()
+          const currentSid = signals.overrideSessionId() ?? api.kv.get<string>(`${KV_PREFIX}.session`, "")
+          const options = unique.map((c, i) => ({
+            title: `${i + 1}. ${c.title}`,
+            value: c.value,
+            description: c.description,
+          }))
+          // 首尾各放一个"回到主会话"，长列表时顶部底部均可直达
+          const backValue = "__main__"
+          const backTitle = `\u2500 ${zh ? "\u56DE\u5230\u4E3B\u4F1A\u8BDD" : "Back to Main"}`
+          options.unshift({ title: backTitle, value: backValue, description: "" })
+          options.push({ title: backTitle, value: backValue, description: "" })
+          const currentIdx = currentSid ? options.findIndex(o => o.value === currentSid) : -1
+          dialog?.replace(() => (
+            <api.ui.DialogSelect
+              title={zh ? "选择子代理" : "Select Sub-Agent"}
+              options={options}
+              current={currentIdx >= 0 ? options[currentIdx].value : undefined}
+              onSelect={(opt) => {
+                if (opt.value === backValue) {
+                  signals.setOverrideSessionId(undefined)
+                  api.kv.set(`${KV_PREFIX}.session`, "")
+                  api.ui.toast({ message: zh ? "已切回主会话" : "Switched to main session" })
+                } else {
+                  signals.setOverrideSessionId(opt.value)
+                  api.kv.set(`${KV_PREFIX}.session`, opt.value)
+                  api.ui.toast({ message: (zh ? "已切换至子代理: " : "Showing sub-agent: ") + opt.value.slice(0, 24) + "\u2026" })
+                }
+                dialog?.clear()
+              }}
+            />
+          ))
+        } else {
+          // ── 无子代理 → DialogPrompt 手动粘贴 ──
+          const zh = langZH()
+          dialog?.replace(() => (
+            <api.ui.DialogPrompt
+              title={signals.overrideSessionId() ? zh ? "切换子代理" : "Switch Sub" : zh ? "查看子代理缓存" : "View Sub Cache"}
+              description={() => <text>{zh ? "未找到子代理，请手动粘贴 Session ID" : "No sub-agents found. Paste a Session ID manually"}</text>}
+              placeholder="ses_..."
+              value={signals.overrideSessionId() ?? api.kv.get<string>(`${KV_PREFIX}.session`, "") ?? ""}
+              onConfirm={(val) => {
+                const sid = val.trim()
+                if (sid) {
+                  signals.setOverrideSessionId(sid)
+                  api.kv.set(`${KV_PREFIX}.session`, sid)
+                  api.ui.toast({ message: (langZH() ? "已切换至子代理: " : "Showing sub-agent: ") + sid.slice(0, 24) + "\u2026" })
+                }
+                dialog?.clear()
+              }}
+              onCancel={() => dialog?.clear()}
+            />
+          ))
+        }
+      },
+    },
+    {
+      title: "Cache: Back to Main",
+      value: "cache.session.back",
+      description: "Return to main session stats",
+      slash: { name: "cache-session-back" },
+      onSelect: (dialog) => {
+        signals.setOverrideSessionId(undefined)
+        api.kv.set(`${KV_PREFIX}.session`, "")
+        api.ui.toast({ message: langZH() ? "已切回主会话" : "Switched to main session" })
+        dialog?.clear()
+      },
+    },
   ])
 }