@codingame/monaco-vscode-chat-service-override 25.1.2 → 26.0.0

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

@@ -1,7 +1,7 @@
 import { CancellationToken } from "@codingame/monaco-vscode-api/vscode/vs/base/common/cancellation";
 import { Event } from "@codingame/monaco-vscode-api/vscode/vs/base/common/event";
 import { Disposable, IDisposable } from "@codingame/monaco-vscode-api/vscode/vs/base/common/lifecycle";
-import { IObservable } from "@codingame/monaco-vscode-api/vscode/vs/base/common/observable";
+import { IObservable, IReader } from "@codingame/monaco-vscode-api/vscode/vs/base/common/observable";
 import { ThemeIcon } from "@codingame/monaco-vscode-api/vscode/vs/base/common/themables";
 import { IAccessibilityService } from "@codingame/monaco-vscode-api/vscode/vs/platform/accessibility/common/accessibility.service";
 import { IAccessibilitySignalService } from "@codingame/monaco-vscode-api/vscode/vs/platform/accessibilitySignal/browser/accessibilitySignalService.service";
@@ -13,12 +13,15 @@ import { ILogService } from "@codingame/monaco-vscode-api/vscode/vs/platform/log
 import { IStorageService } from "@codingame/monaco-vscode-api/vscode/vs/platform/storage/common/storage.service";
 import { ITelemetryService } from "@codingame/monaco-vscode-api/vscode/vs/platform/telemetry/common/telemetry.service";
 import { IExtensionService } from "@codingame/monaco-vscode-api/vscode/vs/workbench/services/extensions/common/extensions.service";
+import { ChatRequestToolReferenceEntry } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/attachments/chatVariableEntries";
 import { IVariableReference } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/chatModes";
+import { IChatToolInvocation } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/chatService/chatService";
 import { IChatService } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/chatService/chatService.service";
-import { ChatRequestToolReferenceEntry } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/attachments/chatVariableEntries";
+import { ILanguageModelChatMetadata } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/languageModels";
 import { ILanguageModelToolsConfirmationService } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/tools/languageModelToolsConfirmationService.service";
-import { CountTokensCallback, IToolAndToolSetEnablementMap, IToolData, IToolImpl, IToolInvocation, IToolResult, ToolDataSource, ToolSet } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/tools/languageModelToolsService";
+import { CountTokensCallback, IBeginToolCallOptions, IToolAndToolSetEnablementMap, IToolData, IToolImpl, IToolInvocation, IToolResult, ToolDataSource, ToolSet, IToolSet, IToolInvokedEvent } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/tools/languageModelToolsService";
 import { ILanguageModelToolsService } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/tools/languageModelToolsService.service";
