@vscode/chat-lib 0.0.1

package/dist/src/_internal/platform/authentication/node/copilotTokenManager.d.ts

@@ -0,0 +1,80 @@
+import { Disposable } from '../../../util/vs/base/common/lifecycle';
+import { SyncDescriptor } from '../../../util/vs/platform/instantiation/common/descriptors';
+import { IConfigurationService } from '../../configuration/common/configurationService';
+import { ICAPIClientService } from '../../endpoint/common/capiClient';
+import { IDomainService } from '../../endpoint/common/domainService';
+import { IEnvService } from '../../env/common/envService';
+import { BaseOctoKitService } from '../../github/common/githubService';
+import { ILogService } from '../../log/common/logService';
+import { IFetcherService } from '../../networking/common/fetcherService';
+import { ITelemetryService } from '../../telemetry/common/telemetry';
+import { CopilotToken, ExtendedTokenInfo, TokenInfoOrError } from '../common/copilotToken';
+import { CheckCopilotToken, ICopilotTokenManager, NotGitHubLoginFailed } from '../common/copilotTokenManager';
+export declare const tokenErrorString = "Tests: either GITHUB_PAT or GITHUB_OAUTH_TOKEN must be set. Run \"npm run get_token\" to get one.";
+export declare function getStaticGitHubToken(): string;
+export declare function getOrCreateTestingCopilotTokenManager(): SyncDescriptor<ICopilotTokenManager & CheckCopilotToken>;
+export declare abstract class BaseCopilotTokenManager extends Disposable implements ICopilotTokenManager {
+    protected readonly _baseOctokitservice: BaseOctoKitService;
+    protected readonly _logService: ILogService;
+    protected readonly _telemetryService: ITelemetryService;
+    protected readonly _domainService: IDomainService;
+    protected readonly _capiClientService: ICAPIClientService;
+    protected readonly _fetcherService: IFetcherService;
+    protected readonly _envService: IEnvService;
+    readonly _serviceBrand: undefined;
+    protected _isDisposed: boolean;
+    private readonly _copilotTokenRefreshEmitter;
+    readonly onDidCopilotTokenRefresh: import("../../../util/vs/base/common/event").Event<void>;
+    constructor(_baseOctokitservice: BaseOctoKitService, _logService: ILogService, _telemetryService: ITelemetryService, _domainService: IDomainService, _capiClientService: ICAPIClientService, _fetcherService: IFetcherService, _envService: IEnvService);
+    private _copilotToken;
+    get copilotToken(): ExtendedTokenInfo | undefined;
+    set copilotToken(token: ExtendedTokenInfo | undefined);
+    abstract getCopilotToken(force?: boolean): Promise<CopilotToken>;
+    resetCopilotToken(httpError?: number): void;
+    /**
+     * Fetches a Copilot token from the GitHub token.
+     * @param githubToken A GitHub token to mint a Copilot token from.
+     * @returns A Copilot token info or an error.
+     * @todo this should be not be public, but it is for now to allow testing.
+     */
+    authFromGitHubToken(githubToken: string, ghUsername: string): Promise<TokenInfoOrError & NotGitHubLoginFailed>;
+    private fetchCopilotToken;
+    private fetchCopilotUserInfo;
+}
+/**
+ * A `CopilotTokenManager` that always returns the same token.
+ * Mostly only useful for short periods, e.g. tests or single completion requests,
+ * as these tokens typically expire after a few hours.
+ * @todo Move this to a test layer
+ */
+export declare class FixedCopilotTokenManager extends BaseCopilotTokenManager implements CheckCopilotToken {
+    private _completionsToken;
+    constructor(_completionsToken: string, logService: ILogService, telemetryService: ITelemetryService, capiClientService: ICAPIClientService, domainService: IDomainService, fetcherService: IFetcherService, envService: IEnvService);
+    set completionsToken(token: string);
+    get completionsToken(): string;
+    getCopilotToken(): Promise<CopilotToken>;
+    checkCopilotToken(): Promise<{
+        status: 'OK';
+    }>;
+}
+/**
+ * Given a GitHub token, return a Copilot token, refreshing it as needed.
+ * The caller that initializes the object is responsible for checking telemetry consent before
+ * using the object.
+ */
+export declare class CopilotTokenManagerFromGitHubToken extends BaseCopilotTokenManager implements CheckCopilotToken {
+    private readonly githubToken;
+    private readonly githubUsername;
+    protected readonly configurationService: IConfigurationService;
+    constructor(githubToken: string, githubUsername: string, logService: ILogService, telemetryService: ITelemetryService, domainService: IDomainService, capiClientService: ICAPIClientService, fetcherService: IFetcherService, envService: IEnvService, configurationService: IConfigurationService);
+    getCopilotToken(force?: boolean): Promise<CopilotToken>;
+    checkCopilotToken(): Promise<({
+        kind: "failure";
+    } & import("../common/copilotToken").TokenError & {
+        kind: "failure";
+        reason: Exclude<import("../common/copilotToken").TokenErrorReason, "GitHubLoginFailed">;
+    }) | {
+        status: "OK";
+    }>;
+}
+//# sourceMappingURL=copilotTokenManager.d.ts.map

