@jupyterlite/ai 0.9.1 → 0.10.0

package/lib/chat-model.js

@@ -4,9 +4,8 @@ import { UUID } from '@lumino/coreutils';
 import { Signal } from '@lumino/signaling';
 import { AI_AVATAR } from './icons';
 /**
- * AI Chat Model implementation that provides chat functionality with OpenAI agents,
- * tool integration, and MCP server support.
- * Extends the base AbstractChatModel to provide AI-powered conversations.
+ * AI Chat Model implementation that provides chat functionality tool integration,
+ * and MCP server support.
  */
 export class AIChatModel extends AbstractChatModel {
     /**
@@ -25,6 +24,7 @@ export class AIChatModel extends AbstractChatModel {
         this._settingsModel = options.settingsModel;
         this._user = options.user;
         this._agentManager = options.agentManager;
+        this._trans = options.trans;
         // Listen for agent events
         this._agentManager.agentEvent.connect(this._onAgentEvent, this);
         // Listen for settings changes to update chat behavior
@@ -90,7 +90,7 @@ export class AIChatModel extends AbstractChatModel {
      */
     clearMessages = () => {
         this.messagesDeleted(0, this.messages.length);
-        this._pendingToolCalls.clear();
+        this._toolContexts.clear();
         this._agentManager.clearHistory();
     };
     /**
@@ -127,7 +127,6 @@ export class AIChatModel extends AbstractChatModel {
             let enhancedMessage = message.body;
             if (this.input.attachments.length > 0) {
                 const attachmentContents = await this._processAttachments(this.input.attachments);
-                // Clear attachments right after  processing
                 this.input.clearAttachments();
                 if (attachmentContents.length > 0) {
                     enhancedMessage +=
@@ -152,56 +151,6 @@ export class AIChatModel extends AbstractChatModel {
             this.updateWriters([]);
         }
     }
-    /**
-     * Approves a tool call and updates the UI accordingly.
-     * @param interruptionId The interruption ID to approve
-     * @param messageId Optional message ID for UI updates
-     */
-    async approveToolCall(interruptionId, messageId) {
-        await this._agentManager.approveToolCall(interruptionId);
-        // Update the tool call box to show "Approved" status
-        if (messageId) {
-            this._updateToolCallBoxStatus(messageId, 'Approved', true);
-        }
-    }
-    /**
-     * Rejects a tool call and updates the UI accordingly.
-     * @param interruptionId The interruption ID to reject
-     * @param messageId Optional message ID for UI updates
-     */
-    async rejectToolCall(interruptionId, messageId) {
-        await this._agentManager.rejectToolCall(interruptionId);
-        // Update the tool call box to show "Rejected" status
-        if (messageId) {
-            this._updateToolCallBoxStatus(messageId, 'Rejected', false);
-        }
-    }
-    /**
-     * Approves all tools in a group.
-     * @param groupId The group ID containing the tool calls
-     * @param interruptionIds Array of interruption IDs to approve
-     * @param messageId Optional message ID for UI updates
-     */
-    async approveGroupedToolCalls(groupId, interruptionIds, messageId) {
-        await this._agentManager.approveGroupedToolCalls(groupId, interruptionIds);
-        // Update the grouped approval message to show approved status
-        if (messageId) {
-            this._updateGroupedApprovalStatus(messageId, 'Tools approved', true);
-        }
-    }
-    /**
-     * Rejects all tools in a group.
-     * @param groupId The group ID containing the tool calls
-     * @param interruptionIds Array of interruption IDs to reject
-     * @param messageId Optional message ID for UI updates
-     */
-    async rejectGroupedToolCalls(groupId, interruptionIds, messageId) {
-        await this._agentManager.rejectGroupedToolCalls(groupId, interruptionIds);
-        // Update the grouped approval message to show rejected status
-        if (messageId) {
-            this._updateGroupedApprovalStatus(messageId, 'Tools rejected', false);
-        }
-    }
     /**
      * Gets the AI user information for system messages.
      */
@@ -243,11 +192,11 @@ export class AIChatModel extends AbstractChatModel {
             case 'tool_call_complete':
                 this._handleToolCallCompleteEvent(event);
                 break;
-            case 'tool_approval_required':
-                this._handleToolApprovalRequired(event);
+            case 'tool_approval_request':
+                this._handleToolApprovalRequest(event);
                 break;
-            case 'grouped_approval_required':
-                this._handleGroupedApprovalRequired(event);
+            case 'tool_approval_resolved':
+                this._handleToolApprovalResolved(event);
                 break;
             case 'error':
                 this._handleErrorEvent(event);
@@ -298,174 +247,101 @@ export class AIChatModel extends AbstractChatModel {
      * @param event Event containing the tool call start data
      */
     _handleToolCallStartEvent(event) {
-        const toolCallMessageId = UUID.uuid4();
+        const messageId = UUID.uuid4();
+        const context = {
+            toolCallId: event.data.callId,
+            messageId,
+            toolName: event.data.toolName,
+            input: event.data.input,
+            status: 'pending'
+        };
+        this._toolContexts.set(event.data.callId, context);
         const toolCallMessage = {
-            body: `<details class="jp-ai-tool-call jp-ai-tool-pending">
-<summary class="jp-ai-tool-header">
-<div class="jp-ai-tool-icon">⚡</div>
-<div class="jp-ai-tool-title">${event.data.toolName}</div>
-<div class="jp-ai-tool-status jp-ai-tool-status-pending">Running...</div>
-</summary>
-<div class="jp-ai-tool-body">
-<div class="jp-ai-tool-section">
-<div class="jp-ai-tool-label">Input</div>
-<pre class="jp-ai-tool-code"><code>${event.data.input}</code></pre>
-</div>
-</div>
-</details>`,
+            body: Private.buildToolCallHtml({
+                toolName: context.toolName,
+                input: context.input,
+                status: context.status,
+                trans: this._trans
+            }),
             sender: this._getAIUser(),
-            id: toolCallMessageId,
+            id: messageId,
             time: Date.now() / 1000,
             type: 'msg',
             raw_time: false
         };
-        if (event.data.callId) {
-            this._pendingToolCalls.set(event.data.callId, toolCallMessageId);
-        }
         this.messageAdded(toolCallMessage);
     }
     /**
      * Handles the completion of a tool call execution.
-     * @param event Event containing the tool call completion data
      */
     _handleToolCallCompleteEvent(event) {
-        const messageId = this._pendingToolCalls.get(event.data.callId);
-        if (messageId) {
-            const existingMessageIndex = this.messages.findIndex(msg => msg.id === messageId);
-            if (existingMessageIndex !== -1) {
-                const existingMessage = this.messages[existingMessageIndex];
-                const inputJson = existingMessage.body.match(/<code>([\s\S]*?)<\/code>/)?.[1] || '';
-                const statusClass = event.data.isError
-                    ? 'jp-ai-tool-error'
-                    : 'jp-ai-tool-completed';
-                const statusText = event.data.isError ? 'Error' : 'Completed';
-                const statusColor = event.data.isError
-                    ? 'jp-ai-tool-status-error'
-                    : 'jp-ai-tool-status-completed';
-                const updatedMessage = {
-                    ...existingMessage,
-                    body: `<details class="jp-ai-tool-call ${statusClass}">
-<summary class="jp-ai-tool-header">
-<div class="jp-ai-tool-icon">⚡</div>
-<div class="jp-ai-tool-title">${event.data.toolName}</div>
-<div class="jp-ai-tool-status ${statusColor}">${statusText}</div>
-</summary>
-<div class="jp-ai-tool-body">
-<div class="jp-ai-tool-section">
-<div class="jp-ai-tool-label">Input</div>
-<pre class="jp-ai-tool-code"><code>${inputJson}</code></pre>
-</div>
-<div class="jp-ai-tool-section">
-<div class="jp-ai-tool-label">${event.data.isError ? 'Error' : 'Result'}</div>
-<pre class="jp-ai-tool-code"><code>${event.data.output}</code></pre>
-</div>
-</div>
-</details>`
-                };
-                this.messageAdded(updatedMessage);
-                this._pendingToolCalls.delete(event.data.callId);
-            }
-        }
+        const status = event.data.isError ? 'error' : 'completed';
+        this._updateToolCallUI(event.data.callId, status, event.data.output);
+        this._toolContexts.delete(event.data.callId);
     }
     /**
-     * Handles tool approval requests from the AI agent.
-     * @param event Event containing the tool approval request data
+     * Handles error events from the AI agent.
      */
-    _handleToolApprovalRequired(event) {
-        // Handle single tool approval - either update existing tool call message or create new approval message
-        if (event.data.callId) {
-            const messageId = this._pendingToolCalls.get(event.data.callId);
-            if (messageId) {
-                const existingMessageIndex = this.messages.findIndex(msg => msg.id === messageId);
-                if (existingMessageIndex !== -1) {
-                    const existingMessage = this.messages[existingMessageIndex];
-                    const assistantName = this._getAIUser().display_name;
-                    const updatedMessage = {
-                        ...existingMessage,
-                        body: `<details class="jp-ai-tool-call jp-ai-tool-pending" open>
-<summary class="jp-ai-tool-header">
-<div class="jp-ai-tool-icon">⚡</div>
-<div class="jp-ai-tool-title">${event.data.toolName}</div>
-<div class="jp-ai-tool-status jp-ai-tool-status-pending">Needs Approval</div>
-</summary>
-<div class="jp-ai-tool-body">
-<div class="jp-ai-tool-section">
-<div class="jp-ai-tool-label">${assistantName} wants to execute this tool. Do you approve?</div>
-<pre class="jp-ai-tool-code"><code>${event.data.toolInput}</code></pre>
-</div>
-[APPROVAL_BUTTONS:${event.data.interruptionId}]
-</div>
-</details>`
-                    };
-                    this.messageAdded(updatedMessage);
-                    this.updateWriters([]);
-                    return;
-                }
-            }
-        }
-        // Fallback: create separate approval message
-        const approvalMessageId = UUID.uuid4();
-        const assistantName = this._getAIUser().display_name;
-        const approvalMessage = {
-            body: `**🤖 Tool Approval Required: ${event.data.toolName}**
-
-${assistantName} wants to execute this tool. Do you approve?
-
-\`\`\`json
-${event.data.toolInput}
-\`\`\`
-
-[APPROVAL_BUTTONS:${event.data.interruptionId}]`,
+    _handleErrorEvent(event) {
+        this.messageAdded({
+            body: `Error generating response: ${event.data.error.message}`,
             sender: this._getAIUser(),
-            id: approvalMessageId,
+            id: UUID.uuid4(),
             time: Date.now() / 1000,
             type: 'msg',
             raw_time: false
-        };
-        this.messageAdded(approvalMessage);
-        this.updateWriters([]); // Stop showing "AI is writing"
+        });
     }
     /**
-     * Handles grouped tool approval requests from the AI agent.
-     * @param event Event containing the grouped tool approval request data
+     * Handles tool approval request events from the AI agent.
      */
-    _handleGroupedApprovalRequired(event) {
-        const assistantName = this._getAIUser().display_name;
-        const approvalMessageId = UUID.uuid4();
-        const toolsList = event.data.approvals
-            .map((info, index) => `**${index + 1}. ${info.toolName}**\n\`\`\`json\n${info.toolInput}\n\`\`\`\n`)
-            .join('\n\n');
-        const approvalMessage = {
-            body: `**🤖 Multiple Tool Approvals Required**
-
-${assistantName} wants to execute ${event.data.approvals.length} tools. Do you approve?
-
-${toolsList}
-
-[GROUP_APPROVAL_BUTTONS:${event.data.groupId}:${event.data.approvals.map(info => info.interruptionId).join(',')}]`,
-            sender: this._getAIUser(),
-            id: approvalMessageId,
-            time: Date.now() / 1000,
-            type: 'msg',
-            raw_time: false
-        };
-        this.messageAdded(approvalMessage);
-        this.updateWriters([]); // Stop showing "AI is writing"
+    _handleToolApprovalRequest(event) {
+        const context = this._toolContexts.get(event.data.toolCallId);
+        if (!context) {
+            return;
+        }
+        context.approvalId = event.data.approvalId;
+        context.input = JSON.stringify(event.data.args, null, 2);
+        this._updateToolCallUI(event.data.toolCallId, 'awaiting_approval');
     }
     /**
-     * Handles error events from the AI agent.
-     * @param event Event containing the error information
+     * Handles tool approval resolved events from the AI agent.
      */
-    _handleErrorEvent(event) {
-        const errorMessage = {
-            body: `Error generating response: ${event.data.error.message}`,
-            sender: this._getAIUser(),
-            id: UUID.uuid4(),
-            time: Date.now() / 1000,
-            type: 'msg',
-            raw_time: false
-        };
-        this.messageAdded(errorMessage);
+    _handleToolApprovalResolved(event) {
+        const context = Array.from(this._toolContexts.values()).find(ctx => ctx.approvalId === event.data.approvalId);
+        if (!context) {
+            return;
+        }
+        const status = event.data.approved ? 'approved' : 'rejected';
+        this._updateToolCallUI(context.toolCallId, status);
+        if (!event.data.approved) {
+            this._toolContexts.delete(context.toolCallId);
+        }
+    }
+    /**
+     * Updates a tool call's UI with new status and optional output.
+     */
+    _updateToolCallUI(toolCallId, status, output) {
+        const context = this._toolContexts.get(toolCallId);
+        if (!context) {
+            return;
+        }
+        const existingMessage = this.messages.find(msg => msg.id === context.messageId);
+        if (!existingMessage) {
+            return;
+        }
+        context.status = status;
+        this.messageAdded({
+            ...existingMessage,
+            body: Private.buildToolCallHtml({
+                toolName: context.toolName,
+                input: context.input,
+                status: context.status,
+                output,
+                approvalId: context.approvalId,
+                trans: this._trans
+            })
+        });
     }
     /**
      * Processes file attachments and returns their content as formatted strings.
@@ -508,16 +384,34 @@ ${toolsList}
             return null;
         }
         try {
-            const model = await this.input.documentManager?.services.contents.get(attachment.value);
-            if (!model || model.type !== 'notebook') {
-                return null;
+            // Try reading from live notebook if open
+            const widget = this.input.documentManager?.findWidget(attachment.value);
+            let cellData;
+            let kernelLang = 'text';
+            const ymodel = widget?.context.model.sharedModel;
+            if (ymodel) {
+                const nb = ymodel.toJSON();
+                cellData = nb.cells;
+                const lang = nb.metadata.language_info?.name ||
+                    nb.metadata.kernelspec?.language ||
+                    'text';
+                kernelLang = String(lang);
+            }
+            else {
+                // Fallback: reading from disk
+                const model = await this.input.documentManager?.services.contents.get(attachment.value);
+                if (!model || model.type !== 'notebook') {
+                    return null;
+                }
+                cellData = model.content.cells ?? [];
+                kernelLang =
+                    model.content.metadata.language_info?.name ||
+                        model.content.metadata.kernelspec?.language ||
+                        'text';
             }
-            const kernelLang = model.content?.metadata?.language_info?.name ||
-                model.content?.metadata?.kernelspec?.language ||
-                'text';
             const selectedCells = attachment.cells
                 .map(cellInfo => {
-                const cell = model.content.cells.find((c) => c.id === cellInfo.id);
+                const cell = cellData.find(c => c.id === cellInfo.id);
                 if (!cell) {
                     return null;
                 }
@@ -548,9 +442,9 @@ ${toolsList}
                             case 'application/vnd.jupyter.widget-view+json':
                                 return `Widget: ${value.model_id ?? 'unknown model'}`;
                             case 'image/png':
-                                return `![image](data:image/png;base64,${value.slice(0, 100)}...)`;
+                                return `![image](data:image/png;base64,${String(value).slice(0, 100)}...)`;
                             case 'image/jpeg':
-                                return `![image](data:image/jpeg;base64,${value.slice(0, 100)}...)`;
+                                return `![image](data:image/jpeg;base64,${String(value).slice(0, 100)}...)`;
                             case 'image/svg+xml':
                                 return String(value).slice(0, 500) + '...\n[svg truncated]';
                             case 'text/html':
@@ -575,8 +469,9 @@ ${toolsList}
                 }
                 let outputs = '';
                 if (cellType === 'code' && Array.isArray(cell.outputs)) {
-                    outputs = cell.outputs
-                        .map((output) => {
+                    const outputsArray = cell.outputs;
+                    outputs = outputsArray
+                        .map(output => {
                         if (output.output_type === 'stream') {
                             return output.text;
                         }
@@ -630,34 +525,37 @@ ${toolsList}
             return null;
         }
         try {
-            const model = await this.input.documentManager?.services.contents.get(attachment.value);
-            if (!model?.content) {
+            // Try reading from an open widget first
+            const widget = this.input.documentManager?.findWidget(attachment.value);
+            if (widget && widget.context && widget.context.model) {
+                const model = widget.context.model;
+                const ymodel = model.sharedModel;
+                if (typeof ymodel.getSource === 'function') {
+                    const source = ymodel.getSource();
+                    return typeof source === 'string'
+                        ? source
+                        : JSON.stringify(source, null, 2);
+                }
+            }
+            // If not open, load from disk
+            const diskModel = await this.input.documentManager?.services.contents.get(attachment.value);
+            if (!diskModel?.content) {
                 return null;
             }
-            if (model.type === 'file') {
+            if (diskModel.type === 'file') {
                 // Regular file content
-                return model.content;
+                return diskModel.content;
             }
-            else if (model.type === 'notebook') {
-                // Clear outputs from notebook cells before sending to LLM
-                // TODO: make this configurable?
-                const cells = model.content.cells.map((cell) => {
-                    const cleanCell = { ...cell };
-                    if (cleanCell.outputs) {
-                        cleanCell.outputs = [];
-                    }
-                    if (cleanCell.execution_count) {
-                        cleanCell.execution_count = null;
-                    }
-                    return cleanCell;
-                });
-                const notebookModel = {
-                    cells,
-                    metadata: model.metadata || {},
-                    nbformat: model.nbformat || 4,
-                    nbformat_minor: model.nbformat_minor || 4
+            if (diskModel.type === 'notebook') {
+                const cleaned = {
+                    ...diskModel,
+                    cells: diskModel.content.cells.map((cell) => ({
+                        ...cell,
+                        outputs: [],
+                        execution_count: null
+                    }))
                 };
-                return JSON.stringify(notebookModel);
+                return JSON.stringify(cleaned);
             }
             return null;
         }
@@ -666,80 +564,122 @@ ${toolsList}
             return null;
         }
     }
+    // Private fields
+    _settingsModel;
+    _user;
+    _toolContexts = new Map();
+    _agentManager;
+    _currentStreamingMessage = null;
+    _nameChanged = new Signal(this);
+    _trans;
+}
+var Private;
+(function (Private) {
+    function escapeHtml(value) {
+        // Prefer the same native escaping approach used in JupyterLab itself
+        // (e.g. `@jupyterlab/completer`).
+        if (typeof document !== 'undefined') {
+            const node = document.createElement('span');
+            node.textContent = value;
+            return node.innerHTML;
+        }
+        // Fallback
+        return value
+            .replace(/&/g, '&amp;')
+            .replace(/</g, '&lt;')
+            .replace(/>/g, '&gt;')
+            .replace(/"/g, '&quot;')
+            .replace(/'/g, '&#39;');
+    }
+    Private.escapeHtml = escapeHtml;
+    const STATUS_CONFIG = {
+        pending: {
+            cssClass: 'jp-ai-tool-pending',
+            statusClass: 'jp-ai-tool-status-pending'
+        },
+        awaiting_approval: {
+            cssClass: 'jp-ai-tool-pending',
+            statusClass: 'jp-ai-tool-status-approval',
+            open: true
+        },
+        approved: {
+            cssClass: 'jp-ai-tool-pending',
+            statusClass: 'jp-ai-tool-status-completed'
+        },
+        rejected: {
+            cssClass: 'jp-ai-tool-error',
+            statusClass: 'jp-ai-tool-status-error'
+        },
+        completed: {
+            cssClass: 'jp-ai-tool-completed',
+            statusClass: 'jp-ai-tool-status-completed'
+        },
+        error: {
+            cssClass: 'jp-ai-tool-error',
+            statusClass: 'jp-ai-tool-status-error'
+        }
+    };
     /**
-     * Updates the status display of a grouped approval message.
-     * @param messageId The message ID to update
-     * @param status The status text to display
-     * @param isSuccess Whether the action was successful
+     * Returns the translated status text for a given tool status.
      */
-    _updateGroupedApprovalStatus(messageId, status, isSuccess) {
-        const existingMessageIndex = this.messages.findIndex(msg => msg.id === messageId);
-        if (existingMessageIndex !== -1) {
-            const existingMessage = this.messages[existingMessageIndex];
-            // Extract tool count and names from existing message
-            const toolCountMatch = existingMessage.body.match(/execute (\d+) tools/);
-            const toolCount = toolCountMatch ? toolCountMatch[1] : 'multiple';
-            const statusIcon = isSuccess ? '✅' : '❌';
-            const statusClass = isSuccess ? 'approved' : 'rejected';
-            const updatedMessage = {
-                ...existingMessage,
-                body: `**${statusIcon} Group Tool Approval: ${status}**
-
-The request to execute ${toolCount} tools has been **${statusClass}**.
-
-<div class="jp-ai-group-approval-${statusClass}">
-Status: ${status}
-</div>`
-            };
-            this.messageAdded(updatedMessage);
+    const getStatusText = (status, trans) => {
+        switch (status) {
+            case 'pending':
+                return trans.__('Running...');
+            case 'awaiting_approval':
+                return trans.__('Awaiting Approval');
+            case 'approved':
+                return trans.__('Approved - Executing...');
+            case 'rejected':
+                return trans.__('Rejected');
+            case 'completed':
+                return trans.__('Completed');
+            case 'error':
+                return trans.__('Error');
         }
-    }
+    };
     /**
-     * Updates the status display of a tool call box.
-     * @param messageId The message ID to update
-     * @param status The status text to display
-     * @param isSuccess Whether the action was successful
+     * Builds HTML for a tool call display.
      */
-    _updateToolCallBoxStatus(messageId, status, isSuccess) {
-        const existingMessageIndex = this.messages.findIndex(msg => msg.id === messageId);
-        if (existingMessageIndex !== -1) {
-            const existingMessage = this.messages[existingMessageIndex];
-            // Extract tool name and input from existing message
-            const toolNameMatch = existingMessage.body.match(/<div class="jp-ai-tool-title">([^<]+)<\/div>/);
-            const toolName = toolNameMatch ? toolNameMatch[1] : 'Unknown Tool';
-            const codeMatch = existingMessage.body.match(/<code>([\s\S]*?)<\/code>/);
-            const toolInput = codeMatch ? codeMatch[1] : '{}';
-            // Determine styling based on status
-            const statusClass = isSuccess
-                ? 'jp-ai-tool-completed'
-                : 'jp-ai-tool-error';
-            const statusColor = isSuccess
-                ? 'jp-ai-tool-status-completed'
-                : 'jp-ai-tool-status-error';
-            const updatedMessage = {
-                ...existingMessage,
-                body: `<details class="jp-ai-tool-call ${statusClass}">
+    function buildToolCallHtml(options) {
+        const { toolName, input, status, output, approvalId, trans } = options;
+        const config = STATUS_CONFIG[status];
+        const statusText = getStatusText(status, trans);
+        const escapedToolName = escapeHtml(toolName);
+        const escapedInput = escapeHtml(input);
+        const openAttr = config.open ? ' open' : '';
+        let bodyContent = `
+<div class="jp-ai-tool-section">
+<div class="jp-ai-tool-label">${trans.__('Input')}</div>
+<pre class="jp-ai-tool-code"><code>${escapedInput}</code></pre>
+</div>`;
+        // Add approval buttons if awaiting approval
+        if (status === 'awaiting_approval' && approvalId) {
+            bodyContent += `
+<div class="jp-ai-tool-approval-buttons jp-ai-approval-id--${approvalId}">
+<button class="jp-ai-approval-btn jp-ai-approval-approve">${trans.__('Approve')}</button>
+<button class="jp-ai-approval-btn jp-ai-approval-reject">${trans.__('Reject')}</button>
+</div>`;
+        }
+        // Add output/result section if provided
+        if (output !== undefined) {
+            const escapedOutput = escapeHtml(output);
+            const label = status === 'error' ? trans.__('Error') : trans.__('Result');
+            bodyContent += `
+<div class="jp-ai-tool-section">
+<div class="jp-ai-tool-label">${label}</div>
+<pre class="jp-ai-tool-code"><code>${escapedOutput}</code></pre>
+</div>`;
+        }
+        return `<details class="jp-ai-tool-call ${config.cssClass}"${openAttr}>
 <summary class="jp-ai-tool-header">
 <div class="jp-ai-tool-icon">⚡</div>
-<div class="jp-ai-tool-title">${toolName}</div>
-<div class="jp-ai-tool-status ${statusColor}">${status}</div>
+<div class="jp-ai-tool-title">${escapedToolName}</div>
+<div class="jp-ai-tool-status ${config.statusClass}">${statusText}</div>
 </summary>
-<div class="jp-ai-tool-body">
-<div class="jp-ai-tool-section">
-<div class="jp-ai-tool-label">Input</div>
-<pre class="jp-ai-tool-code"><code>${toolInput}</code></pre>
+<div class="jp-ai-tool-body">${bodyContent}
 </div>
-</div>
-</details>`
-            };
-            this.messageAdded(updatedMessage);
-        }
+</details>`;
     }
-    // Private fields
-    _settingsModel;
-    _user;
-    _pendingToolCalls = new Map();
-    _agentManager;
-    _currentStreamingMessage = null;
-    _nameChanged = new Signal(this);
-}
+    Private.buildToolCallHtml = buildToolCallHtml;
+})(Private || (Private = {}));