@codingame/monaco-vscode-extensions-service-override 37.1.0 → 37.3.0

package/vscode/src/vs/workbench/api/browser/mainThreadLanguageModels.js

@@ -1,356 +0,0 @@
-
-import { __decorate, __param } from '@codingame/monaco-vscode-api/external/tslib/tslib.es6';
-import { DeferredPromise, AsyncIterableSource } from '@codingame/monaco-vscode-api/vscode/vs/base/common/async';
-import { VSBuffer } from '@codingame/monaco-vscode-api/vscode/vs/base/common/buffer';
-import { CancellationTokenSource } from '@codingame/monaco-vscode-api/vscode/vs/base/common/cancellation';
-import { toErrorMessage } from '@codingame/monaco-vscode-api/vscode/vs/base/common/errorMessage';
-import { CancellationError, transformErrorFromSerialization, transformErrorForSerialization } from '@codingame/monaco-vscode-api/vscode/vs/base/common/errors';
-import { Emitter, Event } from '@codingame/monaco-vscode-api/vscode/vs/base/common/event';
-import { equalSets } from '@codingame/monaco-vscode-api/vscode/vs/base/common/collections';
-import { Disposable, DisposableStore, DisposableMap, toDisposable } 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 { ExtensionIdentifier } from '@codingame/monaco-vscode-api/vscode/vs/platform/extensions/common/extensions';
-import { ILogService } from '@codingame/monaco-vscode-api/vscode/vs/platform/log/common/log.service';
-import { IProductService } from '@codingame/monaco-vscode-api/vscode/vs/platform/product/common/productService.service';
-import { resizeImage } from '@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/browser/chatImageUtils';
-import { ILanguageModelIgnoredFilesService } from '@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/ignoredFiles.service';
-import { ILanguageModelsService } from '@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/languageModels.service';
-import { IAuthenticationAccessService } from '@codingame/monaco-vscode-api/vscode/vs/workbench/services/authentication/browser/authenticationAccessService.service';
-import { INTERNAL_AUTH_PROVIDER_PREFIX } from '@codingame/monaco-vscode-api/vscode/vs/workbench/services/authentication/common/authentication';
-import { IAuthenticationService } from '@codingame/monaco-vscode-api/vscode/vs/workbench/services/authentication/common/authentication.service';
-import { extHostNamedCustomer } from '../../services/extensions/common/extHostCustomers.js';
-import { IExtensionService } from '@codingame/monaco-vscode-api/vscode/vs/workbench/services/extensions/common/extensions.service';
-import { SerializableObjectWithBuffers } from '@codingame/monaco-vscode-api/vscode/vs/workbench/services/extensions/common/proxyIdentifier';
-import { ExtHostContext, MainContext } from '@codingame/monaco-vscode-api/vscode/vs/workbench/api/common/extHost.protocol';
-import { LanguageModelError } from '@codingame/monaco-vscode-api/vscode/vs/workbench/api/common/extHostTypes';
-
-class RequestCancellationTokenSource extends Disposable {
-    constructor(parent, onCancellationRequested) {
-        super();
-        this._source = this._register(( new CancellationTokenSource(parent)));
-        if (onCancellationRequested) {
-            this._register(this._source.token.onCancellationRequested(onCancellationRequested));
-        }
-    }
-    get token() {
-        return this._source.token;
-    }
-    cancel() {
-        this._source.cancel();
-    }
-}
-let MainThreadLanguageModels = class MainThreadLanguageModels {
-    constructor(
-        extHostContext,
-        _chatProviderService,
-        _logService,
-        _productService,
-        _authenticationService,
-        _authenticationAccessService,
-        _extensionService,
-        _ignoredFilesService
-    ) {
-        this._chatProviderService = _chatProviderService;
-        this._logService = _logService;
-        this._productService = _productService;
-        this._authenticationService = _authenticationService;
-        this._authenticationAccessService = _authenticationAccessService;
-        this._extensionService = _extensionService;
-        this._ignoredFilesService = _ignoredFilesService;
-        this._store = ( new DisposableStore());
-        this._providerRegistrations = ( new DisposableMap());
-        this._lmProviderChange = ( new Emitter());
-        this._pendingProgress = ( new Map());
-        this._pendingCancelCTS = ( new DisposableMap());
-        this._ignoredFileProviderRegistrations = ( new DisposableMap());
-        this._proxy = ( extHostContext.getProxy(ExtHostContext.ExtHostChatProvider));
-        let lastModelIds = ( new Set(this._chatProviderService.getLanguageModelIds()));
-        this._store.add(this._chatProviderService.onDidChangeLanguageModels(() => {
-            const currentModelIds = ( new Set(this._chatProviderService.getLanguageModelIds()));
-            if (equalSets(lastModelIds, currentModelIds)) {
-                return;
-            }
-            lastModelIds = currentModelIds;
-            this._proxy.$onChatModelsChange();
-        }));
-    }
-    dispose() {
-        this._lmProviderChange.dispose();
-        this._providerRegistrations.dispose();
-        this._pendingProgress.clear();
-        this._pendingCancelCTS.dispose();
-        this._ignoredFileProviderRegistrations.dispose();
-        this._store.dispose();
-    }
-    $registerLanguageModelProvider(vendor) {
-        const disposables = ( new DisposableStore());
-        try {
-            disposables.add(this._chatProviderService.registerLanguageModelProvider(vendor, {
-                onDidChange: Event.filter(this._lmProviderChange.event, e => e.vendor === vendor, disposables),
-                provideLanguageModelChatInfo: async (options, token) => {
-                    const modelsAndIdentifiers = await this._proxy.$provideLanguageModelChatInfo(vendor, options, token);
-                    const copilotExtensionId = this._productService.defaultChatAgent?.chatExtensionId;
-                    return ( modelsAndIdentifiers.map(m => {
-                        if (m.metadata.auth) {
-                            disposables.add(
-                                this._registerAuthenticationProvider(m.metadata.extension, m.metadata.auth)
-                            );
-                        }
-                        if (m.metadata.isBYOK !== undefined) {
-                            return m;
-                        }
-                        const isBuiltinCopilot = !!copilotExtensionId && ExtensionIdentifier.equals(m.metadata.extension, copilotExtensionId);
-                        return {
-                            ...m,
-                            metadata: {
-                                ...m.metadata,
-                                isBYOK: !isBuiltinCopilot
-                            }
-                        };
-                    }));
-                },
-                sendChatRequest: async (modelId, messages, from, options, token) => {
-                    const requestId = (Math.random() * 1e6) | 0;
-                    const defer = ( new DeferredPromise());
-                    defer.p.catch(() => {});
-                    const stream = new AsyncIterableSource();
-                    try {
-                        this._pendingProgress.set(requestId, {
-                            defer,
-                            stream
-                        });
-                        const cts = ( new RequestCancellationTokenSource(token, () => {
-                            this._proxy.$cancelLanguageModelChatRequest(requestId);
-                        }));
-                        this._pendingCancelCTS.set(requestId, cts);
-                        await Promise.all(( messages.flatMap(msg => msg.content).filter(part => part.type === "image_url").map(async part => {
-                            part.value.data = VSBuffer.wrap(await resizeImage(part.value.data.buffer));
-                        })));
-                        if (token.isCancellationRequested) {
-                            this._pendingProgress.delete(requestId);
-                            this._pendingCancelCTS.deleteAndDispose(requestId);
-                            const err = ( new CancellationError());
-                            stream.reject(err);
-                            defer.error(err);
-                            return {
-                                result: defer.p,
-                                stream: stream.asyncIterable
-                            };
-                        }
-                        await this._proxy.$startChatRequest(modelId, requestId, from, ( new SerializableObjectWithBuffers(messages)), options, cts.token);
-                    } catch (err) {
-                        this._pendingProgress.delete(requestId);
-                        this._pendingCancelCTS.deleteAndDispose(requestId);
-                        throw err;
-                    }
-                    return {
-                        result: defer.p,
-                        stream: stream.asyncIterable
-                    };
-                },
-                provideTokenCount: (modelId, str, token) => {
-                    return this._proxy.$provideTokenLength(modelId, str, token);
-                }
-            }));
-            this._providerRegistrations.set(vendor, disposables);
-        } catch (err) {
-            disposables.dispose();
-            throw err;
-        }
-    }
-    $onLMProviderChange(vendor) {
-        this._lmProviderChange.fire({
-            vendor
-        });
-    }
-    async $reportResponsePart(requestId, chunk) {
-        const data = this._pendingProgress.get(requestId);
-        this._logService.trace("[LM] report response PART", Boolean(data), requestId, chunk);
-        if (data) {
-            data.stream.emitOne(chunk.value);
-        }
-    }
-    async $reportResponseDone(requestId, err) {
-        const data = this._pendingProgress.get(requestId);
-        this._logService.trace("[LM] report response DONE", Boolean(data), requestId, err);
-        if (data) {
-            this._pendingProgress.delete(requestId);
-            this._pendingCancelCTS.deleteAndDispose(requestId);
-            if (err) {
-                const error = LanguageModelError.tryDeserialize(err) ?? transformErrorFromSerialization(err);
-                data.stream.reject(error);
-                data.defer.error(error);
-            } else {
-                data.stream.resolve();
-                data.defer.complete(undefined);
-            }
-        }
-    }
-    $unregisterProvider(vendor) {
-        this._providerRegistrations.deleteAndDispose(vendor);
-    }
-    $cancelLanguageModelChatRequest(requestId) {
-        this._pendingCancelCTS.get(requestId)?.cancel();
-    }
-    $selectChatModels(selector) {
-        return this._chatProviderService.selectLanguageModels(selector);
-    }
-    async $tryStartChatRequest(extension, modelIdentifier, requestId, messages, options, token) {
-        this._logService.trace("[CHAT] request STARTED", extension.value, requestId);
-        const cts = ( new RequestCancellationTokenSource(token));
-        this._pendingCancelCTS.set(requestId, cts);
-        let response;
-        try {
-            response = await this._chatProviderService.sendChatRequest(modelIdentifier, extension, messages.value, options, cts.token);
-        } catch (err) {
-            this._logService.error("[CHAT] request FAILED", extension.value, requestId, err);
-            this._pendingCancelCTS.deleteAndDispose(requestId);
-            throw err;
-        }
-        const streaming = (async () => {
-            try {
-                for await (const part of response.stream) {
-                    this._logService.trace("[CHAT] request PART", extension.value, requestId, part);
-                    await this._proxy.$acceptResponsePart(requestId, ( new SerializableObjectWithBuffers(part)));
-                }
-                this._logService.trace("[CHAT] request DONE", extension.value, requestId);
-            } catch (err) {
-                this._logService.error(
-                    "[CHAT] extension request ERRORED in STREAM",
-                    toErrorMessage(err, true),
-                    extension.value,
-                    requestId
-                );
-                this._proxy.$acceptResponseDone(requestId, transformErrorForSerialization(err));
-            }
-        })();
-        Promise.allSettled([response.result, streaming]).then(() => {
-            this._logService.debug("[CHAT] extension request DONE", extension.value, requestId);
-            this._pendingCancelCTS.deleteAndDispose(requestId);
-            this._proxy.$acceptResponseDone(requestId, undefined);
-        }, err => {
-            this._logService.error(
-                "[CHAT] extension request ERRORED",
-                toErrorMessage(err, true),
-                extension.value,
-                requestId
-            );
-            this._pendingCancelCTS.deleteAndDispose(requestId);
-            this._proxy.$acceptResponseDone(requestId, transformErrorForSerialization(err));
-        });
-    }
-    $countTokens(modelId, value, token) {
-        return this._chatProviderService.computeTokenLength(modelId, value, token);
-    }
-    _registerAuthenticationProvider(extension, auth) {
-        const authProviderId = INTERNAL_AUTH_PROVIDER_PREFIX + extension.value;
-        if (this._authenticationService.getProviderIds().includes(authProviderId)) {
-            return Disposable.None;
-        }
-        const accountLabel = auth.accountLabel ?? ( localize(3012, "Language Models"));
-        const disposables = ( new DisposableStore());
-        const provider = ( new LanguageModelAccessAuthProvider(authProviderId, auth.providerLabel, accountLabel));
-        this._authenticationService.registerAuthenticationProvider(authProviderId, provider);
-        disposables.add(toDisposable(() => {
-            this._authenticationService.unregisterAuthenticationProvider(authProviderId);
-            provider.dispose();
-        }));
-        disposables.add(
-            this._authenticationAccessService.onDidChangeExtensionSessionAccess(async e => {
-                const allowedExtensions = this._authenticationAccessService.readAllowedExtensions(authProviderId, accountLabel);
-                const accessList = [];
-                for (const allowedExtension of allowedExtensions) {
-                    const from = await this._extensionService.getExtension(allowedExtension.id);
-                    if (from) {
-                        accessList.push({
-                            from: from.identifier,
-                            to: extension,
-                            enabled: allowedExtension.allowed ?? true
-                        });
-                    }
-                }
-                this._proxy.$updateModelAccesslist(accessList);
-            })
-        );
-        return disposables;
-    }
-    $fileIsIgnored(uri, token) {
-        return this._ignoredFilesService.fileIsIgnored(URI.revive(uri), token);
-    }
-    $registerFileIgnoreProvider(handle) {
-        this._ignoredFileProviderRegistrations.set(handle, this._ignoredFilesService.registerIgnoredFileProvider({
-            isFileIgnored: async (uri, token) => this._proxy.$isFileIgnored(handle, uri, token)
-        }));
-    }
-    $unregisterFileIgnoreProvider(handle) {
-        this._ignoredFileProviderRegistrations.deleteAndDispose(handle);
-    }
-};
-MainThreadLanguageModels = __decorate(
-    [extHostNamedCustomer(MainContext.MainThreadLanguageModels), ( __param(1, ILanguageModelsService)), ( __param(2, ILogService)), ( __param(3, IProductService)), ( __param(4, IAuthenticationService)), ( __param(5, IAuthenticationAccessService)), ( __param(6, IExtensionService)), ( __param(7, ILanguageModelIgnoredFilesService))],
-    MainThreadLanguageModels
-);
-class LanguageModelAccessAuthProvider {
-    constructor(id, label, _accountLabel) {
-        this.id = id;
-        this.label = label;
-        this._accountLabel = _accountLabel;
-        this.supportsMultipleAccounts = false;
-        this._onDidChangeSessions = ( new Emitter());
-        this.onDidChangeSessions = this._onDidChangeSessions.event;
-    }
-    async getSessions(scopes) {
-        if (scopes === undefined && !this._session) {
-            return [];
-        }
-        if (this._session) {
-            return [this._session];
-        }
-        return [await this.createSession(scopes || [])];
-    }
-    async createSession(scopes) {
-        this._session = this._createFakeSession(scopes);
-        this._onDidChangeSessions.fire({
-            added: [this._session],
-            changed: [],
-            removed: []
-        });
-        return this._session;
-    }
-    removeSession(sessionId) {
-        if (this._session) {
-            this._onDidChangeSessions.fire({
-                added: [],
-                changed: [],
-                removed: [this._session]
-            });
-            this._session = undefined;
-        }
-        return Promise.resolve();
-    }
-    confirmation(extensionName, _recreatingSession) {
-        return localize(
-            3013,
-            "The extension '{0}' wants to access the language models provided by {1}.",
-            extensionName,
-            this.label
-        );
-    }
-    _createFakeSession(scopes) {
-        return {
-            id: "fake-session",
-            account: {
-                id: this.id,
-                label: this._accountLabel
-            },
-            accessToken: "fake-access-token",
-            scopes
-        };
-    }
-    dispose() {
-        this._onDidChangeSessions.dispose();
-    }
-}
-
-export { MainThreadLanguageModels };

