@vscode/chat-lib 0.0.1 → 0.0.2

package/src/_internal/platform/languageServer/common/languageContextService.ts

@@ -1,186 +0,0 @@
-/*---------------------------------------------------------------------------------------------
- *  Copyright (c) Microsoft Corporation. All rights reserved.
- *  Licensed under the MIT License. See License.txt in the project root for license information.
- *--------------------------------------------------------------------------------------------*/
-
-import type * as vscode from 'vscode';
-import { createServiceIdentifier } from '../../../util/common/services';
-
-export const ILanguageContextService = createServiceIdentifier<ILanguageContextService>('ILanguageContextService');
-
-export enum ContextKind {
-	Snippet = 'snippet',
-	Trait = 'trait'
-}
-
-/**
- * A context item represents a piece of code usually taken
- * from a source file.
- */
-export interface SnippetContext {
-	/**
-	 * The kind of the context.
-	 */
-	kind: ContextKind.Snippet;
-
-	/**
-	 * The priority of the snippet. Value range is [0, 1].
-	 */
-	priority: number;
-
-	/**
-	 * The main source the snippet is extracted from.
-	 */
-	uri: vscode.Uri;
-
-	/**
-	 * Additional sources if available.
-	 */
-	additionalUris?: vscode.Uri[];
-
-	/**
-	 * The actual snippet value.
-	 */
-	value: string;
-}
-
-export interface TraitContext {
-	/**
-	 * The kind of the context.
-	 */
-	kind: ContextKind.Trait;
-
-	/**
-	 * The priority of the context.
-	 */
-	priority: number;
-
-	/**
-	 * The name of the trait.
-	 */
-	name: string;
-
-	/**
-	 * The value of the trait.
-	 */
-	value: string;
-}
-
-export type ContextItem = SnippetContext | TraitContext;
-
-export enum KnownSources {
-	unknown = 'unknown',
-	sideCar = 'sideCar',
-	completion = 'completion',
-	populateCache = 'populateCache',
-	nes = 'nes',
-	chat = 'chat',
-	fix = 'fix'
-}
-
-export enum TriggerKind {
-	unknown = 'unknown',
-	selection = 'selection',
-	completion = 'completion',
-}
-
-export type RequestContext = {
-	/**
-	 * A unique request id.
-	 */
-	requestId: string;
-
-	/**
-	 * The opportunity ID provided by VS Code core.
-	 */
-	opportunityId?: string;
-
-	/**
-	 * The time budget in milliseconds to compute the context.
-	 */
-	timeBudget?: number;
-
-	/**
-	 * The token budget to compute the context.
-	 */
-	tokenBudget?: number;
-
-	/**
-	 * The source of the request.
-	 */
-	source?: KnownSources | string;
-
-	/**
-	 * The
-	 */
-	trigger?: TriggerKind | undefined;
-
-	/**
-	 * A list of proposed edits that should be applied before computing the context.
-	 */
-	proposedEdits?: { edit: vscode.TextEdit; source?: 'selectedCompletionInfo' }[];
-
-	/**
-	 * If provided the telemetry will be sampled. A value of 1 will log every request, a value of
-	 * 5 will log every 5th request, a value of 10 will log every 10th request, etc. If not provided
-	 * all telemetry will be logged.
-	 */
-	sampleTelemetry?: number;
-};
-
-export interface ILanguageContextService {
-	readonly _serviceBrand: undefined;
-
-	/**
-	 * Checks whether is language server context is activated for the
-	 * given text document or language.
-	 */
-	isActivated(documentOrLanguageId: vscode.TextDocument | string): Promise<boolean>;
-
-	/**
-	 * Populates the cache with context information for the given document and position.
-	 *
-	 * @param document The document to populate the cache for.
-	 * @param position The position in the document to populate the cache for.
-	 * @param context The context for the request.
-	 */
-	populateCache(document: vscode.TextDocument, position: vscode.Position, context: RequestContext): Promise<void>;
-
-	/**
-	 * Retrieves the context for the given document and position.
-	 *
-	 * @param document The document to retrieve the context for.
-	 * @param position The position in the document to retrieve the context for.
-	 * @param context The context for the request.
-	 * @param token A cancellation token.
-	 * @returns A promise that resolves to an array of context items.
-	 */
-	getContext(document: vscode.TextDocument, position: vscode.Position, context: RequestContext, token: vscode.CancellationToken): AsyncIterable<ContextItem>;
-
-	/**
-	 * Retrieves the context for the given document and position when a request timeout is reached.
-	 * This method is called when the `getContext` request takes too long to complete. Items
-	 * returned by this method are usually coming from a cache and might be not a 100% accurate.
-	 *
-	 * This method is optional and should complete execution quickly, ideally without any
-	 * block or long-running operations.
-	 *
-	 * @param document The document to retrieve the context for.
-	 * @param position The position in the document to retrieve the context for.
-	 * @param context The context for the request.
-	 * @returns An array of `ContextItem` or `undefined`.
-	 */
-	getContextOnTimeout(document: vscode.TextDocument, position: vscode.Position, context: RequestContext): readonly ContextItem[] | undefined;
-}
-
-class EmptyAsyncIterable<T> implements AsyncIterable<T> {
-	public async *[Symbol.asyncIterator](): AsyncIterator<T> {
-	}
-}
-export const NullLanguageContextService: ILanguageContextService = {
-	_serviceBrand: undefined,
-	isActivated: async () => false,
-	populateCache: async () => { },
-	getContext: () => new EmptyAsyncIterable<ContextItem>(),
-	getContextOnTimeout: () => [],
-};

