@theia/ai-chat-ui 1.63.0-next.0 → 1.63.0-next.52

package/lib/browser/chat-node-toolbar-action-contribution.js

@@ -53,6 +53,10 @@ var ChatNodeToolbarCommands;
         id: 'chat:node:toolbar:cancel-request',
         category: CHAT_NODE_TOOLBAR_CATEGORY,
     }, '', CHAT_NODE_TOOLBAR_CATEGORY_KEY);
+    ChatNodeToolbarCommands.RETRY = core_1.Command.toLocalizedCommand({
+        id: 'chat:node:toolbar:retry-message',
+        category: CHAT_NODE_TOOLBAR_CATEGORY,
+    }, 'Retry', CHAT_NODE_TOOLBAR_CATEGORY_KEY);
 })(ChatNodeToolbarCommands || (exports.ChatNodeToolbarCommands = ChatNodeToolbarCommands = {}));
 class DefaultChatNodeToolbarActionContribution {
     getToolbarActions(node) {
@@ -71,6 +75,15 @@ class DefaultChatNodeToolbarActionContribution {
                 }];
         }
         else {
+            const shouldShowRetry = node.response.isError || node.response.isCanceled;
+            if (shouldShowRetry) {
+                return [{
+                        commandId: ChatNodeToolbarCommands.RETRY.id,
+                        icon: (0, browser_1.codicon)('refresh'),
+                        tooltip: core_1.nls.localize('theia/ai/chat-ui/node/toolbar/retry', 'Retry'),
+                        priority: -1 // Higher priority to show it first
+                    }];
+            }
             return [];
         }
     }

package/lib/browser/chat-node-toolbar-action-contribution.js.map

@@ -1 +1 @@
-{"version":3,"file":"chat-node-toolbar-action-contribution.js","sourceRoot":"","sources":["../../src/browser/chat-node-toolbar-action-contribution.ts"],"names":[],"mappings":";;;AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;AAChF,sCAA2C;AAC3C,qDAAkD;AAClD,qDAA4E;AAC5E,4CAA0D;AAqB1D;;;;;;;;;;;;;;;;;;;GAmBG;AACU,QAAA,iCAAiC,GAAG,MAAM,CAAC,mCAAmC,CAAC,CAAC;AAQ7F,IAAiB,uBAAuB,CAavC;AAbD,WAAiB,uBAAuB;IACpC,MAAM,0BAA0B,GAAG,iBAAiB,CAAC;IACrD,MAAM,8BAA8B,GAAG,UAAG,CAAC,aAAa,CAAC,0BAA0B,CAAC,CAAC;IAExE,4BAAI,GAAG,cAAO,CAAC,kBAAkB,CAAC;QAC3C,EAAE,EAAE,gCAAgC;QACpC,QAAQ,EAAE,0BAA0B;KACvC,EAAE,EAAE,EAAE,8BAA8B,CAAC,CAAC;IAE1B,8BAAM,GAAG,cAAO,CAAC,kBAAkB,CAAC;QAC7C,EAAE,EAAE,kCAAkC;QACtC,QAAQ,EAAE,0BAA0B;KACvC,EAAE,EAAE,EAAE,8BAA8B,CAAC,CAAC;AAC3C,CAAC,EAbgB,uBAAuB,uCAAvB,uBAAuB,QAavC;AAED,MAAa,wCAAwC;IACjD,iBAAiB,CAAC,IAAgC;QAC9C,IAAI,IAAA,8BAAa,EAAC,IAAI,CAAC,EAAE,CAAC;YACtB,IAAI,kCAAwB,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;gBACnD,OAAO,CAAC;wBACJ,SAAS,EAAE,uBAAuB,CAAC,MAAM,CAAC,EAAE;wBAC5C,IAAI,EAAE,IAAA,iBAAO,EAAC,OAAO,CAAC;wBACtB,OAAO,EAAE,UAAG,CAAC,QAAQ,CAAC,sCAAsC,EAAE,QAAQ,CAAC;qBAC1E,CAAC,CAAC;YACP,CAAC;YACD,OAAO,CAAC;oBACJ,SAAS,EAAE,uBAAuB,CAAC,IAAI,CAAC,EAAE;oBAC1C,IAAI,EAAE,IAAA,iBAAO,EAAC,MAAM,CAAC;oBACrB,OAAO,EAAE,UAAG,CAAC,QAAQ,CAAC,oCAAoC,EAAE,MAAM,CAAC;iBACtE,CAAC,CAAC;QACP,CAAC;aAAM,CAAC;YACJ,OAAO,EAAE,CAAC;QACd,CAAC;IACL,CAAC;CACJ;AAnBD,4FAmBC"}
+{"version":3,"file":"chat-node-toolbar-action-contribution.js","sourceRoot":"","sources":["../../src/browser/chat-node-toolbar-action-contribution.ts"],"names":[],"mappings":";;;AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;AAChF,sCAA2C;AAC3C,qDAAkD;AAClD,qDAA4E;AAC5E,4CAA0D;AAqB1D;;;;;;;;;;;;;;;;;;;GAmBG;AACU,QAAA,iCAAiC,GAAG,MAAM,CAAC,mCAAmC,CAAC,CAAC;AAQ7F,IAAiB,uBAAuB,CAkBvC;AAlBD,WAAiB,uBAAuB;IACpC,MAAM,0BAA0B,GAAG,iBAAiB,CAAC;IACrD,MAAM,8BAA8B,GAAG,UAAG,CAAC,aAAa,CAAC,0BAA0B,CAAC,CAAC;IAExE,4BAAI,GAAG,cAAO,CAAC,kBAAkB,CAAC;QAC3C,EAAE,EAAE,gCAAgC;QACpC,QAAQ,EAAE,0BAA0B;KACvC,EAAE,EAAE,EAAE,8BAA8B,CAAC,CAAC;IAE1B,8BAAM,GAAG,cAAO,CAAC,kBAAkB,CAAC;QAC7C,EAAE,EAAE,kCAAkC;QACtC,QAAQ,EAAE,0BAA0B;KACvC,EAAE,EAAE,EAAE,8BAA8B,CAAC,CAAC;IAE1B,6BAAK,GAAG,cAAO,CAAC,kBAAkB,CAAC;QAC5C,EAAE,EAAE,iCAAiC;QACrC,QAAQ,EAAE,0BAA0B;KACvC,EAAE,OAAO,EAAE,8BAA8B,CAAC,CAAC;AAChD,CAAC,EAlBgB,uBAAuB,uCAAvB,uBAAuB,QAkBvC;AAED,MAAa,wCAAwC;IACjD,iBAAiB,CAAC,IAAgC;QAC9C,IAAI,IAAA,8BAAa,EAAC,IAAI,CAAC,EAAE,CAAC;YACtB,IAAI,kCAAwB,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;gBACnD,OAAO,CAAC;wBACJ,SAAS,EAAE,uBAAuB,CAAC,MAAM,CAAC,EAAE;wBAC5C,IAAI,EAAE,IAAA,iBAAO,EAAC,OAAO,CAAC;wBACtB,OAAO,EAAE,UAAG,CAAC,QAAQ,CAAC,sCAAsC,EAAE,QAAQ,CAAC;qBAC1E,CAAC,CAAC;YACP,CAAC;YACD,OAAO,CAAC;oBACJ,SAAS,EAAE,uBAAuB,CAAC,IAAI,CAAC,EAAE;oBAC1C,IAAI,EAAE,IAAA,iBAAO,EAAC,MAAM,CAAC;oBACrB,OAAO,EAAE,UAAG,CAAC,QAAQ,CAAC,oCAAoC,EAAE,MAAM,CAAC;iBACtE,CAAC,CAAC;QACP,CAAC;aAAM,CAAC;YACJ,MAAM,eAAe,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC;YAC1E,IAAI,eAAe,EAAE,CAAC;gBAClB,OAAO,CAAC;wBACJ,SAAS,EAAE,uBAAuB,CAAC,KAAK,CAAC,EAAE;wBAC3C,IAAI,EAAE,IAAA,iBAAO,EAAC,SAAS,CAAC;wBACxB,OAAO,EAAE,UAAG,CAAC,QAAQ,CAAC,qCAAqC,EAAE,OAAO,CAAC;wBACrE,QAAQ,EAAE,CAAC,CAAC,CAAC,mCAAmC;qBACnD,CAAC,CAAC;YACP,CAAC;YACD,OAAO,EAAE,CAAC;QACd,CAAC;IACL,CAAC;CACJ;AA5BD,4FA4BC"}

package/lib/browser/chat-response-renderer/delegation-response-renderer.d.ts

@@ -0,0 +1,14 @@
+/// <reference types="react" />
+import { ChatResponseContent } from '@theia/ai-chat';
+import { ChatResponsePartRenderer } from '../chat-response-part-renderer';
+import * as React from '@theia/core/shared/react';
+import { DelegationResponseContent } from '@theia/ai-chat/lib/browser/delegation-response-content';
+import { ResponseNode } from '../chat-tree-view';
+import { SubChatWidgetFactory } from '../chat-tree-view/sub-chat-widget';
+export declare class DelegationResponseRenderer implements ChatResponsePartRenderer<DelegationResponseContent> {
+    subChatWidgetFactory: SubChatWidgetFactory;
+    canHandle(response: ChatResponseContent): number;
+    render(response: DelegationResponseContent, parentNode: ResponseNode): React.ReactNode;
+    private renderExpandableNode;
+}
+//# sourceMappingURL=delegation-response-renderer.d.ts.map