package/dist/src/_internal/platform/authentication/node/copilotTokenManager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"copilotTokenManager.d.ts","sourceRoot":"","sources":["../../../../../../src/_internal/platform/authentication/node/copilotTokenManager.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,UAAU,EAAgB,MAAM,wCAAwC,CAAC;AAClF,OAAO,EAAE,cAAc,EAAE,MAAM,4DAA4D,CAAC;AAC5F,OAAO,EAAE,qBAAqB,EAAE,MAAM,iDAAiD,CAAC;AACxF,OAAO,EAAE,kBAAkB,EAAE,MAAM,kCAAkC,CAAC;AACtE,OAAO,EAAE,cAAc,EAAE,MAAM,qCAAqC,CAAC;AACrE,OAAO,EAAE,WAAW,EAAE,MAAM,6BAA6B,CAAC;AAC1D,OAAO,EAAE,kBAAkB,EAAgB,MAAM,mCAAmC,CAAC;AAErF,OAAO,EAAE,WAAW,EAAE,MAAM,6BAA6B,CAAC;AAC1D,OAAO,EAAgB,eAAe,EAA8B,MAAM,wCAAwC,CAAC;AACnH,OAAO,EAAE,iBAAiB,EAAE,MAAM,kCAAkC,CAAC;AAErE,OAAO,EAAE,YAAY,EAAmB,iBAAiB,EAAa,gBAAgB,EAAuB,MAAM,wBAAwB,CAAC;AAC5I,OAAO,EAAE,iBAAiB,EAAE,oBAAoB,EAAE,oBAAoB,EAAc,MAAM,+BAA+B,CAAC;AAE1H,eAAO,MAAM,gBAAgB,sGAAoG,CAAC;AAElI,wBAAgB,oBAAoB,WAQnC;AAED,wBAAgB,qCAAqC,IAAI,cAAc,CAAC,oBAAoB,GAAG,iBAAiB,CAAC,CAYhH;AAGD,8BAAsB,uBAAwB,SAAQ,UAAW,YAAW,oBAAoB;IAW9F,SAAS,CAAC,QAAQ,CAAC,mBAAmB,EAAE,kBAAkB;IAC1D,SAAS,CAAC,QAAQ,CAAC,WAAW,EAAE,WAAW;IAC3C,SAAS,CAAC,QAAQ,CAAC,iBAAiB,EAAE,iBAAiB;IACvD,SAAS,CAAC,QAAQ,CAAC,cAAc,EAAE,cAAc;IACjD,SAAS,CAAC,QAAQ,CAAC,kBAAkB,EAAE,kBAAkB;IACzD,SAAS,CAAC,QAAQ,CAAC,eAAe,EAAE,eAAe;IACnD,SAAS,CAAC,QAAQ,CAAC,WAAW,EAAE,WAAW;IAhB5C,SAAiB,aAAa,EAAE,SAAS,CAAC;IAE1C,SAAS,CAAC,WAAW,UAAS;IAG9B,OAAO,CAAC,QAAQ,CAAC,2BAA2B,CAAuC;IACnF,QAAQ,CAAC,wBAAwB,2DAA0C;gBAIvD,mBAAmB,EAAE,kBAAkB,EACvC,WAAW,EAAE,WAAW,EACxB,iBAAiB,EAAE,iBAAiB,EACpC,cAAc,EAAE,cAAc,EAC9B,kBAAkB,EAAE,kBAAkB,EACtC,eAAe,EAAE,eAAe,EAChC,WAAW,EAAE,WAAW;IAO5C,OAAO,CAAC,aAAa,CAAgC;IACrD,IAAI,YAAY,IAAI,iBAAiB,GAAG,SAAS,CAEhD;IACD,IAAI,YAAY,CAAC,KAAK,EAAE,iBAAiB,GAAG,SAAS,EAKpD;IAID,QAAQ,CAAC,eAAe,CAAC,KAAK,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,YAAY,CAAC;IAIhE,iBAAiB,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI;IAQ3C;;;;;OAKG;IACG,mBAAmB,CACxB,WAAW,EAAE,MAAM,EACnB,UAAU,EAAE,MAAM,GAChB,OAAO,CAAC,gBAAgB,GAAG,oBAAoB,CAAC;YAgFrC,iBAAiB;YAWjB,oBAAoB;CAUlC;AAID;;;;;GAKG;AAEH,qBAAa,wBAAyB,SAAQ,uBAAwB,YAAW,iBAAiB;IAEhG,OAAO,CAAC,iBAAiB;gBAAjB,iBAAiB,EAAE,MAAM,EACpB,UAAU,EAAE,WAAW,EACjB,gBAAgB,EAAE,iBAAiB,EAClC,iBAAiB,EAAE,kBAAkB,EACzC,aAAa,EAAE,cAAc,EAC5B,cAAc,EAAE,eAAe,EACnC,UAAU,EAAE,WAAW;IAMrC,IAAI,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAGjC;IACD,IAAI,gBAAgB,IAAI,MAAM,CAE7B;IAEK,eAAe,IAAI,OAAO,CAAC,YAAY,CAAC;IAIxC,iBAAiB,IAAI,OAAO,CAAC;QAAE,MAAM,EAAE,IAAI,CAAA;KAAE,CAAC;CAIpD;AAMD;;;;GAIG;AACH,qBAAa,kCAAmC,SAAQ,uBAAwB,YAAW,iBAAiB;IAG1G,OAAO,CAAC,QAAQ,CAAC,WAAW;IAC5B,OAAO,CAAC,QAAQ,CAAC,cAAc;IAOR,SAAS,CAAC,QAAQ,CAAC,oBAAoB,EAAE,qBAAqB;gBARpE,WAAW,EAAE,MAAM,EACnB,cAAc,EAAE,MAAM,EAC1B,UAAU,EAAE,WAAW,EACjB,gBAAgB,EAAE,iBAAiB,EACtC,aAAa,EAAE,cAAc,EACzB,iBAAiB,EAAE,kBAAkB,EACxC,cAAc,EAAE,eAAe,EACnC,UAAU,EAAE,WAAW,EACM,oBAAoB,EAAE,qBAAqB;IAKhF,eAAe,CAAC,KAAK,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,YAAY,CAAC;IAavD,iBAAiB;;;;;;gBAQE,IAAI;;CAK7B"}

package/dist/src/_internal/platform/authentication/node/copilotTokenManager.js

