@codingame/monaco-vscode-katex-common 29.1.1 → 30.0.0

package/vscode/src/vs/workbench/contrib/chat/browser/actions/chatToolActions.js

@@ -0,0 +1,208 @@
+
+import { CancellationTokenSource } from '@codingame/monaco-vscode-api/vscode/vs/base/common/cancellation';
+import { Codicon } from '@codingame/monaco-vscode-api/vscode/vs/base/common/codicons';
+import { Iterable } from '@codingame/monaco-vscode-api/vscode/vs/base/common/iterator';
+import { KeyMod, KeyCode } from '@codingame/monaco-vscode-api/vscode/vs/base/common/keyCodes';
+import '@codingame/monaco-vscode-api/vscode/vs/base/common/observableInternal/index';
+import { localize2, localize } from '@codingame/monaco-vscode-api/vscode/vs/nls';
+import { Action2, MenuId, registerAction2 } from '@codingame/monaco-vscode-api/vscode/vs/platform/actions/common/actions';
+import { ContextKeyExpr } from '@codingame/monaco-vscode-api/vscode/vs/platform/contextkey/common/contextkey';
+import { IInstantiationService } from '@codingame/monaco-vscode-api/vscode/vs/platform/instantiation/common/instantiation';
+import { KeybindingWeight } from '@codingame/monaco-vscode-api/vscode/vs/platform/keybinding/common/keybindingsRegistry';
+import { ITelemetryService } from '@codingame/monaco-vscode-api/vscode/vs/platform/telemetry/common/telemetry.service';
+import { ChatContextKeys } from '@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/actions/chatContextKeys';
+import { IChatToolInvocation, ToolConfirmKind } from '@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/chatService/chatService';
+import { isResponseVM } from '@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/model/chatViewModel';
+import { ChatModeKind } from '@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/constants';
+import { IChatWidgetService } from '@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/browser/chat.service';
+import { ToolsScope } from '../widget/input/chatSelectedTools.js';
+import { CHAT_CATEGORY } from '@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/browser/actions/chatActions';
+import { AcceptToolConfirmationActionId, SkipToolConfirmationActionId } from '@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/browser/actions/chatConstants';
+import { showToolsPicker } from './chatToolPicker.js';
+import { autorun } from '@codingame/monaco-vscode-api/vscode/vs/base/common/observableInternal/reactions/autorun';
+
+class ToolConfirmationAction extends Action2 {
+    run(accessor, context) {
+        const chatWidgetService = accessor.get(IChatWidgetService);
+        const widget = context?.sessionResource ? chatWidgetService.getWidgetBySessionResource(context.sessionResource) : chatWidgetService.lastFocusedWidget;
+        const lastItem = widget?.viewModel?.getItems().at(-1);
+        if (!isResponseVM(lastItem)) {
+            return;
+        }
+        for (const item of lastItem.model.response.value) {
+            const state = item.kind === "toolInvocation" ? item.state.get() : undefined;
+            if (state?.type === IChatToolInvocation.StateKind.WaitingForConfirmation || state?.type === IChatToolInvocation.StateKind.WaitingForPostApproval) {
+                state.confirm(this.getReason());
+                break;
+            }
+        }
+        widget?.focusInput();
+    }
+}
+class AcceptToolConfirmation extends ToolConfirmationAction {
+    constructor() {
+        super({
+            id: AcceptToolConfirmationActionId,
+            title: ( localize2(5014, "Accept")),
+            f1: false,
+            category: CHAT_CATEGORY,
+            keybinding: {
+                when: ( ContextKeyExpr.and(ChatContextKeys.inChatSession, ChatContextKeys.Editing.hasToolConfirmation)),
+                primary: KeyMod.CtrlCmd | KeyCode.Enter,
+                weight: KeybindingWeight.WorkbenchContrib + 1
+            }
+        });
+    }
+    getReason() {
+        return {
+            type: ToolConfirmKind.UserAction
+        };
+    }
+}
+class SkipToolConfirmation extends ToolConfirmationAction {
+    constructor() {
+        super({
+            id: SkipToolConfirmationActionId,
+            title: ( localize2(5015, "Skip")),
+            f1: false,
+            category: CHAT_CATEGORY,
+            keybinding: {
+                when: ( ContextKeyExpr.and(ChatContextKeys.inChatSession, ChatContextKeys.Editing.hasToolConfirmation)),
+                primary: KeyMod.CtrlCmd | KeyCode.Enter | KeyMod.Alt,
+                weight: KeybindingWeight.WorkbenchContrib + 1
+            }
+        });
+    }
+    getReason() {
+        return {
+            type: ToolConfirmKind.Skipped
+        };
+    }
+}
+class ConfigureToolsAction extends Action2 {
+    static {
+        this.ID = "workbench.action.chat.configureTools";
+    }
+    constructor() {
+        super({
+            id: ConfigureToolsAction.ID,
+            title: ( localize(5016, "Configure Tools...")),
+            icon: Codicon.settings,
+            f1: false,
+            category: CHAT_CATEGORY,
+            precondition: ( ChatContextKeys.chatModeKind.isEqualTo(ChatModeKind.Agent)),
+            menu: [{
+                when: ( ContextKeyExpr.and(( ChatContextKeys.chatModeKind.isEqualTo(ChatModeKind.Agent)), ( ChatContextKeys.lockedToCodingAgent.negate()))),
+                id: MenuId.ChatInput,
+                group: "navigation",
+                order: 100
+            }]
+        });
+    }
+    async run(accessor, ...args) {
+        const instaService = accessor.get(IInstantiationService);
+        const chatWidgetService = accessor.get(IChatWidgetService);
+        const telemetryService = accessor.get(ITelemetryService);
+        let widget = chatWidgetService.lastFocusedWidget;
+        if (!widget) {
+            widget = this.extractWidget(args);
+        }
+        if (!widget) {
+            return;
+        }
+        const source = this.extractSource(args) ?? "chatInput";
+        let placeholder;
+        let description;
+        const {
+            entriesScope,
+            entriesMap
+        } = widget.input.selectedToolsModel;
+        switch (entriesScope) {
+        case ToolsScope.Session:
+            placeholder = ( localize(5017, "Select tools for this chat session"));
+            description = ( localize(5018, "The selected tools were configured only for this chat session."));
+            break;
+        case ToolsScope.Agent:
+            placeholder = ( localize(5019, "Select tools for this custom agent"));
+            description = ( localize(
+                5020,
+                "The selected tools are configured by the '{0}' custom agent. Changes to the tools will be applied to the custom agent file as well.",
+                widget.input.currentModeObs.get().label.get()
+            ));
+            break;
+        case ToolsScope.Agent_ReadOnly:
+            placeholder = ( localize(5021, "Select tools for this custom agent"));
+            description = ( localize(
+                5022,
+                "The selected tools are configured by the '{0}' custom agent. Changes to the tools will only be used for this session and will not change the '{0}' custom agent.",
+                widget.input.currentModeObs.get().label.get()
+            ));
+            break;
+        case ToolsScope.Global:
+            placeholder = ( localize(5023, "Select tools that are available to chat."));
+            description = ( localize(
+                5024,
+                "The selected tools will be applied globally for all chat sessions that use the default agent."
+            ));
+            break;
+        }
+        const cts = ( new CancellationTokenSource());
+        const initialMode = widget.input.currentModeObs.get();
+        const modeListener = autorun(reader => {
+            if (initialMode.id !== widget.input.currentModeObs.read(reader).id) {
+                cts.cancel();
+            }
+        });
+        try {
+            const result = await instaService.invokeFunction(
+                showToolsPicker,
+                placeholder,
+                source,
+                description,
+                () => entriesMap.get(),
+                widget.input.selectedLanguageModel.get()?.metadata,
+                cts.token
+            );
+            if (result) {
+                widget.input.selectedToolsModel.set(result, false);
+            }
+        } finally {
+            modeListener.dispose();
+            cts.dispose();
+        }
+        const tools = widget.input.selectedToolsModel.entriesMap.get();
+        telemetryService.publicLog2("chat/selectedTools", {
+            total: tools.size,
+            enabled: Iterable.reduce(tools, (prev, [_, enabled]) => enabled ? prev + 1 : prev, 0)
+        });
+    }
+    extractWidget(args) {
+        function isChatActionContext(obj) {
+            return !!obj && typeof obj === "object" && !!obj.widget;
+        }
+        for (const arg of args) {
+            if (isChatActionContext(arg)) {
+                return arg.widget;
+            }
+        }
+        return undefined;
+    }
+    extractSource(args) {
+        function isChatActionSource(obj) {
+            return !!obj && typeof obj === "object" && !!obj.source;
+        }
+        for (const arg of args) {
+            if (isChatActionSource(arg)) {
+                return arg.source;
+            }
+        }
+        return undefined;
+    }
+}
+function registerChatToolActions() {
+    registerAction2(AcceptToolConfirmation);
+    registerAction2(SkipToolConfirmation);
+    registerAction2(ConfigureToolsAction);
+}
+
+export { ConfigureToolsAction, registerChatToolActions };

