@deepseek-harness-tui/dsh-tui 0.11.1 → 0.11.2

package/lib/types/dsh-adapter/subagents.js

@@ -1,3 +1,4 @@
+import { addUsageToCostBuckets, cloneCostBuckets, emptyCostBuckets, isDeepSeekOfficialProvider, isPeakHour, priceForModel } from '../deepseekPricing.js';
 import { toolResultPayload } from './compat/messages.js';
 const MAX_OUTPUT_EVENTS = 160;
 const MAX_OUTPUT_LINES = 160;
@@ -6,6 +7,12 @@ export class SubagentActivityStore {
     sessionToAgent = new Map();
     listeners = new Set();
     streams = new Map();
+    /** 会话级 durable usage 费用桶，key = `${provider}\u0000${model}`。 */
+    costByModel = new Map();
+    /** 非官方 provider / 未收录模型的用量副本（只展示 token，不计金额）。 */
+    costUnpriced = emptyCostBuckets();
+    /** 每个 agent 已消费的最大 durable 事件 seq（同一事件重投不重复计费）。 */
+    costSeq = new Map();
     commitLine(agentId, kind, text) {
         const state = this.states.get(agentId);
         if (!state)
@@ -241,8 +248,12 @@ export class SubagentActivityStore {
             case 'assistant/message': {
                 if (Array.isArray(data.stream))
                     this.settleAssistant(agentId, data.message?.content, data.turn, data.step, ev.seq);
-                if (data.usage)
+                if (data.usage) {
                     this.setTokens(agentId, data.usage);
+                    // durable usage 才进费用累计：live assistant/chunk 的 usage 只更新
+                    // 展示 token（见 setTokens 调用点），两路同时到达也只计这一路。
+                    this.accumulateCost(agentId, data.usage, ev.time, ev.seq);
+                }
                 break;
             }
             case 'assistant/attempt': {
@@ -320,6 +331,66 @@ export class SubagentActivityStore {
         state.tokens = { ...state.tokens, input, output, total: usage.total ?? ((input ?? 0) + (output ?? 0) || undefined) };
         this.notify();
     }
+    /** durable usage 的计价字段；全零/非法返回 undefined（不建零桶）。 */
+    costDeltaOf(usage) {
+        if (usage === null || typeof usage !== 'object')
+            return undefined;
+        const value = usage;
+        const count = (raw) => typeof raw === 'number' && Number.isFinite(raw) && raw > 0 ? raw : 0;
+        const delta = {
+            input: count(value.inputTokens ?? value.input),
+            output: count(value.outputTokens ?? value.output),
+            cacheRead: count(value.cacheReadTokens),
+            cacheWrite: count(value.cacheWriteTokens),
+        };
+        if (delta.input === 0 && delta.output === 0 && delta.cacheRead === 0 && delta.cacheWrite === 0)
+            return undefined;
+        return delta;
+    }
+    /** 一笔 durable usage 按发生时刻落 peak/idle 桶，model/provider 取事件
+     *  发生时的子代理身份；非官方/未收录同时进 unpriced 副本。
+     *
+     *  `seq` 是同一 binding 内的事件序号：重放/回放把同一条 durable 事件再投
+     *  一次时只计第一笔（重投即双计会让"并入子代理"的估算凭空变大）。判据与
+     *  `settleAssistant` 的 `settledSeq` 同源但**各自独立**——两处都要消费
+     *  同一个 seq，共用字段会让后消费的那处被前一处挡掉。 */
+    accumulateCost(agentId, usage, time, seq) {
+        const state = this.states.get(agentId);
+        const delta = this.costDeltaOf(usage);
+        if (state === undefined || delta === undefined)
+            return;
+        if (typeof seq === 'number') {
+            const consumed = this.costSeq.get(agentId);
+            if (consumed !== undefined && seq <= consumed)
+                return;
+            this.costSeq.set(agentId, seq);
+        }
+        const provider = state.provider ?? 'subagent';
+        const model = state.model ?? provider;
+        const key = `${provider}\u0000${model}`;
+        let entry = this.costByModel.get(key);
+        if (entry === undefined) {
+            entry = { provider, model, buckets: emptyCostBuckets() };
+            this.costByModel.set(key, entry);
+        }
+        const at = typeof time === 'number' ? new Date(time) : new Date();
+        const peak = isPeakHour(at);
+        addUsageToCostBuckets(entry.buckets, delta, peak);
+        if (!isDeepSeekOfficialProvider(provider) || priceForModel(model) === undefined) {
+            addUsageToCostBuckets(this.costUnpriced, delta, peak);
+        }
+    }
+    /** durable 用量的费用快照（深拷贝）：entries 供计价，unpriced 供展示。 */
+    costSnapshot() {
+        return {
+            entries: Array.from(this.costByModel.values()).map(entry => ({
+                provider: entry.provider,
+                model: entry.model,
+                buckets: cloneCostBuckets(entry.buckets),
+            })),
+            unpriced: cloneCostBuckets(this.costUnpriced),
+        };
+    }
     onCompleted(agentId, summary, stopReason = 'completed', endedAt) { this.finish(agentId, 'completed', stopReason, summary, endedAt); }
     onFailed(agentId, error, endedAt) { this.finish(agentId, 'failed', error, undefined, endedAt); }
     onCancelled(agentId, reason = 'cancelled', summary, endedAt) { this.finish(agentId, 'cancelled', reason, summary, endedAt); }
@@ -351,11 +422,15 @@ export class SubagentActivityStore {
     }
     /** Drop all tracked subagents and session links (session swap: the old
      * session's children were disposed with their parent — nothing may keep
-     * routing events of a session the channel no longer projects). */
+     * routing events of a session the channel no longer projects). The
+     * session-scoped cost buckets reset with them. */
     reset() {
         this.states.clear();
         this.sessionToAgent.clear();
         this.streams.clear();
+        this.costByModel.clear();
+        this.costUnpriced = emptyCostBuckets();
+        this.costSeq.clear();
         this.notify();
     }
     snapshot() { return Array.from(this.states.values()).map(state => ({ ...state, output: [...state.output], outputEvents: [...state.outputEvents], toolCalls: state.toolCalls.map(tool => ({ ...tool })), tokens: state.tokens ? { ...state.tokens } : undefined })); }

package/lib/types/herdr.d.ts

@@ -21,6 +21,8 @@ export interface HerdrIntegrationOptions {
     readonly reportTimeoutMs?: number;
     readonly releaseTimeoutMs?: number;
     readonly retryDelaysMs?: readonly number[];
+    /** Milliseconds. Tests pin this so a later report can jump ahead of another pane reporter. */
+    readonly now?: () => number;
 }
 /** Report this TUI's lifecycle to the owning Herdr pane when Herdr launches it. */
 export declare function attachHerdrIntegration(options: HerdrIntegrationOptions): HerdrIntegration | undefined;

package/lib/types/herdr.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"herdr.d.ts","sourceRoot":"","sources":["../../src/herdr.ts"],"names":[],"mappings":"AAAA,OAAO,EAAmB,KAAK,qBAAqB,EAAE,MAAM,4BAA4B,CAAA;AAQxF,UAAU,YAAY;IACpB,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAA;IACzB,SAAS,CAAC,QAAQ,EAAE,MAAM,IAAI,GAAG,MAAM,IAAI,CAAA;CAC5C;AAED,UAAU,aAAa;IACrB,SAAS,CAAC,QAAQ,EAAE,MAAM,IAAI,GAAG,MAAM,IAAI,CAAA;IAC3C,WAAW,IAAI,OAAO,GAAG,IAAI,CAAA;CAC9B;AAED,KAAK,UAAU,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,MAAM,EAAE,KAAK,OAAO,CAAC,qBAAqB,CAAC,CAAA;AAE3F,MAAM,WAAW,gBAAgB;IAC/B,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,CAAA;IACxB,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,CAAA;CACzB;AAED,MAAM,WAAW,uBAAuB;IACtC,QAAQ,CAAC,OAAO,EAAE,YAAY,CAAA;IAC9B,QAAQ,CAAC,SAAS,EAAE,aAAa,CAAA;IACjC,QAAQ,CAAC,SAAS,EAAE,aAAa,CAAA;IACjC,QAAQ,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,UAAU,CAAA;IAChC,QAAQ,CAAC,GAAG,CAAC,EAAE,UAAU,CAAA;IACzB,QAAQ,CAAC,eAAe,CAAC,EAAE,MAAM,CAAA;IACjC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,MAAM,CAAA;IAClC,QAAQ,CAAC,aAAa,CAAC,EAAE,SAAS,MAAM,EAAE,CAAA;CAC3C;AAkBD,mFAAmF;AACnF,wBAAgB,sBAAsB,CACpC,OAAO,EAAE,uBAAuB,GAC/B,gBAAgB,GAAG,SAAS,CAgH9B"}
+{"version":3,"file":"herdr.d.ts","sourceRoot":"","sources":["../../src/herdr.ts"],"names":[],"mappings":"AAAA,OAAO,EAAmB,KAAK,qBAAqB,EAAE,MAAM,4BAA4B,CAAA;AAQxF,UAAU,YAAY;IACpB,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAA;IACzB,SAAS,CAAC,QAAQ,EAAE,MAAM,IAAI,GAAG,MAAM,IAAI,CAAA;CAC5C;AAED,UAAU,aAAa;IACrB,SAAS,CAAC,QAAQ,EAAE,MAAM,IAAI,GAAG,MAAM,IAAI,CAAA;IAC3C,WAAW,IAAI,OAAO,GAAG,IAAI,CAAA;CAC9B;AAED,KAAK,UAAU,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,MAAM,EAAE,KAAK,OAAO,CAAC,qBAAqB,CAAC,CAAA;AAE3F,MAAM,WAAW,gBAAgB;IAC/B,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,CAAA;IACxB,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,CAAA;CACzB;AAED,MAAM,WAAW,uBAAuB;IACtC,QAAQ,CAAC,OAAO,EAAE,YAAY,CAAA;IAC9B,QAAQ,CAAC,SAAS,EAAE,aAAa,CAAA;IACjC,QAAQ,CAAC,SAAS,EAAE,aAAa,CAAA;IACjC,QAAQ,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,UAAU,CAAA;IAChC,QAAQ,CAAC,GAAG,CAAC,EAAE,UAAU,CAAA;IACzB,QAAQ,CAAC,eAAe,CAAC,EAAE,MAAM,CAAA;IACjC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,MAAM,CAAA;IAClC,QAAQ,CAAC,aAAa,CAAC,EAAE,SAAS,MAAM,EAAE,CAAA;IAC1C,8FAA8F;IAC9F,QAAQ,CAAC,GAAG,CAAC,EAAE,MAAM,MAAM,CAAA;CAC5B;AAkBD,mFAAmF;AACnF,wBAAgB,sBAAsB,CACpC,OAAO,EAAE,uBAAuB,GAC/B,gBAAgB,GAAG,SAAS,CAsH9B"}

package/lib/types/herdr.js

@@ -33,7 +33,13 @@ export function attachHerdrIntegration(options) {
             return Promise.reject(new Error('Herdr command could not be started'));
         }
     };
-    let sequence = Date.now() * 1000;
+    const now = options.now ?? Date.now;
+    let sequence = 0;
+    const nextSequence = () => {
+        const clock = now() * 1000;
+        sequence = clock > sequence ? clock : sequence + 1;
+        return String(sequence);
+    };
     let lastConfirmedReport = '';
     let disposed = false;
     let running = false;
@@ -73,7 +79,7 @@ export function attachHerdrIntegration(options) {
                     '--agent', 'dsh-tui',
                     '--state', state,
                     ...(blocked ? ['--message', 'Waiting for user input'] : []),
-                    '--seq', String(++sequence),
+                    '--seq', nextSequence(),
                 ]), reportTimeoutMs);
                 if (disposed)
                     break;
@@ -124,7 +130,7 @@ export function attachHerdrIntegration(options) {
                 'pane', 'release-agent', paneId,
                 '--source', 'custom:dsh-tui',
                 '--agent', 'dsh-tui',
-                '--seq', String(++sequence),
+                '--seq', nextSequence(),
             ]);
             disposePromise = withTimeout(release, releaseTimeoutMs).then(() => undefined);
             return disposePromise;