@toddzheng024/dscode-bundle 0.7.1 → 0.7.3

package/plugins/session-cards/index.mjs

@@ -3,19 +3,22 @@ import { homedir } from 'node:os';
 import { BlockAssembler, createUserMessage } from '@deepseek-ai/dsh-llm';
 import { SessionCards } from './manager.mjs';
 import { TOPIC_PROMPT } from './content.mjs';
+import { chargeTo } from '../session-metrics/attribution.mjs';
 export const name = 'dscode-session-cards';
 export const inject = ['sessions', 'llm'];
 export function apply(ctx, config = {}) {
   const home = process.env.DSH_HOME ?? process.env.DSCODE_HOME ?? join(homedir(), '.local/share/dscode-hub');
-  const cards = new SessionCards({ root: join(home, 'session-cards'), config, generate: async (input, route, signal) => {
+  const cards = new SessionCards({ root: join(home, 'session-cards'), config, generate: async (input, route, signal, sessionId) => {
     const assembler = new BlockAssembler(); let finished = false, usage;
-    for await (const chunk of ctx.llm.stream({ ...route, reasoningEffort: 'low', system: TOPIC_PROMPT,
-      messages: [createUserMessage({ content: [{ type: 'text', text: JSON.stringify(input) }], source: { kind: 'plugin', plugin: name } })],
-      maxTokens: 2000, signal })) {
-      signal.throwIfAborted(); assembler.push(chunk);
-      if (chunk.type === 'finish') finished = true;
-      if (chunk.type === 'usage') usage = chunk.usage;
-    }
+    await chargeTo(sessionId, 'session-card', async () => {
+      for await (const chunk of ctx.llm.stream({ ...route, reasoningEffort: 'low', system: TOPIC_PROMPT,
+        messages: [createUserMessage({ content: [{ type: 'text', text: JSON.stringify(input) }], source: { kind: 'plugin', plugin: name } })],
+        maxTokens: 2000, signal })) {
+        signal.throwIfAborted(); assembler.push(chunk);
+        if (chunk.type === 'finish') finished = true;
+        if (chunk.type === 'usage') usage = chunk.usage;
+      }
+    });
     if (!finished || assembler.finish.kind !== 'stop') throw Error('Incomplete topic response');
     const blocks = assembler.blocks();
     if (blocks.some(b => !['text', 'reasoning'].includes(b.type))) throw Error('Unexpected topic tool call');

package/plugins/session-cards/manager.mjs

@@ -105,7 +105,7 @@ export class SessionCards {
     try {
       const signal = AbortSignal.any([this.stop.signal, controller.signal, AbortSignal.timeout(this.config.timeoutMs)]);
       const result = await this.generate({ messages, topicCount: this.config.topicCount },
-        { provider: this.config.provider ?? state.route.provider, model: this.config.model ?? state.route.model }, signal);
+        { provider: this.config.provider ?? state.route.provider, model: this.config.model ?? state.route.model }, signal, state.session.id);
       usage = result.usage;
       signal.throwIfAborted();
       if (this.states.get(state.session.id) !== state || hash !== this.hash(state)) return;

package/plugins/session-metrics/attribution.mjs

@@ -0,0 +1,19 @@
+// Charge model calls a plugin makes for a session (review, memory, session cards,
+// /doctor) to that session's cost ledger without putting the session on the wire:
+// a `sessionId` request option also drives session-log delivery, provider cache
+// affinity and agent-only prompt shaping, which these calls must not trigger. The
+// ledger middleware runs when the caller iterates the stream, so it reads the charge
+// from the caller's async context.
+import { AsyncLocalStorage } from 'node:async_hooks';
+
+const charges = new AsyncLocalStorage();
+
+/** Run `fn` with its model calls charged to `sessionId` under `purpose`; no session runs it uncharged. */
+export function chargeTo(sessionId, purpose, fn) {
+  return sessionId ? charges.run({ sessionId, purpose }, fn) : fn();
+}
+
+/** The charge of the model call being iterated, if any. */
+export function currentCharge() {
+  return charges.getStore();
+}

package/plugins/session-metrics/balance.mjs

@@ -0,0 +1,71 @@
+// Remaining provider balance, refreshed on a long cache. DeepSeek's response
+// also carries the trusted clock: its `Date` header anchors peak/off-peak pricing
+// without a second network call. OpenRouter has no peak window, so only its
+// remaining credits are read.
+const SOURCES = {
+  'deepseek-official': { url: 'https://api.deepseek.com/user/balance', env: 'DEEPSEEK_API_KEY', parse: body => parseBalance(body), clock: true },
+  openrouter: { url: 'https://openrouter.ai/api/v1/credits', env: 'OPENROUTER_API_KEY', parse: body => parseOpenRouterCredits(body), clock: false },
+};
+export const BALANCE_PROVIDERS = Object.freeze(Object.keys(SOURCES));
+const CACHE_MS = 5 * 60 * 1000;
+const RETRY_MS = 60 * 1000;
+const snapshots = new Map();
+let clock = { anchor: null, skewMs: 0 };
+const snapshotOf = provider => snapshots.get(provider) ?? { balance: null, fetchedAt: 0, pending: false };
+
+/** Best-effort positive USD balance from a `/user/balance` body. */
+export function parseBalance(body) {
+  const infos = Array.isArray(body?.balance_infos) ? body.balance_infos : [];
+  const usd = infos.find(info => String(info?.currency ?? '').toUpperCase() === 'USD') ?? infos[0];
+  const total = Number(usd?.total_balance);
+  if (!Number.isFinite(total) || total < 0) return null;
+  return body?.is_available === false ? null : total;
+}
+
+/** Remaining USD credits from an OpenRouter `/credits` body: purchased minus used. */
+export function parseOpenRouterCredits(body) {
+  const credits = Number(body?.data?.total_credits), used = Number(body?.data?.total_usage);
+  if (body?.data?.total_credits == null || body?.data?.total_usage == null) return null;
+  if (!Number.isFinite(credits) || !Number.isFinite(used) || credits < 0 || used < 0) return null;
+  return Math.max(0, credits - used);
+}
+
+export function balanceNow(provider = 'deepseek-official') { return snapshotOf(provider).balance; }
+/** Clock anchored to the last DeepSeek balance response's `Date` header, else the local one. */
+export function trustedNow() {
+  const local = Date.now();
+  return clock.anchor === null ? local : local + clock.skewMs;
+}
+
+/** Refresh one provider at most once per cache window; never throws into the render path. */
+export async function refreshBalance(options = {}) {
+  const provider = options.provider ?? 'deepseek-official';
+  const source = SOURCES[provider];
+  if (!source) return null;
+  const snapshot = snapshotOf(provider);
+  const now = Date.now();
+  if (snapshot.pending || now - snapshot.fetchedAt < CACHE_MS) return snapshot.balance;
+  const retrySoon = () => Date.now() - (CACHE_MS - RETRY_MS);
+  const key = options.key ?? process.env[source.env];
+  if (!key) { snapshots.set(provider, { ...snapshot, fetchedAt: retrySoon(), pending: false }); return snapshot.balance; }
+  const fetchImpl = options.fetch ?? globalThis.fetch;
+  if (typeof fetchImpl !== 'function') return snapshot.balance;
+  snapshots.set(provider, { ...snapshot, pending: true });
+  try {
+    const response = await fetchImpl(source.url, { headers: { Authorization: `Bearer ${key}`, Accept: 'application/json' } });
+    const header = source.clock ? response.headers?.get?.('date') : null;
+    const anchor = header ? Date.parse(header) : NaN;
+    const body = await response.json();
+    const parsed = response.ok ? source.parse(body) : null;
+    if (Number.isFinite(anchor)) clock = { anchor, skewMs: anchor - Date.now() };
+    snapshots.set(provider, {
+      balance: parsed === null && response.ok ? null : parsed ?? snapshot.balance,
+      fetchedAt: parsed === null ? retrySoon() : Date.now(),
+      pending: false,
+    });
+  } catch {
+    // A transient failure keeps the last known balance and retries sooner.
+    snapshots.set(provider, { ...snapshot, fetchedAt: retrySoon(), pending: false });
+  }
+  return snapshotOf(provider).balance;
+}

package/plugins/session-metrics/index.mjs

@@ -1,8 +1,14 @@
 import { randomUUID } from 'node:crypto';
 import { appendMetric } from './store.mjs';
-import { estimateCost, PRICE_VERSION } from './pricing.mjs';
+import { estimateCost, priceVersionFor } from './pricing.mjs';
 import { setMetricSource } from './view.mjs';
+import { BALANCE_PROVIDERS, refreshBalance } from './balance.mjs';
+import { providerSpec } from '../providers/catalog.mjs';
 import { createWindowRate } from './rate.mjs';
+import { currentCharge } from './attribution.mjs';
+
+// Chunks that carry generated output; the first one marks time to first token.
+const OUTPUT_CHUNKS = new Set(['text-delta', 'reasoning-delta', 'tool-call-delta']);
 export const name = 'dscode-session-metrics';
 export const inject = ['llm', 'agents', 'tokenMeter', 'sessionProjections'];
 export function apply(ctx) {
@@ -21,29 +27,53 @@ export function apply(ctx) {
     return { ...value, currentTps: liveRate.get(session) };
   }));
   const home = process.env.DSH_HOME;
+  // Remaining balance is best-effort decoration: resolve the key lazily (never
+  // inject the credentials service, so a missing one cannot fail startup), keep
+  // the request on a five-minute cache, and never let it reach the render path.
+  const credentials = ctx.get?.('credentials');
+  const refresh = () => Promise.all(BALANCE_PROVIDERS.map(async provider => {
+    try {
+      const ref = providerSpec(provider).credentialRef;
+      const resolved = await credentials?.resolve?.(ref);
+      const key = typeof resolved === 'string' ? resolved : resolved?.value;
+      await refreshBalance({ provider, key: key ?? process.env[ref] });
+    } catch {
+      /* balance stays unknown */
+    }
+  }));
+  refresh();
+  const balanceTimer = setInterval(refresh, 5 * 60 * 1000);
+  if (typeof balanceTimer.unref === 'function') balanceTimer.unref();
+  ctx.effect(() => () => clearInterval(balanceTimer));
   const record = (id, entry) => { try { appendMetric(home, id, entry); } catch { ctx.logger.warn('Session cost telemetry could not be saved.'); } };
   ctx.on('llm/stream', async function* (options, next) {
-    if (!options.sessionId || !home) { yield* next(); return; }
+    // Plugin calls made for a session carry no sessionId on the wire; they are charged through the async context.
+    const charge = options.sessionId ? undefined : currentCharge();
+    const sessionId = options.sessionId ?? charge?.sessionId;
+    if (!sessionId || !home) { yield* next(); return; }
     const id = randomUUID(), time = Date.now();
-    const recipients = new Set([options.sessionId]);
-    let child = ctx.agents.get(options.sessionId);
+    const purpose = options.purpose ?? charge?.purpose ?? 'agent';
+    const recipients = new Set([sessionId]);
+    let child = ctx.agents.get(sessionId);
     while (child?.session.header.origin === 'subagent' && child.session.header.parentSession && !recipients.has(child.session.header.parentSession)) {
       recipients.add(child.session.header.parentSession);
       child = ctx.agents.get(child.session.header.parentSession);
     }
-    const save = entry => { for (const recipient of recipients) record(recipient, { ...entry, sessionId: options.sessionId }); };
-    save( { kind: 'start', id, time, provider: options.provider, model: options.model, purpose: options.purpose ?? 'agent' });
-    let usage;
-    const liveSession = !options.purpose || options.purpose === 'agent' ? ctx.agents.get(options.sessionId)?.session : undefined;
+    const save = entry => { for (const recipient of recipients) record(recipient, { ...entry, sessionId }); };
+    save({ kind: 'start', id, time, provider: options.provider, model: options.model, purpose });
+    let usage, firstTokenTime;
+    const liveSession = purpose === 'agent' ? ctx.agents.get(sessionId)?.session : undefined;
     try {
       for await (const chunk of next()) {
         if (chunk.type === 'usage') usage = chunk.usage;
+        else if (firstTokenTime === undefined && OUTPUT_CHUNKS.has(chunk.type)) firstTokenTime = Date.now();
         if (liveSession) liveRate.add(liveSession, chunk);
         yield chunk;
       }
     } finally {
       if (liveSession) liveRate.calibrate(liveSession, usage?.outputTokens);
-      save({ kind: 'end', id, time, usage: usage ?? null, cost: estimateCost(options.provider, options.model, usage, time), priceVersion: PRICE_VERSION });
+      // `time` stays the start (it prices the call); `endTime` and `firstTokenTime` time it.
+      save({ kind: 'end', id, time, endTime: Date.now(), ...(firstTokenTime === undefined ? {} : { firstTokenTime }), usage: usage ?? null, cost: estimateCost(options.provider, options.model, usage, time), priceVersion: priceVersionFor(options.provider) });
     }
   });
 }

package/plugins/session-metrics/pricing.mjs

@@ -2,17 +2,51 @@
 // https://api-docs.deepseek.com/quick_start/pricing/
 export const PRICE_SOURCE = 'https://api-docs.deepseek.com/quick_start/pricing/';
 export const PRICE_VERSION = 'deepseek-2026-09-11';
+// OpenRouter list prices for the DeepSeek models `/provider openrouter` declares,
+// from the pinned pi-ai 0.85.1 catalog (its OpenRouter /models snapshot).
+// OpenRouter bills no peak window. [cache read, input, output].
+export const OPENROUTER_PRICE_VERSION = 'openrouter-pi-ai-0.85.1';
+const OPENROUTER_PRICES = {
+  'deepseek/deepseek-v4-flash': [0.017052, 0.08526, 0.17052],
+  'deepseek/deepseek-v4-pro': [0.074196, 0.890358, 1.780716],
+  'deepseek/deepseek-v4-flash-vision-exp': [0.007, 0.22, 0.66],
+};
+
+/** The price table a provider's ledger entries are estimated with. */
+export function priceVersionFor(provider) {
+  return provider === 'openrouter' ? OPENROUTER_PRICE_VERSION : PRICE_VERSION;
+}
+
 export function estimateCost(provider, model, usage, time) {
-  if (provider !== 'deepseek-official' || !usage || !Number.isFinite(time)) return null;
+  if (!usage || !Number.isFinite(time)) return null;
+  if (provider === 'openrouter') return OPENROUTER_PRICES[model] ? charge(usage, OPENROUTER_PRICES[model]) : null;
+  if (provider !== 'deepseek-official') return null;
   // Earlier requests require an older price table; never back-price them at today's rate.
   if (time < Date.UTC(2026, 8, 11)) return null;
   const flash = ['deepseek-flash', 'deepseek-v4-flash', 'deepseek-v4-flash-vision-exp'].includes(model)
     || model === 'deepseek-v4-pro' && time >= Date.UTC(2026, 8, 14, 4);
   if (!flash && model !== 'deepseek-v4-pro') return null;
-  const date = new Date(time), day = date.getUTCDay(), hour = date.getUTCHours();
-  const peak = day >= 1 && day <= 5 && (hour >= 1 && hour < 4 || hour >= 6 && hour < 10);
-  const [read, input, output] = flash ? [0.003, 0.15, 0.6] : [0.022, 0.66, 1.98];
+  const cost = charge(usage, flash ? [0.003, 0.15, 0.6] : [0.022, 0.66, 1.98]);
+  return cost === null ? null : cost * (isPeak(time) ? 2 : 1);
+}
+
+function charge(usage, [read, input, output]) {
   const values = [usage.inputTokens, usage.outputTokens, usage.cacheReadTokens ?? 0, usage.cacheWriteTokens ?? 0];
   if (!values.every(n => Number.isFinite(n) && n >= 0) || values[3] !== 0) return null;
-  return (values[0] * input + values[1] * output + values[2] * read) * (peak ? 2 : 1) / 1e6;
+  return (values[0] * input + values[1] * output + values[2] * read) / 1e6;
+}
+
+/**
+ * Peak/off-peak window in UTC: weekdays 01:00-04:00 and 06:00-10:00.
+ * @param time - Unix ms timestamp.
+ * @returns Whether that instant bills at the peak rate.
+ */
+export function isPeak(time) {
+  const date = new Date(time), day = date.getUTCDay(), hour = date.getUTCHours();
+  return day >= 1 && day <= 5 && (hour >= 1 && hour < 4 || hour >= 6 && hour < 10);
+}
+
+/** Footer marker for the billing window: fire for peak, snowflake for off-peak. */
+export function peakEmoji(time) {
+  return isPeak(time) ? '🔥' : '❄️';
 }

package/plugins/session-metrics/view.mjs

@@ -1,7 +1,9 @@
 import { readMetrics } from './store.mjs';
 import { t } from '../i18n/messages.mjs';
-import { estimateCost } from './pricing.mjs';
+import { estimateCost, peakEmoji } from './pricing.mjs';
+import { balanceNow, trustedNow } from './balance.mjs';
 import { sessionAverageTps } from './rate.mjs';
+import { providerOfHeader } from '../providers/catalog.mjs';
 let source;
 export function setMetricSource(next) { source = next; return () => { if (source === next) source = undefined; }; }
 export function summarize(rows, events = [], corrupt = false) {
@@ -37,31 +39,50 @@ export function summarize(rows, events = [], corrupt = false) {
   }
   return { cost, unknown, calls, pending, cache: input > 0 && !cacheUnknown ? Math.min(100, hit / input * 100) : null };
 }
-/** Terminal columns of a string: East Asian wide characters (including the ｜ separator) take two. */
+/** Terminal columns of a string: East Asian wide characters (such as the cache label's CJK glyphs) take two. */
 export function displayWidth(text) {
   let width = 0;
-  for (const char of text) width += /[\u1100-\u115f\u2e80-\ua4cf\uac00-\ud7a3\uf900-\ufaff\ufe30-\ufe4f\uff00-\uff60\uffe0-\uffe6]/.test(char) ? 2 : 1;
+  for (const char of text) width += /[\u1100-\u115f\u2e80-\ua4cf\uac00-\ud7a3\uf900-\ufaff\ufe30-\ufe4f\uff00-\uff60\uffe0-\uffe6\u2600-\u27bf\u{1f300}-\u{1faff}]/u.test(char) ? 2 : 1;
   return width;
 }
-export function formatFooter(metrics, context, columns = 80, rates, locale = 'en') {
+export function formatFooter(metrics, context, columns = 80, rates, locale = 'en', header = '', provider = providerOfHeader(header) ?? 'deepseek-official') {
   const label = key => t(locale, key);
   const ctx = Number.isFinite(context) ? `${Math.round(context)}%` : '--';
   const cache = metrics.cache === null ? '--' : `${metrics.cache.toFixed(1)}%`;
-  const dollars = metrics.unknown && metrics.cost === 0 ? '--' : `~$${metrics.cost.toFixed(metrics.cost < 1 ? 4 : 2)}${metrics.unknown ? '+' : ''}${metrics.pending ? '…' : ''}`;
-  const parts = rates ? [
+  // The balance belongs to the provider the header names; only DeepSeek's official route bills by a peak window.
+  const balance = balanceNow(provider);
+  const spend = metrics.unknown && metrics.cost === 0 ? '--' : `$${metrics.cost.toFixed(2)}${metrics.unknown ? '+' : ''}${metrics.pending ? '…' : ''}`;
+  const dollars = `${spend} / ${balance === null ? '$--' : '$' + balance.toFixed(2)}${provider === 'deepseek-official' ? ' ' + peakEmoji(trustedNow()) : ''}`;
+  const base = rates ? [
     `${label('footer.current')}: ${Number.isFinite(rates.current) ? '~' + rates.current.toFixed(1) : '--'} tps`,
     `${label('footer.average')}: ${Number.isFinite(rates.average) ? rates.average.toFixed(1) : '--'} tps`,
-    `${label('footer.context')}: ${ctx}`, dollars, `${label('footer.cache')} ${cache}`,
-  ] : [`${label('footer.context')}: ${ctx}`, dollars, `${label('footer.cache')} ${cache}`];
-  for (let count = parts.length; count > 0; count--) {
-    const value = parts.slice(0, count).join(' ｜ ');
-    if (displayWidth(value) <= columns) return value;
+    `${label('footer.context')}: ${ctx}`, dollars, `${label('footer.cache')}: ${cache}`,
+  ] : [`${label('footer.context')}: ${ctx}`, dollars, `${label('footer.cache')}: ${cache}`];
+  // Narrow terminals shed the quietest figures first: average, cache hit, current. The
+  // model header then falls back to its bare `model @ effort` form, then context goes,
+  // and only then the header itself — the running cost is the last thing standing.
+  const drops = rates ? [1, 4, 0, 2] : [2, 0];
+  const offset = header === '' ? 0 : 1;
+  const parts = header === '' ? base : [header, ...base];
+  const short = header.replace(/^[^:]+: /, '');
+  const heads = header === '' ? [''] : short === header ? [header] : [header, short];
+  const render = ({ omit, head }) => parts
+    .map((part, index) => (index === 0 && header !== '' ? head : part))
+    .filter((part, index) => part !== '' && !omit.has(index))
+    .join(' | ');
+  for (let dropped = 0; dropped <= drops.length; dropped++) {
+    const omit = new Set(drops.slice(0, dropped).map(index => index + offset));
+    for (const head of heads) {
+      const value = render({ omit, head });
+      if (displayWidth(value) <= columns) return value;
+    }
   }
+  const floor = render({ omit: new Set(drops.map(index => index + offset)), head: '' });
   let clipped = '';
-  for (const char of parts[0]) { if (displayWidth(clipped + char) > columns) break; clipped += char; }
+  for (const char of floor) { if (displayWidth(clipped + char) > columns) break; clipped += char; }
   return clipped;
 }
-export function footerFor(id, stats, columns, locale = 'en') {
+export function footerFor(id, stats, columns, header = '', locale = 'en') {
   try {
     const data = id ? source?.(id) : undefined;
     const ledger = id && process.env.DSH_HOME ? readMetrics(process.env.DSH_HOME, id) : { rows: [], corrupt: false };
@@ -69,6 +90,6 @@ export function footerFor(id, stats, columns, locale = 'en') {
     const used = data?.used;
     const capacity = data?.capacity ?? stats.contextWindow;
     const average = sessionAverageTps(data?.events ?? []);
-    return formatFooter(summary, Number.isFinite(used) && capacity > 0 ? used / capacity * 100 : undefined, columns, { current: data?.currentTps, average }, locale);
-  } catch { return formatFooter({ cost: 0, unknown: true, cache: null }, undefined, columns, { current: null, average: null }, locale); }
+    return formatFooter(summary, Number.isFinite(used) && capacity > 0 ? used / capacity * 100 : undefined, columns, { current: data?.currentTps, average }, locale, header);
+  } catch { return formatFooter({ cost: 0, unknown: true, cache: null }, undefined, columns, { current: null, average: null }, locale, header); }
 }

package/plugins/tui-tools/doctor.mjs

@@ -4,6 +4,7 @@ import { Logger } from '@deepseek-ai/cordis';
 import { BlockAssembler, createUserMessage } from '@deepseek-ai/dsh-llm';
 import { redact } from '../auto-review/policy.mjs';
 import { t, readLanguage } from '../i18n/messages.mjs';
+import { chargeTo } from '../session-metrics/attribution.mjs';
 const L = (key, params) => t(readLanguage(), key, params);
 
 const MAX_LOG_BYTES = 1024 * 1024;
@@ -118,7 +119,7 @@ export function localDoctorReport(evidence) {
   return `${L('doctor.evidence', { traces: evidence.traces.length, logs: evidence.logs.length })}${evidence.logs.length ? '' : evidence.logCoverage ?? ''}\n${local.length ? local.slice(-8).join('\n') : L('doctor.noFindings')}\n${near300.length >= 2 ? `${L('doctor.nearTimeout', { count: near300.length })}\n` : ''}${evidence.logs.slice(-5).map(l => `${time(l.time)} ${l.level} ${l.source}: ${l.detail}`).join('\n')}`;
 }
 
-export async function analyzeDoctorEvidence(ctx, evidence, route, signal, { model = true } = {}) {
+export async function analyzeDoctorEvidence(ctx, evidence, route, signal, { model = true, sessionId } = {}) {
   const fallback = localDoctorReport(evidence);
   if (!model) return fallback;
   if (!route?.provider || !route?.model) return `${fallback}\n${L('doctor.noRoute')}`;
@@ -126,11 +127,13 @@ export async function analyzeDoctorEvidence(ctx, evidence, route, signal, { mode
   const deadline = AbortSignal.any([signal ?? new AbortController().signal, AbortSignal.timeout(45000)]);
   try {
     let finished = false;
-    for await (const chunk of ctx.llm.stream({ provider: route.provider, model: route.model, reasoningEffort: 'low', maxTokens: 4096, system: SYSTEM,
-      messages: [createUserMessage({ content: [{ type: 'text', text: JSON.stringify(evidence) }], source: { kind: 'plugin', plugin: 'dscode-doctor' } })], signal: deadline })) {
-      deadline.throwIfAborted(); assembler.push(chunk);
-      if (chunk.type === 'finish') finished = true;
-    }
+    await chargeTo(sessionId, 'doctor', async () => {
+      for await (const chunk of ctx.llm.stream({ provider: route.provider, model: route.model, reasoningEffort: 'low', maxTokens: 4096, system: SYSTEM,
+        messages: [createUserMessage({ content: [{ type: 'text', text: JSON.stringify(evidence) }], source: { kind: 'plugin', plugin: 'dscode-doctor' } })], signal: deadline })) {
+        deadline.throwIfAborted(); assembler.push(chunk);
+        if (chunk.type === 'finish') finished = true;
+      }
+    });
     if (!finished || assembler.finish.kind !== 'stop') throw Error(`model response was incomplete (${safe(JSON.stringify(assembler.finish ?? { kind: 'no finish' }))})`);
     const blocks = assembler.blocks();
     if (blocks.some(b => !['text', 'reasoning'].includes(b.type))) throw Error('model returned unexpected tool output');

package/plugins/tui-tools/index.mjs

@@ -92,7 +92,7 @@ export function apply(ctx) {
       `Session: ${session.id}`, `Workspace: ${session.header.cwd ?? process.cwd()}`,
       `Agent: ${agent.status} | preset: ${session.header.agentPreset ?? 'standard'}`,
       `Model: ${route?.provider ?? 'default'} / ${route?.model ?? 'default'}`,
-      `Effort: ${route?.reasoningEffort ?? 'model default'}${route?.reasoningEffort === 'ultra' ? ' (DeepSeek wire: max; collaboration enabled)' : ''}`,
+      `Effort: ${route?.reasoningEffort ?? 'model default'}${route?.reasoningEffort === 'ultra' ? ` (${route.provider === 'openrouter' ? 'OpenRouter wire: xhigh' : 'DeepSeek wire: max'}; collaboration enabled)` : ''}`,
       `Permission: ${show(ctx.permissionPresets.current(session))}`,
       `Tokens: ${show(usage ?? 'no provider usage yet')}`,
       `Context: ${pressure?.pressureTokens ?? pressure?.surfaceTokens ?? '?'} / ${pressure?.contextWindow ?? '?'} tokens`,
@@ -120,7 +120,7 @@ export function apply(ctx) {
     const evidence = await collectDoctorEvidence(ctx, { agent, signal });
     if (action === 'preview') return ok(JSON.stringify(evidence, null, 2));
     const route = agent.session.requestHeader()?.config ?? agent.options;
-    return ok(`${health}\n\n${await analyzeDoctorEvidence(ctx, evidence, route, signal, { model: action !== 'local' })}`);
+    return ok(`${health}\n\n${await analyzeDoctorEvidence(ctx, evidence, route, signal, { model: action !== 'local', sessionId: agent.session.id })}`);
   });
   register('mcp', 'MCP list, tools <id>, enable/disable/reconnect <id>', async ({ agent, rawInput }) => {
     const [action = 'list', id, extra] = rawInput.trim().split(/\s+/).filter(Boolean);

package/plugins/ultra/policy.mjs

@@ -16,6 +16,22 @@ export function flashRequest(options, messages) {
   return copy;
 }
 
+/**
+ * The same shaping for pi-ai routes (OpenRouter), on harness-format options:
+ * Ultra sends max and adds the collaboration policy; below Ultra the delegation
+ * tools are not offered. Returns a copy; the logged input is never mutated.
+ */
+export function piAiRequest(options) {
+  const ultra = options.reasoningEffort === 'ultra';
+  const hidden = ultra ? ['workflow', 'ralph'] : ['subagent', 'subagent_fork', 'workflow', 'ralph'];
+  const next = { ...options, ...(options.tools ? { tools: options.tools.filter(tool => !hidden.includes(tool.name)) } : {}), ...(ultra ? { reasoningEffort: 'max' } : {}) };
+  if (!ultra || options.purpose || !options.tools?.some(t => t.name === 'subagent' || t.name === 'subagent_fork')) return next;
+  if (typeof next.system === 'string') return { ...next, system: next.system + '\n\n' + ULTRA_POLICY };
+  const [first, ...rest] = next.messages ?? [];
+  if (first?.role === 'system' && Array.isArray(first.content)) return { ...next, messages: [{ ...first, content: [...first.content, { type: 'text', text: '\n\n' + ULTRA_POLICY }] }, ...rest] };
+  return { ...next, messages: [{ role: 'system', content: [{ type: 'text', text: ULTRA_POLICY }] }, ...(next.messages ?? [])] };
+}
+
 export function ultraRequest(options, messages) {
   if (options.reasoningEffort !== 'ultra' || options.purpose || !options.tools?.some(t => t.name === 'subagent' || t.name === 'subagent_fork')) return messages;
   const copy = messages.map(m => ({ ...m }));

package/vendor/persistent/index.js

@@ -1,3 +1,6 @@
+// dscode-stdin-stall-v2
+const STDIN_STALL_MS = 5000;
+const STDIN_STALL_NOTE = "\n[Interrupted after 5s with no output: the command held the terminal without producing anything, and this tool cannot supply terminal input. Give it stdin (a heredoc, `< file` or `< /dev/null`) or use a non-interactive form.]";
 // dscode-shell-reset-v1
 import { randomUUID } from "node:crypto";
 import z from "@deepseek-ai/schemastery";
@@ -255,6 +258,9 @@ async function executeCommand(ctx, shells, owner, command, config, upstream) {
 		const id = await shells.get(owner, commandDeadline.signal);
 		const marker = markers();
 		const wrapped = wrapCommand(command, marker);
+		const startedAt = Date.now();
+		let stdinWaited = false;
+		let stdinStalled = false;
 		let first = true;
 		let fallback = "";
 		let fallbackTruncated = false;
@@ -298,10 +304,21 @@ async function executeCommand(ctx, shells, owner, command, config, upstream) {
 			}
 			if (latest.text.includes(marker.end)) {
 				const complete = commandOutput(retainedScrollback(ctx, owner, id, latest), marker);
-				if (complete !== void 0) return renderCaptured(complete, config.maxOutputChars);
+				if (complete !== void 0) return renderCaptured(complete, config.maxOutputChars) + (stdinStalled ? STDIN_STALL_NOTE : "");
 			}
 			if (result.sessionStatus.kind === "exited") return await respondToSessionExit(ctx, shells, owner, id, result.sessionStatus, marker, fallback, fallbackTruncated, config);
-			if (result.waitReason === "stdin_read") return renderCaptured(partialOutput(retainedScrollback(ctx, owner, id, latest), marker, fallback, fallbackTruncated), config.maxOutputChars);
+			const partial = renderCaptured(partialOutput(retainedScrollback(ctx, owner, id, latest), marker, fallback, fallbackTruncated), config.maxOutputChars);
+			if (result.waitReason === "stdin_read") stdinWaited = true;
+			if (result.waitReason === "stdin_read" && partial.length > 0) return partial;
+			if (stdinWaited && partial.length === 0 && !stdinStalled && Date.now() - startedAt >= STDIN_STALL_MS) {
+				stdinStalled = true;
+				try {
+					await ctx.terminals.signal(owner, id, "SIGINT");
+				} catch (_unsignallableForeground) {
+					stdinStalled = false;
+					return partial;
+				}
+			}
 			await pause();
 		}
 	} catch (e_1) {

package/vendor/pi-ai/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 DeepSeek
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.