@@ -0,0 +1,274 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CopilotTokenManagerFromGitHubToken = exports.FixedCopilotTokenManager = exports.BaseCopilotTokenManager = exports.tokenErrorString = void 0;
+exports.getStaticGitHubToken = getStaticGitHubToken;
+exports.getOrCreateTestingCopilotTokenManager = getOrCreateTestingCopilotTokenManager;
+const copilot_api_1 = require("@vscode/copilot-api");
+const event_1 = require("../../../util/vs/base/common/event");
+const lifecycle_1 = require("../../../util/vs/base/common/lifecycle");
+const descriptors_1 = require("../../../util/vs/platform/instantiation/common/descriptors");
+const configurationService_1 = require("../../configuration/common/configurationService");
+const capiClient_1 = require("../../endpoint/common/capiClient");
+const domainService_1 = require("../../endpoint/common/domainService");
+const envService_1 = require("../../env/common/envService");
+const githubService_1 = require("../../github/common/githubService");
+const nullOctokitServiceImpl_1 = require("../../github/common/nullOctokitServiceImpl");
+const logService_1 = require("../../log/common/logService");
+const fetcherService_1 = require("../../networking/common/fetcherService");
+const telemetry_1 = require("../../telemetry/common/telemetry");
+const telemetryData_1 = require("../../telemetry/common/telemetryData");
+const copilotToken_1 = require("../common/copilotToken");
+const copilotTokenManager_1 = require("../common/copilotTokenManager");
+exports.tokenErrorString = `Tests: either GITHUB_PAT or GITHUB_OAUTH_TOKEN must be set. Run "npm run get_token" to get one.`;
+function getStaticGitHubToken() {
+    if (process.env.GITHUB_PAT) {
+        return process.env.GITHUB_PAT;
+    }
+    if (process.env.GITHUB_OAUTH_TOKEN) {
+        return process.env.GITHUB_OAUTH_TOKEN;
+    }
+    throw new Error(exports.tokenErrorString);
+}
+function getOrCreateTestingCopilotTokenManager() {
+    let result;
+    if (process.env.GITHUB_PAT) {
+        result = new descriptors_1.SyncDescriptor(FixedCopilotTokenManager, [process.env.GITHUB_PAT]);
+    }
+    if (process.env.GITHUB_OAUTH_TOKEN) {
+        result = new descriptors_1.SyncDescriptor(CopilotTokenManagerFromGitHubToken, [process.env.GITHUB_OAUTH_TOKEN]);
+    }
+    if (!result) {
+        throw new Error(exports.tokenErrorString);
+    }
+    return result;
+}
+//TODO: Move this to common
+class BaseCopilotTokenManager extends lifecycle_1.Disposable {
+    //#endregion
+    constructor(_baseOctokitservice, _logService, _telemetryService, _domainService, _capiClientService, _fetcherService, _envService) {
+        super();
+        this._baseOctokitservice = _baseOctokitservice;
+        this._logService = _logService;
+        this._telemetryService = _telemetryService;
+        this._domainService = _domainService;
+        this._capiClientService = _capiClientService;
+        this._fetcherService = _fetcherService;
+        this._envService = _envService;
+        this._isDisposed = false;
+        //#region Events
+        this._copilotTokenRefreshEmitter = this._register(new event_1.Emitter());
+        this.onDidCopilotTokenRefresh = this._copilotTokenRefreshEmitter.event;
+        this._register((0, lifecycle_1.toDisposable)(() => this._isDisposed = true));
+    }
+    get copilotToken() {
+        return this._copilotToken;
+    }
+    set copilotToken(token) {
+        if (token !== this._copilotToken) {
+            this._copilotToken = token;
+            this._copilotTokenRefreshEmitter.fire();
+        }
+    }
+    //#endregion
+    //#region Public methods
+    resetCopilotToken(httpError) {
+        if (httpError !== undefined) {
+            this._telemetryService.sendGHTelemetryEvent('auth.reset_token_' + httpError);
+        }
+        this._logService.debug(`Resetting copilot token on HTTP error ${httpError || 'unknown'}`);
+        this.copilotToken = undefined;
+    }
+    /**
+     * Fetches a Copilot token from the GitHub token.
+     * @param githubToken A GitHub token to mint a Copilot token from.
+     * @returns A Copilot token info or an error.
+     * @todo this should be not be public, but it is for now to allow testing.
+     */
+    async authFromGitHubToken(githubToken, ghUsername) {
+        this._telemetryService.sendGHTelemetryEvent('auth.new_login');
+        const [response, userInfo] = await Promise.all([
+            this.fetchCopilotToken(githubToken),
+            this.fetchCopilotUserInfo(githubToken)
+        ]);
+        if (!response) {
+            this._logService.warn('Failed to get copilot token');
+            this._telemetryService.sendGHTelemetryErrorEvent('auth.request_failed');
+            return { kind: 'failure', reason: 'FailedToGetToken' };
+        }
+        // FIXME: Unverified type after inputting response
+        const tokenInfo = await (0, fetcherService_1.jsonVerboseError)(response);
+        if (!tokenInfo) {
+            this._logService.warn('Failed to get copilot token');
+            this._telemetryService.sendGHTelemetryErrorEvent('auth.request_read_failed');
+            return { kind: 'failure', reason: 'FailedToGetToken' };
+        }
+        if (response.status === 401) {
+            this._logService.warn('Failed to get copilot token due to 401 status');
+            this._telemetryService.sendGHTelemetryErrorEvent('auth.unknown_401');
+            return { kind: 'failure', reason: 'HTTP401' };
+        }
+        if (response.status === 403 && tokenInfo.message?.startsWith('API rate limit exceeded')) {
+            this._logService.warn('Failed to get copilot token due to exceeding API rate limit');
+            this._telemetryService.sendGHTelemetryErrorEvent('auth.rate_limited');
+            return { kind: 'failure', reason: 'RateLimited' };
+        }
+        if (!response.ok || !tokenInfo.token) {
+            this._logService.warn(`Invalid copilot token: missing token: ${response.status} ${response.statusText}`);
+            const data = telemetryData_1.TelemetryData.createAndMarkAsIssued({
+                status: response.status.toString(),
+                status_text: response.statusText,
+            });
+            this._telemetryService.sendGHTelemetryErrorEvent('auth.invalid_token', data.properties, data.measurements);
+            const error_details = tokenInfo.error_details;
+            return { kind: 'failure', reason: 'NotAuthorized', ...error_details };
+        }
+        const expires_at = tokenInfo.expires_at;
+        // some users have clocks adjusted ahead, expires_at will immediately be less than current clock time;
+        // adjust expires_at to the refresh time + a buffer to avoid expiring the token before the refresh can fire.
+        tokenInfo.expires_at = (0, copilotTokenManager_1.nowSeconds)() + tokenInfo.refresh_in + 60; // extra buffer to allow refresh to happen successfully
+        // extend the token envelope
+        const login = ghUsername ?? 'unknown';
+        let isVscodeTeamMember = false;
+        // VS Code team members are guaranteed to be a part of an internal org so we can check that first to minimize API calls
+        if ((0, copilotToken_1.containsInternalOrg)(tokenInfo.organization_list ?? [])) {
+            isVscodeTeamMember = !!(await this._baseOctokitservice.getTeamMembershipWithToken(githubService_1.VSCodeTeamId, githubToken, login));
+        }
+        const extendedInfo = {
+            ...tokenInfo,
+            copilot_plan: userInfo.copilot_plan,
+            quota_snapshots: userInfo.quota_snapshots,
+            quota_reset_date: userInfo.quota_reset_date,
+            username: login,
+            isVscodeTeamMember,
+        };
+        const telemetryData = telemetryData_1.TelemetryData.createAndMarkAsIssued({}, {
+            adjusted_expires_at: tokenInfo.expires_at,
+            expires_at: expires_at, // track original expires_at
+            current_time: (0, copilotTokenManager_1.nowSeconds)(),
+        });
+        this._telemetryService.sendGHTelemetryEvent('auth.new_token', telemetryData.properties, telemetryData.measurements);
+        return { kind: 'success', ...extendedInfo };
+    }
+    //#endregion
+    //#region Private methods
+    async fetchCopilotToken(githubToken) {
+        const options = {
+            headers: {
+                Authorization: `token ${githubToken}`,
+                'X-GitHub-Api-Version': '2025-04-01'
+            },
+            verifyJSONAndRetry: true,
+        };
+        return await this._capiClientService.makeRequest(options, { type: copilot_api_1.RequestType.CopilotToken });
+    }
+    async fetchCopilotUserInfo(githubToken) {
+        const response = await this._capiClientService.makeRequest({
+            headers: {
+                Authorization: `token ${githubToken}`,
+                'X-GitHub-Api-Version': '2025-04-01',
+            }
+        }, { type: copilot_api_1.RequestType.CopilotUserInfo });
+        const data = await response.json();
+        return data;
+    }
+}
+exports.BaseCopilotTokenManager = BaseCopilotTokenManager;
+//#region FixedCopilotTokenManager
+/**
+ * A `CopilotTokenManager` that always returns the same token.
+ * Mostly only useful for short periods, e.g. tests or single completion requests,
+ * as these tokens typically expire after a few hours.
+ * @todo Move this to a test layer
+ */
+let FixedCopilotTokenManager = class FixedCopilotTokenManager extends BaseCopilotTokenManager {
+    constructor(_completionsToken, logService, telemetryService, capiClientService, domainService, fetcherService, envService) {
+        super(new nullOctokitServiceImpl_1.NullBaseOctoKitService(capiClientService, fetcherService, logService, telemetryService), logService, telemetryService, domainService, capiClientService, fetcherService, envService);
+        this._completionsToken = _completionsToken;
+        this.copilotToken = { token: _completionsToken, expires_at: 0, refresh_in: 0, username: 'fixedTokenManager', isVscodeTeamMember: false, copilot_plan: 'unknown' };
+    }
+    set completionsToken(token) {
+        this._completionsToken = token;
+        this.copilotToken = { token, expires_at: 0, refresh_in: 0, username: 'fixedTokenManager', isVscodeTeamMember: false, copilot_plan: 'unknown' };
+    }
+    get completionsToken() {
+        return this._completionsToken;
+    }
+    async getCopilotToken() {
+        return new copilotToken_1.CopilotToken(this.copilotToken);
+    }
+    async checkCopilotToken() {
+        // assume it's valid
+        return { status: 'OK' };
+    }
+};
+exports.FixedCopilotTokenManager = FixedCopilotTokenManager;
+exports.FixedCopilotTokenManager = FixedCopilotTokenManager = __decorate([
+    __param(1, logService_1.ILogService),
+    __param(2, telemetry_1.ITelemetryService),
+    __param(3, capiClient_1.ICAPIClientService),
+    __param(4, domainService_1.IDomainService),
+    __param(5, fetcherService_1.IFetcherService),
+    __param(6, envService_1.IEnvService)
+], FixedCopilotTokenManager);
+//#endregion
+//#region CopilotTokenManagerFromGitHubToken
+/**
+ * Given a GitHub token, return a Copilot token, refreshing it as needed.
+ * The caller that initializes the object is responsible for checking telemetry consent before
+ * using the object.
+ */
+let CopilotTokenManagerFromGitHubToken = class CopilotTokenManagerFromGitHubToken extends BaseCopilotTokenManager {
+    constructor(githubToken, githubUsername, logService, telemetryService, domainService, capiClientService, fetcherService, envService, configurationService) {
+        super(new nullOctokitServiceImpl_1.NullBaseOctoKitService(capiClientService, fetcherService, logService, telemetryService), logService, telemetryService, domainService, capiClientService, fetcherService, envService);
+        this.githubToken = githubToken;
+        this.githubUsername = githubUsername;
+        this.configurationService = configurationService;
+    }
+    async getCopilotToken(force) {
+        if (!this.copilotToken || this.copilotToken.expires_at < (0, copilotTokenManager_1.nowSeconds)() - (60 * 5 /* 5min */) || force) {
+            const tokenResult = await this.authFromGitHubToken(this.githubToken, this.githubUsername);
+            if (tokenResult.kind === 'failure') {
+                throw Error(`Failed to get copilot token: ${tokenResult.reason.toString()} ${tokenResult.message ?? ''}`);
+            }
+            this.copilotToken = { ...tokenResult };
+        }
+        return new copilotToken_1.CopilotToken(this.copilotToken);
+    }
+    async checkCopilotToken() {
+        if (!this.copilotToken || this.copilotToken.expires_at < (0, copilotTokenManager_1.nowSeconds)()) {
+            const tokenResult = await this.authFromGitHubToken(this.githubToken, this.githubUsername);
+            if (tokenResult.kind === 'failure') {
+                return tokenResult;
+            }
+            this.copilotToken = { ...tokenResult };
+        }
+        const result = {
+            status: 'OK',
+        };
+        return result;
+    }
+};
+exports.CopilotTokenManagerFromGitHubToken = CopilotTokenManagerFromGitHubToken;
+exports.CopilotTokenManagerFromGitHubToken = CopilotTokenManagerFromGitHubToken = __decorate([
+    __param(2, logService_1.ILogService),
+    __param(3, telemetry_1.ITelemetryService),
+    __param(4, domainService_1.IDomainService),
+    __param(5, capiClient_1.ICAPIClientService),
+    __param(6, fetcherService_1.IFetcherService),
+    __param(7, envService_1.IEnvService),
+    __param(8, configurationService_1.IConfigurationService)
+], CopilotTokenManagerFromGitHubToken);
+//# sourceMappingURL=copilotTokenManager.js.map

