@theia/ai-chat-ui 1.71.0-next.8 → 1.71.1

package/lib/browser/chat-token-usage-indicator-util.js

@@ -0,0 +1,138 @@
+"use strict";
+// *****************************************************************************
+// Copyright (C) 2026 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CHAT_CONTEXT_WINDOW_SIZE = void 0;
+exports.isAboveTokenUsageWarningThreshold = isAboveTokenUsageWarningThreshold;
+exports.decideTokenUsageWarning = decideTokenUsageWarning;
+exports.formatTokenCount = formatTokenCount;
+exports.getUsageColorClass = getUsageColorClass;
+exports.computeSessionTokenUsage = computeSessionTokenUsage;
+exports.getLatestTokenUsage = getLatestTokenUsage;
+exports.buildBarTooltip = buildBarTooltip;
+const nls_1 = require("@theia/core/lib/common/nls");
+/**
+ * Provisional context window size used as the denominator for the indicator bar
+ * fill and the tooltip's "Total: X / Y" display until per-model context sizes
+ * are available. See issue #17323 comments for context.
+ */
+exports.CHAT_CONTEXT_WINDOW_SIZE = 200000;
+/** Returns true when the given total has crossed (>=) the configured warning threshold. */
+function isAboveTokenUsageWarningThreshold(totalTokens, threshold) {
+    return totalTokens > 0 && totalTokens >= threshold;
+}
+/**
+ * Pure decision function for whether to show the token usage warning for a session.
+ * Callers are expected to short-circuit before invoking this when the warning feature
+ * is disabled, so this helper is not concerned with the enabled state.
+ * - `reset`: usage is below the threshold; any prior "already notified" state for this session should be cleared.
+ * - `skip`:  we already notified for this session while still above the threshold.
+ * - `notify`: warning should be shown now and the session marked as notified.
+ */
+function decideTokenUsageWarning(args) {
+    if (!isAboveTokenUsageWarningThreshold(args.totalTokens, args.threshold)) {
+        return 'reset';
+    }
+    if (args.alreadyNotified) {
+        return 'skip';
+    }
+    return 'notify';
+}
+function formatTokenCount(count) {
+    if (count === undefined || count === 0) {
+        return '-';
+    }
+    if (count >= 1000) {
+        return `${(count / 1000).toFixed(1)}k`;
+    }
+    return count.toString();
+}
+/**
+ * Returns the CSS class for the token usage indicator based on the current
+ * total, the configured warning threshold, and the assumed context window size.
+ * Yellow band: [threshold, contextWindowSize). Red band: [contextWindowSize, ∞).
+ */
+function getUsageColorClass(totalTokens, threshold, contextWindowSize = exports.CHAT_CONTEXT_WINDOW_SIZE) {
+    if (totalTokens === 0) {
+        return 'token-usage-none';
+    }
+    if (totalTokens < threshold) {
+        return 'token-usage-green';
+    }
+    if (totalTokens < contextWindowSize) {
+        return 'token-usage-yellow';
+    }
+    return 'token-usage-red';
+}
+function computeSessionTokenUsage(chatModel) {
+    if (!chatModel) {
+        return 0;
+    }
+    const requests = chatModel.getRequests();
+    for (let i = requests.length - 1; i >= 0; i--) {
+        const usage = requests[i].response.tokenUsage;
+        if (usage) {
+            return usage.inputTokens
+                + usage.outputTokens
+                + (usage.cacheCreationInputTokens ?? 0)
+                + (usage.cacheReadInputTokens ?? 0);
+        }
+    }
+    return 0;
+}
+function getLatestTokenUsage(chatModel) {
+    if (!chatModel) {
+        return undefined;
+    }
+    const requests = chatModel.getRequests();
+    for (let i = requests.length - 1; i >= 0; i--) {
+        const usage = requests[i].response.tokenUsage;
+        if (usage) {
+            return usage;
+        }
+    }
+    return undefined;
+}
+function buildBarTooltip(usage, totalTokens, threshold) {
+    if (!usage) {
+        return undefined;
+    }
+    const lines = [
+        `**${nls_1.nls.localize('theia/ai/chat-ui/tokenUsageLabel', 'Token Usage')}**`
+    ];
+    const colorClass = getUsageColorClass(totalTokens, threshold);
+    if (colorClass === 'token-usage-yellow') {
+        lines.push(`⚠ ${nls_1.nls.localize('theia/ai/chat-ui/tokenUsageWarning', 'Token usage warning threshold reached.')}`, '');
+    }
+    else if (colorClass === 'token-usage-red') {
+        lines.push(`⚠ ${nls_1.nls.localize('theia/ai/chat-ui/tokenUsageOverflow', 'Token usage well past the warning threshold. Consider compacting or starting a new session.')}`, '');
+    }
+    lines.push(`${nls_1.nls.localizeByDefault('Input: {0}', formatTokenCount(usage.inputTokens))} | ${nls_1.nls.localizeByDefault('Output: {0}', formatTokenCount(usage.outputTokens))}`);
+    const cacheParts = [];
+    if (usage.cacheReadInputTokens) {
+        cacheParts.push(nls_1.nls.localize('theia/ai/chat-ui/tokenUsageTooltipCacheRead', 'Cache read: {0}', formatTokenCount(usage.cacheReadInputTokens)));
+    }
+    if (usage.cacheCreationInputTokens) {
+        cacheParts.push(nls_1.nls.localize('theia/ai/chat-ui/tokenUsageTooltipCacheCreate', 'Cache creation: {0}', formatTokenCount(usage.cacheCreationInputTokens)));
+    }
+    if (cacheParts.length > 0) {
+        lines.push(cacheParts.join(' | '));
+    }
+    const percentage = Math.round((totalTokens / exports.CHAT_CONTEXT_WINDOW_SIZE) * 100);
+    lines.push(nls_1.nls.localize('theia/ai/chat-ui/tokenUsageTooltipTotal', 'Total: {0} / {1} ({2}%)', formatTokenCount(totalTokens), formatTokenCount(exports.CHAT_CONTEXT_WINDOW_SIZE), percentage));
+    return { value: lines.join('  \n') };
+}
+//# sourceMappingURL=chat-token-usage-indicator-util.js.map