package/lib/browser/chat-response-renderer/delegation-response-renderer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"delegation-response-renderer.d.ts","sourceRoot":"","sources":["../../../src/browser/chat-response-renderer/delegation-response-renderer.tsx"],"names":[],"mappings":";AAgBA,OAAO,EAAyB,mBAAmB,EAAqB,MAAM,gBAAgB,CAAC;AAC/F,OAAO,EAAE,wBAAwB,EAAE,MAAM,gCAAgC,CAAC;AAC1E,OAAO,KAAK,KAAK,MAAM,0BAA0B,CAAC;AAClD,OAAO,EAAE,yBAAyB,EAA+B,MAAM,wDAAwD,CAAC;AAChI,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEjD,OAAO,EAAE,oBAAoB,EAAE,MAAM,mCAAmC,CAAC;AAGzE,qBACa,0BAA2B,YAAW,wBAAwB,CAAC,yBAAyB,CAAC;IAGlG,oBAAoB,EAAE,oBAAoB,CAAC;IAE3C,SAAS,CAAC,QAAQ,EAAE,mBAAmB,GAAG,MAAM;IAMhD,MAAM,CAAC,QAAQ,EAAE,yBAAyB,EAAE,UAAU,EAAE,YAAY,GAAG,KAAK,CAAC,SAAS;IAItF,OAAO,CAAC,oBAAoB;CAQ/B"}

package/lib/browser/chat-response-renderer/delegation-response-renderer.js

@@ -0,0 +1,144 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DelegationResponseRenderer = void 0;
+const tslib_1 = require("tslib");
+// *****************************************************************************
+// Copyright (C) 2025 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+const inversify_1 = require("@theia/core/shared/inversify");
+const React = require("@theia/core/shared/react");
+const delegation_response_content_1 = require("@theia/ai-chat/lib/browser/delegation-response-content");
+const sub_chat_widget_1 = require("../chat-tree-view/sub-chat-widget");
+const core_1 = require("@theia/core");
+let DelegationResponseRenderer = class DelegationResponseRenderer {
+    canHandle(response) {
+        if ((0, delegation_response_content_1.isDelegationResponseContent)(response)) {
+            return 10;
+        }
+        return -1;
+    }
+    render(response, parentNode) {
+        return this.renderExpandableNode(response, parentNode);
+    }
+    renderExpandableNode(response, parentNode) {
+        return React.createElement(DelegatedChat, { response: response.response, agentId: response.agentId, prompt: response.prompt, parentNode: parentNode, subChatWidgetFactory: this.subChatWidgetFactory });
+    }
+};
+exports.DelegationResponseRenderer = DelegationResponseRenderer;
+tslib_1.__decorate([
+    (0, inversify_1.inject)(sub_chat_widget_1.SubChatWidgetFactory),
+    tslib_1.__metadata("design:type", Function)
+], DelegationResponseRenderer.prototype, "subChatWidgetFactory", void 0);
+exports.DelegationResponseRenderer = DelegationResponseRenderer = tslib_1.__decorate([
+    (0, inversify_1.injectable)()
+], DelegationResponseRenderer);
+class DelegatedChat extends React.Component {
+    constructor(props) {
+        super(props);
+        this.toDispose = new core_1.DisposableCollection();
+        this.state = {
+            node: undefined
+        };
+        this.widget = props.subChatWidgetFactory();
+    }
+    componentDidMount() {
+        // Start rendering as soon as the response is created (streaming mode)
+        this.props.response.responseCreated.then(chatModel => {
+            const node = mapResponseToNode(chatModel, this.props.parentNode);
+            this.setState({ node });
+            // Listen for changes to update the rendering as the response streams in
+            const changeListener = () => {
+                // Force re-render when the response content changes
+                this.forceUpdate();
+            };
+            this.toDispose.push(chatModel.onDidChange(changeListener));
+        }).catch(error => {
+            console.error('Failed to create delegated chat response:', error);
+            // Still try to handle completion in case of partial success
+        });
+        // Keep the completion handling for final cleanup if needed
+        this.props.response.responseCompleted.then(() => {
+            // Final update when response is complete
+            this.forceUpdate();
+        }).catch(error => {
+            console.error('Error in delegated chat response completion:', error);
+            // Force update anyway to show any partial content or error state
+            this.forceUpdate();
+        });
+    }
+    componentWillUnmount() {
+        this.toDispose.dispose();
+    }
+    render() {
+        var _a, _b, _c, _d, _e, _f;
+        const { agentId, prompt } = this.props;
+        const hasNode = !!this.state.node;
+        const isComplete = (_b = (_a = this.state.node) === null || _a === void 0 ? void 0 : _a.response.isComplete) !== null && _b !== void 0 ? _b : false;
+        const isCanceled = (_d = (_c = this.state.node) === null || _c === void 0 ? void 0 : _c.response.isCanceled) !== null && _d !== void 0 ? _d : false;
+        const isError = (_f = (_e = this.state.node) === null || _e === void 0 ? void 0 : _e.response.isError) !== null && _f !== void 0 ? _f : false;
+        let statusIcon = '';
+        let statusText = '';
+        if (hasNode) {
+            if (isComplete) {
+                statusIcon = 'codicon-check';
+                statusText = 'completed';
+            }
+            else if (isCanceled) {
+                statusIcon = 'codicon-cancel';
+                statusText = 'canceled';
+            }
+            else if (isError) {
+                statusIcon = 'codicon-error';
+                statusText = 'error';
+            }
+            else {
+                statusIcon = 'codicon-loading';
+                statusText = 'generating...';
+            }
+        }
+        else {
+            statusIcon = 'codicon-loading';
+            statusText = 'starting...';
+        }
+        return (React.createElement("div", { className: "theia-delegation-container" },
+            React.createElement("details", { className: "delegation-response-details" },
+                React.createElement("summary", { className: "delegation-summary" },
+                    React.createElement("div", { className: "delegation-header" },
+                        React.createElement("span", { className: "delegation-agent" },
+                            React.createElement("strong", null, "Agent:"),
+                            " ",
+                            agentId),
+                        React.createElement("span", { className: "delegation-status" },
+                            React.createElement("span", { className: `codicon ${statusIcon} delegation-status-icon` }),
+                            React.createElement("span", { className: "delegation-status-text" }, statusText)))),
+                React.createElement("div", { className: "delegation-content" },
+                    React.createElement("div", { className: "delegation-prompt-section" },
+                        React.createElement("strong", null, "Delegated prompt:"),
+                        React.createElement("div", { className: "delegation-prompt" }, prompt)),
+                    React.createElement("div", { className: "delegation-response-section" },
+                        React.createElement("strong", null, "Response:"),
+                        React.createElement("div", { className: 'delegation-response-placeholder' }, hasNode && this.state.node ? this.widget.renderChatResponse(this.state.node) :
+                            React.createElement("div", { className: "theia-ChatContentInProgress" }, "Starting delegation...")))))));
+    }
+}
+function mapResponseToNode(response, parentNode) {
+    return {
+        id: response.id,
+        parent: parentNode,
+        response,
+        sessionId: parentNode.sessionId
+    };
+}
+//# sourceMappingURL=delegation-response-renderer.js.map