package/dist/src/_internal/platform/authentication/node/copilotTokenManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"copilotTokenManager.js","sourceRoot":"","sources":["../../../../../../src/_internal/platform/authentication/node/copilotTokenManager.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;;;;;;;;;;AAqBhG,oDAQC;AAED,sFAYC;AAzCD,qDAAkD;AAClD,8DAA6D;AAC7D,sEAAkF;AAClF,4FAA4F;AAC5F,0FAAwF;AACxF,iEAAsE;AACtE,uEAAqE;AACrE,4DAA0D;AAC1D,qEAAqF;AACrF,uFAAoF;AACpF,4DAA0D;AAC1D,2EAAmH;AACnH,gEAAqE;AACrE,wEAAqE;AACrE,yDAA4I;AAC5I,uEAA0H;AAE7G,QAAA,gBAAgB,GAAG,iGAAiG,CAAC;AAElI,SAAgB,oBAAoB;IACnC,IAAI,OAAO,CAAC,GAAG,CAAC,UAAU,EAAE,CAAC;QAC5B,OAAO,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC;IAC/B,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,CAAC;QACpC,OAAO,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;IACvC,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,wBAAgB,CAAC,CAAC;AACnC,CAAC;AAED,SAAgB,qCAAqC;IACpD,IAAI,MAA4E,CAAC;IACjF,IAAI,OAAO,CAAC,GAAG,CAAC,UAAU,EAAE,CAAC;QAC5B,MAAM,GAAG,IAAI,4BAAc,CAAC,wBAAwB,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;IACjF,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,CAAC;QACpC,MAAM,GAAG,IAAI,4BAAc,CAAC,kCAAkC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC,CAAC;IACnG,CAAC;IACD,IAAI,CAAC,MAAM,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CAAC,wBAAgB,CAAC,CAAC;IACnC,CAAC;IACD,OAAO,MAAM,CAAC;AACf,CAAC;AAED,2BAA2B;AAC3B,MAAsB,uBAAwB,SAAQ,sBAAU;IAS/D,YAAY;IACZ,YACoB,mBAAuC,EACvC,WAAwB,EACxB,iBAAoC,EACpC,cAA8B,EAC9B,kBAAsC,EACtC,eAAgC,EAChC,WAAwB;QAE3C,KAAK,EAAE,CAAC;QARW,wBAAmB,GAAnB,mBAAmB,CAAoB;QACvC,gBAAW,GAAX,WAAW,CAAa;QACxB,sBAAiB,GAAjB,iBAAiB,CAAmB;QACpC,mBAAc,GAAd,cAAc,CAAgB;QAC9B,uBAAkB,GAAlB,kBAAkB,CAAoB;QACtC,oBAAe,GAAf,eAAe,CAAiB;QAChC,gBAAW,GAAX,WAAW,CAAa;QAdlC,gBAAW,GAAG,KAAK,CAAC;QAE9B,gBAAgB;QACC,gCAA2B,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,eAAO,EAAQ,CAAC,CAAC;QAC1E,6BAAwB,GAAG,IAAI,CAAC,2BAA2B,CAAC,KAAK,CAAC;QAa1E,IAAI,CAAC,SAAS,CAAC,IAAA,wBAAY,EAAC,GAAG,EAAE,CAAC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,CAAC,CAAC;IAC7D,CAAC;IAID,IAAI,YAAY;QACf,OAAO,IAAI,CAAC,aAAa,CAAC;IAC3B,CAAC;IACD,IAAI,YAAY,CAAC,KAAoC;QACpD,IAAI,KAAK,KAAK,IAAI,CAAC,aAAa,EAAE,CAAC;YAClC,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;YAC3B,IAAI,CAAC,2BAA2B,CAAC,IAAI,EAAE,CAAC;QACzC,CAAC;IACF,CAAC;IAMD,YAAY;IACZ,wBAAwB;IACxB,iBAAiB,CAAC,SAAkB;QACnC,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAC7B,IAAI,CAAC,iBAAiB,CAAC,oBAAoB,CAAC,mBAAmB,GAAG,SAAS,CAAC,CAAC;QAC9E,CAAC;QACD,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,yCAAyC,SAAS,IAAI,SAAS,EAAE,CAAC,CAAC;QAC1F,IAAI,CAAC,YAAY,GAAG,SAAS,CAAC;IAC/B,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,mBAAmB,CACxB,WAAmB,EACnB,UAAkB;QAElB,IAAI,CAAC,iBAAiB,CAAC,oBAAoB,CAAC,gBAAgB,CAAC,CAAC;QAC9D,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YAC9C,IAAI,CAAC,iBAAiB,CAAC,WAAW,CAAC;YACnC,IAAI,CAAC,oBAAoB,CAAC,WAAW,CAAC;SACtC,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,EAAE,CAAC;YACf,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YACrD,IAAI,CAAC,iBAAiB,CAAC,yBAAyB,CAAC,qBAAqB,CAAC,CAAC;YACxE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,kBAAkB,EAAE,CAAC;QACxD,CAAC;QAED,kDAAkD;QAClD,MAAM,SAAS,GAA0B,MAAM,IAAA,iCAAgB,EAAC,QAAQ,CAAC,CAAC;QAC1E,IAAI,CAAC,SAAS,EAAE,CAAC;YAChB,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YACrD,IAAI,CAAC,iBAAiB,CAAC,yBAAyB,CAAC,0BAA0B,CAAC,CAAC;YAC7E,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,kBAAkB,EAAE,CAAC;QACxD,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YAC7B,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,+CAA+C,CAAC,CAAC;YACvE,IAAI,CAAC,iBAAiB,CAAC,yBAAyB,CAAC,kBAAkB,CAAC,CAAC;YACrE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;QAC/C,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,IAAI,SAAS,CAAC,OAAO,EAAE,UAAU,CAAC,yBAAyB,CAAC,EAAE,CAAC;YACzF,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,6DAA6D,CAAC,CAAC;YACrF,IAAI,CAAC,iBAAiB,CAAC,yBAAyB,CAAC,mBAAmB,CAAC,CAAC;YACtE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;QACnD,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;YACtC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,yCAAyC,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;YACzG,MAAM,IAAI,GAAG,6BAAa,CAAC,qBAAqB,CAAC;gBAChD,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC,QAAQ,EAAE;gBAClC,WAAW,EAAE,QAAQ,CAAC,UAAU;aAChC,CAAC,CAAC;YACH,IAAI,CAAC,iBAAiB,CAAC,yBAAyB,CAAC,oBAAoB,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;YAC3G,MAAM,aAAa,GAAG,SAAS,CAAC,aAAa,CAAC;YAC9C,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,eAAe,EAAE,GAAG,aAAa,EAAE,CAAC;QACvE,CAAC;QAED,MAAM,UAAU,GAAG,SAAS,CAAC,UAAU,CAAC;QACxC,sGAAsG;QACtG,4GAA4G;QAC5G,SAAS,CAAC,UAAU,GAAG,IAAA,gCAAU,GAAE,GAAG,SAAS,CAAC,UAAU,GAAG,EAAE,CAAC,CAAC,uDAAuD;QAExH,4BAA4B;QAC5B,MAAM,KAAK,GAAG,UAAU,IAAI,SAAS,CAAC;QACtC,IAAI,kBAAkB,GAAG,KAAK,CAAC;QAC/B,uHAAuH;QACvH,IAAI,IAAA,kCAAmB,EAAC,SAAS,CAAC,iBAAiB,IAAI,EAAE,CAAC,EAAE,CAAC;YAC5D,kBAAkB,GAAG,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC,0BAA0B,CAAC,4BAAY,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC;QACtH,CAAC;QACD,MAAM,YAAY,GAAsB;YACvC,GAAG,SAAS;YACZ,YAAY,EAAE,QAAQ,CAAC,YAAY;YACnC,eAAe,EAAE,QAAQ,CAAC,eAAe;YACzC,gBAAgB,EAAE,QAAQ,CAAC,gBAAgB;YAC3C,QAAQ,EAAE,KAAK;YACf,kBAAkB;SAClB,CAAC;QACF,MAAM,aAAa,GAAG,6BAAa,CAAC,qBAAqB,CACxD,EAAE,EACF;YACC,mBAAmB,EAAE,SAAS,CAAC,UAAU;YACzC,UAAU,EAAE,UAAU,EAAE,4BAA4B;YACpD,YAAY,EAAE,IAAA,gCAAU,GAAE;SAC1B,CACD,CAAC;QAEF,IAAI,CAAC,iBAAiB,CAAC,oBAAoB,CAAC,gBAAgB,EAAE,aAAa,CAAC,UAAU,EAAE,aAAa,CAAC,YAAY,CAAC,CAAC;QAEpH,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,GAAG,YAAY,EAAE,CAAC;IAC7C,CAAC;IAED,YAAY;IAEZ,yBAAyB;IACjB,KAAK,CAAC,iBAAiB,CAAC,WAAmB;QAClD,MAAM,OAAO,GAAiB;YAC7B,OAAO,EAAE;gBACR,aAAa,EAAE,SAAS,WAAW,EAAE;gBACrC,sBAAsB,EAAE,YAAY;aACpC;YACD,kBAAkB,EAAE,IAAI;SACxB,CAAC;QACF,OAAO,MAAM,IAAI,CAAC,kBAAkB,CAAC,WAAW,CAAW,OAAO,EAAE,EAAE,IAAI,EAAE,yBAAW,CAAC,YAAY,EAAE,CAAC,CAAC;IACzG,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAAC,WAAmB;QACrD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,WAAW,CAAW;YACpE,OAAO,EAAE;gBACR,aAAa,EAAE,SAAS,WAAW,EAAE;gBACrC,sBAAsB,EAAE,YAAY;aACpC;SACD,EAAE,EAAE,IAAI,EAAE,yBAAW,CAAC,eAAe,EAAE,CAAC,CAAC;QAC1C,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnC,OAAO,IAAI,CAAC;IACb,CAAC;CACD;AA/JD,0DA+JC;AAED,kCAAkC;AAElC;;;;;GAKG;AAEI,IAAM,wBAAwB,GAA9B,MAAM,wBAAyB,SAAQ,uBAAuB;IACpE,YACS,iBAAyB,EACpB,UAAuB,EACjB,gBAAmC,EAClC,iBAAqC,EACzC,aAA6B,EAC5B,cAA+B,EACnC,UAAuB;QAEpC,KAAK,CAAC,IAAI,+CAAsB,CAAC,iBAAiB,EAAE,cAAc,EAAE,UAAU,EAAE,gBAAgB,CAAC,EAAE,UAAU,EAAE,gBAAgB,EAAE,aAAa,EAAE,iBAAiB,EAAE,cAAc,EAAE,UAAU,CAAC,CAAC;QARvL,sBAAiB,GAAjB,iBAAiB,CAAQ;QASjC,IAAI,CAAC,YAAY,GAAG,EAAE,KAAK,EAAE,iBAAiB,EAAE,UAAU,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,QAAQ,EAAE,mBAAmB,EAAE,kBAAkB,EAAE,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC;IACnK,CAAC;IAED,IAAI,gBAAgB,CAAC,KAAa;QACjC,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAC;QAC/B,IAAI,CAAC,YAAY,GAAG,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,QAAQ,EAAE,mBAAmB,EAAE,kBAAkB,EAAE,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC;IAChJ,CAAC;IACD,IAAI,gBAAgB;QACnB,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,eAAe;QACpB,OAAO,IAAI,2BAAY,CAAC,IAAI,CAAC,YAAa,CAAC,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,iBAAiB;QACtB,oBAAoB;QACpB,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IACzB,CAAC;CACD,CAAA;AA9BY,4DAAwB;mCAAxB,wBAAwB;IAGlC,WAAA,wBAAW,CAAA;IACX,WAAA,6BAAiB,CAAA;IACjB,WAAA,+BAAkB,CAAA;IAClB,WAAA,8BAAc,CAAA;IACd,WAAA,gCAAe,CAAA;IACf,WAAA,wBAAW,CAAA;GARD,wBAAwB,CA8BpC;AAED,YAAY;AAEZ,4CAA4C;AAE5C;;;;GAIG;AACI,IAAM,kCAAkC,GAAxC,MAAM,kCAAmC,SAAQ,uBAAuB;IAE9E,YACkB,WAAmB,EACnB,cAAsB,EAC1B,UAAuB,EACjB,gBAAmC,EACtC,aAA6B,EACzB,iBAAqC,EACxC,cAA+B,EACnC,UAAuB,EACM,oBAA2C;QAErF,KAAK,CAAC,IAAI,+CAAsB,CAAC,iBAAiB,EAAE,cAAc,EAAE,UAAU,EAAE,gBAAgB,CAAC,EAAE,UAAU,EAAE,gBAAgB,EAAE,aAAa,EAAE,iBAAiB,EAAE,cAAc,EAAE,UAAU,CAAC,CAAC;QAV9K,gBAAW,GAAX,WAAW,CAAQ;QACnB,mBAAc,GAAd,cAAc,CAAQ;QAOG,yBAAoB,GAApB,oBAAoB,CAAuB;IAGtF,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,KAAe;QACpC,IAAI,CAAC,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,YAAY,CAAC,UAAU,GAAG,IAAA,gCAAU,GAAE,GAAG,CAAC,EAAE,GAAG,CAAC,CAAC,UAAU,CAAC,IAAI,KAAK,EAAE,CAAC;YACtG,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;YAC1F,IAAI,WAAW,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;gBACpC,MAAM,KAAK,CACV,gCAAgC,WAAW,CAAC,MAAM,CAAC,QAAQ,EAAE,IAAI,WAAW,CAAC,OAAO,IAAI,EAAE,EAAE,CAC5F,CAAC;YACH,CAAC;YACD,IAAI,CAAC,YAAY,GAAG,EAAE,GAAG,WAAW,EAAE,CAAC;QACxC,CAAC;QACD,OAAO,IAAI,2BAAY,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,iBAAiB;QACtB,IAAI,CAAC,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,YAAY,CAAC,UAAU,GAAG,IAAA,gCAAU,GAAE,EAAE,CAAC;YACvE,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;YAC1F,IAAI,WAAW,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;gBACpC,OAAO,WAAW,CAAC;YACpB,CAAC;YACD,IAAI,CAAC,YAAY,GAAG,EAAE,GAAG,WAAW,EAAE,CAAC;QACxC,CAAC;QACD,MAAM,MAAM,GAAqB;YAChC,MAAM,EAAE,IAAI;SACZ,CAAC;QACF,OAAO,MAAM,CAAC;IACf,CAAC;CACD,CAAA;AA1CY,gFAAkC;6CAAlC,kCAAkC;IAK5C,WAAA,wBAAW,CAAA;IACX,WAAA,6BAAiB,CAAA;IACjB,WAAA,8BAAc,CAAA;IACd,WAAA,+BAAkB,CAAA;IAClB,WAAA,gCAAe,CAAA;IACf,WAAA,wBAAW,CAAA;IACX,WAAA,4CAAqB,CAAA;GAXX,kCAAkC,CA0C9C"}