package/vscode/src/vs/workbench/contrib/chat/browser/actions/chatToolPicker.d.ts

@@ -0,0 +1,22 @@
+import { CancellationToken } from "@codingame/monaco-vscode-api/vscode/vs/base/common/cancellation";
+import { ServicesAccessor } from "@codingame/monaco-vscode-api/vscode/vs/platform/instantiation/common/instantiation";
+import { ILanguageModelChatMetadata } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/languageModels";
+import { IToolData, IToolSet } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/tools/languageModelToolsService";
+/**
+ * New QuickTree implementation of the tools picker.
+ * Uses IQuickTree to provide a true hierarchical tree structure with:
+ * - Collapsible nodes for buckets and toolsets
+ * - Checkbox state management with parent-child relationships
+ * - Special handling for MCP servers (server as bucket, tools as direct children)
+ * - Built-in filtering and search capabilities
+ *
+ * @param accessor - Service accessor for dependency injection
+ * @param placeHolder - Placeholder text shown in the picker
+ * @param description - Optional description text shown in the picker
+ * @param toolsEntries - Optional initial selection state for tools and toolsets
+ * @param modelId - Optional model ID to filter tools by supported models
+ * @param onUpdate - Optional callback fired when the selection changes
+ * @param token - Optional cancellation token to close the picker when cancelled
+ * @returns Promise resolving to the final selection map, or undefined if cancelled
+ */
+export declare function showToolsPicker(accessor: ServicesAccessor, placeHolder: string, source: string, description?: string, getToolsEntries?: () => ReadonlyMap<IToolSet | IToolData, boolean>, model?: ILanguageModelChatMetadata | undefined, token?: CancellationToken): Promise<ReadonlyMap<IToolSet | IToolData, boolean> | undefined>;