package/lib/browser/chat-response-renderer/delegation-response-renderer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"delegation-response-renderer.js","sourceRoot":"","sources":["../../../src/browser/chat-response-renderer/delegation-response-renderer.tsx"],"names":[],"mappings":";;;;AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;AAChF,4DAAkE;AAGlE,kDAAkD;AAClD,wGAAgI;AAGhI,uEAAyE;AACzE,sCAAmD;AAG5C,IAAM,0BAA0B,GAAhC,MAAM,0BAA0B;IAKnC,SAAS,CAAC,QAA6B;QACnC,IAAI,IAAA,yDAA2B,EAAC,QAAQ,CAAC,EAAE,CAAC;YACxC,OAAO,EAAE,CAAC;QACd,CAAC;QACD,OAAO,CAAC,CAAC,CAAC;IACd,CAAC;IACD,MAAM,CAAC,QAAmC,EAAE,UAAwB;QAChE,OAAO,IAAI,CAAC,oBAAoB,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;IAC3D,CAAC;IAEO,oBAAoB,CAAC,QAAmC,EAAE,UAAwB;QACtF,OAAO,oBAAC,aAAa,IACjB,QAAQ,EAAE,QAAQ,CAAC,QAAQ,EAC3B,OAAO,EAAE,QAAQ,CAAC,OAAO,EACzB,MAAM,EAAE,QAAQ,CAAC,MAAM,EACvB,UAAU,EAAE,UAAU,EACtB,oBAAoB,EAAE,IAAI,CAAC,oBAAoB,GAAI,CAAC;IAC5D,CAAC;CACJ,CAAA;AAvBY,gEAA0B;AAGnC;IADC,IAAA,kBAAM,EAAC,sCAAoB,CAAC;;wEACc;qCAHlC,0BAA0B;IADtC,IAAA,sBAAU,GAAE;GACA,0BAA0B,CAuBtC;AAcD,MAAM,aAAc,SAAQ,KAAK,CAAC,SAAiD;IAI/E,YAAY,KAAyB;QACjC,KAAK,CAAC,KAAK,CAAC,CAAC;QAHA,cAAS,GAAG,IAAI,2BAAoB,EAAE,CAAC;QAIpD,IAAI,CAAC,KAAK,GAAG;YACT,IAAI,EAAE,SAAS;SAClB,CAAC;QACF,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,oBAAoB,EAAE,CAAC;IAC/C,CAAC;IAEQ,iBAAiB;QACtB,sEAAsE;QACtE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE;YACjD,MAAM,IAAI,GAAG,iBAAiB,CAAC,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YACjE,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;YAExB,wEAAwE;YACxE,MAAM,cAAc,GAAG,GAAG,EAAE;gBACxB,oDAAoD;gBACpD,IAAI,CAAC,WAAW,EAAE,CAAC;YACvB,CAAC,CAAC;YACF,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE;YACb,OAAO,CAAC,KAAK,CAAC,2CAA2C,EAAE,KAAK,CAAC,CAAC;YAClE,4DAA4D;QAChE,CAAC,CAAC,CAAC;QAEH,2DAA2D;QAC3D,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,EAAE;YAC5C,yCAAyC;YACzC,IAAI,CAAC,WAAW,EAAE,CAAC;QACvB,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE;YACb,OAAO,CAAC,KAAK,CAAC,8CAA8C,EAAE,KAAK,CAAC,CAAC;YACrE,iEAAiE;YACjE,IAAI,CAAC,WAAW,EAAE,CAAC;QACvB,CAAC,CAAC,CAAC;IACP,CAAC;IAEQ,oBAAoB;QACzB,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;IAC7B,CAAC;IAEQ,MAAM;;QACX,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC;QACvC,MAAM,OAAO,GAAG,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;QAClC,MAAM,UAAU,GAAG,MAAA,MAAA,IAAI,CAAC,KAAK,CAAC,IAAI,0CAAE,QAAQ,CAAC,UAAU,mCAAI,KAAK,CAAC;QACjE,MAAM,UAAU,GAAG,MAAA,MAAA,IAAI,CAAC,KAAK,CAAC,IAAI,0CAAE,QAAQ,CAAC,UAAU,mCAAI,KAAK,CAAC;QACjE,MAAM,OAAO,GAAG,MAAA,MAAA,IAAI,CAAC,KAAK,CAAC,IAAI,0CAAE,QAAQ,CAAC,OAAO,mCAAI,KAAK,CAAC;QAE3D,IAAI,UAAU,GAAG,EAAE,CAAC;QACpB,IAAI,UAAU,GAAG,EAAE,CAAC;QACpB,IAAI,OAAO,EAAE,CAAC;YACV,IAAI,UAAU,EAAE,CAAC;gBACb,UAAU,GAAG,eAAe,CAAC;gBAC7B,UAAU,GAAG,WAAW,CAAC;YAC7B,CAAC;iBAAM,IAAI,UAAU,EAAE,CAAC;gBACpB,UAAU,GAAG,gBAAgB,CAAC;gBAC9B,UAAU,GAAG,UAAU,CAAC;YAC5B,CAAC;iBAAM,IAAI,OAAO,EAAE,CAAC;gBACjB,UAAU,GAAG,eAAe,CAAC;gBAC7B,UAAU,GAAG,OAAO,CAAC;YACzB,CAAC;iBAAM,CAAC;gBACJ,UAAU,GAAG,iBAAiB,CAAC;gBAC/B,UAAU,GAAG,eAAe,CAAC;YACjC,CAAC;QACL,CAAC;aAAM,CAAC;YACJ,UAAU,GAAG,iBAAiB,CAAC;YAC/B,UAAU,GAAG,aAAa,CAAC;QAC/B,CAAC;QAED,OAAO,CACH,6BAAK,SAAS,EAAC,4BAA4B;YACvC,iCAAS,SAAS,EAAC,6BAA6B;gBAC5C,iCAAS,SAAS,EAAC,oBAAoB;oBACnC,6BAAK,SAAS,EAAC,mBAAmB;wBAC9B,8BAAM,SAAS,EAAC,kBAAkB;4BAC9B,6CAAuB;;4BAAE,OAAO,CAC7B;wBACP,8BAAM,SAAS,EAAC,mBAAmB;4BAC/B,8BAAM,SAAS,EAAE,WAAW,UAAU,yBAAyB,GAAS;4BACxE,8BAAM,SAAS,EAAC,wBAAwB,IAAE,UAAU,CAAQ,CACzD,CACL,CACA;gBACV,6BAAK,SAAS,EAAC,oBAAoB;oBAC/B,6BAAK,SAAS,EAAC,2BAA2B;wBACtC,wDAAkC;wBAClC,6BAAK,SAAS,EAAC,mBAAmB,IAAE,MAAM,CAAO,CAC/C;oBACN,6BAAK,SAAS,EAAC,6BAA6B;wBACxC,gDAA0B;wBAC1B,6BAAK,SAAS,EAAC,iCAAiC,IAC3C,OAAO,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;4BAC3E,6BAAK,SAAS,EAAC,6BAA6B,6BAA6B,CAE3E,CACJ,CACJ,CACA,CACR,CACT,CAAC;IACN,CAAC;CACJ;AAED,SAAS,iBAAiB,CAAC,QAA2B,EAAE,UAAwB;IAC5E,OAAO;QACH,EAAE,EAAE,QAAQ,CAAC,EAAE;QACf,MAAM,EAAE,UAA0C;QAClD,QAAQ;QACR,SAAS,EAAE,UAAU,CAAC,SAAS;KAClC,CAAC;AACN,CAAC"}

package/lib/browser/chat-response-renderer/index.d.ts

@@ -8,4 +8,6 @@ export * from './text-part-renderer';
 export * from './toolcall-part-renderer';
 export * from './thinking-part-renderer';
 export * from './progress-part-renderer';
+export * from './tool-confirmation';
+export * from './delegation-response-renderer';
 //# sourceMappingURL=index.d.ts.map

package/lib/browser/chat-response-renderer/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/browser/chat-response-renderer/index.ts"],"names":[],"mappings":"AAeA,cAAc,yBAAyB,CAAC;AACxC,cAAc,sBAAsB,CAAC;AACrC,cAAc,yBAAyB,CAAC;AACxC,cAAc,uBAAuB,CAAC;AACtC,cAAc,mCAAmC,CAAC;AAClD,cAAc,0BAA0B,CAAC;AACzC,cAAc,sBAAsB,CAAC;AACrC,cAAc,0BAA0B,CAAC;AACzC,cAAc,0BAA0B,CAAC;AACzC,cAAc,0BAA0B,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/browser/chat-response-renderer/index.ts"],"names":[],"mappings":"AAeA,cAAc,yBAAyB,CAAC;AACxC,cAAc,sBAAsB,CAAC;AACrC,cAAc,yBAAyB,CAAC;AACxC,cAAc,uBAAuB,CAAC;AACtC,cAAc,mCAAmC,CAAC;AAClD,cAAc,0BAA0B,CAAC;AACzC,cAAc,sBAAsB,CAAC;AACrC,cAAc,0BAA0B,CAAC;AACzC,cAAc,0BAA0B,CAAC;AACzC,cAAc,0BAA0B,CAAC;AACzC,cAAc,qBAAqB,CAAC;AACpC,cAAc,gCAAgC,CAAC"}

package/lib/browser/chat-response-renderer/index.js

@@ -26,4 +26,6 @@ tslib_1.__exportStar(require("./text-part-renderer"), exports);
 tslib_1.__exportStar(require("./toolcall-part-renderer"), exports);
 tslib_1.__exportStar(require("./thinking-part-renderer"), exports);
 tslib_1.__exportStar(require("./progress-part-renderer"), exports);
+tslib_1.__exportStar(require("./tool-confirmation"), exports);
+tslib_1.__exportStar(require("./delegation-response-renderer"), exports);
 //# sourceMappingURL=index.js.map

package/lib/browser/chat-response-renderer/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/browser/chat-response-renderer/index.ts"],"names":[],"mappings":";;;AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;AAChF,kEAAwC;AACxC,+DAAqC;AACrC,kEAAwC;AACxC,gEAAsC;AACtC,4EAAkD;AAClD,mEAAyC;AACzC,+DAAqC;AACrC,mEAAyC;AACzC,mEAAyC;AACzC,mEAAyC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/browser/chat-response-renderer/index.ts"],"names":[],"mappings":";;;AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;AAChF,kEAAwC;AACxC,+DAAqC;AACrC,kEAAwC;AACxC,gEAAsC;AACtC,4EAAkD;AAClD,mEAAyC;AACzC,+DAAqC;AACrC,mEAAyC;AACzC,mEAAyC;AACzC,mEAAyC;AACzC,8DAAoC;AACpC,yEAA+C"}

package/lib/browser/chat-response-renderer/tool-confirmation.d.ts

@@ -0,0 +1,17 @@
+/// <reference types="react" />
+import * as React from '@theia/core/shared/react';
+import { ToolCallChatResponseContent } from '@theia/ai-chat/lib/common';
+/**
+ * States the tool confirmation component can be in
+ */
+export type ToolConfirmationState = 'waiting' | 'allowed' | 'denied';
+export interface ToolConfirmationProps {
+    response: ToolCallChatResponseContent;
+    onAllow: (mode?: 'once' | 'session' | 'forever') => void;
+    onDeny: (mode?: 'once' | 'session' | 'forever') => void;
+}
+/**
+ * Component that displays approval/denial buttons for tool execution
+ */
+export declare const ToolConfirmation: React.FC<ToolConfirmationProps>;
+//# sourceMappingURL=tool-confirmation.d.ts.map

