@codingame/monaco-vscode-chat-service-override 34.1.3 → 35.0.1

package/vscode/src/vs/workbench/contrib/chat/browser/chatQuotaNotification.js

@@ -1,21 +1,39 @@
 
 import { __decorate, __param } from '@codingame/monaco-vscode-api/external/tslib/tslib.es6';
-import { Disposable, DisposableStore } from '@codingame/monaco-vscode-api/vscode/vs/base/common/lifecycle';
 import { safeIntl } from '@codingame/monaco-vscode-api/vscode/vs/base/common/date';
+import { createMarkdownCommandLink, MarkdownString } from '@codingame/monaco-vscode-api/vscode/vs/base/common/htmlContent';
+import { Disposable, DisposableStore } from '@codingame/monaco-vscode-api/vscode/vs/base/common/lifecycle';
+import { URI } from '@codingame/monaco-vscode-api/vscode/vs/base/common/uri';
 import { localize } from '@codingame/monaco-vscode-api/vscode/vs/nls';
+import { CommandsRegistry } from '@codingame/monaco-vscode-api/vscode/vs/platform/commands/common/commands';
 import { IContextKeyService } from '@codingame/monaco-vscode-api/vscode/vs/platform/contextkey/common/contextkey.service';
-import { StorageScope } from '@codingame/monaco-vscode-api/vscode/vs/platform/storage/common/storage';
+import { ILogService } from '@codingame/monaco-vscode-api/vscode/vs/platform/log/common/log.service';
+import { IOpenerService } from '@codingame/monaco-vscode-api/vscode/vs/platform/opener/common/opener.service';
+import { StorageScope, StorageTarget } from '@codingame/monaco-vscode-api/vscode/vs/platform/storage/common/storage';
 import { IStorageService } from '@codingame/monaco-vscode-api/vscode/vs/platform/storage/common/storage.service';
+import { ITelemetryService } from '@codingame/monaco-vscode-api/vscode/vs/platform/telemetry/common/telemetry.service';
+import { IWorkbenchAssignmentService } from '@codingame/monaco-vscode-api/vscode/vs/workbench/services/assignment/common/assignmentService.service';
 import { ChatEntitlement } from '@codingame/monaco-vscode-api/vscode/vs/workbench/services/chat/common/chatEntitlementService';
 import { IChatEntitlementService } from '@codingame/monaco-vscode-api/vscode/vs/workbench/services/chat/common/chatEntitlementService.service';
-import { SELECTED_MODEL_STORAGE_KEY_PREFIX, getSelectedModelVendor } from '../common/chatSelectedModel.js';
-import { COPILOT_VENDOR_ID } from '@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/languageModels';
+import { SELECTED_MODEL_STORAGE_KEY_PREFIX, isSelectedModelCopilot } from '@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/chatSelectedModel';
 import { ILanguageModelsService } from '@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/languageModels.service';
 import { ChatInputNotificationSeverity } from '@codingame/monaco-vscode-katex-common/vscode/vs/workbench/contrib/chat/browser/widget/input/chatInputNotificationService';
 import { IChatInputNotificationService } from '@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/browser/widget/input/chatInputNotificationService.service';
 
 const QUOTA_NOTIFICATION_ID = "copilot.quotaStatus";
 const THRESHOLDS = [50, 75, 90, 95];