package/lib/browser/chat-token-usage-indicator-util.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-token-usage-indicator-util.js","sourceRoot":"","sources":["../../src/browser/chat-token-usage-indicator-util.ts"],"names":[],"mappings":";AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;;;AAgBhF,8EAEC;AAUD,0DAYC;AAED,4CAQC;AAOD,gDAWC;AAED,4DAeC;AAED,kDAYC;AAED,0CAmCC;AArID,oDAAiD;AAGjD;;;;GAIG;AACU,QAAA,wBAAwB,GAAG,MAAM,CAAC;AAI/C,2FAA2F;AAC3F,SAAgB,iCAAiC,CAAC,WAAmB,EAAE,SAAiB;IACpF,OAAO,WAAW,GAAG,CAAC,IAAI,WAAW,IAAI,SAAS,CAAC;AACvD,CAAC;AAED;;;;;;;GAOG;AACH,SAAgB,uBAAuB,CAAC,IAIvC;IACG,IAAI,CAAC,iCAAiC,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;QACvE,OAAO,OAAO,CAAC;IACnB,CAAC;IACD,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,OAAO,MAAM,CAAC;IAClB,CAAC;IACD,OAAO,QAAQ,CAAC;AACpB,CAAC;AAED,SAAgB,gBAAgB,CAAC,KAAyB;IACtD,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;QACrC,OAAO,GAAG,CAAC;IACf,CAAC;IACD,IAAI,KAAK,IAAI,IAAI,EAAE,CAAC;QAChB,OAAO,GAAG,CAAC,KAAK,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IAC3C,CAAC;IACD,OAAO,KAAK,CAAC,QAAQ,EAAE,CAAC;AAC5B,CAAC;AAED;;;;GAIG;AACH,SAAgB,kBAAkB,CAAC,WAAmB,EAAE,SAAiB,EAAE,oBAA4B,gCAAwB;IAC3H,IAAI,WAAW,KAAK,CAAC,EAAE,CAAC;QACpB,OAAO,kBAAkB,CAAC;IAC9B,CAAC;IACD,IAAI,WAAW,GAAG,SAAS,EAAE,CAAC;QAC1B,OAAO,mBAAmB,CAAC;IAC/B,CAAC;IACD,IAAI,WAAW,GAAG,iBAAiB,EAAE,CAAC;QAClC,OAAO,oBAAoB,CAAC;IAChC,CAAC;IACD,OAAO,iBAAiB,CAAC;AAC7B,CAAC;AAED,SAAgB,wBAAwB,CAAC,SAAqB;IAC1D,IAAI,CAAC,SAAS,EAAE,CAAC;QACb,OAAO,CAAC,CAAC;IACb,CAAC;IACD,MAAM,QAAQ,GAAG,SAAS,CAAC,WAAW,EAAE,CAAC;IACzC,KAAK,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5C,MAAM,KAAK,GAAmC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC;QAC9E,IAAI,KAAK,EAAE,CAAC;YACR,OAAO,KAAK,CAAC,WAAW;kBAClB,KAAK,CAAC,YAAY;kBAClB,CAAC,KAAK,CAAC,wBAAwB,IAAI,CAAC,CAAC;kBACrC,CAAC,KAAK,CAAC,oBAAoB,IAAI,CAAC,CAAC,CAAC;QAC5C,CAAC;IACL,CAAC;IACD,OAAO,CAAC,CAAC;AACb,CAAC;AAED,SAAgB,mBAAmB,CAAC,SAAqB;IACrD,IAAI,CAAC,SAAS,EAAE,CAAC;QACb,OAAO,SAAS,CAAC;IACrB,CAAC;IACD,MAAM,QAAQ,GAAG,SAAS,CAAC,WAAW,EAAE,CAAC;IACzC,KAAK,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5C,MAAM,KAAK,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC;QAC9C,IAAI,KAAK,EAAE,CAAC;YACR,OAAO,KAAK,CAAC;QACjB,CAAC;IACL,CAAC;IACD,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,SAAgB,eAAe,CAAC,KAAqC,EAAE,WAAmB,EAAE,SAAiB;IACzG,IAAI,CAAC,KAAK,EAAE,CAAC;QACT,OAAO,SAAS,CAAC;IACrB,CAAC;IACD,MAAM,KAAK,GAAa;QACpB,KAAK,SAAG,CAAC,QAAQ,CAAC,kCAAkC,EAAE,aAAa,CAAC,IAAI;KAC3E,CAAC;IACF,MAAM,UAAU,GAAG,kBAAkB,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;IAC9D,IAAI,UAAU,KAAK,oBAAoB,EAAE,CAAC;QACtC,KAAK,CAAC,IAAI,CAAC,KAAK,SAAG,CAAC,QAAQ,CAAC,oCAAoC,EAAE,wCAAwC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;IACxH,CAAC;SAAM,IAAI,UAAU,KAAK,iBAAiB,EAAE,CAAC;QAC1C,KAAK,CAAC,IAAI,CAAC,KAAK,SAAG,CAAC,QAAQ,CAAC,qCAAqC,EAAE,6FAA6F,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;IAC9K,CAAC;IACD,KAAK,CAAC,IAAI,CAAC,GAAG,SAAG,CAAC,iBAAiB,CAAC,YAAY,EAC5C,gBAAgB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,MAAM,SAAG,CAAC,iBAAiB,CAAC,aAAa,EACzE,gBAAgB,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC,CAAC;IACjD,MAAM,UAAU,GAAa,EAAE,CAAC;IAChC,IAAI,KAAK,CAAC,oBAAoB,EAAE,CAAC;QAC7B,UAAU,CAAC,IAAI,CAAC,SAAG,CAAC,QAAQ,CAAC,6CAA6C,EAAE,iBAAiB,EAAE,gBAAgB,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC;IAClJ,CAAC;IACD,IAAI,KAAK,CAAC,wBAAwB,EAAE,CAAC;QACjC,UAAU,CAAC,IAAI,CAAC,SAAG,CAAC,QAAQ,CAAC,+CAA+C,EAAE,qBAAqB,EAAE,gBAAgB,CAAC,KAAK,CAAC,wBAAwB,CAAC,CAAC,CAAC,CAAC;IAC5J,CAAC;IACD,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxB,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;IACvC,CAAC;IACD,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,WAAW,GAAG,gCAAwB,CAAC,GAAG,GAAG,CAAC,CAAC;IAC9E,KAAK,CAAC,IAAI,CAAC,SAAG,CAAC,QAAQ,CACnB,yCAAyC,EACzC,yBAAyB,EACzB,gBAAgB,CAAC,WAAW,CAAC,EAC7B,gBAAgB,CAAC,gCAAwB,CAAC,EAC1C,UAAU,CACb,CAAC,CAAC;IACH,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;AACzC,CAAC"}

package/lib/browser/chat-token-usage-indicator-util.spec.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=chat-token-usage-indicator-util.spec.d.ts.map

package/lib/browser/chat-token-usage-indicator-util.spec.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-token-usage-indicator-util.spec.d.ts","sourceRoot":"","sources":["../../src/browser/chat-token-usage-indicator-util.spec.ts"],"names":[],"mappings":""}

package/lib/browser/chat-token-usage-indicator-util.spec.js

@@ -0,0 +1,226 @@
+"use strict";
+// *****************************************************************************
+// Copyright (C) 2026 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+Object.defineProperty(exports, "__esModule", { value: true });
+const chai_1 = require("chai");
+const core_1 = require("@theia/core");
+const chat_token_usage_indicator_util_1 = require("./chat-token-usage-indicator-util");
+const THRESHOLD = 100;
+const CONTEXT_WINDOW = 200;
+function createMockRequest(tokenUsage, isComplete = true) {
+    return {
+        response: {
+            tokenUsage,
+            isComplete
+        }
+    };
+}
+function createMockChatModel(requests) {
+    const emitter = new core_1.Emitter();
+    return {
+        onDidChange: emitter.event,
+        id: 'test-model',
+        getRequests: () => requests,
+    };
+}
+describe('ChatTokenUsageIndicator', () => {
+    describe('formatTokenCount', () => {
+        it('should return "-" for undefined', () => {
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.formatTokenCount)(undefined)).to.equal('-');
+        });
+        it('should return "-" for 0', () => {
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.formatTokenCount)(0)).to.equal('-');
+        });
+        it('should format numbers below 1000 as-is', () => {
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.formatTokenCount)(500)).to.equal('500');
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.formatTokenCount)(999)).to.equal('999');
+        });
+        it('should format numbers >= 1000 with k suffix and one decimal', () => {
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.formatTokenCount)(1000)).to.equal('1.0k');
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.formatTokenCount)(1500)).to.equal('1.5k');
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.formatTokenCount)(125000)).to.equal('125.0k');
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.formatTokenCount)(200000)).to.equal('200.0k');
+        });
+    });
+    describe('getUsageColorClass', () => {
+        it('returns none for 0 tokens regardless of threshold', () => {
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.getUsageColorClass)(0, THRESHOLD, CONTEXT_WINDOW)).to.equal('token-usage-none');
+        });
+        it('returns green for tokens below the threshold', () => {
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.getUsageColorClass)(1, THRESHOLD, CONTEXT_WINDOW)).to.equal('token-usage-green');
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.getUsageColorClass)(THRESHOLD - 1, THRESHOLD, CONTEXT_WINDOW)).to.equal('token-usage-green');
+        });
+        it('returns yellow for tokens at or above the threshold but below the context window size', () => {
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.getUsageColorClass)(THRESHOLD, THRESHOLD, CONTEXT_WINDOW)).to.equal('token-usage-yellow');
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.getUsageColorClass)(CONTEXT_WINDOW - 1, THRESHOLD, CONTEXT_WINDOW)).to.equal('token-usage-yellow');
+        });
+        it('returns red for tokens at or above the context window size', () => {
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.getUsageColorClass)(CONTEXT_WINDOW, THRESHOLD, CONTEXT_WINDOW)).to.equal('token-usage-red');
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.getUsageColorClass)(CONTEXT_WINDOW + 1, THRESHOLD, CONTEXT_WINDOW)).to.equal('token-usage-red');
+        });
+    });
+    describe('computeSessionTokenUsage', () => {
+        it('should return 0 when chatModel is undefined', () => {
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.computeSessionTokenUsage)(undefined)).to.equal(0);
+        });
+        it('should return 0 when there are no requests', () => {
+            const model = createMockChatModel([]);
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.computeSessionTokenUsage)(model)).to.equal(0);
+        });
+        it('should return 0 when requests have no token usage', () => {
+            const model = createMockChatModel([
+                createMockRequest(undefined),
+                createMockRequest(undefined)
+            ]);
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.computeSessionTokenUsage)(model)).to.equal(0);
+        });
+        it('should sum all tokens from a single request', () => {
+            const model = createMockChatModel([
+                createMockRequest({ inputTokens: 1000, outputTokens: 500 })
+            ]);
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.computeSessionTokenUsage)(model)).to.equal(1500);
+        });
+        it('should return only the last request tokens, not the sum', () => {
+            const model = createMockChatModel([
+                createMockRequest({ inputTokens: 1000, outputTokens: 500 }),
+                createMockRequest({ inputTokens: 2000, outputTokens: 800 })
+            ]);
+            // Only the last request: 2000+800 = 2800
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.computeSessionTokenUsage)(model)).to.equal(2800);
+        });
+        it('should return last request tokens from 3 requests (2 complete, 1 in-progress)', () => {
+            const model = createMockChatModel([
+                createMockRequest({ inputTokens: 1000, outputTokens: 500 }, true),
+                createMockRequest({ inputTokens: 2000, outputTokens: 800 }, true),
+                createMockRequest({ inputTokens: 3000, outputTokens: 200 }, false) // in-progress
+            ]);
+            // Only the last request: 3000+200 = 3200
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.computeSessionTokenUsage)(model)).to.equal(3200);
+        });
+        it('should skip requests without token usage and return last with usage', () => {
+            const model = createMockChatModel([
+                createMockRequest({ inputTokens: 1000, outputTokens: 500 }),
+                createMockRequest(undefined),
+                createMockRequest({ inputTokens: 3000, outputTokens: 1000 })
+            ]);
+            // Last request with usage: 3000+1000 = 4000
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.computeSessionTokenUsage)(model)).to.equal(4000);
+        });
+        it('should include cache creation and cache read tokens in the total', () => {
+            const model = createMockChatModel([
+                createMockRequest({
+                    inputTokens: 1000,
+                    outputTokens: 500,
+                    cacheCreationInputTokens: 200,
+                    cacheReadInputTokens: 300
+                })
+            ]);
+            // 1000+500+200+300 = 2000
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.computeSessionTokenUsage)(model)).to.equal(2000);
+        });
+        it('should return only last request cache tokens, not sum across requests', () => {
+            const model = createMockChatModel([
+                createMockRequest({
+                    inputTokens: 1000,
+                    outputTokens: 500,
+                    cacheCreationInputTokens: 200,
+                    cacheReadInputTokens: 300
+                }),
+                createMockRequest({
+                    inputTokens: 500,
+                    outputTokens: 100,
+                    cacheCreationInputTokens: 50,
+                    cacheReadInputTokens: 50
+                })
+            ]);
+            // Only last request: 500+100+50+50 = 700
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.computeSessionTokenUsage)(model)).to.equal(700);
+        });
+        it('should return in-progress request token usage when it is last', () => {
+            const model = createMockChatModel([
+                createMockRequest({ inputTokens: 1000, outputTokens: 500 }, true),
+                createMockRequest({ inputTokens: 5000, outputTokens: 200 }, false) // in-progress
+            ]);
+            // Only last request: 5000+200 = 5200
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.computeSessionTokenUsage)(model)).to.equal(5200);
+        });
+        it('should return total from in-progress request when it is the only request', () => {
+            const model = createMockChatModel([
+                createMockRequest({ inputTokens: 5000, outputTokens: 200 }, false)
+            ]);
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.computeSessionTokenUsage)(model)).to.equal(5200);
+        });
+        it('should skip trailing requests without usage and find last with usage', () => {
+            const model = createMockChatModel([
+                createMockRequest({ inputTokens: 1000, outputTokens: 500 }),
+                createMockRequest({ inputTokens: 3000, outputTokens: 1000 }),
+                createMockRequest(undefined)
+            ]);
+            // Last request with usage: 3000+1000 = 4000
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.computeSessionTokenUsage)(model)).to.equal(4000);
+        });
+    });
+    describe('isAboveTokenUsageWarningThreshold', () => {
+        it('is false for 0 tokens', () => {
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.isAboveTokenUsageWarningThreshold)(0, THRESHOLD)).to.equal(false);
+        });
+        it('is false for tokens below the threshold', () => {
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.isAboveTokenUsageWarningThreshold)(THRESHOLD - 1, THRESHOLD)).to.equal(false);
+        });
+        it('is true for tokens at the threshold', () => {
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.isAboveTokenUsageWarningThreshold)(THRESHOLD, THRESHOLD)).to.equal(true);
+        });
+        it('is true for tokens above the threshold', () => {
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.isAboveTokenUsageWarningThreshold)(THRESHOLD + 1, THRESHOLD)).to.equal(true);
+        });
+    });
+    describe('decideTokenUsageWarning', () => {
+        it('resets when usage is below the threshold', () => {
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.decideTokenUsageWarning)({
+                totalTokens: 50,
+                threshold: THRESHOLD,
+                alreadyNotified: true
+            })).to.equal('reset');
+        });
+        it('resets when usage is zero', () => {
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.decideTokenUsageWarning)({
+                totalTokens: 0,
+                threshold: THRESHOLD,
+                alreadyNotified: true
+            })).to.equal('reset');
+        });
+        it('skips when the session has already been notified', () => {
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.decideTokenUsageWarning)({
+                totalTokens: THRESHOLD + 50,
+                threshold: THRESHOLD,
+                alreadyNotified: true
+            })).to.equal('skip');
+        });
+        it('notifies when usage crosses the threshold and the session has not been notified yet', () => {
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.decideTokenUsageWarning)({
+                totalTokens: THRESHOLD,
+                threshold: THRESHOLD,
+                alreadyNotified: false
+            })).to.equal('notify');
+            (0, chai_1.expect)((0, chat_token_usage_indicator_util_1.decideTokenUsageWarning)({
+                totalTokens: THRESHOLD * 2,
+                threshold: THRESHOLD,
+                alreadyNotified: false
+            })).to.equal('notify');
+        });
+    });
+});
+//# sourceMappingURL=chat-token-usage-indicator-util.spec.js.map