package/lib/browser/chat-response-renderer/tool-confirmation.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-confirmation.d.ts","sourceRoot":"","sources":["../../../src/browser/chat-response-renderer/tool-confirmation.tsx"],"names":[],"mappings":";AAgBA,OAAO,KAAK,KAAK,MAAM,0BAA0B,CAAC;AAGlD,OAAO,EAAE,2BAA2B,EAAE,MAAM,2BAA2B,CAAC;AAExE;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAAG,SAAS,GAAG,SAAS,GAAG,QAAQ,CAAC;AAErE,MAAM,WAAW,qBAAqB;IAClC,QAAQ,EAAE,2BAA2B,CAAC;IACtC,OAAO,EAAE,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS,KAAK,IAAI,CAAC;IACzD,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS,KAAK,IAAI,CAAC;CAC3D;AAED;;GAEG;AACH,eAAO,MAAM,gBAAgB,EAAE,KAAK,CAAC,EAAE,CAAC,qBAAqB,CAyI5D,CAAC"}

package/lib/browser/chat-response-renderer/tool-confirmation.js

@@ -0,0 +1,120 @@
+"use strict";
+// *****************************************************************************
+// Copyright (C) 2025 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ToolConfirmation = void 0;
+const React = require("@theia/core/shared/react");
+const nls_1 = require("@theia/core/lib/common/nls");
+const browser_1 = require("@theia/core/lib/browser");
+/**
+ * Component that displays approval/denial buttons for tool execution
+ */
+const ToolConfirmation = ({ response, onAllow, onDeny }) => {
+    const [state, setState] = React.useState('waiting');
+    // Track selected mode for each action
+    const [allowMode, setAllowMode] = React.useState('once');
+    const [denyMode, setDenyMode] = React.useState('once');
+    const [dropdownOpen, setDropdownOpen] = React.useState(undefined);
+    const handleAllow = React.useCallback(() => {
+        setState('allowed');
+        onAllow(allowMode);
+    }, [onAllow, allowMode]);
+    const handleDeny = React.useCallback(() => {
+        setState('denied');
+        onDeny(denyMode);
+    }, [onDeny, denyMode]);
+    if (state === 'allowed') {
+        return (React.createElement("div", { className: "theia-tool-confirmation-status allowed" },
+            React.createElement("span", { className: (0, browser_1.codicon)('check') }),
+            " ",
+            nls_1.nls.localize('theia/ai/chat-ui/toolconfirmation/allowed', 'Tool execution allowed')));
+    }
+    if (state === 'denied') {
+        return (React.createElement("div", { className: "theia-tool-confirmation-status denied" },
+            React.createElement("span", { className: (0, browser_1.codicon)('close') }),
+            " ",
+            nls_1.nls.localize('theia/ai/chat-ui/toolconfirmation/denied', 'Tool execution denied')));
+    }
+    // Helper for dropdown options
+    const MODES = ['once', 'session', 'forever'];
+    // Unified labels for both main button and dropdown, as requested
+    const modeLabel = (type, mode) => {
+        if (type === 'allow') {
+            switch (mode) {
+                case 'once': return nls_1.nls.localize('theia/ai/chat-ui/toolconfirmation/allow', 'Allow');
+                case 'session': return nls_1.nls.localize('theia/ai/chat-ui/toolconfirmation/allow-session', 'Allow for this Chat');
+                case 'forever': return nls_1.nls.localize('theia/ai/chat-ui/toolconfirmation/allow-forever', 'Always Allow');
+            }
+        }
+        else {
+            switch (mode) {
+                case 'once': return nls_1.nls.localize('theia/ai/chat-ui/toolconfirmation/deny', 'Deny');
+                case 'session': return nls_1.nls.localize('theia/ai/chat-ui/toolconfirmation/deny-session', 'Deny for this Chat');
+                case 'forever': return nls_1.nls.localize('theia/ai/chat-ui/toolconfirmation/deny-forever', 'Always Deny');
+            }
+        }
+    };
+    // Main button label is always the same as the dropdown label for the selected mode
+    const mainButtonLabel = modeLabel; // Use the same function for both
+    // Tooltips for dropdown options
+    const modeTooltip = (type, mode) => {
+        if (type === 'allow') {
+            switch (mode) {
+                case 'once': return nls_1.nls.localize('theia/ai/chat-ui/toolconfirmation/allow-tooltip', 'Allow this tool call once');
+                case 'session': return nls_1.nls.localize('theia/ai/chat-ui/toolconfirmation/allow-session-tooltip', 'Allow all calls of this tool for this chat session');
+                case 'forever': return nls_1.nls.localize('theia/ai/chat-ui/toolconfirmation/allow-forever-tooltip', 'Always allow this tool');
+            }
+        }
+        else {
+            switch (mode) {
+                case 'once': return nls_1.nls.localize('theia/ai/chat-ui/toolconfirmation/deny-tooltip', 'Deny this tool call once');
+                case 'session': return nls_1.nls.localize('theia/ai/chat-ui/toolconfirmation/deny-session-tooltip', 'Deny all calls of this tool for this chat session');
+                case 'forever': return nls_1.nls.localize('theia/ai/chat-ui/toolconfirmation/deny-forever-tooltip', 'Always deny this tool');
+            }
+        }
+    };
+    // Split button for approve/deny
+    const renderSplitButton = (type) => {
+        const selectedMode = type === 'allow' ? allowMode : denyMode;
+        const setMode = type === 'allow' ? setAllowMode : setDenyMode;
+        const handleMain = type === 'allow' ? handleAllow : handleDeny;
+        const otherModes = MODES.filter(m => m !== selectedMode);
+        return (React.createElement("div", { className: `theia-tool-confirmation-split-button ${type}`, style: { display: 'inline-flex', position: 'relative' } },
+            React.createElement("button", { className: `theia-button ${type === 'allow' ? 'primary' : 'secondary'} theia-tool-confirmation-main-btn`, onClick: handleMain }, mainButtonLabel(type, selectedMode)),
+            React.createElement("button", { className: `theia-button ${type === 'allow' ? 'primary' : 'secondary'} theia-tool-confirmation-chevron-btn`, onClick: () => setDropdownOpen(dropdownOpen === type ? undefined : type), "aria-haspopup": "true", "aria-expanded": dropdownOpen === type, tabIndex: 0, title: type === 'allow' ? 'More Allow Options' : 'More Deny Options' },
+                React.createElement("span", { className: (0, browser_1.codicon)('chevron-down') })),
+            dropdownOpen === type && (React.createElement("ul", { className: "theia-tool-confirmation-dropdown-menu", onMouseLeave: () => setDropdownOpen(undefined) }, otherModes.map(mode => (React.createElement("li", { key: mode, className: "theia-tool-confirmation-dropdown-item", onClick: () => {
+                    setMode(mode);
+                    setDropdownOpen(undefined);
+                }, title: modeTooltip(type, mode) }, modeLabel(type, mode))))))));
+    };
+    return (React.createElement("div", { className: "theia-tool-confirmation" },
+        React.createElement("div", { className: "theia-tool-confirmation-header" },
+            React.createElement("span", { className: (0, browser_1.codicon)('shield') }),
+            " ",
+            nls_1.nls.localize('theia/ai/chat-ui/toolconfirmation/header', 'Confirm Tool Execution')),
+        React.createElement("div", { className: "theia-tool-confirmation-info" },
+            React.createElement("div", { className: "theia-tool-confirmation-name" },
+                React.createElement("span", { className: "label" },
+                    nls_1.nls.localize('theia/ai/chat-ui/toolconfirmation/tool', 'Tool'),
+                    ":"),
+                React.createElement("span", { className: "value" }, response.name))),
+        React.createElement("div", { className: "theia-tool-confirmation-actions" },
+            renderSplitButton('deny'),
+            renderSplitButton('allow'))));
+};
+exports.ToolConfirmation = ToolConfirmation;
+//# sourceMappingURL=tool-confirmation.js.map