+const TRAJECTORY_NUDGE_SPEC = {
+    treatmentName: "config.chatQuotaTrajectoryNudge",
+    shownStorageKey: "chat.quotaTrajectory.shownPeriod",
+    averageDailyUsageThreshold: 4.5,
+    minimumPercentUsed: 10,
+    maximumPercentUsed: 35,
+    billingPeriodDays: 30,
+    msPerDay: 24 * 60 * 60 * 1000,
+    learnMoreUrl: "https://aka.ms/token-usage-tips",
+    learnMoreCommandId: "workbench.action.chat.learnMoreAboutCreditUsage"
+};
+const QUOTA_EXHAUSTED_DISMISSED_STORAGE_KEY = "chat.quotaNotification.exhaustedDismissed";
 let ChatQuotaNotificationContribution = class ChatQuotaNotificationContribution extends Disposable {
     static {
         this.ID = "workbench.contrib.chatQuotaNotification";
@@ -25,7 +43,10 @@ let ChatQuotaNotificationContribution = class ChatQuotaNotificationContribution
         _chatInputNotificationService,
         _contextKeyService,
         _languageModelsService,
-        _storageService
+        _storageService,
+        _assignmentService,
+        _telemetryService,
+        _logService
     ) {
         super();
         this._chatEntitlementService = _chatEntitlementService;
@@ -33,7 +54,11 @@ let ChatQuotaNotificationContribution = class ChatQuotaNotificationContribution
         this._contextKeyService = _contextKeyService;
         this._languageModelsService = _languageModelsService;
         this._storageService = _storageService;
+        this._assignmentService = _assignmentService;
+        this._telemetryService = _telemetryService;
+        this._logService = _logService;
         this._showingExhausted = false;
+        this._trajectoryAssignmentRequested = false;
         this._register(
             this._chatEntitlementService.onDidChangeQuotaRemaining(() => this._update())
         );
@@ -41,6 +66,10 @@ let ChatQuotaNotificationContribution = class ChatQuotaNotificationContribution
             this._chatEntitlementService.onDidChangeQuotaExceeded(() => this._update())
         );
         this._register(this._chatEntitlementService.onDidChangeEntitlement(() => this._update()));
+        this._register(CommandsRegistry.registerCommand(
+            TRAJECTORY_NUDGE_SPEC.learnMoreCommandId,
+            accessor => this._handleCreditEfficiencyLearnMoreCommand(accessor)
+        ));
         const storageListener = this._register(( new DisposableStore()));
         this._register(
             this._storageService.onDidChangeValue(StorageScope.APPLICATION, undefined, storageListener)(e => {
@@ -49,8 +78,32 @@ let ChatQuotaNotificationContribution = class ChatQuotaNotificationContribution
                 }
             })
         );
+        this._register(this._chatInputNotificationService.onDidDismiss(id => {
+            if (id === QUOTA_NOTIFICATION_ID && this._showingExhausted) {
+                this._setExhaustedDismissed();
+            }
+        }));
         this._update();
     }