package/lib/browser/chat-token-usage-indicator-util.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-token-usage-indicator-util.spec.js","sourceRoot":"","sources":["../../src/browser/chat-token-usage-indicator-util.spec.ts"],"names":[],"mappings":";AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;;AAEhF,+BAA8B;AAC9B,sCAA6C;AAE7C,uFAM2C;AAE3C,MAAM,SAAS,GAAG,GAAG,CAAC;AACtB,MAAM,cAAc,GAAG,GAAG,CAAC;AAE3B,SAAS,iBAAiB,CAAC,UAA+B,EAAE,UAAU,GAAG,IAAI;IACzE,OAAO;QACH,QAAQ,EAAE;YACN,UAAU;YACV,UAAU;SACmB;KACpC,CAAC;AACN,CAAC;AAED,SAAS,mBAAmB,CAAC,QAAqC;IAC9D,MAAM,OAAO,GAAG,IAAI,cAAO,EAAW,CAAC;IACvC,OAAO;QACH,WAAW,EAAE,OAAO,CAAC,KAAuB;QAC5C,EAAE,EAAE,YAAY;QAChB,WAAW,EAAE,GAAG,EAAE,CAAC,QAA8B;KACvC,CAAC;AACnB,CAAC;AAED,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACrC,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACvC,IAAA,aAAM,EAAC,IAAA,kDAAgB,EAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;YAC/B,IAAA,aAAM,EAAC,IAAA,kDAAgB,EAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAC9C,IAAA,aAAM,EAAC,IAAA,kDAAgB,EAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAC9C,IAAA,aAAM,EAAC,IAAA,kDAAgB,EAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;YACnE,IAAA,aAAM,EAAC,IAAA,kDAAgB,EAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAChD,IAAA,aAAM,EAAC,IAAA,kDAAgB,EAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAChD,IAAA,aAAM,EAAC,IAAA,kDAAgB,EAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YACpD,IAAA,aAAM,EAAC,IAAA,kDAAgB,EAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YACzD,IAAA,aAAM,EAAC,IAAA,oDAAkB,EAAC,CAAC,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;QAC1F,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACpD,IAAA,aAAM,EAAC,IAAA,oDAAkB,EAAC,CAAC,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;YACvF,IAAA,aAAM,EAAC,IAAA,oDAAkB,EAAC,SAAS,GAAG,CAAC,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;QACvG,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uFAAuF,EAAE,GAAG,EAAE;YAC7F,IAAA,aAAM,EAAC,IAAA,oDAAkB,EAAC,SAAS,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC;YAChG,IAAA,aAAM,EAAC,IAAA,oDAAkB,EAAC,cAAc,GAAG,CAAC,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC;QAC7G,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;YAClE,IAAA,aAAM,EAAC,IAAA,oDAAkB,EAAC,cAAc,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;YAClG,IAAA,aAAM,EAAC,IAAA,oDAAkB,EAAC,cAAc,GAAG,CAAC,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;QAC1G,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACtC,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACnD,IAAA,aAAM,EAAC,IAAA,0DAAwB,EAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YAClD,MAAM,KAAK,GAAG,mBAAmB,CAAC,EAAE,CAAC,CAAC;YACtC,IAAA,aAAM,EAAC,IAAA,0DAAwB,EAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YACzD,MAAM,KAAK,GAAG,mBAAmB,CAAC;gBAC9B,iBAAiB,CAAC,SAAS,CAAC;gBAC5B,iBAAiB,CAAC,SAAS,CAAC;aAC/B,CAAC,CAAC;YACH,IAAA,aAAM,EAAC,IAAA,0DAAwB,EAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACnD,MAAM,KAAK,GAAG,mBAAmB,CAAC;gBAC9B,iBAAiB,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,GAAG,EAAE,CAAC;aAC9D,CAAC,CAAC;YACH,IAAA,aAAM,EAAC,IAAA,0DAAwB,EAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YAC/D,MAAM,KAAK,GAAG,mBAAmB,CAAC;gBAC9B,iBAAiB,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,GAAG,EAAE,CAAC;gBAC3D,iBAAiB,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,GAAG,EAAE,CAAC;aAC9D,CAAC,CAAC;YACH,yCAAyC;YACzC,IAAA,aAAM,EAAC,IAAA,0DAAwB,EAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+EAA+E,EAAE,GAAG,EAAE;YACrF,MAAM,KAAK,GAAG,mBAAmB,CAAC;gBAC9B,iBAAiB,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,GAAG,EAAE,EAAE,IAAI,CAAC;gBACjE,iBAAiB,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,GAAG,EAAE,EAAE,IAAI,CAAC;gBACjE,iBAAiB,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,GAAG,EAAE,EAAE,KAAK,CAAC,CAAC,cAAc;aACpF,CAAC,CAAC;YACH,yCAAyC;YACzC,IAAA,aAAM,EAAC,IAAA,0DAAwB,EAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;YAC3E,MAAM,KAAK,GAAG,mBAAmB,CAAC;gBAC9B,iBAAiB,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,GAAG,EAAE,CAAC;gBAC3D,iBAAiB,CAAC,SAAS,CAAC;gBAC5B,iBAAiB,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;aAC/D,CAAC,CAAC;YACH,4CAA4C;YAC5C,IAAA,aAAM,EAAC,IAAA,0DAAwB,EAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;YACxE,MAAM,KAAK,GAAG,mBAAmB,CAAC;gBAC9B,iBAAiB,CAAC;oBACd,WAAW,EAAE,IAAI;oBACjB,YAAY,EAAE,GAAG;oBACjB,wBAAwB,EAAE,GAAG;oBAC7B,oBAAoB,EAAE,GAAG;iBAC5B,CAAC;aACL,CAAC,CAAC;YACH,0BAA0B;YAC1B,IAAA,aAAM,EAAC,IAAA,0DAAwB,EAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;YAC7E,MAAM,KAAK,GAAG,mBAAmB,CAAC;gBAC9B,iBAAiB,CAAC;oBACd,WAAW,EAAE,IAAI;oBACjB,YAAY,EAAE,GAAG;oBACjB,wBAAwB,EAAE,GAAG;oBAC7B,oBAAoB,EAAE,GAAG;iBAC5B,CAAC;gBACF,iBAAiB,CAAC;oBACd,WAAW,EAAE,GAAG;oBAChB,YAAY,EAAE,GAAG;oBACjB,wBAAwB,EAAE,EAAE;oBAC5B,oBAAoB,EAAE,EAAE;iBAC3B,CAAC;aACL,CAAC,CAAC;YACH,yCAAyC;YACzC,IAAA,aAAM,EAAC,IAAA,0DAAwB,EAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACrE,MAAM,KAAK,GAAG,mBAAmB,CAAC;gBAC9B,iBAAiB,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,GAAG,EAAE,EAAE,IAAI,CAAC;gBACjE,iBAAiB,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,GAAG,EAAE,EAAE,KAAK,CAAC,CAAC,cAAc;aACpF,CAAC,CAAC;YACH,qCAAqC;YACrC,IAAA,aAAM,EAAC,IAAA,0DAAwB,EAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;YAChF,MAAM,KAAK,GAAG,mBAAmB,CAAC;gBAC9B,iBAAiB,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,GAAG,EAAE,EAAE,KAAK,CAAC;aACrE,CAAC,CAAC;YACH,IAAA,aAAM,EAAC,IAAA,0DAAwB,EAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;YAC5E,MAAM,KAAK,GAAG,mBAAmB,CAAC;gBAC9B,iBAAiB,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,GAAG,EAAE,CAAC;gBAC3D,iBAAiB,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;gBAC5D,iBAAiB,CAAC,SAAS,CAAC;aAC/B,CAAC,CAAC;YACH,4CAA4C;YAC5C,IAAA,aAAM,EAAC,IAAA,0DAAwB,EAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC/C,EAAE,CAAC,uBAAuB,EAAE,GAAG,EAAE;YAC7B,IAAA,aAAM,EAAC,IAAA,mEAAiC,EAAC,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC5E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YAC/C,IAAA,aAAM,EAAC,IAAA,mEAAiC,EAAC,SAAS,GAAG,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACxF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC3C,IAAA,aAAM,EAAC,IAAA,mEAAiC,EAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACnF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAC9C,IAAA,aAAM,EAAC,IAAA,mEAAiC,EAAC,SAAS,GAAG,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACvF,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAChD,IAAA,aAAM,EAAC,IAAA,yDAAuB,EAAC;gBAC3B,WAAW,EAAE,EAAE;gBACf,SAAS,EAAE,SAAS;gBACpB,eAAe,EAAE,IAAI;aACxB,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;YACjC,IAAA,aAAM,EAAC,IAAA,yDAAuB,EAAC;gBAC3B,WAAW,EAAE,CAAC;gBACd,SAAS,EAAE,SAAS;gBACpB,eAAe,EAAE,IAAI;aACxB,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YACxD,IAAA,aAAM,EAAC,IAAA,yDAAuB,EAAC;gBAC3B,WAAW,EAAE,SAAS,GAAG,EAAE;gBAC3B,SAAS,EAAE,SAAS;gBACpB,eAAe,EAAE,IAAI;aACxB,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qFAAqF,EAAE,GAAG,EAAE;YAC3F,IAAA,aAAM,EAAC,IAAA,yDAAuB,EAAC;gBAC3B,WAAW,EAAE,SAAS;gBACtB,SAAS,EAAE,SAAS;gBACpB,eAAe,EAAE,KAAK;aACzB,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YACvB,IAAA,aAAM,EAAC,IAAA,yDAAuB,EAAC;gBAC3B,WAAW,EAAE,SAAS,GAAG,CAAC;gBAC1B,SAAS,EAAE,SAAS;gBACpB,eAAe,EAAE,KAAK;aACzB,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAC3B,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC"}