package/lib/browser/chat-response-renderer/tool-confirmation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-confirmation.js","sourceRoot":"","sources":["../../../src/browser/chat-response-renderer/tool-confirmation.tsx"],"names":[],"mappings":";AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;;;AAEhF,kDAAkD;AAClD,oDAAiD;AACjD,qDAAkD;AAclD;;GAEG;AACI,MAAM,gBAAgB,GAAoC,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE;IAC/F,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,KAAK,CAAC,QAAQ,CAAwB,SAAS,CAAC,CAAC;IAC3E,sCAAsC;IACtC,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,KAAK,CAAC,QAAQ,CAAiC,MAAM,CAAC,CAAC;IACzF,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG,KAAK,CAAC,QAAQ,CAAiC,MAAM,CAAC,CAAC;IACvF,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,KAAK,CAAC,QAAQ,CAA+B,SAAS,CAAC,CAAC;IAEhG,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE;QACvC,QAAQ,CAAC,SAAS,CAAC,CAAC;QACpB,OAAO,CAAC,SAAS,CAAC,CAAC;IACvB,CAAC,EAAE,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC;IAEzB,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE;QACtC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACnB,MAAM,CAAC,QAAQ,CAAC,CAAC;IACrB,CAAC,EAAE,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC;IAEvB,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACtB,OAAO,CACH,6BAAK,SAAS,EAAC,wCAAwC;YACnD,8BAAM,SAAS,EAAE,IAAA,iBAAO,EAAC,OAAO,CAAC,GAAS;;YAAE,SAAG,CAAC,QAAQ,CAAC,2CAA2C,EAAE,wBAAwB,CAAC,CAC7H,CACT,CAAC;IACN,CAAC;IAED,IAAI,KAAK,KAAK,QAAQ,EAAE,CAAC;QACrB,OAAO,CACH,6BAAK,SAAS,EAAC,uCAAuC;YAClD,8BAAM,SAAS,EAAE,IAAA,iBAAO,EAAC,OAAO,CAAC,GAAS;;YAAE,SAAG,CAAC,QAAQ,CAAC,0CAA0C,EAAE,uBAAuB,CAAC,CAC3H,CACT,CAAC;IACN,CAAC;IAED,8BAA8B;IAC9B,MAAM,KAAK,GAA0C,CAAC,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;IACpF,iEAAiE;IACjE,MAAM,SAAS,GAAG,CAAC,IAAsB,EAAE,IAAoC,EAAE,EAAE;QAC/E,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;YACnB,QAAQ,IAAI,EAAE,CAAC;gBACX,KAAK,MAAM,CAAC,CAAC,OAAO,SAAG,CAAC,QAAQ,CAAC,yCAAyC,EAAE,OAAO,CAAC,CAAC;gBACrF,KAAK,SAAS,CAAC,CAAC,OAAO,SAAG,CAAC,QAAQ,CAAC,iDAAiD,EAAE,qBAAqB,CAAC,CAAC;gBAC9G,KAAK,SAAS,CAAC,CAAC,OAAO,SAAG,CAAC,QAAQ,CAAC,iDAAiD,EAAE,cAAc,CAAC,CAAC;YAC3G,CAAC;QACL,CAAC;aAAM,CAAC;YACJ,QAAQ,IAAI,EAAE,CAAC;gBACX,KAAK,MAAM,CAAC,CAAC,OAAO,SAAG,CAAC,QAAQ,CAAC,wCAAwC,EAAE,MAAM,CAAC,CAAC;gBACnF,KAAK,SAAS,CAAC,CAAC,OAAO,SAAG,CAAC,QAAQ,CAAC,gDAAgD,EAAE,oBAAoB,CAAC,CAAC;gBAC5G,KAAK,SAAS,CAAC,CAAC,OAAO,SAAG,CAAC,QAAQ,CAAC,gDAAgD,EAAE,aAAa,CAAC,CAAC;YACzG,CAAC;QACL,CAAC;IACL,CAAC,CAAC;IACF,mFAAmF;IACnF,MAAM,eAAe,GAAG,SAAS,CAAC,CAAC,iCAAiC;IAEpE,gCAAgC;IAChC,MAAM,WAAW,GAAG,CAAC,IAAsB,EAAE,IAAoC,EAAE,EAAE;QACjF,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;YACnB,QAAQ,IAAI,EAAE,CAAC;gBACX,KAAK,MAAM,CAAC,CAAC,OAAO,SAAG,CAAC,QAAQ,CAAC,iDAAiD,EAAE,2BAA2B,CAAC,CAAC;gBACjH,KAAK,SAAS,CAAC,CAAC,OAAO,SAAG,CAAC,QAAQ,CAAC,yDAAyD,EAAE,oDAAoD,CAAC,CAAC;gBACrJ,KAAK,SAAS,CAAC,CAAC,OAAO,SAAG,CAAC,QAAQ,CAAC,yDAAyD,EAAE,wBAAwB,CAAC,CAAC;YAC7H,CAAC;QACL,CAAC;aAAM,CAAC;YACJ,QAAQ,IAAI,EAAE,CAAC;gBACX,KAAK,MAAM,CAAC,CAAC,OAAO,SAAG,CAAC,QAAQ,CAAC,gDAAgD,EAAE,0BAA0B,CAAC,CAAC;gBAC/G,KAAK,SAAS,CAAC,CAAC,OAAO,SAAG,CAAC,QAAQ,CAAC,wDAAwD,EAAE,mDAAmD,CAAC,CAAC;gBACnJ,KAAK,SAAS,CAAC,CAAC,OAAO,SAAG,CAAC,QAAQ,CAAC,wDAAwD,EAAE,uBAAuB,CAAC,CAAC;YAC3H,CAAC;QACL,CAAC;IACL,CAAC,CAAC;IAEF,gCAAgC;IAChC,MAAM,iBAAiB,GAAG,CAAC,IAAsB,EAAE,EAAE;QACjD,MAAM,YAAY,GAAG,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC;QAC7D,MAAM,OAAO,GAAG,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,WAAW,CAAC;QAC9D,MAAM,UAAU,GAAG,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,UAAU,CAAC;QAC/D,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,YAAY,CAAC,CAAC;QACzD,OAAO,CACH,6BAAK,SAAS,EAAE,wCAAwC,IAAI,EAAE,EAC1D,KAAK,EAAE,EAAE,OAAO,EAAE,aAAa,EAAE,QAAQ,EAAE,UAAU,EAAE;YACvD,gCACI,SAAS,EAAE,gBAAgB,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,WAAW,mCAAmC,EACxG,OAAO,EAAE,UAAU,IAElB,eAAe,CAAC,IAAI,EAAE,YAAY,CAAC,CAC/B;YACT,gCACI,SAAS,EAAE,gBAAgB,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,WAAW,sCAAsC,EAC3G,OAAO,EAAE,GAAG,EAAE,CAAC,eAAe,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,mBAC1D,MAAM,mBACL,YAAY,KAAK,IAAI,EACpC,QAAQ,EAAE,CAAC,EACX,KAAK,EAAE,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,mBAAmB;gBAEpE,8BAAM,SAAS,EAAE,IAAA,iBAAO,EAAC,cAAc,CAAC,GAAS,CAC5C;YACR,YAAY,KAAK,IAAI,IAAI,CACtB,4BACI,SAAS,EAAC,uCAAuC,EACjD,YAAY,EAAE,GAAG,EAAE,CAAC,eAAe,CAAC,SAAS,CAAC,IAE7C,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CACpB,4BACI,GAAG,EAAE,IAAI,EACT,SAAS,EAAC,uCAAuC,EACjD,OAAO,EAAE,GAAG,EAAE;oBACV,OAAO,CAAC,IAAI,CAAC,CAAC;oBACd,eAAe,CAAC,SAAS,CAAC,CAAC;gBAC/B,CAAC,EACD,KAAK,EAAE,WAAW,CAAC,IAAI,EAAE,IAAI,CAAC,IAE7B,SAAS,CAAC,IAAI,EAAE,IAAI,CAAC,CACrB,CACR,CAAC,CACD,CACR,CACC,CACT,CAAC;IACN,CAAC,CAAC;IAEF,OAAO,CACH,6BAAK,SAAS,EAAC,yBAAyB;QACpC,6BAAK,SAAS,EAAC,gCAAgC;YAC3C,8BAAM,SAAS,EAAE,IAAA,iBAAO,EAAC,QAAQ,CAAC,GAAS;;YAAE,SAAG,CAAC,QAAQ,CAAC,0CAA0C,EAAE,wBAAwB,CAAC,CAC7H;QACN,6BAAK,SAAS,EAAC,8BAA8B;YACzC,6BAAK,SAAS,EAAC,8BAA8B;gBACzC,8BAAM,SAAS,EAAC,OAAO;oBAAE,SAAG,CAAC,QAAQ,CAAC,wCAAwC,EAAE,MAAM,CAAC;wBAAS;gBAChG,8BAAM,SAAS,EAAC,OAAO,IAAE,QAAQ,CAAC,IAAI,CAAQ,CAC5C,CACJ;QACN,6BAAK,SAAS,EAAC,iCAAiC;YAC3C,iBAAiB,CAAC,MAAM,CAAC;YACzB,iBAAiB,CAAC,OAAO,CAAC,CACzB,CACJ,CACT,CAAC;AACN,CAAC,CAAC;AAzIW,QAAA,gBAAgB,oBAyI3B"}

package/lib/browser/chat-response-renderer/toolcall-part-renderer.d.ts

@@ -2,9 +2,13 @@
 import { ChatResponsePartRenderer } from '../chat-response-part-renderer';
 import { ChatResponseContent, ToolCallChatResponseContent } from '@theia/ai-chat/lib/common';
 import { ReactNode } from '@theia/core/shared/react';
+import { ToolConfirmationManager, ToolConfirmationMode } from '@theia/ai-chat/lib/browser/chat-tool-preferences';
+import { ResponseNode } from '../chat-tree-view';
 export declare class ToolCallPartRenderer implements ChatResponsePartRenderer<ToolCallChatResponseContent> {
+    protected toolConfirmationManager: ToolConfirmationManager;
     canHandle(response: ChatResponseContent): number;
-    render(response: ToolCallChatResponseContent): ReactNode;
+    render(response: ToolCallChatResponseContent, parentNode: ResponseNode): ReactNode;
+    protected getToolConfirmationSettings(responseId: string, chatId: string): ToolConfirmationMode;
     protected renderCollapsibleArguments(args: string | undefined): ReactNode;
     private prettyPrintArgs;
     private tryPrettyPrintJson;

package/lib/browser/chat-response-renderer/toolcall-part-renderer.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"toolcall-part-renderer.d.ts","sourceRoot":"","sources":["../../../src/browser/chat-response-renderer/toolcall-part-renderer.tsx"],"names":[],"mappings":";AAgBA,OAAO,EAAE,wBAAwB,EAAE,MAAM,gCAAgC,CAAC;AAE1E,OAAO,EAAE,mBAAmB,EAAE,2BAA2B,EAAE,MAAM,2BAA2B,CAAC;AAC7F,OAAO,EAAE,SAAS,EAAE,MAAM,0BAA0B,CAAC;AAIrD,qBACa,oBAAqB,YAAW,wBAAwB,CAAC,2BAA2B,CAAC;IAE9F,SAAS,CAAC,QAAQ,EAAE,mBAAmB,GAAG,MAAM;IAOhD,MAAM,CAAC,QAAQ,EAAE,2BAA2B,GAAG,SAAS;IAmBxD,SAAS,CAAC,0BAA0B,CAAC,IAAI,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS;IAazE,OAAO,CAAC,eAAe;IASvB,OAAO,CAAC,kBAAkB;CAsB7B"}
+{"version":3,"file":"toolcall-part-renderer.d.ts","sourceRoot":"","sources":["../../../src/browser/chat-response-renderer/toolcall-part-renderer.tsx"],"names":[],"mappings":";AAgBA,OAAO,EAAE,wBAAwB,EAAE,MAAM,gCAAgC,CAAC;AAE1E,OAAO,EAAE,mBAAmB,EAAE,2BAA2B,EAAE,MAAM,2BAA2B,CAAC;AAC7F,OAAO,EAAE,SAAS,EAAE,MAAM,0BAA0B,CAAC;AAKrD,OAAO,EAAE,uBAAuB,EAAE,oBAAoB,EAAE,MAAM,kDAAkD,CAAC;AACjH,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEjD,qBACa,oBAAqB,YAAW,wBAAwB,CAAC,2BAA2B,CAAC;IAG9F,SAAS,CAAC,uBAAuB,EAAE,uBAAuB,CAAC;IAE3D,SAAS,CAAC,QAAQ,EAAE,mBAAmB,GAAG,MAAM;IAOhD,MAAM,CAAC,QAAQ,EAAE,2BAA2B,EAAE,UAAU,EAAE,YAAY,GAAG,SAAS;IAYlF,SAAS,CAAC,2BAA2B,CAAC,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,oBAAoB;IAI/F,SAAS,CAAC,0BAA0B,CAAC,IAAI,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS;IAazE,OAAO,CAAC,eAAe;IASvB,OAAO,CAAC,kBAAkB;CAsB7B"}

