1
>
/*---------------------------------------------------------------------------------------------
languageModelToolsService.ts
2
>
* Copyright (c) Microsoft Corporation. All rights reserved.
3
>
* Licensed under the MIT License. See License.txt in the project root for license information.
4
>
*--------------------------------------------------------------------------------------------*/
5
>
6
>
import { Separator } from '../../../../../base/common/actions.js';
7
>
import { VSBuffer } from '../../../../../base/common/buffer.js';
8
>
import { CancellationToken } from '../../../../../base/common/cancellation.js';
9
>
import { Event } from '../../../../../base/common/event.js';
10
>
import { IMarkdownString } from '../../../../../base/common/htmlContent.js';
11
>
import { Iterable } from '../../../../../base/common/iterator.js';
12
>
import { IJSONSchema } from '../../../../../base/common/jsonSchema.js';
13
>
import { Disposable, IDisposable, toDisposable } from '../../../../../base/common/lifecycle.js';
14
>
import { Schemas } from '../../../../../base/common/network.js';
15
>
import { derived, IObservable, IReader, ITransaction, ObservableSet } from '../../../../../base/common/observable.js';
16
>
import { ThemeIcon } from '../../../../../base/common/themables.js';
17
>
import { URI } from '../../../../../base/common/uri.js';
18
>
import { Location } from '../../../../../editor/common/languages.js';
19
>
import { localize } from '../../../../../nls.js';
20
>
import { ConfirmationOption } from '../../../../../platform/agentHost/common/state/protocol/state.js';
21
>
import { ContextKeyExpression, IContextKeyService } from '../../../../../platform/contextkey/common/contextkey.js';
22
>
import { ExtensionIdentifier } from '../../../../../platform/extensions/common/extensions.js';
23
>
import { ByteSize } from '../../../../../platform/files/common/files.js';
24
>
import { createDecorator } from '../../../../../platform/instantiation/common/instantiation.js';
25
>
import { IProgress } from '../../../../../platform/progress/common/progress.js';
26
>
import { ChatRequestToolReferenceEntry } from '../attachments/chatVariableEntries.js';
27
>
import { IVariableReference } from '../chatModes.js';
28
>
import { ConfirmedReason, IChatAgentFeedbackReviewConfirmationData, IChatAutomationConfigurationData, IChatAutomationConfiguredData, IChatExtensionsContent, IChatModifiedFilesConfirmationData, IChatSearchToolInvocationData, IChatSessionCreatedData, IChatSimpleToolInvocationData, IChatSubagentToolInvocationData, IChatTodoListContent, IChatToolInputInvocationData, IChatToolInvocation, type IChatTerminalToolInvocationData } from '../chatService/chatService.js';
29
>
import { ILanguageModelChatMetadata, LanguageModelPartAudience } from '../languageModels.js';
30
>
import { UserSelectedTools } from '../participants/chatAgents.js';
31
>
import { PromptElementJSON, stringifyPromptElementJSON } from './promptTsxTypes.js';
32
>
33
>
/**
34
>
* Selector for matching language models by vendor, family, version, or id.
35
>
* Used to filter tools to specific models or model families.
36
>
*/
37
>
export interface ILanguageModelChatSelector {
38
>
readonly vendor?: string;
39
>
readonly family?: string;
40
>
readonly version?: string;
41
>
readonly id?: string;
42
>
}
43
>
44
>
export interface IToolData {
45
>
readonly id: string;
46
>
readonly source: ToolDataSource;
47
>
readonly toolReferenceName?: string;
48
>
readonly legacyToolReferenceFullNames?: readonly string[];
49
>
readonly icon?: { dark: URI; light?: URI } | ThemeIcon;
50
>
readonly when?: ContextKeyExpression;
51
>
readonly tags?: readonly string[];
52
>
readonly displayName: string;
53
>
readonly userDescription?: string;
54
>
readonly modelDescription: string;
55
>
readonly inputSchema?: IJSONSchema;
56
>
readonly canBeReferencedInPrompt?: boolean;
57
>
/**
58
>
* True if the tool runs in the (possibly remote) workspace, false if it runs
59
>
* on the host, undefined if known.
60
>
*/
61
>
readonly runsInWorkspace?: boolean;
62
>
readonly alwaysDisplayInputOutput?: boolean;
63
>
/** True if this tool might ask for pre-approval */
64
>
readonly canRequestPreApproval?: boolean;
65
>
/** True if this tool might ask for post-approval */
66
>
readonly canRequestPostApproval?: boolean;
67
>
/**
68
>
* Model selectors that this tool is available for.
69
>
* If defined, the tool is only available when the selected model matches one of the selectors.
70
>
*/
71
>
readonly models?: readonly ILanguageModelChatSelector[];
72
>
}
73
>
74
>
/**
75
>
* Check if a tool matches the given model metadata based on the tool's `models` selectors.
76
>
* If the tool has no `models` defined, it matches all models.
77
>
* If model is undefined, model-specific filtering is skipped (tool is included).
78
>
*/
79
>
export function toolMatchesModel(toolData: IToolData, model: ILanguageModelChatMetadata | undefined): boolean {
80
// If no model selectors are defined, the tool is available for all models
81
if (!toolData.models || toolData.models.length === 0) {