package/dist/src/_internal/platform/chat/common/chatMLFetcher.d.ts

@@ -0,0 +1,48 @@
+import type { CancellationToken } from 'vscode';
+import { AsyncIterableObject } from '../../../util/vs/base/common/async';
+import { Event } from '../../../util/vs/base/common/event';
+import { FinishedCallback, IResponseDelta, OptionalChatRequestParams } from '../../networking/common/fetch';
+import { IChatEndpoint, IMakeChatRequestOptions } from '../../networking/common/networking';
+import { ChatResponse, ChatResponses } from './commonTypes';
+export interface Source {
+    readonly extensionId?: string;
+}
+export interface IResponsePart {
+    readonly text: string;
+    readonly delta: IResponseDelta;
+}
+export interface IFetchMLOptions extends IMakeChatRequestOptions {
+    endpoint: IChatEndpoint;
+    requestOptions: OptionalChatRequestParams;
+}
+export declare const IChatMLFetcher: import("../../../util/common/services").ServiceIdentifier<IChatMLFetcher>;
+export interface IChatMLFetcher {
+    readonly _serviceBrand: undefined;
+    readonly onDidMakeChatMLRequest: Event<{
+        readonly model: string;
+        readonly source?: Source;
+        readonly tokenCount?: number;
+    }>;
+    fetchOne(options: IFetchMLOptions, token: CancellationToken): Promise<ChatResponse>;
+    /**
+     * Note: the returned array of strings may be less than `n` (e.g., in case there were errors during streaming)
+     */
+    fetchMany(options: IFetchMLOptions, token: CancellationToken): Promise<ChatResponses>;
+}
+export declare class FetchStreamSource {
+    private _stream;
+    private _paused?;
+    private _seenAnnotationTypes;
+    get stream(): AsyncIterableObject<IResponsePart>;
+    constructor();
+    pause(): void;
+    unpause(): void;
+    update(text: string, delta: IResponseDelta): void;
+    resolve(): void;
+}
+export declare class FetchStreamRecorder {
+    readonly callback: FinishedCallback;
+    readonly deltas: IResponseDelta[];
+    constructor(callback: FinishedCallback | undefined);
+}
+//# sourceMappingURL=chatMLFetcher.d.ts.map

