@codingame/monaco-vscode-chat-service-override 11.1.2 → 12.0.0

package/vscode/src/vs/workbench/contrib/chat/common/languageModels.js

@@ -1,172 +0,0 @@
-import { __decorate, __param } from 'vscode/external/tslib/tslib.es6.js';
-import { Emitter } from 'vscode/vscode/vs/base/common/event';
-import { Iterable } from 'vscode/vscode/vs/base/common/iterator';
-import { DisposableStore, toDisposable } from 'vscode/vscode/vs/base/common/lifecycle';
-import { isFalsyOrWhitespace } from 'vscode/vscode/vs/base/common/strings';
-import { localize } from 'vscode/vscode/vs/nls';
-import { IContextKeyService } from 'vscode/vscode/vs/platform/contextkey/common/contextkey.service';
-import { ExtensionIdentifier } from 'vscode/vscode/vs/platform/extensions/common/extensions';
-import { ILogService } from 'vscode/vscode/vs/platform/log/common/log.service';
-import { isProposedApiEnabled } from 'vscode/vscode/vs/workbench/services/extensions/common/extensions';
-import { IExtensionService } from 'vscode/vscode/vs/workbench/services/extensions/common/extensions.service';
-import { ExtensionsRegistry } from 'vscode/vscode/vs/workbench/services/extensions/common/extensionsRegistry';
-import { CONTEXT_LANGUAGE_MODELS_ARE_USER_SELECTABLE } from 'vscode/vscode/vs/workbench/contrib/chat/common/chatContextKeys';
-
-const languageModelType = {
-    type: 'object',
-    properties: {
-        vendor: {
-            type: 'string',
-            description: ( localize(3182, "A globally unique vendor of language models."))
-        }
-    }
-};
-const languageModelExtensionPoint = ExtensionsRegistry.registerExtensionPoint({
-    extensionPoint: 'languageModels',
-    jsonSchema: {
-        description: ( localize(3183, "Contribute language models of a specific vendor.")),
-        oneOf: [
-            languageModelType,
-            {
-                type: 'array',
-                items: languageModelType
-            }
-        ]
-    },
-    activationEventsGenerator: (contribs, result) => {
-        for (const contrib of contribs) {
-            result.push(`onLanguageModelChat:${contrib.vendor}`);
-        }
-    }
-});
-let LanguageModelsService = class LanguageModelsService {
-    constructor(_extensionService, _logService, _contextKeyService) {
-        this._extensionService = _extensionService;
-        this._logService = _logService;
-        this._contextKeyService = _contextKeyService;
-        this._store = ( (new DisposableStore()));
-        this._providers = ( (new Map()));
-        this._vendors = ( (new Set()));
-        this._onDidChangeProviders = this._store.add(( (new Emitter())));
-        this.onDidChangeLanguageModels = this._onDidChangeProviders.event;
-        this._hasUserSelectableModels = CONTEXT_LANGUAGE_MODELS_ARE_USER_SELECTABLE.bindTo(this._contextKeyService);
-        this._store.add(languageModelExtensionPoint.setHandler((extensions) => {
-            this._vendors.clear();
-            for (const extension of extensions) {
-                if (!isProposedApiEnabled(extension.description, 'chatProvider')) {
-                    extension.collector.error(( localize(3184, "This contribution point requires the 'chatProvider' proposal.")));
-                    continue;
-                }
-                for (const item of Iterable.wrap(extension.value)) {
-                    if (( (this._vendors.has(item.vendor)))) {
-                        extension.collector.error(( localize(
-                            3185,
-                            "The vendor '{0}' is already registered and cannot be registered twice",
-                            item.vendor
-                        )));
-                        continue;
-                    }
-                    if (isFalsyOrWhitespace(item.vendor)) {
-                        extension.collector.error(( localize(3186, "The vendor field cannot be empty.")));
-                        continue;
-                    }
-                    if (item.vendor.trim() !== item.vendor) {
-                        extension.collector.error(( localize(3187, "The vendor field cannot start or end with whitespace.")));
-                        continue;
-                    }
-                    this._vendors.add(item.vendor);
-                }
-            }
-            const removed = [];
-            for (const [identifier, value] of this._providers) {
-                if (!( (this._vendors.has(value.metadata.vendor)))) {
-                    this._providers.delete(identifier);
-                    removed.push(identifier);
-                }
-            }
-            if (removed.length > 0) {
-                this._onDidChangeProviders.fire({ removed });
-            }
-        }));
-    }
-    dispose() {
-        this._store.dispose();
-        this._providers.clear();
-    }
-    getLanguageModelIds() {
-        return Array.from(( (this._providers.keys())));
-    }
-    lookupLanguageModel(identifier) {
-        return this._providers.get(identifier)?.metadata;
-    }
-    async selectLanguageModels(selector) {
-        if (selector.vendor) {
-            await this._extensionService.activateByEvent(`onLanguageModelChat:${selector.vendor}}`);
-        }
-        else {
-            const all = ( (Array.from(this._vendors).map(
-                vendor => this._extensionService.activateByEvent(`onLanguageModelChat:${vendor}`)
-            )));
-            await Promise.all(all);
-        }
-        const result = [];
-        for (const [identifier, model] of this._providers) {
-            if ((selector.vendor === undefined || model.metadata.vendor === selector.vendor)
-                && (selector.family === undefined || model.metadata.family === selector.family)
-                && (selector.version === undefined || model.metadata.version === selector.version)
-                && (selector.identifier === undefined || model.metadata.id === selector.identifier)
-                && (!model.metadata.targetExtensions || ( (model.metadata.targetExtensions.some(candidate => ExtensionIdentifier.equals(candidate, selector.extension)))))) {
-                result.push(identifier);
-            }
-        }
-        this._logService.trace('[LM] selected language models', selector, result);
-        return result;
-    }
-    registerLanguageModelChat(identifier, provider) {
-        this._logService.trace('[LM] registering language model chat', identifier, provider.metadata);
-        if (!( (this._vendors.has(provider.metadata.vendor)))) {
-            throw ( (new Error(`Chat response provider uses UNKNOWN vendor ${provider.metadata.vendor}.`)));
-        }
-        if (( (this._providers.has(identifier)))) {
-            throw ( (new Error(
-                `Chat response provider with identifier ${identifier} is already registered.`
-            )));
-        }
-        this._providers.set(identifier, provider);
-        this._onDidChangeProviders.fire({ added: [{ identifier, metadata: provider.metadata }] });
-        this.updateUserSelectableModelsContext();
-        return toDisposable(() => {
-            this.updateUserSelectableModelsContext();
-            if (this._providers.delete(identifier)) {
-                this._onDidChangeProviders.fire({ removed: [identifier] });
-                this._logService.trace('[LM] UNregistered language model chat', identifier, provider.metadata);
-            }
-        });
-    }
-    updateUserSelectableModelsContext() {
-        const hasUserSelectableModels = ( (Array.from(( (this._providers.values()))).some(p => p.metadata.isUserSelectable && !p.metadata.isDefault)));
-        const hasDefaultModel = ( (Array.from(( (this._providers.values()))).some(p => p.metadata.isDefault)));
-        this._hasUserSelectableModels.set(hasUserSelectableModels && hasDefaultModel);
-    }
-    async sendChatRequest(identifier, from, messages, options, token) {
-        const provider = this._providers.get(identifier);
-        if (!provider) {
-            throw ( (new Error(`Chat response provider with identifier ${identifier} is not registered.`)));
-        }
-        return provider.sendChatRequest(messages, from, options, token);
-    }
-    computeTokenLength(identifier, message, token) {
-        const provider = this._providers.get(identifier);
-        if (!provider) {
-            throw ( (new Error(`Chat response provider with identifier ${identifier} is not registered.`)));
-        }
-        return provider.provideTokenCount(message, token);
-    }
-};
-LanguageModelsService = ( (__decorate([
-    ( (__param(0, IExtensionService))),
-    ( (__param(1, ILogService))),
-    ( (__param(2, IContextKeyService)))
-], LanguageModelsService)));
-
-export { LanguageModelsService, languageModelExtensionPoint };