package/vscode/src/vs/workbench/api/browser/mainThreadMcp.d.ts

@@ -1,108 +0,0 @@
-import { Disposable } from "@codingame/monaco-vscode-api/vscode/vs/base/common/lifecycle";
-import { URI, UriComponents } from "@codingame/monaco-vscode-api/vscode/vs/base/common/uri";
-import { IContextKeyService } from "@codingame/monaco-vscode-api/vscode/vs/platform/contextkey/common/contextkey.service";
-import { IConfigurationService } from "@codingame/monaco-vscode-api/vscode/vs/platform/configuration/common/configuration.service";
-import { IDialogService } from "@codingame/monaco-vscode-api/vscode/vs/platform/dialogs/common/dialogs.service";
-import { LogLevel } from "@codingame/monaco-vscode-api/vscode/vs/platform/log/common/log";
-import { ITelemetryService } from "@codingame/monaco-vscode-api/vscode/vs/platform/telemetry/common/telemetry.service";
-import { ISecretStorageService } from "@codingame/monaco-vscode-api/vscode/vs/platform/secrets/common/secrets.service";
-import { IWorkbenchMcpGatewayService } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/mcp/common/mcpGatewayService.service";
-import { IMcpRegistry } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/mcp/common/mcpRegistryTypes.service";
-import { McpCollectionDefinition, McpConnectionState, McpServerDefinition } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/mcp/common/mcpTypes";
-import { IAuthenticationMcpAccessService } from "@codingame/monaco-vscode-api/vscode/vs/workbench/services/authentication/browser/authenticationMcpAccessService.service";
-import { IAuthenticationMcpService } from "@codingame/monaco-vscode-api/vscode/vs/workbench/services/authentication/browser/authenticationMcpService.service";
-import { IAuthenticationMcpUsageService } from "@codingame/monaco-vscode-api/vscode/vs/workbench/services/authentication/browser/authenticationMcpUsageService.service";
-import { IAuthenticationService } from "@codingame/monaco-vscode-api/vscode/vs/workbench/services/authentication/common/authentication.service";
-import { IDynamicAuthenticationProviderStorageService } from "@codingame/monaco-vscode-api/vscode/vs/workbench/services/authentication/common/dynamicAuthenticationProviderStorage.service";
-import { IExtensionService } from "@codingame/monaco-vscode-api/vscode/vs/workbench/services/extensions/common/extensions.service";
-import { IExtHostContext } from "../../services/extensions/common/extHostCustomers.js";
-import { IMcpAuthenticationDetails, IMcpAuthenticationOptions, IAuthMetadataSource, MainThreadMcpShape } from "@codingame/monaco-vscode-api/vscode/vs/workbench/api/common/extHost.protocol";
-export declare class MainThreadMcp extends Disposable implements MainThreadMcpShape {
-    private readonly _extHostContext;
-    private readonly _mcpRegistry;
-    private readonly dialogService;
-    private readonly _authenticationService;
-    private readonly authenticationMcpServersService;
-    private readonly authenticationMCPServerAccessService;
-    private readonly authenticationMCPServerUsageService;
-    private readonly _dynamicAuthenticationProviderStorageService;
-    private readonly _extensionService;
-    private readonly _contextKeyService;
-    private readonly _telemetryService;
-    private readonly _mcpGatewayService;
-    private readonly _configurationService;
-    private readonly _secretStorageService;
-    private _serverIdCounter;
-    private readonly _servers;
-    private readonly _serverDefinitions;
-    private readonly _serverAuthTracking;
-    private readonly _proxy;
-    private readonly _collectionDefinitions;
-    private readonly _gateways;
-    constructor(_extHostContext: IExtHostContext, _mcpRegistry: IMcpRegistry, dialogService: IDialogService, _authenticationService: IAuthenticationService, authenticationMcpServersService: IAuthenticationMcpService, authenticationMCPServerAccessService: IAuthenticationMcpAccessService, authenticationMCPServerUsageService: IAuthenticationMcpUsageService, _dynamicAuthenticationProviderStorageService: IDynamicAuthenticationProviderStorageService, _extensionService: IExtensionService, _contextKeyService: IContextKeyService, _telemetryService: ITelemetryService, _mcpGatewayService: IWorkbenchMcpGatewayService, _configurationService: IConfigurationService, _secretStorageService: ISecretStorageService);
-    private _publishServerDefinitions;
-    $upsertMcpCollection(collection: McpCollectionDefinition.FromExtHost, serversDto: McpServerDefinition.Serialized[]): void;
-    $deleteMcpCollection(collectionId: string): void;
-    $onDidChangeState(id: number, update: McpConnectionState): void;
-    $onDidPublishLog(id: number, level: LogLevel, log: string): void;
-    $onDidReceiveMessage(id: number, message: string): void;
-    $getTokenForProviderId(id: number, providerId: string, scopes: string[], options?: IMcpAuthenticationOptions): Promise<string | undefined>;
-    $getTokenFromServerMetadata(id: number, authDetails: IMcpAuthenticationDetails, { errorOnUserInteraction, forceNewRegistration, clientId }?: IMcpAuthenticationOptions): Promise<string | undefined>;
-    private _ensureXaaIssuer;
-    private _getSessionForProvider;
-    private continueWithIncorrectAccountPrompt;
-    private _onDidChangeAuthSessions;
-    $logMcpAuthSetup(data: IAuthMetadataSource): void;
-    $startMcpGateway(chatSessionResource?: UriComponents): Promise<{
-        servers: {
-            label: string;
-            address: URI;
-        }[];
-        gatewayId: string;
-    } | undefined>;
-    $disposeMcpGateway(gatewayId: string): void;
-    private loginPrompt;
-    dispose(): void;
-}
-/**
- * The context needed to re-acquire a token for a tracked MCP server, captured when the
- * session was first established. The tracker holds this opaquely and replays it verbatim on
- * re-validation so the silent token request targets the same authority / resource / audience
- * that the original sign-in used. Dropping the authorization server here would let the provider
- * fall back to a default authority (e.g. the Microsoft provider's `organizations` tenant) and
- * request a token against the wrong tenant.
- */
-export interface IMcpServerAuthContext {
-    readonly authorizationServer?: URI;
-    readonly clientId?: string;
-    readonly resource?: string;
-    readonly audience?: string;
-}
-/**
- * Tracks which MCP servers are using which authentication providers.
- * Organized by provider ID for efficient lookup when auth sessions change.
- */
-export declare class McpServerAuthTracker {
-    private readonly _tracking;
-    /**
-     * Track authentication for a server with a specific provider.
-     * Replaces any existing tracking for this server/provider combination.
-     */
-    track(providerId: string, serverId: number, scopes: string[], context: IMcpServerAuthContext): void;
-    /**
-     * Remove all authentication tracking for a server across all providers.
-     */
-    untrack(serverId: number): void;
-    /**
-     * Get all servers using a specific authentication provider.
-     */
-    get(providerId: string): ReadonlyArray<{
-        serverId: number;
-        scopes: string[];
-        context: IMcpServerAuthContext;
-    }> | undefined;
-    /**
-     * Clear all tracking data.
-     */
-    clear(): void;
-}