package/dist/src/_internal/platform/chat/common/chatMLFetcher.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatMLFetcher.d.ts","sourceRoot":"","sources":["../../../../../../src/_internal/platform/chat/common/chatMLFetcher.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,QAAQ,CAAC;AAEhD,OAAO,EAAE,mBAAmB,EAAuB,MAAM,oCAAoC,CAAC;AAC9F,OAAO,EAAE,KAAK,EAAE,MAAM,oCAAoC,CAAC;AAC3D,OAAO,EAAE,gBAAgB,EAAE,cAAc,EAAE,yBAAyB,EAAE,MAAM,+BAA+B,CAAC;AAC5G,OAAO,EAAE,aAAa,EAAE,uBAAuB,EAAE,MAAM,oCAAoC,CAAC;AAC5F,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAE5D,MAAM,WAAW,MAAM;IACtB,QAAQ,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC;CAC9B;AAED,MAAM,WAAW,aAAa;IAC7B,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,KAAK,EAAE,cAAc,CAAC;CAC/B;AAED,MAAM,WAAW,eAAgB,SAAQ,uBAAuB;IAC/D,QAAQ,EAAE,aAAa,CAAC;IACxB,cAAc,EAAE,yBAAyB,CAAC;CAC1C;AAGD,eAAO,MAAM,cAAc,2EAA4D,CAAC;AAExF,MAAM,WAAW,cAAc;IAE9B,QAAQ,CAAC,aAAa,EAAE,SAAS,CAAC;IAElC,QAAQ,CAAC,sBAAsB,EAAE,KAAK,CAAC;QAAE,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAE3H,QAAQ,CAAC,OAAO,EAAE,eAAe,EAAE,KAAK,EAAE,iBAAiB,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;IAEpF;;OAEG;IACH,SAAS,CAAC,OAAO,EAAE,eAAe,EAAE,KAAK,EAAE,iBAAiB,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC;CACtF;AAED,qBAAa,iBAAiB;IAE7B,OAAO,CAAC,OAAO,CAA4C;IAC3D,OAAO,CAAC,OAAO,CAAC,CAAgC;IAGhD,OAAO,CAAC,oBAAoB,CAAqB;IAEjD,IAAW,MAAM,IAAI,mBAAmB,CAAC,aAAa,CAAC,CAEtD;;IAID,KAAK;IAIL,OAAO;IAgBP,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,cAAc,GAAG,IAAI;IAsBjD,OAAO,IAAI,IAAI;CAQf;AAED,qBAAa,mBAAmB;IAC/B,SAAgB,QAAQ,EAAE,gBAAgB,CAAC;IAC3C,SAAgB,MAAM,EAAE,cAAc,EAAE,CAAM;gBAG7C,QAAQ,EAAE,gBAAgB,GAAG,SAAS;CAQvC"}