package/src/_internal/platform/languages/common/languageDiagnosticsService.ts

@@ -1,67 +0,0 @@
-/*---------------------------------------------------------------------------------------------
- *  Copyright (c) Microsoft Corporation. All rights reserved.
- *  Licensed under the MIT License. See License.txt in the project root for license information.
- *--------------------------------------------------------------------------------------------*/
-
-import type * as vscode from 'vscode';
-import { createServiceIdentifier } from '../../../util/common/services';
-import { isEqual } from '../../../util/vs/base/common/resources';
-import { DiagnosticSeverity } from '../../../util/common/test/shims/vscodeTypesShim';
-export const ILanguageDiagnosticsService = createServiceIdentifier<ILanguageDiagnosticsService>('ILanguageDiagnosticService');
-
-export interface ILanguageDiagnosticsService {
-	_serviceBrand: undefined;
-	onDidChangeDiagnostics: vscode.Event<vscode.DiagnosticChangeEvent>;
-	getDiagnostics(resource: vscode.Uri): vscode.Diagnostic[];
-	getAllDiagnostics(): [vscode.Uri, vscode.Diagnostic[]][];
-	waitForNewDiagnostics(resource: vscode.Uri, token: vscode.CancellationToken, timeout?: number): Promise<vscode.Diagnostic[]>;
-}
-
-export abstract class AbstractLanguageDiagnosticsService implements ILanguageDiagnosticsService {
-
-	declare readonly _serviceBrand: undefined;
-
-	abstract onDidChangeDiagnostics: vscode.Event<vscode.DiagnosticChangeEvent>;
-
-	abstract getDiagnostics(resource: vscode.Uri): vscode.Diagnostic[];
-	abstract getAllDiagnostics(): [vscode.Uri, vscode.Diagnostic[]][];
-
-	waitForNewDiagnostics(resource: vscode.Uri, token: vscode.CancellationToken, timeout: number = 5000): Promise<vscode.Diagnostic[]> {
-		let onCancellationRequest: vscode.Disposable;
-		let diagnosticsChangeListener: vscode.Disposable;
-		let timer: any;
-		return new Promise<vscode.Diagnostic[]>((resolve) => {
-			onCancellationRequest = token.onCancellationRequested(() => resolve([]));
-			timer = setTimeout(() => resolve(this.getDiagnostics(resource)), timeout);
-			diagnosticsChangeListener = this.onDidChangeDiagnostics(e => {
-				for (const uri of e.uris) {
-					if (isEqual(uri, resource)) {
-						resolve(this.getDiagnostics(resource));
-						break;
-					}
-				}
-			});
-		}).finally(() => {
-			onCancellationRequest.dispose();
-			diagnosticsChangeListener.dispose();
-			clearTimeout(timer);
-		});
-	}
-}
-
-/**
-* Smallest range covering all of the diagnostics
-* @param diagnostics diagnostics to cover
-* @returns minimal covering range
-*/
-export function rangeSpanningDiagnostics(diagnostics: vscode.Diagnostic[]): vscode.Range {
-	return diagnostics.map(d => d.range).reduce((a, b) => a.union(b));
-}
-
-export function isError(diagnostics: vscode.Diagnostic) {
-	return diagnostics.severity === DiagnosticSeverity.Error;
-}
-
-export function getDiagnosticsAtSelection(diagnostics: vscode.Diagnostic[], selection: vscode.Range, severities: DiagnosticSeverity[] = [DiagnosticSeverity.Error, DiagnosticSeverity.Warning]): vscode.Diagnostic | undefined {
-	return diagnostics.find(d => d.range.contains(selection) && severities.includes(d.severity));
-}