+import { URI } from "@codingame/monaco-vscode-api/vscode/vs/base/common/uri";
 export declare const globalAutoApproveDescription: import("@codingame/monaco-vscode-api/vscode/vs/nls").ILocalizedString;
 export declare class LanguageModelToolsService extends Disposable implements ILanguageModelToolsService {
     private readonly _instantiationService;
@@ -37,19 +40,24 @@ export declare class LanguageModelToolsService extends Disposable implements ILa
     readonly vscodeToolSet: ToolSet;
     readonly executeToolSet: ToolSet;
     readonly readToolSet: ToolSet;
+    readonly agentToolSet: ToolSet;
     private readonly _onDidChangeTools;
     readonly onDidChangeTools: Event<void>;
     private readonly _onDidPrepareToolCallBecomeUnresponsive;
     readonly onDidPrepareToolCallBecomeUnresponsive: Event<{
-        sessionId: string;
+        sessionResource: URI;
         toolData: IToolData;
     }>;
+    private readonly _onDidInvokeTool;
+    readonly onDidInvokeTool: Event<IToolInvokedEvent>;
     /** Throttle tools updates because it sends all tools and runs on context key updates */
     private readonly _onDidChangeToolsScheduler;
     private readonly _tools;
     private readonly _toolContextKeys;
     private readonly _ctxToolsCount;
     private readonly _callsByRequestId;
+    /** Pending tool calls in the streaming phase, keyed by toolCallId */
+    private readonly _pendingToolCalls;
     private readonly _isAgentModeEnabled;
     constructor(_instantiationService: IInstantiationService, _extensionService: IExtensionService, _contextKeyService: IContextKeyService, _chatService: IChatService, _dialogService: IDialogService, _telemetryService: ITelemetryService, _logService: ILogService, _configurationService: IConfigurationService, _accessibilityService: IAccessibilityService, _accessibilitySignalService: IAccessibilitySignalService, _storageService: IStorageService, _confirmationService: ILanguageModelToolsConfirmationService);
     /**
@@ -64,13 +72,16 @@ export declare class LanguageModelToolsService extends Disposable implements ILa
     private _refreshAllToolContextKeys;
     registerToolImplementation(id: string, tool: IToolImpl): IDisposable;
     registerTool(toolData: IToolData, tool: IToolImpl): IDisposable;
-    getTools(includeDisabled?: boolean): Iterable<IToolData>;
-    readonly toolsObservable: IObservable<readonly IToolData[]>;
+    getTools(model: ILanguageModelChatMetadata | undefined): Iterable<IToolData>;
+    observeTools(model: ILanguageModelChatMetadata | undefined): IObservable<readonly IToolData[]>;
+    getAllToolsIncludingDisabled(): Iterable<IToolData>;
     getTool(id: string): IToolData | undefined;
-    private _getToolEntry;
-    getToolByName(name: string, includeDisabled?: boolean): IToolData | undefined;
+    getToolByName(name: string): IToolData | undefined;
     invokeTool(dto: IToolInvocation, countTokens: CountTokensCallback, token: CancellationToken): Promise<IToolResult>;
     private prepareToolInvocation;
+    beginToolCall(options: IBeginToolCallOptions): IChatToolInvocation | undefined;
+    private _callHandleToolStream;
+    updateToolStream(toolCallId: string, partialInput: unknown, token: CancellationToken): Promise<void>;
     private playAccessibilitySignal;
     private ensureToolDetails;
     private formatToolInput;
@@ -91,11 +102,12 @@ export declare class LanguageModelToolsService extends Disposable implements ILa
      * @param fullReferenceNames A list of tool or toolset by their full reference names that are enabled.
      * @returns A map of tool or toolset instances to their enablement state.
      */
-    toToolAndToolSetEnablementMap(fullReferenceNames: readonly string[], _target: string | undefined): IToolAndToolSetEnablementMap;
+    toToolAndToolSetEnablementMap(fullReferenceNames: readonly string[], _target: string | undefined, model: ILanguageModelChatMetadata | undefined): IToolAndToolSetEnablementMap;
     toFullReferenceNames(map: IToolAndToolSetEnablementMap): string[];
     toToolReferences(variableReferences: readonly IVariableReference[]): ChatRequestToolReferenceEntry[];
     private readonly _toolSets;
     readonly toolSets: IObservable<Iterable<ToolSet>>;
+    getToolSetsForModel(model: ILanguageModelChatMetadata | undefined, reader?: IReader): Iterable<IToolSet>;
     getToolSet(id: string): ToolSet | undefined;
     getToolSetByName(name: string): ToolSet | undefined;
     getSpecedToolSetName(referenceName: string): string;
@@ -104,12 +116,10 @@ export declare class LanguageModelToolsService extends Disposable implements ILa
         description?: string;
         legacyFullNames?: string[];
     }): ToolSet & IDisposable;
-    readonly toolsWithFullReferenceName: import("@codingame/monaco-vscode-api/vscode/vs/base/common/observable").IObservableWithChange<[
-        IToolData | ToolSet,
-        string
-    ][], void>;
+    private readonly allToolsIncludingDisableObs;
+    private readonly toolsWithFullReferenceName;
     getFullReferenceNames(): Iterable<string>;
     getDeprecatedFullReferenceNames(): Map<string, Set<string>>;
     getToolByFullReferenceName(fullReferenceName: string): IToolData | ToolSet | undefined;
-    getFullReferenceName(tool: IToolData | ToolSet, toolSet?: ToolSet): string;
+    getFullReferenceName(tool: IToolData | IToolSet, toolSet?: IToolSet): string;
 }