package/dist/src/_internal/platform/chat/common/chatMLFetcher.js

@@ -0,0 +1,77 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FetchStreamRecorder = exports.FetchStreamSource = exports.IChatMLFetcher = void 0;
+const services_1 = require("../../../util/common/services");
+const async_1 = require("../../../util/vs/base/common/async");
+exports.IChatMLFetcher = (0, services_1.createServiceIdentifier)('IChatMLFetcher');
+class FetchStreamSource {
+    get stream() {
+        return this._stream.asyncIterable;
+    }
+    constructor() {
+        this._stream = new async_1.AsyncIterableSource();
+        // This means that we will only show one instance of each annotation type, but the IDs are not correct and there is no other way
+        this._seenAnnotationTypes = new Set();
+    }
+    pause() {
+        this._paused ??= [];
+    }
+    unpause() {
+        const toEmit = this._paused;
+        if (!toEmit) {
+            return;
+        }
+        this._paused = undefined;
+        for (const part of toEmit) {
+            if (part) {
+                this.update(part.text, part.delta);
+            }
+            else {
+                this.resolve();
+            }
+        }
+    }
+    update(text, delta) {
+        if (this._paused) {
+            this._paused.push({ text, delta });
+            return;
+        }
+        if (delta.codeVulnAnnotations) {
+            // We can only display vulnerabilities inside codeblocks, and it's ok to discard annotations that fell outside of them
+            const numTripleBackticks = text.match(/(^|\n)```/g)?.length ?? 0;
+            const insideCodeblock = numTripleBackticks % 2 === 1;
+            if (!insideCodeblock || text.match(/(^|\n)```\w*\s*$/)) { // Not inside a codeblock, or right on the start triple-backtick of a codeblock
+                delta.codeVulnAnnotations = undefined;
+            }
+        }
+        if (delta.codeVulnAnnotations) {
+            delta.codeVulnAnnotations = delta.codeVulnAnnotations.filter(annotation => !this._seenAnnotationTypes.has(annotation.details.type));
+            delta.codeVulnAnnotations.forEach(annotation => this._seenAnnotationTypes.add(annotation.details.type));
+        }
+        this._stream.emitOne({ text, delta });
+    }
+    resolve() {
+        if (this._paused) {
+            this._paused.push(undefined);
+            return;
+        }
+        this._stream.resolve();
+    }
+}
+exports.FetchStreamSource = FetchStreamSource;
+class FetchStreamRecorder {
+    constructor(callback) {
+        this.deltas = [];
+        this.callback = async (text, index, delta) => {
+            const result = callback ? await callback(text, index, delta) : undefined;
+            this.deltas.push(delta);
+            return result;
+        };
+    }
+}
+exports.FetchStreamRecorder = FetchStreamRecorder;
+//# sourceMappingURL=chatMLFetcher.js.map

package/dist/src/_internal/platform/chat/common/chatMLFetcher.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatMLFetcher.js","sourceRoot":"","sources":["../../../../../../src/_internal/platform/chat/common/chatMLFetcher.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAGhG,4DAAwE;AACxE,8DAA8F;AAqBjF,QAAA,cAAc,GAAG,IAAA,kCAAuB,EAAiB,gBAAgB,CAAC,CAAC;AAgBxF,MAAa,iBAAiB;IAQ7B,IAAW,MAAM;QAChB,OAAO,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC;IACnC,CAAC;IAED;QAVQ,YAAO,GAAG,IAAI,2BAAmB,EAAiB,CAAC;QAG3D,gIAAgI;QACxH,yBAAoB,GAAG,IAAI,GAAG,EAAU,CAAC;IAMjC,CAAC;IAEjB,KAAK;QACJ,IAAI,CAAC,OAAO,KAAK,EAAE,CAAC;IACrB,CAAC;IAED,OAAO;QACN,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,IAAI,CAAC,MAAM,EAAE,CAAC;YACb,OAAO;QACR,CAAC;QAED,IAAI,CAAC,OAAO,GAAG,SAAS,CAAC;QACzB,KAAK,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;YAC3B,IAAI,IAAI,EAAE,CAAC;gBACV,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;YACpC,CAAC;iBAAM,CAAC;gBACP,IAAI,CAAC,OAAO,EAAE,CAAC;YAChB,CAAC;QACF,CAAC;IACF,CAAC;IAED,MAAM,CAAC,IAAY,EAAE,KAAqB;QACzC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;YACnC,OAAO;QACR,CAAC;QAED,IAAI,KAAK,CAAC,mBAAmB,EAAE,CAAC;YAC/B,sHAAsH;YACtH,MAAM,kBAAkB,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,MAAM,IAAI,CAAC,CAAC;YACjE,MAAM,eAAe,GAAG,kBAAkB,GAAG,CAAC,KAAK,CAAC,CAAC;YACrD,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC,+EAA+E;gBACxI,KAAK,CAAC,mBAAmB,GAAG,SAAS,CAAC;YACvC,CAAC;QACF,CAAC;QAED,IAAI,KAAK,CAAC,mBAAmB,EAAE,CAAC;YAC/B,KAAK,CAAC,mBAAmB,GAAG,KAAK,CAAC,mBAAmB,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;YACpI,KAAK,CAAC,mBAAmB,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;QACzG,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IACvC,CAAC;IAED,OAAO;QACN,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC7B,OAAO;QACR,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;IACxB,CAAC;CACD;AAhED,8CAgEC;AAED,MAAa,mBAAmB;IAI/B,YACC,QAAsC;QAHvB,WAAM,GAAqB,EAAE,CAAC;QAK7C,IAAI,CAAC,QAAQ,GAAG,KAAK,EAAE,IAAY,EAAE,KAAa,EAAE,KAAqB,EAA+B,EAAE;YACzG,MAAM,MAAM,GAAG,QAAQ,CAAC,CAAC,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACzE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACxB,OAAO,MAAM,CAAC;QACf,CAAC,CAAC;IACH,CAAC;CACD;AAbD,kDAaC"}

package/dist/src/_internal/platform/chat/common/chatQuotaService.d.ts