package/src/_internal/platform/languages/common/testLanguageDiagnosticsService.ts

@@ -1,28 +0,0 @@
-/*---------------------------------------------------------------------------------------------
- *  Copyright (c) Microsoft Corporation. All rights reserved.
- *  Licensed under the MIT License. See License.txt in the project root for license information.
- *--------------------------------------------------------------------------------------------*/
-
-import type { Diagnostic, DiagnosticChangeEvent, Uri } from 'vscode';
-import { Emitter } from '../../../util/vs/base/common/event';
-import { ResourceMap } from '../../../util/vs/base/common/map';
-import { AbstractLanguageDiagnosticsService } from './languageDiagnosticsService';
-
-export class TestLanguageDiagnosticsService extends AbstractLanguageDiagnosticsService {
-	private diagnosticsMap = new ResourceMap<Diagnostic[]>();
-	private _onDidChangeDiagnostics = new Emitter<DiagnosticChangeEvent>();
-	public readonly onDidChangeDiagnostics = this._onDidChangeDiagnostics.event;
-
-	setDiagnostics(resource: Uri, diagnostics: Diagnostic[]): void {
-		this.diagnosticsMap.set(resource, diagnostics);
-		this._onDidChangeDiagnostics.fire({ uris: [resource] });
-	}
-
-	override getDiagnostics(resource: Uri): Diagnostic[] {
-		return this.diagnosticsMap.get(resource) || [];
-	}
-
-	override getAllDiagnostics(): [Uri, Diagnostic[]][] {
-		return Array.from(this.diagnosticsMap.entries());
-	}
-}

package/src/_internal/platform/log/common/logService.ts