package/lib/browser/chat-response-renderer/toolcall-part-renderer.js

@@ -20,7 +20,10 @@ const tslib_1 = require("tslib");
 const inversify_1 = require("@theia/core/shared/inversify");
 const common_1 = require("@theia/ai-chat/lib/common");
 const nls_1 = require("@theia/core/lib/common/nls");
+const browser_1 = require("@theia/core/lib/browser");
 const React = require("@theia/core/shared/react");
+const tool_confirmation_1 = require("./tool-confirmation");
+const chat_tool_preferences_1 = require("@theia/ai-chat/lib/browser/chat-tool-preferences");
 let ToolCallPartRenderer = class ToolCallPartRenderer {
     canHandle(response) {
         if (common_1.ToolCallChatResponseContent.is(response)) {
@@ -28,24 +31,13 @@ let ToolCallPartRenderer = class ToolCallPartRenderer {
         }
         return -1;
     }
-    render(response) {
-        return (React.createElement("h4", { className: 'theia-toolCall' }, response.finished ? (React.createElement("details", null,
-            React.createElement("summary", null,
-                nls_1.nls.localize('theia/ai/chat-ui/toolcall-part-renderer/finished', 'Ran'),
-                " ",
-                response.name,
-                "(",
-                this.renderCollapsibleArguments(response.arguments),
-                ")"),
-            React.createElement("pre", null, this.tryPrettyPrintJson(response)))) : (React.createElement("span", null,
-            React.createElement(Spinner, null),
-            " ",
-            nls_1.nls.localizeByDefault('Running'),
-            " ",
-            response.name,
-            "(",
-            this.renderCollapsibleArguments(response.arguments),
-            ")"))));
+    render(response, parentNode) {
+        const chatId = parentNode.sessionId;
+        const confirmationMode = response.name ? this.getToolConfirmationSettings(response.name, chatId) : chat_tool_preferences_1.ToolConfirmationMode.DISABLED;
+        return React.createElement(ToolCallContent, { response: response, confirmationMode: confirmationMode, toolConfirmationManager: this.toolConfirmationManager, chatId: chatId, renderCollapsibleArguments: this.renderCollapsibleArguments.bind(this), tryPrettyPrintJson: this.tryPrettyPrintJson.bind(this) });
+    }
+    getToolConfirmationSettings(responseId, chatId) {
+        return this.toolConfirmationManager.getConfirmationMode(responseId, chatId);
     }
     renderCollapsibleArguments(args) {
         if (!args || !args.trim() || args.trim() === '{}') {
@@ -84,8 +76,80 @@ let ToolCallPartRenderer = class ToolCallPartRenderer {
     }
 };
 exports.ToolCallPartRenderer = ToolCallPartRenderer;
+tslib_1.__decorate([
+    (0, inversify_1.inject)(chat_tool_preferences_1.ToolConfirmationManager),
+    tslib_1.__metadata("design:type", chat_tool_preferences_1.ToolConfirmationManager)
+], ToolCallPartRenderer.prototype, "toolConfirmationManager", void 0);
 exports.ToolCallPartRenderer = ToolCallPartRenderer = tslib_1.__decorate([
     (0, inversify_1.injectable)()
 ], ToolCallPartRenderer);
-const Spinner = () => (React.createElement("i", { className: "fa fa-spinner fa-spin" }));
+const Spinner = () => (React.createElement("span", { className: `${(0, browser_1.codicon)('loading')} theia-animation-spin` }));
+/**
+ * A function component to handle tool call rendering and confirmation
+ */
+const ToolCallContent = ({ response, confirmationMode, toolConfirmationManager, chatId, tryPrettyPrintJson, renderCollapsibleArguments }) => {
+    const [confirmationState, setConfirmationState] = React.useState('waiting');
+    React.useEffect(() => {
+        if (confirmationMode === chat_tool_preferences_1.ToolConfirmationMode.ALWAYS_ALLOW) {
+            response.confirm();
+            setConfirmationState('allowed');
+            return;
+        }
+        else if (confirmationMode === chat_tool_preferences_1.ToolConfirmationMode.DISABLED) {
+            response.deny();
+            setConfirmationState('denied');
+            return;
+        }
+        response.confirmed.then(confirmed => {
+            if (confirmed === true) {
+                setConfirmationState('allowed');
+            }
+            else {
+                setConfirmationState('denied');
+            }
+        })
+            .catch(() => {
+            setConfirmationState('denied');
+        });
+    }, [response, confirmationMode]);
+    const handleAllow = React.useCallback((mode = 'once') => {
+        if (mode === 'forever' && response.name) {
+            toolConfirmationManager.setConfirmationMode(response.name, chat_tool_preferences_1.ToolConfirmationMode.ALWAYS_ALLOW);
+        }
+        else if (mode === 'session' && response.name) {
+            toolConfirmationManager.setSessionConfirmationMode(response.name, chat_tool_preferences_1.ToolConfirmationMode.ALWAYS_ALLOW, chatId);
+        }
+        response.confirm();
+    }, [response, toolConfirmationManager, chatId]);
+    const handleDeny = React.useCallback((mode = 'once') => {
+        if (mode === 'forever' && response.name) {
+            toolConfirmationManager.setConfirmationMode(response.name, chat_tool_preferences_1.ToolConfirmationMode.DISABLED);
+        }
+        else if (mode === 'session' && response.name) {
+            toolConfirmationManager.setSessionConfirmationMode(response.name, chat_tool_preferences_1.ToolConfirmationMode.DISABLED, chatId);
+        }
+        response.deny();
+    }, [response, toolConfirmationManager, chatId]);
+    return (React.createElement("div", { className: 'theia-toolCall' },
+        React.createElement("h4", null, confirmationState === 'denied' ? (React.createElement("span", { className: "theia-tool-denied" },
+            React.createElement("span", { className: (0, browser_1.codicon)('error') }),
+            " ",
+            nls_1.nls.localize('theia/ai/chat-ui/toolcall-part-renderer/denied', 'Execution denied'),
+            ": ",
+            response.name)) : response.finished ? (React.createElement("details", null,
+            React.createElement("summary", null,
+                nls_1.nls.localize('theia/ai/chat-ui/toolcall-part-renderer/finished', 'Ran'),
+                " ",
+                response.name,
+                "(",
+                renderCollapsibleArguments(response.arguments),
+                ")"),
+            React.createElement("pre", null, tryPrettyPrintJson(response)))) : (confirmationState === 'allowed' && (React.createElement("span", null,
+            React.createElement(Spinner, null),
+            " ",
+            nls_1.nls.localizeByDefault('Running'),
+            " ",
+            response.name)))),
+        confirmationState === 'waiting' && (React.createElement(tool_confirmation_1.ToolConfirmation, { response: response, onAllow: handleAllow, onDeny: handleDeny }))));
+};
 //# sourceMappingURL=toolcall-part-renderer.js.map

package/lib/browser/chat-response-renderer/toolcall-part-renderer.js.map

@@ -1 +1 @@
-{"version":3,"file":"toolcall-part-renderer.js","sourceRoot":"","sources":["../../../src/browser/chat-response-renderer/toolcall-part-renderer.tsx"],"names":[],"mappings":";AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;;;;AAGhF,4DAA0D;AAC1D,sDAA6F;AAE7F,oDAAiD;AACjD,kDAAkD;AAG3C,IAAM,oBAAoB,GAA1B,MAAM,oBAAoB;IAE7B,SAAS,CAAC,QAA6B;QACnC,IAAI,oCAA2B,CAAC,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC3C,OAAO,EAAE,CAAC;QACd,CAAC;QACD,OAAO,CAAC,CAAC,CAAC;IACd,CAAC;IAED,MAAM,CAAC,QAAqC;QACxC,OAAO,CACH,4BAAI,SAAS,EAAC,gBAAgB,IACzB,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CACjB;YACI;gBAAU,SAAG,CAAC,QAAQ,CAAC,kDAAkD,EAAE,KAAK,CAAC;;gBAAG,QAAQ,CAAC,IAAI;;gBAC3F,IAAI,CAAC,0BAA0B,CAAC,QAAQ,CAAC,SAAS,CAAC;oBAC/C;YACV,iCAAM,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAO,CACxC,CACb,CAAC,CAAC,CAAC,CACA;YACI,oBAAC,OAAO,OAAG;;YAAE,SAAG,CAAC,iBAAiB,CAAC,SAAS,CAAC;;YAAG,QAAQ,CAAC,IAAI;;YAAG,IAAI,CAAC,0BAA0B,CAAC,QAAQ,CAAC,SAAS,CAAC;gBAChH,CACV,CACA,CACR,CAAC;IACN,CAAC;IAES,0BAA0B,CAAC,IAAwB;QACzD,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,IAAI,CAAC,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC;YAChD,OAAO,SAAS,CAAC;QACrB,CAAC;QAED,OAAO,CACH,iCAAS,SAAS,EAAC,uBAAuB;YACtC,iCAAS,SAAS,EAAC,+BAA+B,UAAc;YAChE,kCAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAQ,CACnC,CACb,CAAC;IACN,CAAC;IAEO,eAAe,CAAC,IAAY;QAChC,IAAI,CAAC;YACD,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;QAC1D,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACT,eAAe;YACf,OAAO,IAAI,CAAC;QAChB,CAAC;IACL,CAAC;IAEO,kBAAkB,CAAC,QAAqC;QAC5D,IAAI,eAAe,GAAG,QAAQ,CAAC,MAAM,CAAC;QACtC,IAAI,CAAC;YACD,IAAI,eAAe,EAAE,CAAC;gBAClB,IAAI,OAAO,eAAe,KAAK,QAAQ,EAAE,CAAC;oBACtC,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;gBAClD,CAAC;gBACD,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,eAAe,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;YACpE,CAAC;QACL,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACT,IAAI,OAAO,eAAe,KAAK,QAAQ,EAAE,CAAC;gBACtC,eAAe,GAAG,SAAG,CAAC,QAAQ,CAC1B,0DAA0D,EAC1D,2FAA2F,EAC3F,CAAC,CAAC,OAAO,EACT,eAAe,CAClB,CAAC;YACN,CAAC;YACD,eAAe;QACnB,CAAC;QACD,OAAO,eAAe,CAAC;IAC3B,CAAC;CACJ,CAAA;AAxEY,oDAAoB;+BAApB,oBAAoB;IADhC,IAAA,sBAAU,GAAE;GACA,oBAAoB,CAwEhC;AAED,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC,CAClB,2BAAG,SAAS,EAAC,uBAAuB,GAAK,CAC5C,CAAC"}
+{"version":3,"file":"toolcall-part-renderer.js","sourceRoot":"","sources":["../../../src/browser/chat-response-renderer/toolcall-part-renderer.tsx"],"names":[],"mappings":";AAAA,gFAAgF;AAChF,yCAAyC;AACzC,EAAE;AACF,2EAA2E;AAC3E,mEAAmE;AACnE,wCAAwC;AACxC,EAAE;AACF,4EAA4E;AAC5E,8EAA8E;AAC9E,6EAA6E;AAC7E,yDAAyD;AACzD,uDAAuD;AACvD,EAAE;AACF,gFAAgF;AAChF,gFAAgF;;;;AAGhF,4DAAkE;AAClE,sDAA6F;AAE7F,oDAAiD;AACjD,qDAAkD;AAClD,kDAAkD;AAClD,2DAA8E;AAC9E,4FAAiH;AAI1G,IAAM,oBAAoB,GAA1B,MAAM,oBAAoB;IAK7B,SAAS,CAAC,QAA6B;QACnC,IAAI,oCAA2B,CAAC,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC3C,OAAO,EAAE,CAAC;QACd,CAAC;QACD,OAAO,CAAC,CAAC,CAAC;IACd,CAAC;IAED,MAAM,CAAC,QAAqC,EAAE,UAAwB;QAClE,MAAM,MAAM,GAAG,UAAU,CAAC,SAAS,CAAC;QACpC,MAAM,gBAAgB,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,2BAA2B,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,4CAAoB,CAAC,QAAQ,CAAC;QACjI,OAAO,oBAAC,eAAe,IACnB,QAAQ,EAAE,QAAQ,EAClB,gBAAgB,EAAE,gBAAgB,EAClC,uBAAuB,EAAE,IAAI,CAAC,uBAAuB,EACrD,MAAM,EAAE,MAAM,EACd,0BAA0B,EAAE,IAAI,CAAC,0BAA0B,CAAC,IAAI,CAAC,IAAI,CAAC,EACtE,kBAAkB,EAAE,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,GAAI,CAAC;IACnE,CAAC;IAES,2BAA2B,CAAC,UAAkB,EAAE,MAAc;QACpE,OAAO,IAAI,CAAC,uBAAuB,CAAC,mBAAmB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAChF,CAAC;IAES,0BAA0B,CAAC,IAAwB;QACzD,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,IAAI,CAAC,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC;YAChD,OAAO,SAAS,CAAC;QACrB,CAAC;QAED,OAAO,CACH,iCAAS,SAAS,EAAC,uBAAuB;YACtC,iCAAS,SAAS,EAAC,+BAA+B,UAAc;YAChE,kCAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAQ,CACnC,CACb,CAAC;IACN,CAAC;IAEO,eAAe,CAAC,IAAY;QAChC,IAAI,CAAC;YACD,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;QAC1D,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACT,eAAe;YACf,OAAO,IAAI,CAAC;QAChB,CAAC;IACL,CAAC;IAEO,kBAAkB,CAAC,QAAqC;QAC5D,IAAI,eAAe,GAAG,QAAQ,CAAC,MAAM,CAAC;QACtC,IAAI,CAAC;YACD,IAAI,eAAe,EAAE,CAAC;gBAClB,IAAI,OAAO,eAAe,KAAK,QAAQ,EAAE,CAAC;oBACtC,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;gBAClD,CAAC;gBACD,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,eAAe,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;YACpE,CAAC;QACL,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACT,IAAI,OAAO,eAAe,KAAK,QAAQ,EAAE,CAAC;gBACtC,eAAe,GAAG,SAAG,CAAC,QAAQ,CAC1B,0DAA0D,EAC1D,2FAA2F,EAC3F,CAAC,CAAC,OAAO,EACT,eAAe,CAClB,CAAC;YACN,CAAC;YACD,eAAe;QACnB,CAAC;QACD,OAAO,eAAe,CAAC;IAC3B,CAAC;CACJ,CAAA;AAxEY,oDAAoB;AAGnB;IADT,IAAA,kBAAM,EAAC,+CAAuB,CAAC;sCACG,+CAAuB;qEAAC;+BAHlD,oBAAoB;IADhC,IAAA,sBAAU,GAAE;GACA,oBAAoB,CAwEhC;AAED,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC,CAClB,8BAAM,SAAS,EAAE,GAAG,IAAA,iBAAO,EAAC,SAAS,CAAC,uBAAuB,GAAS,CACzE,CAAC;AAWF;;GAEG;AACH,MAAM,eAAe,GAAmC,CAAC,EAAE,QAAQ,EAAE,gBAAgB,EAAE,uBAAuB,EAAE,MAAM,EAAE,kBAAkB,EAAE,0BAA0B,EAAE,EAAE,EAAE;IACxK,MAAM,CAAC,iBAAiB,EAAE,oBAAoB,CAAC,GAAG,KAAK,CAAC,QAAQ,CAAwB,SAAS,CAAC,CAAC;IAEnG,KAAK,CAAC,SAAS,CAAC,GAAG,EAAE;QACjB,IAAI,gBAAgB,KAAK,4CAAoB,CAAC,YAAY,EAAE,CAAC;YACzD,QAAQ,CAAC,OAAO,EAAE,CAAC;YACnB,oBAAoB,CAAC,SAAS,CAAC,CAAC;YAChC,OAAO;QACX,CAAC;aAAM,IAAI,gBAAgB,KAAK,4CAAoB,CAAC,QAAQ,EAAE,CAAC;YAC5D,QAAQ,CAAC,IAAI,EAAE,CAAC;YAChB,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YAC/B,OAAO;QACX,CAAC;QACD,QAAQ,CAAC,SAAS,CAAC,IAAI,CACnB,SAAS,CAAC,EAAE;YACR,IAAI,SAAS,KAAK,IAAI,EAAE,CAAC;gBACrB,oBAAoB,CAAC,SAAS,CAAC,CAAC;YACpC,CAAC;iBAAM,CAAC;gBACJ,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YACnC,CAAC;QACL,CAAC,CACJ;aACI,KAAK,CAAC,GAAG,EAAE;YACR,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;IACX,CAAC,EAAE,CAAC,QAAQ,EAAE,gBAAgB,CAAC,CAAC,CAAC;IAEjC,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW,CAAC,CAAC,OAAuC,MAAM,EAAE,EAAE;QACpF,IAAI,IAAI,KAAK,SAAS,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC;YACtC,uBAAuB,CAAC,mBAAmB,CAAC,QAAQ,CAAC,IAAI,EAAE,4CAAoB,CAAC,YAAY,CAAC,CAAC;QAClG,CAAC;aAAM,IAAI,IAAI,KAAK,SAAS,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC;YAC7C,uBAAuB,CAAC,0BAA0B,CAAC,QAAQ,CAAC,IAAI,EAAE,4CAAoB,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QACjH,CAAC;QACD,QAAQ,CAAC,OAAO,EAAE,CAAC;IACvB,CAAC,EAAE,CAAC,QAAQ,EAAE,uBAAuB,EAAE,MAAM,CAAC,CAAC,CAAC;IAEhD,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,CAAC,CAAC,OAAuC,MAAM,EAAE,EAAE;QACnF,IAAI,IAAI,KAAK,SAAS,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC;YACtC,uBAAuB,CAAC,mBAAmB,CAAC,QAAQ,CAAC,IAAI,EAAE,4CAAoB,CAAC,QAAQ,CAAC,CAAC;QAC9F,CAAC;aAAM,IAAI,IAAI,KAAK,SAAS,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC;YAC7C,uBAAuB,CAAC,0BAA0B,CAAC,QAAQ,CAAC,IAAI,EAAE,4CAAoB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAC7G,CAAC;QACD,QAAQ,CAAC,IAAI,EAAE,CAAC;IACpB,CAAC,EAAE,CAAC,QAAQ,EAAE,uBAAuB,EAAE,MAAM,CAAC,CAAC,CAAC;IAEhD,OAAO,CACH,6BAAK,SAAS,EAAC,gBAAgB;QAC3B,gCACK,iBAAiB,KAAK,QAAQ,CAAC,CAAC,CAAC,CAC9B,8BAAM,SAAS,EAAC,mBAAmB;YAC/B,8BAAM,SAAS,EAAE,IAAA,iBAAO,EAAC,OAAO,CAAC,GAAS;;YAAE,SAAG,CAAC,QAAQ,CAAC,gDAAgD,EAAE,kBAAkB,CAAC;;YAAI,QAAQ,CAAC,IAAI,CAC5I,CACV,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CACpB;YACI;gBAAU,SAAG,CAAC,QAAQ,CAAC,kDAAkD,EAAE,KAAK,CAAC;;gBAAG,QAAQ,CAAC,IAAI;;gBAC3F,0BAA0B,CAAC,QAAQ,CAAC,SAAS,CAAC;oBAC1C;YACV,iCAAM,kBAAkB,CAAC,QAAQ,CAAC,CAAO,CACnC,CACb,CAAC,CAAC,CAAC,CACA,iBAAiB,KAAK,SAAS,IAAI,CAC/B;YACI,oBAAC,OAAO,OAAG;;YAAE,SAAG,CAAC,iBAAiB,CAAC,SAAS,CAAC;;YAAG,QAAQ,CAAC,IAAI,CAC1D,CACV,CACJ,CACA;QAGJ,iBAAiB,KAAK,SAAS,IAAI,CAChC,oBAAC,oCAAgB,IACb,QAAQ,EAAE,QAAQ,EAClB,OAAO,EAAE,WAAW,EACpB,MAAM,EAAE,UAAU,GACpB,CACL,CACC,CACT,CAAC;AACN,CAAC,CAAC"}

package/lib/browser/chat-tree-view/chat-view-tree-input-widget.d.ts

@@ -1,7 +1,7 @@
 import { AIChatInputWidget, type AIChatInputConfiguration } from '../chat-input-widget';
 import type { EditableRequestNode } from './chat-view-tree-widget';
 import { URI } from '@theia/core';
-import type { ChatRequestModel, EditableChatRequestModel } from '@theia/ai-chat';
+import type { ChatRequestModel, EditableChatRequestModel, ChatHierarchyBranch } from '@theia/ai-chat';
 import type { AIVariableResolutionRequest } from '@theia/ai-core';
 export declare const AIChatTreeInputConfiguration: unique symbol;
 export interface AIChatTreeInputConfiguration extends AIChatInputConfiguration {
@@ -9,6 +9,10 @@ export interface AIChatTreeInputConfiguration extends AIChatInputConfiguration {
 export declare const AIChatTreeInputArgs: unique symbol;
 export interface AIChatTreeInputArgs {
     node: EditableRequestNode;
+    /**
+     * The branch of the chat tree for this request node (used by the input widget for state tracking).
+     */
+    branch?: ChatHierarchyBranch;
     initialValue?: string;
     onQuery: (query: string) => Promise<void>;
     onUnpin?: () => void;
@@ -25,6 +29,7 @@ export declare class AIChatTreeInputWidget extends AIChatInputWidget {
     get requestNode(): EditableRequestNode;
     get request(): EditableChatRequestModel;
     protected init(): void;
+    protected updateBranch(): void;
     protected getResourceUri(): URI;
     addContext(variable: AIVariableResolutionRequest): void;
     protected getContext(): readonly AIVariableResolutionRequest[];

package/lib/browser/chat-tree-view/chat-view-tree-input-widget.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chat-view-tree-input-widget.d.ts","sourceRoot":"","sources":["../../../src/browser/chat-tree-view/chat-view-tree-input-widget.tsx"],"names":[],"mappings":"AAiBA,OAAO,EAAE,iBAAiB,EAAE,KAAK,wBAAwB,EAAE,MAAM,sBAAsB,CAAC;AACxF,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,GAAG,EAAE,MAAM,aAAa,CAAC;AAElC,OAAO,KAAK,EAAE,gBAAgB,EAAE,wBAAwB,EAAE,MAAM,gBAAgB,CAAC;AACjF,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,gBAAgB,CAAC;AAGlE,eAAO,MAAM,4BAA4B,eAAyC,CAAC;AACnF,MAAM,WAAW,4BAA6B,SAAQ,wBAAwB;CAAI;AAElF,eAAO,MAAM,mBAAmB,eAAgC,CAAC;AACjE,MAAM,WAAW,mBAAmB;IAChC,IAAI,EAAE,mBAAmB,CAAC;IAC1B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,OAAO,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1C,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC;IACrB,QAAQ,CAAC,EAAE,CAAC,YAAY,EAAE,gBAAgB,KAAK,IAAI,CAAC;IACpD,iBAAiB,CAAC,EAAE,CAAC,YAAY,EAAE,gBAAgB,KAAK,IAAI,CAAC;IAC7D,wBAAwB,CAAC,EAAE,CAAC,YAAY,EAAE,gBAAgB,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;CACtF;AACD,eAAO,MAAM,sBAAsB,eAAmC,CAAC;AACvE,MAAM,MAAM,sBAAsB,GAAG,CAAC,IAAI,EAAE,mBAAmB,KAAK,qBAAqB,CAAC;AAE1F,qBACa,qBAAsB,SAAQ,iBAAiB;IACxD,OAAuB,EAAE,SAA4B;IAGrD,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,mBAAmB,CAAC;IAE7C,mBAC4B,aAAa,EAAE,4BAA4B,GAAG,SAAS,CAAC;IAEpF,IAAI,WAAW,IAAI,mBAAmB,CAErC;IAED,IAAI,OAAO,IAAI,wBAAwB,CAEtC;cAGkB,IAAI,IAAI,IAAI;cAaZ,cAAc,IAAI,GAAG;IAI/B,UAAU,CAAC,QAAQ,EAAE,2BAA2B,GAAG,IAAI;cAI7C,UAAU,IAAI,SAAS,2BAA2B,EAAE;cAIpD,oBAAoB,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;CAG/D"}
+{"version":3,"file":"chat-view-tree-input-widget.d.ts","sourceRoot":"","sources":["../../../src/browser/chat-tree-view/chat-view-tree-input-widget.tsx"],"names":[],"mappings":"AAiBA,OAAO,EAAE,iBAAiB,EAAE,KAAK,wBAAwB,EAAE,MAAM,sBAAsB,CAAC;AACxF,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,GAAG,EAAE,MAAM,aAAa,CAAC;AAElC,OAAO,KAAK,EAAE,gBAAgB,EAAE,wBAAwB,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AACtG,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,gBAAgB,CAAC;AAGlE,eAAO,MAAM,4BAA4B,eAAyC,CAAC;AACnF,MAAM,WAAW,4BAA6B,SAAQ,wBAAwB;CAAI;AAElF,eAAO,MAAM,mBAAmB,eAAgC,CAAC;AACjE,MAAM,WAAW,mBAAmB;IAChC,IAAI,EAAE,mBAAmB,CAAC;IAC1B;;OAEG;IACH,MAAM,CAAC,EAAE,mBAAmB,CAAC;IAC7B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,OAAO,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC1C,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC;IACrB,QAAQ,CAAC,EAAE,CAAC,YAAY,EAAE,gBAAgB,KAAK,IAAI,CAAC;IACpD,iBAAiB,CAAC,EAAE,CAAC,YAAY,EAAE,gBAAgB,KAAK,IAAI,CAAC;IAC7D,wBAAwB,CAAC,EAAE,CAAC,YAAY,EAAE,gBAAgB,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;CACtF;AACD,eAAO,MAAM,sBAAsB,eAAmC,CAAC;AACvE,MAAM,MAAM,sBAAsB,GAAG,CAAC,IAAI,EAAE,mBAAmB,KAAK,qBAAqB,CAAC;AAE1F,qBACa,qBAAsB,SAAQ,iBAAiB;IACxD,OAAuB,EAAE,SAA4B;IAGrD,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,mBAAmB,CAAC;IAE7C,mBAC4B,aAAa,EAAE,4BAA4B,GAAG,SAAS,CAAC;IAEpF,IAAI,WAAW,IAAI,mBAAmB,CAErC;IAED,IAAI,OAAO,IAAI,wBAAwB,CAEtC;cAGkB,IAAI,IAAI,IAAI;IAoB/B,SAAS,CAAC,YAAY,IAAI,IAAI;cAIX,cAAc,IAAI,GAAG;IAI/B,UAAU,CAAC,QAAQ,EAAE,2BAA2B,GAAG,IAAI;cAI7C,UAAU,IAAI,SAAS,2BAA2B,EAAE;cAIpD,oBAAoB,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;CAG/D"}