@@ -0,0 +1,54 @@
+import { IHeaders } from '../../networking/common/fetcherService';
+/**
+ * This is the quota info we get from the `copilot_internal/user` endpoint.
+ * It is accessed via the copilot token object
+ */
+export interface CopilotUserQuotaInfo {
+    quota_reset_date?: string;
+    quota_snapshots?: {
+        chat: {
+            quota_id: string;
+            entitlement: number;
+            remaining: number;
+            unlimited: boolean;
+            overage_count: number;
+            overage_permitted: boolean;
+            percent_remaining: number;
+        };
+        completions: {
+            quota_id: string;
+            entitlement: number;
+            remaining: number;
+            unlimited: boolean;
+            overage_count: number;
+            overage_permitted: boolean;
+            percent_remaining: number;
+        };
+        premium_interactions: {
+            quota_id: string;
+            entitlement: number;
+            remaining: number;
+            unlimited: boolean;
+            overage_count: number;
+            overage_permitted: boolean;
+            percent_remaining: number;
+        };
+    };
+}
+export interface IChatQuota {
+    quota: number;
+    used: number;
+    unlimited: boolean;
+    overageUsed: number;
+    overageEnabled: boolean;
+    resetDate: Date;
+}
+export interface IChatQuotaService {
+    readonly _serviceBrand: undefined;
+    quotaExhausted: boolean;
+    overagesEnabled: boolean;
+    processQuotaHeaders(headers: IHeaders): void;
+    clearQuota(): void;
+}
+export declare const IChatQuotaService: import("../../../util/common/services").ServiceIdentifier<IChatQuotaService>;
+//# sourceMappingURL=chatQuotaService.d.ts.map

package/dist/src/_internal/platform/chat/common/chatQuotaService.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatQuotaService.d.ts","sourceRoot":"","sources":["../../../../../../src/_internal/platform/chat/common/chatQuotaService.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,QAAQ,EAAE,MAAM,wCAAwC,CAAC;AAElE;;;GAGG;AACH,MAAM,WAAW,oBAAoB;IACpC,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,eAAe,CAAC,EAAE;QACjB,IAAI,EAAE;YACL,QAAQ,EAAE,MAAM,CAAC;YACjB,WAAW,EAAE,MAAM,CAAC;YACpB,SAAS,EAAE,MAAM,CAAC;YAClB,SAAS,EAAE,OAAO,CAAC;YACnB,aAAa,EAAE,MAAM,CAAC;YACtB,iBAAiB,EAAE,OAAO,CAAC;YAC3B,iBAAiB,EAAE,MAAM,CAAC;SAC1B,CAAC;QACF,WAAW,EAAE;YACZ,QAAQ,EAAE,MAAM,CAAC;YACjB,WAAW,EAAE,MAAM,CAAC;YACpB,SAAS,EAAE,MAAM,CAAC;YAClB,SAAS,EAAE,OAAO,CAAC;YACnB,aAAa,EAAE,MAAM,CAAC;YACtB,iBAAiB,EAAE,OAAO,CAAC;YAC3B,iBAAiB,EAAE,MAAM,CAAC;SAC1B,CAAC;QACF,oBAAoB,EAAE;YACrB,QAAQ,EAAE,MAAM,CAAC;YACjB,WAAW,EAAE,MAAM,CAAC;YACpB,SAAS,EAAE,MAAM,CAAC;YAClB,SAAS,EAAE,OAAO,CAAC;YACnB,aAAa,EAAE,MAAM,CAAC;YACtB,iBAAiB,EAAE,OAAO,CAAC;YAC3B,iBAAiB,EAAE,MAAM,CAAC;SAC1B,CAAC;KACF,CAAC;CACF;AAED,MAAM,WAAW,UAAU;IAC1B,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,OAAO,CAAC;IACnB,WAAW,EAAE,MAAM,CAAC;IACpB,cAAc,EAAE,OAAO,CAAC;IACxB,SAAS,EAAE,IAAI,CAAC;CAChB;AAED,MAAM,WAAW,iBAAiB;IACjC,QAAQ,CAAC,aAAa,EAAE,SAAS,CAAC;IAClC,cAAc,EAAE,OAAO,CAAC;IACxB,eAAe,EAAE,OAAO,CAAC;IACzB,mBAAmB,CAAC,OAAO,EAAE,QAAQ,GAAG,IAAI,CAAC;IAC7C,UAAU,IAAI,IAAI,CAAC;CACnB;AAED,eAAO,MAAM,iBAAiB,8EAAkE,CAAC"}

package/dist/src/_internal/platform/chat/common/chatQuotaService.js

@@ -0,0 +1,10 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.IChatQuotaService = void 0;
+const services_1 = require("../../../util/common/services");
+exports.IChatQuotaService = (0, services_1.createServiceIdentifier)('IChatQuotaService');
+//# sourceMappingURL=chatQuotaService.js.map

package/dist/src/_internal/platform/chat/common/chatQuotaService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatQuotaService.js","sourceRoot":"","sources":["../../../../../../src/_internal/platform/chat/common/chatQuotaService.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,4DAAwE;AAyD3D,QAAA,iBAAiB,GAAG,IAAA,kCAAuB,EAAoB,mBAAmB,CAAC,CAAC"}

package/dist/src/_internal/platform/chat/common/chatQuotaServiceImpl.d.ts

@@ -0,0 +1,16 @@
+import { Disposable } from '../../../util/vs/base/common/lifecycle';
+import { IAuthenticationService } from '../../authentication/common/authentication';
+import { IHeaders } from '../../networking/common/fetcherService';
+import { IChatQuotaService } from './chatQuotaService';
+export declare class ChatQuotaService extends Disposable implements IChatQuotaService {
+    private readonly _authService;
+    readonly _serviceBrand: undefined;
+    private _quotaInfo;
+    constructor(_authService: IAuthenticationService);
+    get quotaExhausted(): boolean;
+    get overagesEnabled(): boolean;
+    clearQuota(): void;
+    processQuotaHeaders(headers: IHeaders): void;
+    private processUserInfoQuotaSnapshot;
+}
+//# sourceMappingURL=chatQuotaServiceImpl.d.ts.map

package/dist/src/_internal/platform/chat/common/chatQuotaServiceImpl.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatQuotaServiceImpl.d.ts","sourceRoot":"","sources":["../../../../../../src/_internal/platform/chat/common/chatQuotaServiceImpl.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,UAAU,EAAE,MAAM,wCAAwC,CAAC;AACpE,OAAO,EAAE,sBAAsB,EAAE,MAAM,4CAA4C,CAAC;AACpF,OAAO,EAAE,QAAQ,EAAE,MAAM,wCAAwC,CAAC;AAClE,OAAO,EAAoC,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAEzF,qBAAa,gBAAiB,SAAQ,UAAW,YAAW,iBAAiB;IAIxC,OAAO,CAAC,QAAQ,CAAC,YAAY;IAHjE,SAAiB,aAAa,EAAE,SAAS,CAAC;IAC1C,OAAO,CAAC,UAAU,CAAyB;gBAEU,YAAY,EAAE,sBAAsB;IAOzF,IAAI,cAAc,IAAI,OAAO,CAK5B;IAED,IAAI,eAAe,IAAI,OAAO,CAK7B;IAED,UAAU,IAAI,IAAI;IAIlB,mBAAmB,CAAC,OAAO,EAAE,QAAQ,GAAG,IAAI;IA2C5C,OAAO,CAAC,4BAA4B;CAapC"}