@@ -1,232 +0,0 @@
-/*---------------------------------------------------------------------------------------------
- *  Copyright (c) Microsoft Corporation. All rights reserved.
- *  Licensed under the MIT License. See License.txt in the project root for license information.
- *--------------------------------------------------------------------------------------------*/
-
-import { createServiceIdentifier } from '../../../util/common/services';
-import { Disposable } from '../../../util/vs/base/common/lifecycle';
-
-export const ILogService = createServiceIdentifier<ILogService>('ILogService');
-
-/**
- * Log levels (taken from vscode.d.ts)
- */
-export enum LogLevel {
-
-	/**
-	 * No messages are logged with this level.
-	 */
-	Off = 0,
-
-	/**
-	 * All messages are logged with this level.
-	 */
-	Trace = 1,
-
-	/**
-	 * Messages with debug and higher log level are logged with this level.
-	 */
-	Debug = 2,
-
-	/**
-	 * Messages with info and higher log level are logged with this level.
-	 */
-	Info = 3,
-
-	/**
-	 * Messages with warning and higher log level are logged with this level.
-	 */
-	Warning = 4,
-
-	/**
-	 * Only error messages are logged with this level.
-	 */
-	Error = 5
-}
-
-export interface ILogTarget {
-	logIt(level: LogLevel, metadataStr: string, ...extra: any[]): void;
-	show?(preserveFocus?: boolean): void;
-}
-
-// Simple implementation of a log targe used for logging to the console.
-export class ConsoleLog implements ILogTarget {
-	constructor(private readonly prefix?: string, private readonly minLogLevel: LogLevel = LogLevel.Warning) { }
-
-	logIt(level: LogLevel, metadataStr: string, ...extra: any[]) {
-		if (this.prefix) {
-			metadataStr = `${this.prefix}${metadataStr}`;
-		}
-
-		// Note we don't log INFO or DEBUG messages into console.
-		// They are still logged in the output channel.
-		if (level === LogLevel.Error) {
-			console.error(metadataStr, ...extra);
-		} else if (level === LogLevel.Warning) {
-			console.warn(metadataStr, ...extra);
-		} else if (level >= this.minLogLevel) {
-			console.log(metadataStr, ...extra);
-		}
-	}
-}
-
-export interface ILogService extends ILogger {
-	readonly _serviceBrand: undefined;
-}
-
-/**
- * Mirrors vscode's {@link LogOutputChannel} in terms of available logging functions
- * Args has been ommitted for now in favor of simplifying the interface
- */
-export interface ILogger {
-	trace(message: string): void;
-	debug(message: string): void;
-	info(message: string): void;
-	warn(message: string): void;
-	/**
-	 * Logs an error message. Prefer this method over `error()` when logging exception details.
-	 *
-	 * @param error The Error object that was thrown
-	 * @param message An optional message for context (e.g. "Request error"). Must not contain customer data. **Do not include stack trace or messages from the error object.**
-	*/
-	error(error: string | Error, message?: string): void;
-	show(preserveFocus?: boolean): void;
-}
-
-export class LogServiceImpl extends Disposable implements ILogService {
-	declare _serviceBrand: undefined;
-
-	readonly logger: LoggerImpl;
-
-	constructor(
-		logTargets: ILogTarget[],
-	) {
-		super();
-		this.logger = new LoggerImpl(logTargets);
-	}
-
-	// Delegate logging methods directly to the internal logger
-	trace(message: string): void {
-		this.logger.trace(message);
-	}
-
-	debug(message: string): void {
-		this.logger.debug(message);
-	}
-
-	info(message: string): void {
-		this.logger.info(message);
-	}
-
-	warn(message: string): void {
-		this.logger.warn(message);
-	}
-
-	error(error: string | Error, message?: string): void {
-		this.logger.error(error, message);
-	}
-
-	show(preserveFocus?: boolean): void {
-		this.logger.show(preserveFocus);
-	}
-}
-
-class LoggerImpl implements ILogger {
-	constructor(
-		private readonly _logTargets: ILogTarget[],
-	) { }
-
-	private _logIt(level: LogLevel, message: string): void {
-		LogMemory.addLog(LogLevel[level], message);
-		this._logTargets.forEach(t => t.logIt(level, message));
-	}
-
-	trace(message: string): void {
-		this._logIt(LogLevel.Trace, message);
-	}
-
-	debug(message: string): void {
-		this._logIt(LogLevel.Debug, message);
-	}
-
-	info(message: string): void {
-		this._logIt(LogLevel.Info, message);
-	}
-
-	warn(message: string): void {
-		this._logIt(LogLevel.Warning, message);
-	}
-
-	error(error: string | Error, message?: string): void {
-		this._logIt(LogLevel.Error, collectErrorMessages(error) + (message ? `: ${message}` : ''));
-	}
-
-	show(preserveFocus?: boolean): void {
-		this._logTargets.forEach(t => t.show?.(preserveFocus));
-	}
-}
-
-export function collectErrorMessages(e: any): string {
-	// Collect error messages from nested errors as seen with Node's `fetch`.
-	const seen = new Set<any>();
-	function collect(e: any, indent: string): string {
-		if (!e || !['object', 'string'].includes(typeof e) || seen.has(e)) {
-			return '';
-		}
-		seen.add(e);
-		const message = typeof e === 'string' ? e : (e.stack || e.message || e.code || e.toString?.() || '');
-		const messageStr = message.toString?.() as (string | undefined) || '';
-		return [
-			messageStr ? `${messageStr.split('\n').map(line => `${indent}${line}`).join('\n')}\n` : '',
-			collect(e.cause, indent + '  '),
-			...(Array.isArray(e.errors) ? e.errors.map((e: any) => collect(e, indent + '  ')) : []),
-		].join('');
-	}
-	return collect(e, '')
-		.trim();
-}
-
-export class LogMemory {
-	private static _logs: string[] = [];
-	private static _requestIds: string[] = [];
-	private static readonly MAX_LOGS = 50;
-
-	/**
-	 * Extracts the requestId from a log message if it matches the expected pattern.
-	 * Returns a string in the format 'requestId: {string}' or undefined if not found.
-	 */
-	private static extractRequestIdFromMessage(message: string): string | undefined {
-		const match = message.match(/request done: requestId: \[([0-9a-fA-F-]+)\] model deployment ID: \[/);
-		if (match) {
-			const requestId = match[1];
-			if (!this._requestIds.includes(requestId)) {
-				return requestId;
-			}
-		}
-		return undefined;
-	}
-
-	static addLog(level: string, message: string): void {
-		if (this._logs.length >= this.MAX_LOGS) {
-			this._logs.shift();
-		}
-		this._logs.push(`${level}: ${message}`);
-
-		// Extract and store requestId if present
-		if (this._requestIds.length >= this.MAX_LOGS) {
-			this._requestIds.shift();
-		}
-		const requestId = this.extractRequestIdFromMessage(message);
-		if (requestId) {
-			this._requestIds.push(requestId);
-		}
-	}
-
-	static getLogs(): string[] {
-		return this._logs;
-	}
-
-	static getRequestIds(): string[] {
-		return this._requestIds;
-	}
-}

package/src/_internal/platform/nesFetch/common/completionsAPI.ts

@@ -1,78 +0,0 @@
-/*---------------------------------------------------------------------------------------------
- *  Copyright (c) Microsoft Corporation. All rights reserved.
- *  Licensed under the MIT License. See License.txt in the project root for license information.
- *--------------------------------------------------------------------------------------------*/
-
-/**
- * Represents a completion response from the API. Note: both the streamed and non-streamed response objects share the same shape (unlike the chat endpoint).
- *
- * Based (updated 31.10.2024) on https://platform.openai.com/docs/api-reference/completions/object
- * 		(! except `Choice#finish_reason` can be a null during streaming)
- *      (! omits `id`, `model`, and `created` for performance)
- */
-export interface Completion {
-	/** The list of completion choices the model generated for the input prompt. */
-	choices: Completion.Choice[];
-	/** This fingerprint represents the backend configuration that the model runs with. */
-	system_fingerprint: string;
-	/** The object type, which is always "text_completion". */
-	object: string;
-	/** Usage statistics for the completion request. */
-	usage: Completion.Usage | undefined;
-}
-
-export namespace Completion {
-	export interface Choice {
-		/** The index of the choice. */
-		index: number;
-		/** The reason the model stopped generating tokens. */
-		finish_reason: FinishReason | null;
-		/** The log probabilities of the tokens. */
-		logprobs?: LogProbs | null;
-		/** The generated text. */
-		text: string | undefined;
-	}
-
-	/**
-	 * The reason the model stopped generating tokens.
-	 */
-	export enum FinishReason {
-		/** If the model hit a natural stop point or a provided stop sequence. */
-		Stop = 'stop',
-		/** If the maximum number of tokens specified in the request was reached. */
-		Length = 'length',
-		/** If content was omitted due to a flag from our content filters. */
-		ContentFilter = 'content_filter',
-	}
-
-	export type LogProbs = {
-		/** The list of tokens generated by the model. */
-		tokens: string[];
-		/** The log probabilities of the tokens. */
-		token_logprobs: number[];
-		/** The text offsets of the tokens. */
-		text_offset: number[];
-		/** The top log probabilities of the tokens. */
-		top_logprobs: Record<string, number>[];
-	};
-
-	export interface Usage {
-		/** Number of tokens in the generated completion. */
-		completion_tokens: number;
-		/** Number of tokens in the prompt. */
-		prompt_tokens: number;
-		/** Total number of tokens used in the request (prompt + completion). */
-		total_tokens: number;
-		/** Breakdown of tokens used in a completion. */
-		completion_tokens_details: TokensDetails;
-		/** Breakdown of tokens used in the prompt. */
-		prompt_tokens_details: TokensDetails;
-	}
-
-	export interface TokensDetails {
-		/** Audio input tokens present in the prompt. */
-		audio_tokens: number;
-		/** Tokens generated by the model for reasoning. */
-		reasoning_tokens: number;
-	}
-}