+    async _resolveTrajectoryTreatment(warning) {
+        const treatment = await this._assignmentService.getTreatment(TRAJECTORY_NUDGE_SPEC.treatmentName);
+        this._trajectoryTreatment = treatment;
+        if (treatment !== undefined) {
+            this._logQuotaTrajectoryNudgeEnrolled(treatment, warning);
+        }
+        if (treatment === true) {
+            this._update();
+        }
+    }
+    _requestTrajectoryTreatment(warning) {
+        if (!this._trajectoryAssignmentRequested) {
+            this._trajectoryAssignmentRequested = true;
+            void this._resolveTrajectoryTreatment(warning).catch(error => {
+                this._logService.error(`Failed to resolve ${TRAJECTORY_NUDGE_SPEC.treatmentName}`, error);
+                this._trajectoryAssignmentRequested = false;
+            });
+        }
+    }
     _getRelevantSnapshot() {
         const quotas = this._chatEntitlementService.quotas;
         const entitlement = this._chatEntitlementService.entitlement;
@@ -75,12 +128,17 @@ let ChatQuotaNotificationContribution = class ChatQuotaNotificationContribution
     _update() {
         const entitlement = this._chatEntitlementService.entitlement;
         const isCopilot = this._isCopilotModelSelected();
+        if (this._isQuotaKnownAvailable()) {
+            this._clearExhaustedDismissed();
+        }
         if (!isCopilot) {
             return;
         }
         const isQuotaNotificationEligible = entitlement === ChatEntitlement.Unknown || this._isUBBEligible();
         if (this._isManagedPlan(entitlement) && this._isManagedPlanBlocked()) {
-            this._showManagedPlanBlockedNotification();
+            if (!this._isExhaustedDismissed()) {
+                this._showManagedPlanBlockedNotification();
+            }
             return;
         }
         if (isQuotaNotificationEligible && this._isQuotaUsedUp()) {
@@ -88,12 +146,14 @@ let ChatQuotaNotificationContribution = class ChatQuotaNotificationContribution
             const additionalUsageEnabled = quotas.additionalUsageEnabled ?? false;
             const wasAdditionalUsageEnabled = this._prevAdditionalUsageEnabled;
             this._prevAdditionalUsageEnabled = additionalUsageEnabled;
-            if (additionalUsageEnabled) {
-                if (this._prevQuotaPercentUsed !== undefined || wasAdditionalUsageEnabled === false) {
-                    this._showOverageActivationNotification();
+            if (!this._isExhaustedDismissed()) {
+                if (additionalUsageEnabled) {
+                    if (this._prevQuotaPercentUsed !== undefined || wasAdditionalUsageEnabled === false) {
+                        this._showOverageActivationNotification();
+                    }
+                } else {
+                    this._showExhaustedNotification();
                 }
-            } else {
-                this._showExhaustedNotification();
             }
             const exhaustedSnapshot = this._getRelevantSnapshot();
             if (exhaustedSnapshot && !exhaustedSnapshot.unlimited) {
@@ -102,6 +162,11 @@ let ChatQuotaNotificationContribution = class ChatQuotaNotificationContribution
             return;
         }
         if (isQuotaNotificationEligible) {
+            const trajectoryWarning = this._computeQuotaTrajectoryWarning();
+            if (trajectoryWarning) {
+                this._showQuotaTrajectoryWarning(trajectoryWarning);
+                return;
+            }
             const quotaWarning = this._computeQuotaWarning();
             if (quotaWarning) {
                 this._showQuotaApproachingWarning(quotaWarning);
@@ -128,11 +193,91 @@ let ChatQuotaNotificationContribution = class ChatQuotaNotificationContribution
         this._prevQuotaPercentUsed = percentUsed;
         if (crossed !== undefined) {
             return {
-                percentUsed: Math.floor(percentUsed)
+                percentUsed: Math.floor(percentUsed),
+                threshold: crossed
             };
         }
         return undefined;
     }
+    _computeQuotaTrajectoryWarning() {
+        if (this._isTrajectoryShownInCurrentPeriod()) {
+            return undefined;
+        }
+        const snapshot = this._getRelevantSnapshot();
+        if (!snapshot || snapshot.unlimited || snapshot.percentRemaining <= 0) {
+            return undefined;
+        }
+        const resetDate = this._chatEntitlementService.quotas.resetDate;
+        if (!resetDate) {
+            return undefined;
+        }
+        const resetTime = ( new Date(resetDate)).getTime();
+        if (!Number.isFinite(resetTime)) {
+            return undefined;
+        }
+        const periodStartTime = resetTime - (TRAJECTORY_NUDGE_SPEC.billingPeriodDays * TRAJECTORY_NUDGE_SPEC.msPerDay);
+        const elapsedDays = Math.max(0, (Date.now() - periodStartTime) / TRAJECTORY_NUDGE_SPEC.msPerDay);
+        if (elapsedDays <= 0) {
+            return undefined;
+        }
+        const percentUsed = 100 - snapshot.percentRemaining;
+        if (percentUsed < TRAJECTORY_NUDGE_SPEC.minimumPercentUsed || percentUsed > TRAJECTORY_NUDGE_SPEC.maximumPercentUsed) {
+            return undefined;
+        }
+        const averageDailyUsage = percentUsed / elapsedDays;
+        if (averageDailyUsage < TRAJECTORY_NUDGE_SPEC.averageDailyUsageThreshold) {
+            return undefined;
+        }
+        this._requestTrajectoryTreatment({
+            averageDailyUsage,
+            percentUsed
+        });
+        return this._trajectoryTreatment === true ? {
+            averageDailyUsage,
+            percentUsed
+        } : undefined;
+    }
+    _showQuotaTrajectoryWarning(warning) {
+        this._showingExhausted = false;
+        this._storeTrajectoryShown();
+        const learnMoreLink = createMarkdownCommandLink({
+            text: ( localize(7148, "Learn about optimizing usage")),
+            id: TRAJECTORY_NUDGE_SPEC.learnMoreCommandId,
+            tooltip: ( localize(7149, "Learn about optimizing usage"))
+        });
+        const message = ( localize(
+            7150,
+            "You're likely to exhaust your AI credits before your billing period. {0}.",
+            learnMoreLink
+        ));
+        this._setNotification({
+            id: QUOTA_NOTIFICATION_ID,
+            telemetryId: "quotaTrajectoryNudge",
+            severity: ChatInputNotificationSeverity.Info,
+            message: ( new MarkdownString(message, {
+                isTrusted: {
+                    enabledCommands: [TRAJECTORY_NUDGE_SPEC.learnMoreCommandId]
+                }
+            })),
+            description: undefined,
+            actions: [],
+            dismissible: true,
+            autoDismissOnMessage: false
+        });
+    }
+    async _handleCreditEfficiencyLearnMoreCommand(accessor) {
+        this._telemetryService.publicLog2("chatQuotaTrajectoryNudgeLinkClicked");
+        queueMicrotask(() => this._hideNotification());
+        await accessor.get(IOpenerService).open(( URI.parse(TRAJECTORY_NUDGE_SPEC.learnMoreUrl)));
+    }
+    _logQuotaTrajectoryNudgeEnrolled(treatment, warning) {
+        this._telemetryService.publicLog2("chatQuotaTrajectoryNudgeEnrolled", {
+            treatment,
+            entitlement: ChatEntitlement[this._chatEntitlementService.entitlement],
+            averageDailyUsage: Math.round(warning.averageDailyUsage * 100) / 100,
+            percentUsed: Math.round(warning.percentUsed * 100) / 100
+        });
+    }
     _findCrossedThreshold(current, previous) {
         if (previous === undefined) {
             return undefined;
@@ -153,37 +298,38 @@ let ChatQuotaNotificationContribution = class ChatQuotaNotificationContribution
         let description;
         let actions;
         if (entitlement === ChatEntitlement.Unknown) {
-            description = ( localize(6648, "Sign in to keep going."));
+            description = ( localize(7151, "Sign in to keep going."));
             actions = [{
-                label: ( localize(6649, "Sign In")),
+                label: ( localize(7152, "Sign In")),
                 commandId: "workbench.action.chat.triggerSetup"
             }];
         } else if (entitlement === ChatEntitlement.Free) {
-            description = ( localize(6650, "Upgrade to keep going."));
+            description = ( localize(7153, "Upgrade to keep going."));
             actions = [{
-                label: ( localize(6651, "Upgrade")),
+                label: ( localize(7154, "Upgrade")),
                 commandId: "workbench.action.chat.upgradePlan"
             }];
         } else if (this._isManagedPlan(entitlement)) {
-            description = ( localize(6652, "Contact your admin to increase your limits."));
+            description = ( localize(7155, "Contact your admin to increase your limits."));
             actions = [];
         } else if (hadOverage) {
-            description = ( localize(6653, "Increase your budget to keep building."));
+            description = ( localize(7156, "Increase your budget to keep building."));
             actions = [{
-                label: ( localize(6654, "Manage Budget")),
+                label: ( localize(7157, "Manage Budget")),
                 commandId: "workbench.action.chat.manageAdditionalSpend"
             }];
         } else {
-            description = ( localize(6655, "Manage your budget to keep building."));
+            description = ( localize(7158, "Manage your budget to keep building."));
             actions = [{
-                label: ( localize(6656, "Manage Budget")),
+                label: ( localize(7159, "Manage Budget")),
                 commandId: "workbench.action.chat.manageAdditionalSpend"
             }];
         }
         this._setNotification({
             id: QUOTA_NOTIFICATION_ID,
+            telemetryId: "quotaExhausted",
             severity: ChatInputNotificationSeverity.Info,
-            message: ( localize(6657, "Credit Limit Reached")),
+            message: ( localize(7160, "Credit Limit Reached")),
             description,
             actions,
             dismissible: true,
@@ -194,9 +340,10 @@ let ChatQuotaNotificationContribution = class ChatQuotaNotificationContribution
         this._showingExhausted = true;
         this._setNotification({
             id: QUOTA_NOTIFICATION_ID,
+            telemetryId: "overageActivation",
             severity: ChatInputNotificationSeverity.Info,
-            message: ( localize(6658, "Credit Limit Reached")),
-            description: ( localize(6659, "Additional budget is now covering extra usage.")),
+            message: ( localize(7161, "Credit Limit Reached")),
+            description: ( localize(7162, "Additional budget is now covering extra usage.")),
             actions: [],
             dismissible: true,
             autoDismissOnMessage: true
@@ -209,28 +356,29 @@ let ChatQuotaNotificationContribution = class ChatQuotaNotificationContribution
         let description;
         let actions;
         if (entitlement === ChatEntitlement.Unknown || entitlement === ChatEntitlement.Free) {
-            description = ( localize(6660, "Upgrade to continue past the limit."));
+            description = ( localize(7163, "Upgrade to continue past the limit."));
             actions = [{
-                label: ( localize(6661, "Upgrade")),
+                label: ( localize(7164, "Upgrade")),
                 commandId: "workbench.action.chat.upgradePlan"
             }];
         } else if (this._isManagedPlan(entitlement)) {
-            description = ( localize(6662, "Contact your admin to increase your limits."));
+            description = ( localize(7165, "Contact your admin to increase your limits."));
             actions = [];
         } else if (quotas.additionalUsageEnabled) {
-            description = ( localize(6663, "Additional budget is enabled to cover extra usage."));
+            description = ( localize(7166, "Additional budget is enabled to cover extra usage."));
             actions = [];
         } else {
-            description = ( localize(6664, "Set additional budget to cover extra usage."));
+            description = ( localize(7167, "Set additional budget to cover extra usage."));
             actions = [{
-                label: ( localize(6665, "Manage Budget")),
+                label: ( localize(7168, "Manage Budget")),
                 commandId: "workbench.action.chat.manageAdditionalSpend"
             }];
         }
         this._setNotification({
             id: QUOTA_NOTIFICATION_ID,
+            telemetryId: `quotaApproaching${warning.threshold}`,
             severity: ChatInputNotificationSeverity.Info,
-            message: ( localize(6666, "Credits at {0}%", warning.percentUsed)),
+            message: ( localize(7169, "Credits at {0}%", warning.percentUsed)),
             description,
             actions,
             dismissible: true,
@@ -275,10 +423,11 @@ let ChatQuotaNotificationContribution = class ChatQuotaNotificationContribution
     }
     _showRateLimitWarning(warning) {
         this._showingExhausted = false;
-        const message = warning.type === "session" ? ( localize(6667, "You've used {0}% of your session rate limit.", warning.percentUsed)) : ( localize(6668, "You've used {0}% of your weekly rate limit.", warning.percentUsed));
-        const description = warning.resetDate ? ( localize(6669, "Resets on {0}.", this._formatResetDate(warning.resetDate))) : undefined;
+        const message = warning.type === "session" ? ( localize(7170, "You've used {0}% of your session rate limit.", warning.percentUsed)) : ( localize(7171, "You've used {0}% of your weekly rate limit.", warning.percentUsed));
+        const description = warning.resetDate ? ( localize(7172, "Resets on {0}.", this._formatResetDate(warning.resetDate))) : undefined;
         this._setNotification({
             id: QUOTA_NOTIFICATION_ID,
+            telemetryId: warning.type === "session" ? "sessionRateLimitWarning" : "weeklyRateLimitWarning",
             severity: ChatInputNotificationSeverity.Info,
             message,
             description,
@@ -288,11 +437,7 @@ let ChatQuotaNotificationContribution = class ChatQuotaNotificationContribution
         });
     }
     _isCopilotModelSelected() {
-        const vendor = getSelectedModelVendor(this._contextKeyService, this._storageService, this._languageModelsService);
-        if (!vendor) {
-            return true;
-        }
-        return vendor === COPILOT_VENDOR_ID;
+        return isSelectedModelCopilot(this._contextKeyService, this._storageService, this._languageModelsService);
     }
     _isManagedPlan(entitlement) {
         return entitlement === ChatEntitlement.Business || entitlement === ChatEntitlement.Enterprise;
@@ -305,10 +450,11 @@ let ChatQuotaNotificationContribution = class ChatQuotaNotificationContribution
         this._showingExhausted = true;
         this._setNotification({
             id: QUOTA_NOTIFICATION_ID,
+            telemetryId: "managedPlanBlocked",
             severity: ChatInputNotificationSeverity.Info,
-            message: ( localize(6670, "Usage Blocked")),
+            message: ( localize(7173, "Usage Blocked")),
             description: ( localize(
-                6671,
+                7174,
                 "Your organization or enterprise has exceeded its Copilot budget. Contact your admin to resume usage."
             )),
             actions: [],
@@ -333,6 +479,32 @@ let ChatQuotaNotificationContribution = class ChatQuotaNotificationContribution
             minute: "2-digit"
         }).value.format(resetDate);
     }
+    _getTrajectoryPeriodKey() {
+        const resetDate = this._chatEntitlementService.quotas.resetDate;
+        if (!resetDate) {
+            return undefined;
+        }
+        const date = ( new Date(resetDate));
+        if (!Number.isFinite(date.getTime())) {
+            return undefined;
+        }
+        return `${date.getUTCFullYear()}-${date.getUTCMonth() + 1}`;
+    }
+    _isTrajectoryShownInCurrentPeriod() {
+        const periodKey = this._getTrajectoryPeriodKey();
+        return !!periodKey && this._storageService.get(TRAJECTORY_NUDGE_SPEC.shownStorageKey, StorageScope.APPLICATION) === periodKey;
+    }
+    _storeTrajectoryShown() {
+        const periodKey = this._getTrajectoryPeriodKey();
+        if (periodKey) {
+            this._storageService.store(
+                TRAJECTORY_NUDGE_SPEC.shownStorageKey,
+                periodKey,
+                StorageScope.APPLICATION,
+                StorageTarget.USER
+            );
+        }
+    }
     _setNotification(notification) {
         this._chatInputNotificationService.setNotification(notification);
     }
@@ -340,7 +512,24 @@ let ChatQuotaNotificationContribution = class ChatQuotaNotificationContribution
         this._showingExhausted = false;
         this._chatInputNotificationService.deleteNotification(QUOTA_NOTIFICATION_ID);
     }
+    _isQuotaKnownAvailable() {
+        return !!this._getRelevantSnapshot() && !this._isQuotaUsedUp();
+    }
+    _isExhaustedDismissed() {
+        return this._storageService.getBoolean(QUOTA_EXHAUSTED_DISMISSED_STORAGE_KEY, StorageScope.APPLICATION, false);
+    }
+    _setExhaustedDismissed() {
+        this._storageService.store(
+            QUOTA_EXHAUSTED_DISMISSED_STORAGE_KEY,
+            true,
+            StorageScope.APPLICATION,
+            StorageTarget.MACHINE
+        );
+    }
+    _clearExhaustedDismissed() {
+        this._storageService.remove(QUOTA_EXHAUSTED_DISMISSED_STORAGE_KEY, StorageScope.APPLICATION);
+    }
 };
-ChatQuotaNotificationContribution = ( __decorate([( __param(0, IChatEntitlementService)), ( __param(1, IChatInputNotificationService)), ( __param(2, IContextKeyService)), ( __param(3, ILanguageModelsService)), ( __param(4, IStorageService))], ChatQuotaNotificationContribution));
+ChatQuotaNotificationContribution = ( __decorate([( __param(0, IChatEntitlementService)), ( __param(1, IChatInputNotificationService)), ( __param(2, IContextKeyService)), ( __param(3, ILanguageModelsService)), ( __param(4, IStorageService)), ( __param(5, IWorkbenchAssignmentService)), ( __param(6, ITelemetryService)), ( __param(7, ILogService))], ChatQuotaNotificationContribution));
 
 export { ChatQuotaNotificationContribution };

package/vscode/src/vs/workbench/contrib/chat/browser/chatRepoInfo.js

@@ -128,13 +128,13 @@ let ChatRepoInfoContribution = class ChatRepoInfoContribution extends Disposable
         const registry = ( Registry.as(Extensions.Configuration));
         registry.registerConfiguration({
             id: "chatRepoInfo",
-            title: ( localize(6672, "Chat Repository Info")),
+            title: ( localize(7175, "Chat Repository Info")),
             type: "object",
             properties: {
                 [ChatConfiguration.RepoInfoEnabled]: {
                     type: "boolean",
                     description: ( localize(
-                        6673,
+                        7176,
                         "Controls whether lightweight repository metadata (branch, commit, remotes) is captured when a chat request is submitted for internal diagnostics."
                     )),
                     default: false

package/vscode/src/vs/workbench/contrib/chat/browser/chatResponseFileChangesService.d.ts

@@ -0,0 +1,30 @@
+import { Disposable, IDisposable } from "@codingame/monaco-vscode-api/vscode/vs/base/common/lifecycle";
+import { IObservable } from "@codingame/monaco-vscode-api/vscode/vs/base/common/observable";
+import { URI } from "@codingame/monaco-vscode-api/vscode/vs/base/common/uri";
+import { IEditSessionEntryDiff } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/editing/chatEditingService";
+import { IChatResponseFileChangesService } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/browser/chatResponseFileChangesService.service";
+/**
+ * Supplies the per-response (per-request) file-change diffs rendered by the
+ * "Changed N files" summary under a completed chat response.
+ *
+ * Most chat sessions derive these diffs from their {@link IChatEditingSession}.
+ * Some session types (notably agent host sessions) own an authoritative,
+ * server-computed view of a turn's changes and provide it here instead, so the
+ * summary reflects the same source of truth as the rest of that session's
+ * change UI.
+ */
+export interface IChatResponseFileChangesProvider {
+    /**
+     * Returns an observable of the file-change diffs produced by `requestId`
+     * within `sessionResource`, or `undefined` when this provider cannot
+     * supply changes for that request (in which case the caller falls back to
+     * the chat editing session).
+     */
+    getChangesForRequest(sessionResource: URI, requestId: string): IObservable<readonly IEditSessionEntryDiff[]> | undefined;
+}
+export declare class ChatResponseFileChangesService extends Disposable implements IChatResponseFileChangesService {
+    readonly _serviceBrand: undefined;
+    private readonly _providers;
+    registerProvider(chatSessionType: string, provider: IChatResponseFileChangesProvider): IDisposable;
+    getChangesForRequest(sessionResource: URI, requestId: string): IObservable<readonly IEditSessionEntryDiff[]> | undefined;
+}

package/vscode/src/vs/workbench/contrib/chat/browser/chatResponseFileChangesService.js

@@ -0,0 +1,29 @@
+
+import { Disposable, toDisposable } from '@codingame/monaco-vscode-api/vscode/vs/base/common/lifecycle';
+import { getChatSessionType } from '@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/model/chatUri';
+
+class ChatResponseFileChangesService extends Disposable {
+    constructor() {
+        super(...arguments);
+        this._providers = ( new Map());
+    }
+    registerProvider(chatSessionType, provider) {
+        if (( this._providers.has(chatSessionType))) {
+            throw ( new Error(
+                `A chat response file changes provider is already registered for session type '${chatSessionType}'`
+            ));
+        }
+        this._providers.set(chatSessionType, provider);
+        return toDisposable(() => {
+            if (this._providers.get(chatSessionType) === provider) {
+                this._providers.delete(chatSessionType);
+            }
+        });
+    }
+    getChangesForRequest(sessionResource, requestId) {
+        const provider = this._providers.get(getChatSessionType(sessionResource));
+        return provider?.getChangesForRequest(sessionResource, requestId);
+    }
+}
+
+export { ChatResponseFileChangesService };