package/lib/browser/chat-tree-view/chat-view-tree-input-widget.d.ts

@@ -1,7 +1,7 @@
 import { AIChatInputWidget, type AIChatInputConfiguration } from '../chat-input-widget';
 import type { EditableRequestNode } from './chat-view-tree-widget';
 import { URI } from '@theia/core';
-import type { ChatRequestModel, EditableChatRequestModel, ChatHierarchyBranch } from '@theia/ai-chat';
+import type { ChatModel, ChatRequestModel, EditableChatRequestModel, ChatHierarchyBranch } from '@theia/ai-chat';
 import type { AIVariableResolutionRequest } from '@theia/ai-core';
 export declare const AIChatTreeInputConfiguration: unique symbol;
 export interface AIChatTreeInputConfiguration extends AIChatInputConfiguration {
@@ -42,5 +42,10 @@ export declare class AIChatTreeInputWidget extends AIChatInputWidget {
     getAllVariablesForRequest(): AIVariableResolutionRequest[];
     clearPendingImageAttachments(): void;
     protected deleteContextElement(index: number): void;
+    /**
+     * No-op in edit mode: the token usage warning is owned by the main chat input
+     * widget so we avoid double-notifying when an edit widget is open alongside it.
+     */
+    protected evaluateTokenUsageWarning(_chatModel: ChatModel): void;
 }
 //# sourceMappingURL=chat-view-tree-input-widget.d.ts.map

package/lib/browser/chat-tree-view/chat-view-tree-input-widget.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chat-view-tree-input-widget.d.ts","sourceRoot":"","sources":["../../../src/browser/chat-tree-view/chat-view-tree-input-widget.tsx"],"names":[],"mappings":"AAiBA,OAAO,EAAE,iBAAiB,EAAE,KAAK,wBAAwB,EAAE,MAAM,sBAAsB,CAAC;AACxF,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,GAAG,EAAE,MAAM,aAAa,CAAC;AAElC,OAAO,KAAK,EAAE,gBAAgB,EAAE,wBAAwB,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AACtG,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,gBAAgB,CAAC;AAGlE,eAAO,MAAM,4BAA4B,eAAyC,CAAC;AACnF,MAAM,WAAW,4BAA6B,SAAQ,wBAAwB;CAAI;AAElF,eAAO,MAAM,mBAAmB,eAAgC,CAAC;AACjE,MAAM,WAAW,mBAAmB;IAChC,IAAI,EAAE,mBAAmB,CAAC;IAC1B;;OAEG;IACH,MAAM,CAAC,EAAE,mBAAmB,CAAC;IAC7B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,OAAO,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1C,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC;IACrB,QAAQ,CAAC,EAAE,CAAC,YAAY,EAAE,gBAAgB,KAAK,IAAI,CAAC;IACpD,iBAAiB,CAAC,EAAE,CAAC,YAAY,EAAE,gBAAgB,KAAK,IAAI,CAAC;IAC7D,wBAAwB,CAAC,EAAE,CAAC,YAAY,EAAE,gBAAgB,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;CACtF;AACD,eAAO,MAAM,sBAAsB,eAAmC,CAAC;AACvE,MAAM,MAAM,sBAAsB,GAAG,CAAC,IAAI,EAAE,mBAAmB,KAAK,qBAAqB,CAAC;AAE1F,qBACa,qBAAsB,SAAQ,iBAAiB;IACxD,OAAuB,EAAE,SAA4B;IAGrD,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,mBAAmB,CAAC;IAE7C,mBAC4B,aAAa,EAAE,4BAA4B,GAAG,SAAS,CAAC;IAEpF,IAAI,WAAW,IAAI,mBAAmB,CAErC;IAED,IAAI,OAAO,IAAI,wBAAwB,CAEtC;cAGkB,IAAI,IAAI,IAAI;IAoB/B,SAAS,CAAC,YAAY,IAAI,IAAI;cAIX,cAAc,IAAI,GAAG;IAI/B,UAAU,CAAC,QAAQ,EAAE,2BAA2B,GAAG,IAAI;IAIhE;;;;OAIG;IACM,oBAAoB,CAAC,QAAQ,EAAE,2BAA2B,GAAG,MAAM;cAMzD,UAAU,IAAI,SAAS,2BAA2B,EAAE;IAI9D,yBAAyB,IAAI,2BAA2B,EAAE;IAI1D,4BAA4B,IAAI,IAAI;cAI1B,oBAAoB,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;CAG/D"}
+{"version":3,"file":"chat-view-tree-input-widget.d.ts","sourceRoot":"","sources":["../../../src/browser/chat-tree-view/chat-view-tree-input-widget.tsx"],"names":[],"mappings":"AAiBA,OAAO,EAAE,iBAAiB,EAAE,KAAK,wBAAwB,EAAE,MAAM,sBAAsB,CAAC;AACxF,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,GAAG,EAAE,MAAM,aAAa,CAAC;AAElC,OAAO,KAAK,EAAE,SAAS,EAAE,gBAAgB,EAAE,wBAAwB,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AACjH,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,gBAAgB,CAAC;AAGlE,eAAO,MAAM,4BAA4B,eAAyC,CAAC;AACnF,MAAM,WAAW,4BAA6B,SAAQ,wBAAwB;CAAI;AAElF,eAAO,MAAM,mBAAmB,eAAgC,CAAC;AACjE,MAAM,WAAW,mBAAmB;IAChC,IAAI,EAAE,mBAAmB,CAAC;IAC1B;;OAEG;IACH,MAAM,CAAC,EAAE,mBAAmB,CAAC;IAC7B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,OAAO,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1C,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC;IACrB,QAAQ,CAAC,EAAE,CAAC,YAAY,EAAE,gBAAgB,KAAK,IAAI,CAAC;IACpD,iBAAiB,CAAC,EAAE,CAAC,YAAY,EAAE,gBAAgB,KAAK,IAAI,CAAC;IAC7D,wBAAwB,CAAC,EAAE,CAAC,YAAY,EAAE,gBAAgB,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;CACtF;AACD,eAAO,MAAM,sBAAsB,eAAmC,CAAC;AACvE,MAAM,MAAM,sBAAsB,GAAG,CAAC,IAAI,EAAE,mBAAmB,KAAK,qBAAqB,CAAC;AAE1F,qBACa,qBAAsB,SAAQ,iBAAiB;IACxD,OAAuB,EAAE,SAA4B;IAGrD,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,mBAAmB,CAAC;IAE7C,mBAC4B,aAAa,EAAE,4BAA4B,GAAG,SAAS,CAAC;IAEpF,IAAI,WAAW,IAAI,mBAAmB,CAErC;IAED,IAAI,OAAO,IAAI,wBAAwB,CAEtC;cAGkB,IAAI,IAAI,IAAI;IAoB/B,SAAS,CAAC,YAAY,IAAI,IAAI;cAIX,cAAc,IAAI,GAAG;IAI/B,UAAU,CAAC,QAAQ,EAAE,2BAA2B,GAAG,IAAI;IAIhE;;;;OAIG;IACM,oBAAoB,CAAC,QAAQ,EAAE,2BAA2B,GAAG,MAAM;cAMzD,UAAU,IAAI,SAAS,2BAA2B,EAAE;IAI9D,yBAAyB,IAAI,2BAA2B,EAAE;IAI1D,4BAA4B,IAAI,IAAI;cAI1B,oBAAoB,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAI5D;;;OAGG;cACgB,yBAAyB,CAAC,UAAU,EAAE,SAAS,GAAG,IAAI;CAG5E"}

package/lib/browser/chat-tree-view/chat-view-tree-input-widget.js

@@ -80,6 +80,13 @@ let AIChatTreeInputWidget = class AIChatTreeInputWidget extends chat_input_widge
     deleteContextElement(index) {
         this.request.editContextManager.deleteVariables(index);
     }
+    /**
+     * No-op in edit mode: the token usage warning is owned by the main chat input
+     * widget so we avoid double-notifying when an edit widget is open alongside it.
+     */
+    evaluateTokenUsageWarning(_chatModel) {
+        // Intentionally empty.
+    }
 };
 exports.AIChatTreeInputWidget = AIChatTreeInputWidget;
 tslib_1.__decorate([

package/lib/browser/chat-tree-view/chat-view-tree-input-widget.js.map

@@ -1 +1 @@
-{"version":3,"file":"chat-view-tree-input-widget.js","sourceRoot":"","sources":["../../../src/browser/chat-tree-view/chat-view-tree-input-widget.tsx"],"names":[],"mappings":";AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;;;;AAEhF,4DAA2F;AAC3F,4DAAwF;AAExF,sCAAkC;AAClC,wFAAkF;AAGlF,qDAA8C;AAEjC,QAAA,4BAA4B,GAAG,MAAM,CAAC,8BAA8B,CAAC,CAAC;AAGtE,QAAA,mBAAmB,GAAG,MAAM,CAAC,qBAAqB,CAAC,CAAC;AAcpD,QAAA,sBAAsB,GAAG,MAAM,CAAC,wBAAwB,CAAC,CAAC;AAIhE,IAAM,qBAAqB,GAA3B,MAAM,qBAAsB,SAAQ,qCAAiB;aACjC,OAAE,GAAG,wBAAwB,AAA3B,CAA4B;IAQrD,IAAI,WAAW;QACX,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;IAC1B,CAAC;IAED,IAAI,OAAO;QACP,OAAO,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC;IACpC,CAAC;IAGkB,IAAI;QACnB,KAAK,CAAC,IAAI,EAAE,CAAC;QACb,IAAI,CAAC,YAAY,EAAE,CAAC;QAEpB,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC;QACzC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,GAAG,EAAE;YACjD,IAAI,CAAC,YAAY,EAAE,CAAC;QACxB,CAAC,CAAC,CAAC,CAAC;QAEJ,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,EAAE,aAAG,CAAC,MAAM,EAAE,GAAG,EAAE;YAC5C,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE;YAC/B,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACjB,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;YAC3B,CAAC;QACL,CAAC,CAAC,CAAC;IACP,CAAC;IAES,YAAY;QAClB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;IAC9D,CAAC;IAEkB,cAAc;QAC7B,OAAO,IAAI,UAAG,CAAC,YAAY,IAAI,CAAC,WAAW,CAAC,EAAE,UAAU,8DAA4B,EAAE,CAAC,CAAC;IAC5F,CAAC;IAEQ,UAAU,CAAC,QAAqC;QACrD,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;IAC3D,CAAC;IAED;;;;OAIG;IACM,oBAAoB,CAAC,QAAqC;QAC/D,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QACvD,uFAAuF;QACvF,OAAO,YAAY,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;IACpC,CAAC;IAEkB,UAAU;QACzB,OAAO,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,YAAY,EAAE,CAAC;IAC1D,CAAC;IAEQ,yBAAyB;QAC9B,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,YAAY,EAAE,CAAC,CAAC;IAC/D,CAAC;IAEQ,4BAA4B;QACjC,oEAAoE;IACxE,CAAC;IAEkB,oBAAoB,CAAC,KAAa;QACjD,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IAC3D,CAAC;;AA3EQ,sDAAqB;AAIX;IADlB,IAAA,kBAAM,EAAC,2BAAmB,CAAC;;mDACiB;AAGjB;IAD3B,IAAA,kBAAM,EAAC,oCAA4B,CAAC;IAAE,IAAA,oBAAQ,GAAE;;4DACmC;AAWjE;IADlB,IAAA,yBAAa,GAAE;;;;iDAmBf;gCApCQ,qBAAqB;IADjC,IAAA,sBAAU,GAAE;GACA,qBAAqB,CA4EjC"}
+{"version":3,"file":"chat-view-tree-input-widget.js","sourceRoot":"","sources":["../../../src/browser/chat-tree-view/chat-view-tree-input-widget.tsx"],"names":[],"mappings":";AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;;;;AAEhF,4DAA2F;AAC3F,4DAAwF;AAExF,sCAAkC;AAClC,wFAAkF;AAGlF,qDAA8C;AAEjC,QAAA,4BAA4B,GAAG,MAAM,CAAC,8BAA8B,CAAC,CAAC;AAGtE,QAAA,mBAAmB,GAAG,MAAM,CAAC,qBAAqB,CAAC,CAAC;AAcpD,QAAA,sBAAsB,GAAG,MAAM,CAAC,wBAAwB,CAAC,CAAC;AAIhE,IAAM,qBAAqB,GAA3B,MAAM,qBAAsB,SAAQ,qCAAiB;aACjC,OAAE,GAAG,wBAAwB,AAA3B,CAA4B;IAQrD,IAAI,WAAW;QACX,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;IAC1B,CAAC;IAED,IAAI,OAAO;QACP,OAAO,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC;IACpC,CAAC;IAGkB,IAAI;QACnB,KAAK,CAAC,IAAI,EAAE,CAAC;QACb,IAAI,CAAC,YAAY,EAAE,CAAC;QAEpB,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC;QACzC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,GAAG,EAAE;YACjD,IAAI,CAAC,YAAY,EAAE,CAAC;QACxB,CAAC,CAAC,CAAC,CAAC;QAEJ,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,EAAE,aAAG,CAAC,MAAM,EAAE,GAAG,EAAE;YAC5C,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE;YAC/B,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACjB,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;YAC3B,CAAC;QACL,CAAC,CAAC,CAAC;IACP,CAAC;IAES,YAAY;QAClB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;IAC9D,CAAC;IAEkB,cAAc;QAC7B,OAAO,IAAI,UAAG,CAAC,YAAY,IAAI,CAAC,WAAW,CAAC,EAAE,UAAU,8DAA4B,EAAE,CAAC,CAAC;IAC5F,CAAC;IAEQ,UAAU,CAAC,QAAqC;QACrD,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;IAC3D,CAAC;IAED;;;;OAIG;IACM,oBAAoB,CAAC,QAAqC;QAC/D,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QACvD,uFAAuF;QACvF,OAAO,YAAY,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;IACpC,CAAC;IAEkB,UAAU;QACzB,OAAO,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,YAAY,EAAE,CAAC;IAC1D,CAAC;IAEQ,yBAAyB;QAC9B,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,YAAY,EAAE,CAAC,CAAC;IAC/D,CAAC;IAEQ,4BAA4B;QACjC,oEAAoE;IACxE,CAAC;IAEkB,oBAAoB,CAAC,KAAa;QACjD,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IAC3D,CAAC;IAED;;;OAGG;IACgB,yBAAyB,CAAC,UAAqB;QAC9D,uBAAuB;IAC3B,CAAC;;AAnFQ,sDAAqB;AAIX;IADlB,IAAA,kBAAM,EAAC,2BAAmB,CAAC;;mDACiB;AAGjB;IAD3B,IAAA,kBAAM,EAAC,oCAA4B,CAAC;IAAE,IAAA,oBAAQ,GAAE;;4DACmC;AAWjE;IADlB,IAAA,yBAAa,GAAE;;;;iDAmBf;gCApCQ,qBAAqB;IADjC,IAAA,sBAAU,GAAE;GACA,qBAAqB,CAoFjC"}

package/lib/browser/chat-tree-view/chat-view-tree-widget.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chat-view-tree-widget.d.ts","sourceRoot":"","sources":["../../../src/browser/chat-tree-view/chat-view-tree-widget.tsx"],"names":[],"mappings":"AAeA,OAAO,EACH,SAAS,EACT,gBAAgB,EAChB,SAAS,EACT,gBAAgB,EAChB,mBAAmB,EACnB,iBAAiB,EACjB,WAAW,EACX,wBAAwB,EAIxB,KAAK,WAAW,EAChB,KAAK,mBAAmB,EAC3B,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AACjE,OAAO,EAAE,eAAe,EAAE,oBAAoB,EAAc,oBAAoB,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACtH,OAAO,EAGH,mBAAmB,EACnB,YAAY,EAGZ,SAAS,EACT,aAAa,EACb,SAAS,EACT,QAAQ,EACR,SAAS,EACT,UAAU,EAGb,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,UAAU,EAAE,iBAAiB,EAAE,MAAM,6CAA6C,CAAC;AAQ5F,OAAO,KAAK,KAAK,MAAM,0BAA0B,CAAC;AAElD,OAAO,EAAE,iCAAiC,EAAE,MAAM,0CAA0C,CAAC;AAC7F,OAAO,EAAE,wBAAwB,EAAE,MAAM,gCAAgC,CAAC;AAG1E,OAAO,EAAE,sBAAsB,EAAE,KAAK,qBAAqB,EAAE,MAAM,+BAA+B,CAAC;AAInG,MAAM,WAAW,WAAY,SAAQ,QAAQ;IACzC,OAAO,EAAE,gBAAgB,CAAC;IAC1B,MAAM,EAAE,mBAAmB,CAAC;IAC5B,SAAS,EAAE,MAAM,CAAA;CACpB;AACD,eAAO,MAAM,aAAa,GAAI,MAAM,QAAQ,KAAG,IAAI,IAAI,WAAgC,CAAC;AAExF,MAAM,WAAW,mBAAoB,SAAQ,WAAW;IACpD,OAAO,EAAE,wBAAwB,CAAA;CACpC;AACD,eAAO,MAAM,qBAAqB,GAAI,MAAM,QAAQ,KAAG,IAAI,IAAI,mBAAuF,CAAC;AAGvJ,MAAM,WAAW,YAAa,SAAQ,QAAQ;IAC1C,QAAQ,EAAE,iBAAiB,CAAC;IAC5B,SAAS,EAAE,MAAM,CAAA;CACpB;AACD,eAAO,MAAM,cAAc,GAAI,MAAM,QAAQ,KAAG,IAAI,IAAI,YAAkC,CAAC;AAE3F,wBAAgB,UAAU,CAAC,CAAC,EAAE,KAAK,CAAC,aAAa,GAAG,OAAO,CAE1D;AAED,eAAO,MAAM,0BAA0B,eAAuC,CAAC;AAC/E,MAAM,WAAW,0BAA0B;IACvC,oBAAoB,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC;IACzC,qBAAqB,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC;IAC1C,QAAQ,CAAC,cAAc,CAAC,EAAE,OAAO,CAAC;IAClC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,OAAO,CAAC;IAC5C,QAAQ,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,cAAc,CAAC,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;IACtC,oFAAoF;IACpF,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;CAC9B;AAED,qBACa,kBAAmB,SAAQ,UAAU;IAE9C,MAAM,CAAC,QAAQ,CAAC,EAAE,sBAAsB;IACxC,MAAM,CAAC,QAAQ,CAAC,YAAY,WAA8B;IAG1D,SAAS,CAAC,QAAQ,CAAC,yBAAyB,EAAE,oBAAoB,CAAC,wBAAwB,CAAC,mBAAmB,CAAC,CAAC,CAAC;IAGlH,SAAS,CAAC,QAAQ,CAAC,kCAAkC,EAAE,oBAAoB,CAAC,iCAAiC,CAAC,CAAC;IAG/G,SAAS,CAAC,gBAAgB,EAAE,gBAAgB,CAAC;IAG7C,SAAS,CAAC,QAAQ,CAAC,eAAe,EAAE,iBAAiB,CAAC;IAGtD,SAAS,CAAC,eAAe,EAAE,eAAe,CAAC;IAG3C,SAAS,CAAC,QAAQ,CAAC,aAAa,EAAE,aAAa,CAAC;IAGhD,SAAS,CAAC,YAAY,EAAE,YAAY,CAAC;IAGrC,SAAS,CAAC,QAAQ,CAAC,uBAAuB,EAAE,oBAAoB,CAAC,0BAA0B,CAAC,CAAC;IAG7F,SAAS,CAAC,kBAAkB,EAAE,sBAAsB,CAAC;IAGrD,SAAS,CAAC,QAAQ,CAAC,iBAAiB,EAAE,mBAAmB,CAAC;IAG1D,SAAS,CAAC,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;IAG5C,SAAS,CAAC,QAAQ,CAAC,iBAAiB,EAAE,iBAAiB,CAAC;IAExD,SAAS,CAAC,oBAAoB,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;IAEpD,SAAS,CAAC,QAAQ,CAAC,sBAAsB,uBAA8B;IACvE,eAAe,qBAAqC;IAEpD,SAAS,CAAC,QAAQ,CAAC,UAAU,EAAE,GAAG,CAAC,MAAM,EAAE,qBAAqB,CAAC,CAAa;IAE9E,SAAS,CAAC,kBAAkB,UAAQ;IAEpC,SAAS,CAAC,SAAS,UAAS;IAE5B,SAAS,CAAC,WAAW,EAAE,MAAM,CAAC;IAE9B,8EAA8E;IAC9E,SAAS,CAAC,QAAQ,UAAQ;IAC1B;;OAEG;IACH,SAAS,CAAC,iBAAiB,UAAS;IACpC;;;OAGG;IACH,SAAS,CAAC,0BAA0B,CAAC,EAAE,MAAM,CAAC;IAC9C;;;OAGG;IACH,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,0BAA0B,OAAO;IAE3D,kBAAkB,CAAC,EAAE,CAAC,eAAe,EAAE,OAAO,KAAK,IAAI,CAAC;IAExD,SAAS,CAAC,aAAa,SAAK;IAE5B,IAAI,iBAAiB,CAAC,iBAAiB,EAAE,OAAO,EAG/C;IAED,IAAI,iBAAiB,IAAI,OAAO,CAE/B;gBAGsB,KAAK,EAAE,SAAS,EAChB,KAAK,EAAE,SAAS,EACN,mBAAmB,EAAE,mBAAmB;cAiBtD,IAAI,IAAI,IAAI;IAuCxB,UAAU,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI;IAKzC,SAAS,CAAC,iBAAiB,CAAC,WAAW,EAAE,OAAO,GAAG,IAAI;IAqBvD,sEAAsE;IACtE,SAAS,CAAC,+BAA+B,CAAC,UAAU,EAAE,OAAO,GAAG,IAAI;IA6BpE,SAAS,CAAC,sBAAsB,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI;IAOxD,SAAS,CAAC,mBAAmB,CAAC,WAAW,EAAE,OAAO,GAAG,MAAM;IA0B3D;;;;OAIG;IACH,SAAS,CAAC,kBAAkB,IAAI,OAAO;cAqBpB,UAAU,CAAC,KAAK,EAAE,SAAS,GAAG,KAAK,CAAC,SAAS;IAehE,0EAA0E;IAC1E,SAAS,CAAC,0BAA0B,IAAI,KAAK,CAAC,SAAS;IAYvD,4DAA4D;IAC5D,SAAS,CAAC,+BAA+B,IAAI,IAAI;IAWjD;;OAEG;IACH,SAAS,CAAC,gCAAgC,IAAI,0BAA0B,EAAE;IAK1E;;OAEG;IACH,SAAS,KAAK,sBAAsB,IAAI,0BAA0B,GAAG,SAAS,CAE7E;IAED,SAAS,CAAC,qBAAqB,IAAI,KAAK,CAAC,SAAS;IAQlD,SAAS,CAAC,oBAAoB,IAAI,KAAK,CAAC,SAAS;IAQjD,SAAS,CAAC,+BAA+B,CAAC,cAAc,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,SAAS,EAAE,GAAG,KAAK,CAAC,SAAS;IAkB5G,SAAS,CAAC,gBAAgB,CAAC,MAAM,EAAE,mBAAmB,GAAG,WAAW;IAcpE,SAAS,CAAC,iBAAiB,CAAC,QAAQ,EAAE,iBAAiB,GAAG,YAAY;IAStE,SAAS,CAAC,QAAQ,CAAC,0BAA0B,uBAA8B;IAE3E;;;OAGG;IACI,cAAc,CAAC,SAAS,EAAE,SAAS,GAAG,IAAI;IA4CjD,SAAS,CAAC,sBAAsB,CAAC,OAAO,EAAE,gBAAgB,GAAG,IAAI;cAQ9C,cAAc,IAAI,MAAM,GAAG,SAAS;cASvC,iBAAiB,CAAC,SAAS,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC;cAcnD,UAAU,CACzB,IAAI,EAAE,QAAQ,EACd,KAAK,EAAE,SAAS,GACjB,KAAK,CAAC,SAAS;IAuBlB,SAAS,CAAC,WAAW,CAAC,IAAI,EAAE,WAAW,GAAG,YAAY,GAAG,KAAK,CAAC,SAAS;IA0ExE,SAAS,CAAC,aAAa,CAAC,IAAI,EAAE,WAAW,GAAG,YAAY,GAAG,MAAM;IAQjE,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,WAAW,GAAG,YAAY,GAAG,SAAS,GAAG,SAAS;IAO3E,SAAS,CAAC,qBAAqB,CAAC,IAAI,EAAE,WAAW,GAAG,YAAY,GAAG,MAAM,GAAG,SAAS;IASrF,SAAS,CAAC,YAAY,CAAC,IAAI,EAAE,WAAW,GAAG,YAAY,GAAG,KAAK,CAAC,SAAS;IASzE,SAAS,CAAC,iBAAiB,CAAC,IAAI,EAAE,WAAW,GAAG,KAAK,CAAC,SAAS;IAqC/D,SAAS,CAAC,kBAAkB,CAAC,IAAI,EAAE,YAAY,GAAG,KAAK,CAAC,SAAS;IA+BjE,SAAS,CAAC,2BAA2B,CAAC,OAAO,EAAE,mBAAmB,EAAE,IAAI,EAAE,YAAY,GAAG,KAAK,CAAC,SAAS;IAgBxG,SAAS,CAAC,iBAAiB,CAAC,IAAI,EAAE,QAAQ,GAAG,SAAS,EAAE,KAAK,EAAE,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,GAAG,IAAI;cAUhF,WAAW,CAAC,KAAK,EAAE,aAAa,GAAG,OAAO;IAO7D;;OAEG;cACgB,iBAAiB,IAAI,IAAI;CAM/C"}
+{"version":3,"file":"chat-view-tree-widget.d.ts","sourceRoot":"","sources":["../../../src/browser/chat-tree-view/chat-view-tree-widget.tsx"],"names":[],"mappings":"AAeA,OAAO,EACH,SAAS,EACT,gBAAgB,EAChB,SAAS,EACT,gBAAgB,EAChB,mBAAmB,EACnB,iBAAiB,EACjB,WAAW,EACX,wBAAwB,EAIxB,KAAK,WAAW,EAChB,KAAK,mBAAmB,EAC3B,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AACjE,OAAO,EAAE,eAAe,EAAE,oBAAoB,EAAc,oBAAoB,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACtH,OAAO,EAGH,mBAAmB,EACnB,YAAY,EAGZ,SAAS,EACT,aAAa,EACb,SAAS,EACT,QAAQ,EACR,SAAS,EACT,UAAU,EAGb,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,UAAU,EAAE,iBAAiB,EAAE,MAAM,6CAA6C,CAAC;AAQ5F,OAAO,KAAK,KAAK,MAAM,0BAA0B,CAAC;AAGlD,OAAO,EAAE,iCAAiC,EAAE,MAAM,0CAA0C,CAAC;AAC7F,OAAO,EAAE,wBAAwB,EAAE,MAAM,gCAAgC,CAAC;AAI1E,OAAO,EAAE,sBAAsB,EAAE,KAAK,qBAAqB,EAAE,MAAM,+BAA+B,CAAC;AAInG,MAAM,WAAW,WAAY,SAAQ,QAAQ;IACzC,OAAO,EAAE,gBAAgB,CAAC;IAC1B,MAAM,EAAE,mBAAmB,CAAC;IAC5B,SAAS,EAAE,MAAM,CAAA;CACpB;AACD,eAAO,MAAM,aAAa,GAAI,MAAM,QAAQ,KAAG,IAAI,IAAI,WAAgC,CAAC;AAExF,MAAM,WAAW,mBAAoB,SAAQ,WAAW;IACpD,OAAO,EAAE,wBAAwB,CAAA;CACpC;AACD,eAAO,MAAM,qBAAqB,GAAI,MAAM,QAAQ,KAAG,IAAI,IAAI,mBAAuF,CAAC;AAGvJ,MAAM,WAAW,YAAa,SAAQ,QAAQ;IAC1C,QAAQ,EAAE,iBAAiB,CAAC;IAC5B,SAAS,EAAE,MAAM,CAAA;CACpB;AACD,eAAO,MAAM,cAAc,GAAI,MAAM,QAAQ,KAAG,IAAI,IAAI,YAAkC,CAAC;AAE3F,wBAAgB,UAAU,CAAC,CAAC,EAAE,KAAK,CAAC,aAAa,GAAG,OAAO,CAE1D;AAED,eAAO,MAAM,0BAA0B,eAAuC,CAAC;AAC/E,MAAM,WAAW,0BAA0B;IACvC,oBAAoB,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC;IACzC,qBAAqB,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC;IAC1C,QAAQ,CAAC,cAAc,CAAC,EAAE,OAAO,CAAC;IAClC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,OAAO,CAAC;IAC5C,QAAQ,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,cAAc,CAAC,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;IACtC,oFAAoF;IACpF,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;CAC9B;AAED,qBACa,kBAAmB,SAAQ,UAAU;IAE9C,MAAM,CAAC,QAAQ,CAAC,EAAE,sBAAsB;IACxC,MAAM,CAAC,QAAQ,CAAC,YAAY,WAA8B;IAG1D,SAAS,CAAC,QAAQ,CAAC,yBAAyB,EAAE,oBAAoB,CAAC,wBAAwB,CAAC,mBAAmB,CAAC,CAAC,CAAC;IAGlH,SAAS,CAAC,QAAQ,CAAC,kCAAkC,EAAE,oBAAoB,CAAC,iCAAiC,CAAC,CAAC;IAG/G,SAAS,CAAC,gBAAgB,EAAE,gBAAgB,CAAC;IAG7C,SAAS,CAAC,QAAQ,CAAC,eAAe,EAAE,iBAAiB,CAAC;IAGtD,SAAS,CAAC,eAAe,EAAE,eAAe,CAAC;IAG3C,SAAS,CAAC,QAAQ,CAAC,aAAa,EAAE,aAAa,CAAC;IAGhD,SAAS,CAAC,YAAY,EAAE,YAAY,CAAC;IAGrC,SAAS,CAAC,QAAQ,CAAC,uBAAuB,EAAE,oBAAoB,CAAC,0BAA0B,CAAC,CAAC;IAG7F,SAAS,CAAC,kBAAkB,EAAE,sBAAsB,CAAC;IAGrD,SAAS,CAAC,QAAQ,CAAC,iBAAiB,EAAE,mBAAmB,CAAC;IAG1D,SAAS,CAAC,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;IAG5C,SAAS,CAAC,QAAQ,CAAC,iBAAiB,EAAE,iBAAiB,CAAC;IAExD,SAAS,CAAC,oBAAoB,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;IAEpD,SAAS,CAAC,QAAQ,CAAC,sBAAsB,uBAA8B;IACvE,eAAe,qBAAqC;IAEpD,SAAS,CAAC,QAAQ,CAAC,UAAU,EAAE,GAAG,CAAC,MAAM,EAAE,qBAAqB,CAAC,CAAa;IAE9E,SAAS,CAAC,kBAAkB,UAAQ;IAEpC,SAAS,CAAC,SAAS,UAAS;IAE5B,SAAS,CAAC,WAAW,EAAE,MAAM,CAAC;IAE9B,8EAA8E;IAC9E,SAAS,CAAC,QAAQ,UAAQ;IAC1B;;OAEG;IACH,SAAS,CAAC,iBAAiB,UAAS;IACpC;;;OAGG;IACH,SAAS,CAAC,0BAA0B,CAAC,EAAE,MAAM,CAAC;IAC9C;;;OAGG;IACH,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,0BAA0B,OAAO;IAE3D,kBAAkB,CAAC,EAAE,CAAC,eAAe,EAAE,OAAO,KAAK,IAAI,CAAC;IAExD,SAAS,CAAC,aAAa,SAAK;IAE5B,IAAI,iBAAiB,CAAC,iBAAiB,EAAE,OAAO,EAG/C;IAED,IAAI,iBAAiB,IAAI,OAAO,CAE/B;gBAGsB,KAAK,EAAE,SAAS,EAChB,KAAK,EAAE,SAAS,EACN,mBAAmB,EAAE,mBAAmB;cAiBtD,IAAI,IAAI,IAAI;IAuCxB,UAAU,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI;IAKzC,SAAS,CAAC,iBAAiB,CAAC,WAAW,EAAE,OAAO,GAAG,IAAI;IAqBvD,sEAAsE;IACtE,SAAS,CAAC,+BAA+B,CAAC,UAAU,EAAE,OAAO,GAAG,IAAI;IA6BpE,SAAS,CAAC,sBAAsB,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI;IAOxD,SAAS,CAAC,mBAAmB,CAAC,WAAW,EAAE,OAAO,GAAG,MAAM;IA0B3D;;;;OAIG;IACH,SAAS,CAAC,kBAAkB,IAAI,OAAO;cAqBpB,UAAU,CAAC,KAAK,EAAE,SAAS,GAAG,KAAK,CAAC,SAAS;IAehE,0EAA0E;IAC1E,SAAS,CAAC,0BAA0B,IAAI,KAAK,CAAC,SAAS;IAYvD,4DAA4D;IAC5D,SAAS,CAAC,+BAA+B,IAAI,IAAI;IAWjD;;OAEG;IACH,SAAS,CAAC,gCAAgC,IAAI,0BAA0B,EAAE;IAK1E;;OAEG;IACH,SAAS,KAAK,sBAAsB,IAAI,0BAA0B,GAAG,SAAS,CAE7E;IAED,SAAS,CAAC,qBAAqB,IAAI,KAAK,CAAC,SAAS;IAQlD,SAAS,CAAC,oBAAoB,IAAI,KAAK,CAAC,SAAS;IAQjD,SAAS,CAAC,+BAA+B,CAAC,cAAc,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,SAAS,EAAE,GAAG,KAAK,CAAC,SAAS;IAkB5G,SAAS,CAAC,gBAAgB,CAAC,MAAM,EAAE,mBAAmB,GAAG,WAAW;IAcpE,SAAS,CAAC,iBAAiB,CAAC,QAAQ,EAAE,iBAAiB,GAAG,YAAY;IAStE,SAAS,CAAC,QAAQ,CAAC,0BAA0B,uBAA8B;IAE3E;;;OAGG;IACI,cAAc,CAAC,SAAS,EAAE,SAAS,GAAG,IAAI;IA4CjD,SAAS,CAAC,sBAAsB,CAAC,OAAO,EAAE,gBAAgB,GAAG,IAAI;cAQ9C,cAAc,IAAI,MAAM,GAAG,SAAS;cASvC,iBAAiB,CAAC,SAAS,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC;cAcnD,UAAU,CACzB,IAAI,EAAE,QAAQ,EACd,KAAK,EAAE,SAAS,GACjB,KAAK,CAAC,SAAS;IAuBlB,SAAS,CAAC,WAAW,CAAC,IAAI,EAAE,WAAW,GAAG,YAAY,GAAG,KAAK,CAAC,SAAS;IA2FxE,SAAS,CAAC,aAAa,CAAC,IAAI,EAAE,WAAW,GAAG,YAAY,GAAG,MAAM;IAQjE,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,WAAW,GAAG,YAAY,GAAG,SAAS,GAAG,SAAS;IAO3E,SAAS,CAAC,qBAAqB,CAAC,IAAI,EAAE,WAAW,GAAG,YAAY,GAAG,MAAM,GAAG,SAAS;IASrF,SAAS,CAAC,YAAY,CAAC,IAAI,EAAE,WAAW,GAAG,YAAY,GAAG,KAAK,CAAC,SAAS;IASzE,SAAS,CAAC,iBAAiB,CAAC,IAAI,EAAE,WAAW,GAAG,KAAK,CAAC,SAAS;IAqC/D,SAAS,CAAC,kBAAkB,CAAC,IAAI,EAAE,YAAY,GAAG,KAAK,CAAC,SAAS;IA+BjE,SAAS,CAAC,2BAA2B,CAAC,OAAO,EAAE,mBAAmB,EAAE,IAAI,EAAE,YAAY,GAAG,KAAK,CAAC,SAAS;IAgBxG,SAAS,CAAC,iBAAiB,CAAC,IAAI,EAAE,QAAQ,GAAG,SAAS,EAAE,KAAK,EAAE,KAAK,CAAC,UAAU,CAAC,WAAW,CAAC,GAAG,IAAI;cAUhF,WAAW,CAAC,KAAK,EAAE,aAAa,GAAG,OAAO;IAO7D;;OAEG;cACgB,iBAAiB,IAAI,IAAI;CAM/C"}

package/lib/browser/chat-tree-view/chat-view-tree-widget.js

@@ -29,8 +29,10 @@ const nls_1 = require("@theia/core/lib/common/nls");
 const inversify_1 = require("@theia/core/shared/inversify");
 const React = require("@theia/core/shared/react");
 const image_context_variable_1 = require("@theia/ai-chat/lib/common/image-context-variable");
+const markdown_rendering_1 = require("@theia/core/lib/common/markdown-rendering");
 const chat_node_toolbar_action_contribution_1 = require("../chat-node-toolbar-action-contribution");
 const chat_response_part_renderer_1 = require("../chat-response-part-renderer");
+const chat_token_usage_indicator_util_1 = require("../chat-token-usage-indicator-util");
 const markdown_part_renderer_1 = require("../chat-response-renderer/markdown-part-renderer");
 const chat_progress_message_1 = require("../chat-progress-message");
 const chat_view_tree_input_widget_1 = require("./chat-view-tree-input-widget");
@@ -97,7 +99,7 @@ let ChatViewTreeWidget = class ChatViewTreeWidget extends browser_2.TreeWidget {
         this.addEventListener(this.node, 'focusout', () => this.chatResponseFocusKey.set(false));
         this.toDispose.pushAll([
             this.toDisposeOnChatModelChange,
-            this.activationService.onDidChangeActiveStatus(change => {
+            this.activationService.onDidChangeCanRun(change => {
                 this.chatInputs.forEach(widget => {
                     widget.setEnabled(change);
                 });
@@ -420,9 +422,24 @@ let ChatViewTreeWidget = class ChatViewTreeWidget extends browser_2.TreeWidget {
             React.createElement("div", { className: 'theia-ChatNodeHeader' },
                 React.createElement("div", { className: `theia-AgentAvatar ${this.getAgentIconClassName(node)}` }),
                 React.createElement("h3", { ref: agentLabel, className: 'theia-AgentLabel', onMouseEnter: () => {
-                        if (agentDescription) {
+                        const tokenUsage = (0, exports.isResponseNode)(node) ? node.response.tokenUsage : undefined;
+                        const hasTokenInfo = tokenUsage && (tokenUsage.inputTokens > 0 || tokenUsage.outputTokens > 0);
+                        const tokenInfo = hasTokenInfo
+                            ? `${nls_1.nls.localize('theia/ai/chat-ui/tokenUsageLabel', 'Token Usage')}: ${nls_1.nls.localizeByDefault('Input: {0}', (0, chat_token_usage_indicator_util_1.formatTokenCount)(tokenUsage.inputTokens))} | ${nls_1.nls.localizeByDefault('Output: {0}', (0, chat_token_usage_indicator_util_1.formatTokenCount)(tokenUsage.outputTokens))}`
+                            : undefined;
+                        if (agentDescription || tokenInfo) {
+                            const md = new markdown_rendering_1.MarkdownStringImpl();
+                            if (agentDescription) {
+                                md.appendMarkdown(agentDescription);
+                            }
+                            if (agentDescription && tokenInfo) {
+                                md.appendMarkdown('\n\n---\n\n');
+                            }
+                            if (tokenInfo) {
+                                md.appendMarkdown(tokenInfo);
+                            }
                             this.hoverService.requestHover({
-                                content: agentDescription,
+                                content: md,
                                 target: agentLabel.current,
                                 position: 'right'
                             });