mito-ai 0.1.27__py3-none-any.whl → 0.1.29__py3-none-any.whl

mito_ai-0.1.27.data/data/share/jupyter/labextensions/mito_ai/static/lib_index_js.43d516a828f1937a42bb.js → mito_ai-0.1.29.data/data/share/jupyter/labextensions/mito_ai/static/lib_index_js.8fc39671fbc9ba62e74b.js

@@ -510,6 +510,67 @@ const AlertBlock = ({ content, mitoAIConnectionErrorType }) => {
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (AlertBlock);
 
 
+/***/ }),
+
+/***/ "./lib/Extensions/AiChat/ChatMessage/AssistantCodeBlock.js":
+/*!*****************************************************************!*\
+  !*** ./lib/Extensions/AiChat/ChatMessage/AssistantCodeBlock.js ***!
+  \*****************************************************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
+/* harmony import */ var _PythonCode__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ./PythonCode */ "./lib/Extensions/AiChat/ChatMessage/PythonCode.js");
+/* harmony import */ var _utils_classNames__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../../../utils/classNames */ "./lib/utils/classNames.js");
+/* harmony import */ var _style_CodeBlock_css__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../../../style/CodeBlock.css */ "./style/CodeBlock.css");
+/* harmony import */ var _icons_ExpandIcon__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../../icons/ExpandIcon */ "./lib/icons/ExpandIcon.js");
+/* harmony import */ var _icons_CodeIcon__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../../icons/CodeIcon */ "./lib/icons/CodeIcon.js");
+/* harmony import */ var _CodeBlockToolbar__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ./CodeBlockToolbar */ "./lib/Extensions/AiChat/ChatMessage/CodeBlockToolbar.js");
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+
+
+
+
+
+
+const AssistantCodeBlock = ({ code, isCodeComplete, renderMimeRegistry, previewAICode, acceptAICode, rejectAICode, isLastAiMessage, codeReviewStatus, agentModeEnabled }) => {
+    const [isCodeExpanded, setIsCodeExpanded] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
+    // Memoize calculations
+    const lineCount = (0,react__WEBPACK_IMPORTED_MODULE_0__.useMemo)(() => code.split('\n').length - 2, [code]);
+    const shouldShowToolbar = isLastAiMessage || isCodeComplete;
+    if (agentModeEnabled) {
+        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_2__.classNames)('code-block-container', {
+                'agent-mode': true,
+                'agent-mode-collapsed': !isCodeExpanded
+            }) },
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { onClick: () => setIsCodeExpanded(!isCodeExpanded), className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_2__.classNames)('agent-mode-toggle', {
+                    expanded: isCodeExpanded
+                }) },
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "agent-mode-toggle-content" },
+                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_CodeIcon__WEBPACK_IMPORTED_MODULE_3__["default"], null),
+                    "Generated ",
+                    lineCount,
+                    " lines of code"),
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_ExpandIcon__WEBPACK_IMPORTED_MODULE_4__["default"], { isExpanded: isCodeExpanded })),
+            isCodeExpanded && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_PythonCode__WEBPACK_IMPORTED_MODULE_5__["default"], { code: code, renderMimeRegistry: renderMimeRegistry }))));
+    }
+    else {
+        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "code-block-container" },
+            shouldShowToolbar && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_CodeBlockToolbar__WEBPACK_IMPORTED_MODULE_6__["default"], { code: code, isLastAiMessage: isLastAiMessage, codeReviewStatus: codeReviewStatus, onPreview: previewAICode, onAccept: acceptAICode, onReject: rejectAICode })),
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_PythonCode__WEBPACK_IMPORTED_MODULE_5__["default"], { code: code, renderMimeRegistry: renderMimeRegistry })));
+    }
+};
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (AssistantCodeBlock);
+
+
 /***/ }),
 
 /***/ "./lib/Extensions/AiChat/ChatMessage/ChatDropdown.js":
@@ -874,20 +935,21 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
 /* harmony import */ var _utils_classNames__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../../utils/classNames */ "./lib/utils/classNames.js");
-/* harmony import */ var _CodeBlock__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ./CodeBlock */ "./lib/Extensions/AiChat/ChatMessage/CodeBlock.js");
+/* harmony import */ var _UserCodeBlock__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ./UserCodeBlock */ "./lib/Extensions/AiChat/ChatMessage/UserCodeBlock.js");
+/* harmony import */ var _AssistantCodeBlock__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ./AssistantCodeBlock */ "./lib/Extensions/AiChat/ChatMessage/AssistantCodeBlock.js");
 /* harmony import */ var _AlertBlock__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ./AlertBlock */ "./lib/Extensions/AiChat/ChatMessage/AlertBlock.js");
-/* harmony import */ var _MarkdownBlock__WEBPACK_IMPORTED_MODULE_13__ = __webpack_require__(/*! ./MarkdownBlock */ "./lib/Extensions/AiChat/ChatMessage/MarkdownBlock.js");
+/* harmony import */ var _MarkdownBlock__WEBPACK_IMPORTED_MODULE_14__ = __webpack_require__(/*! ./MarkdownBlock */ "./lib/Extensions/AiChat/ChatMessage/MarkdownBlock.js");
 /* harmony import */ var _utils_strings__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../../utils/strings */ "./lib/utils/strings.js");
-/* harmony import */ var _icons_Pencil__WEBPACK_IMPORTED_MODULE_14__ = __webpack_require__(/*! ../../../icons/Pencil */ "./lib/icons/Pencil.js");
+/* harmony import */ var _icons_Pencil__WEBPACK_IMPORTED_MODULE_15__ = __webpack_require__(/*! ../../../icons/Pencil */ "./lib/icons/Pencil.js");
 /* harmony import */ var _ChatInput__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ./ChatInput */ "./lib/Extensions/AiChat/ChatMessage/ChatInput.js");
-/* harmony import */ var _components_TextAndIconButton__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ../../../components/TextAndIconButton */ "./lib/components/TextAndIconButton.js");
-/* harmony import */ var _icons_PlayButtonIcon__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ../../../icons/PlayButtonIcon */ "./lib/icons/PlayButtonIcon.js");
-/* harmony import */ var _icons_CopyIcon__WEBPACK_IMPORTED_MODULE_11__ = __webpack_require__(/*! ../../../icons/CopyIcon */ "./lib/icons/CopyIcon.js");
-/* harmony import */ var _utils_copyToClipboard__WEBPACK_IMPORTED_MODULE_10__ = __webpack_require__(/*! ../../../utils/copyToClipboard */ "./lib/utils/copyToClipboard.js");
-/* harmony import */ var _components_TextButton__WEBPACK_IMPORTED_MODULE_12__ = __webpack_require__(/*! ../../../components/TextButton */ "./lib/components/TextButton.js");
+/* harmony import */ var _components_TextAndIconButton__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ../../../components/TextAndIconButton */ "./lib/components/TextAndIconButton.js");
+/* harmony import */ var _icons_PlayButtonIcon__WEBPACK_IMPORTED_MODULE_10__ = __webpack_require__(/*! ../../../icons/PlayButtonIcon */ "./lib/icons/PlayButtonIcon.js");
+/* harmony import */ var _icons_CopyIcon__WEBPACK_IMPORTED_MODULE_12__ = __webpack_require__(/*! ../../../icons/CopyIcon */ "./lib/icons/CopyIcon.js");
+/* harmony import */ var _utils_copyToClipboard__WEBPACK_IMPORTED_MODULE_11__ = __webpack_require__(/*! ../../../utils/copyToClipboard */ "./lib/utils/copyToClipboard.js");
+/* harmony import */ var _components_TextButton__WEBPACK_IMPORTED_MODULE_13__ = __webpack_require__(/*! ../../../components/TextButton */ "./lib/components/TextButton.js");
 /* harmony import */ var _style_ChatMessage_css__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../../../style/ChatMessage.css */ "./style/ChatMessage.css");
 /* harmony import */ var _style_MarkdownMessage_css__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../../../../style/MarkdownMessage.css */ "./style/MarkdownMessage.css");
-/* harmony import */ var _components_AgentToolComponents_GetCellOutputToolUI__WEBPACK_IMPORTED_MODULE_15__ = __webpack_require__(/*! ../../../components/AgentToolComponents/GetCellOutputToolUI */ "./lib/components/AgentToolComponents/GetCellOutputToolUI.js");
+/* harmony import */ var _components_AgentToolComponents_GetCellOutputToolUI__WEBPACK_IMPORTED_MODULE_16__ = __webpack_require__(/*! ../../../components/AgentToolComponents/GetCellOutputToolUI */ "./lib/components/AgentToolComponents/GetCellOutputToolUI.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -908,7 +970,8 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
-const ChatMessage = ({ message, messageType, promptType, agentResponse, messageIndex, mitoAIConnectionError, mitoAIConnectionErrorType, notebookTracker, renderMimeRegistry, isLastAiMessage, isLastMessage, operatingSystem, previewAICode, acceptAICode, rejectAICode, onUpdateMessage, contextManager, codeReviewStatus, setNextSteps, }) => {
+
+const ChatMessage = ({ message, messageType, promptType, agentResponse, messageIndex, mitoAIConnectionError, mitoAIConnectionErrorType, notebookTracker, renderMimeRegistry, isLastAiMessage, isLastMessage, operatingSystem, previewAICode, acceptAICode, rejectAICode, onUpdateMessage, contextManager, codeReviewStatus, setNextSteps, agentModeEnabled, }) => {
     const [isEditing, setIsEditing] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
     if (message.role !== 'user' && message.role !== 'assistant') {
         return null;
@@ -958,15 +1021,15 @@ const ChatMessage = ({ message, messageType, promptType, agentResponse, messageI
                 // An empty code will look like this '```python  ```'
                 if (messagePart.length > 14) {
                     return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement((react__WEBPACK_IMPORTED_MODULE_0___default().Fragment), null,
-                        react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_CodeBlock__WEBPACK_IMPORTED_MODULE_7__["default"], { key: index + messagePart, code: messagePart, isCodeComplete: isCodeComplete, role: message.role, renderMimeRegistry: renderMimeRegistry, previewAICode: previewAICode, acceptAICode: acceptAICode, rejectAICode: rejectAICode, isLastAiMessage: isLastAiMessage, codeReviewStatus: codeReviewStatus }),
+                        message.role === 'user' ? (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_UserCodeBlock__WEBPACK_IMPORTED_MODULE_7__["default"], { code: messagePart, renderMimeRegistry: renderMimeRegistry, agentModeEnabled: agentModeEnabled })) : (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_AssistantCodeBlock__WEBPACK_IMPORTED_MODULE_8__["default"], { code: messagePart, isCodeComplete: isCodeComplete, renderMimeRegistry: renderMimeRegistry, previewAICode: previewAICode, acceptAICode: acceptAICode, rejectAICode: rejectAICode, isLastAiMessage: isLastAiMessage, codeReviewStatus: codeReviewStatus, agentModeEnabled: agentModeEnabled })),
                         isLastAiMessage && isCodeComplete && codeReviewStatus === 'chatPreview' &&
                             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'chat-message-buttons' },
-                                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_TextAndIconButton__WEBPACK_IMPORTED_MODULE_8__["default"], { onClick: () => { previewAICode(); }, text: 'Overwrite Active Cell', icon: _icons_PlayButtonIcon__WEBPACK_IMPORTED_MODULE_9__["default"], title: 'Write the Ai generated code to the active cell in the jupyter notebook, replacing the current code.', variant: 'gray', width: 'fit-contents' }),
-                                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_TextAndIconButton__WEBPACK_IMPORTED_MODULE_8__["default"], { onClick: () => { void (0,_utils_copyToClipboard__WEBPACK_IMPORTED_MODULE_10__["default"])(messagePart); }, text: 'Copy', icon: _icons_CopyIcon__WEBPACK_IMPORTED_MODULE_11__["default"], title: 'Copy the Ai generated code to your clipboard', variant: 'gray', width: 'fit-contents' })),
+                                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_TextAndIconButton__WEBPACK_IMPORTED_MODULE_9__["default"], { onClick: () => { previewAICode(); }, text: 'Overwrite Active Cell', icon: _icons_PlayButtonIcon__WEBPACK_IMPORTED_MODULE_10__["default"], title: 'Write the Ai generated code to the active cell in the jupyter notebook, replacing the current code.', variant: 'gray', width: 'fit-contents' }),
+                                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_TextAndIconButton__WEBPACK_IMPORTED_MODULE_9__["default"], { onClick: () => { void (0,_utils_copyToClipboard__WEBPACK_IMPORTED_MODULE_11__["default"])(messagePart); }, text: 'Copy', icon: _icons_CopyIcon__WEBPACK_IMPORTED_MODULE_12__["default"], title: 'Copy the Ai generated code to your clipboard', variant: 'gray', width: 'fit-contents' })),
                         isLastAiMessage && isCodeComplete && codeReviewStatus === 'codeCellPreview' &&
                             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'chat-message-buttons' },
-                                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_TextButton__WEBPACK_IMPORTED_MODULE_12__["default"], { onClick: () => { acceptAICode(); }, text: `Accept code ${operatingSystem === 'mac' ? '⌘Y' : 'Ctrl+Y'}`, title: 'Accept the Ai generated code', variant: 'green', width: 'fit-contents' }),
-                                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_TextButton__WEBPACK_IMPORTED_MODULE_12__["default"], { onClick: () => { rejectAICode(); }, text: `Reject code ${operatingSystem === 'mac' ? '⌘U' : 'Ctrl+U'}`, title: 'Reject the Ai generated code and revert to the previous version of the code cell', variant: 'red', width: 'fit-contents' }))));
+                                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_TextButton__WEBPACK_IMPORTED_MODULE_13__["default"], { onClick: () => { acceptAICode(); }, text: `Accept code ${operatingSystem === 'mac' ? '⌘Y' : 'Ctrl+Y'}`, title: 'Accept the Ai generated code', variant: 'green', width: 'fit-contents' }),
+                                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_TextButton__WEBPACK_IMPORTED_MODULE_13__["default"], { onClick: () => { rejectAICode(); }, text: `Reject code ${operatingSystem === 'mac' ? '⌘U' : 'Ctrl+U'}`, title: 'Reject the Ai generated code and revert to the previous version of the code cell', variant: 'red', width: 'fit-contents' }))));
                 }
                 else {
                     // Return null for empty code blocks
@@ -982,15 +1045,15 @@ const ChatMessage = ({ message, messageType, promptType, agentResponse, messageI
                             }
                         } }, message.role === 'user' && promptType === 'smartDebug' ? (
                     /* Use a pre tag to preserve the newline and indentation of the error message */
-                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("pre", { className: "chat-taskpane-smart-debug-error-message" }, messagePart)) : (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_MarkdownBlock__WEBPACK_IMPORTED_MODULE_13__["default"], { markdown: messagePart, renderMimeRegistry: renderMimeRegistry, notebookTracker: notebookTracker })))));
+                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("pre", { className: "chat-taskpane-smart-debug-error-message" }, messagePart)) : (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_MarkdownBlock__WEBPACK_IMPORTED_MODULE_14__["default"], { markdown: messagePart, renderMimeRegistry: renderMimeRegistry, notebookTracker: notebookTracker })))));
             }
         }),
         editable &&
             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "message-action-buttons" },
                 react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { className: "message-start-editing-button", onClick: handleEditClick, title: "Edit message" },
-                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_Pencil__WEBPACK_IMPORTED_MODULE_14__["default"], null))),
+                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_Pencil__WEBPACK_IMPORTED_MODULE_15__["default"], null))),
         (agentResponse === null || agentResponse === void 0 ? void 0 : agentResponse.type) === 'get_cell_output' &&
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AgentToolComponents_GetCellOutputToolUI__WEBPACK_IMPORTED_MODULE_15__["default"], null)));
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AgentToolComponents_GetCellOutputToolUI__WEBPACK_IMPORTED_MODULE_16__["default"], null)));
 };
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (ChatMessage);
 
@@ -1045,10 +1108,10 @@ const Citation = ({ citationIndex, cellId, line, notebookTracker }) => {
 
 /***/ }),
 
-/***/ "./lib/Extensions/AiChat/ChatMessage/CodeBlock.js":
-/*!********************************************************!*\
-  !*** ./lib/Extensions/AiChat/ChatMessage/CodeBlock.js ***!
-  \********************************************************/
+/***/ "./lib/Extensions/AiChat/ChatMessage/CodeBlockToolbar.js":
+/*!***************************************************************!*\
+  !*** ./lib/Extensions/AiChat/ChatMessage/CodeBlockToolbar.js ***!
+  \***************************************************************/
 /***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
 
 __webpack_require__.r(__webpack_exports__);
@@ -1057,15 +1120,12 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony export */ });
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
-/* harmony import */ var _PythonCode__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ./PythonCode */ "./lib/Extensions/AiChat/ChatMessage/PythonCode.js");
-/* harmony import */ var _style_CodeBlock_css__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../../../style/CodeBlock.css */ "./style/CodeBlock.css");
-/* harmony import */ var _utils_copyToClipboard__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ../../../utils/copyToClipboard */ "./lib/utils/copyToClipboard.js");
-/* harmony import */ var _components_IconButton__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../../components/IconButton */ "./lib/components/IconButton.js");
-/* harmony import */ var _icons_CopyIcon__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ../../../icons/CopyIcon */ "./lib/icons/CopyIcon.js");
-/* harmony import */ var _icons_PlayButtonIcon__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../../icons/PlayButtonIcon */ "./lib/icons/PlayButtonIcon.js");
-/* harmony import */ var _icons_AcceptIcon__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../../../icons/AcceptIcon */ "./lib/icons/AcceptIcon.js");
-/* harmony import */ var _icons_RejectIcon__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ../../../icons/RejectIcon */ "./lib/icons/RejectIcon.js");
-/* harmony import */ var _icons_ExpandIcon__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../../icons/ExpandIcon */ "./lib/icons/ExpandIcon.js");
+/* harmony import */ var _utils_copyToClipboard__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../../../utils/copyToClipboard */ "./lib/utils/copyToClipboard.js");
+/* harmony import */ var _components_IconButton__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../../components/IconButton */ "./lib/components/IconButton.js");
+/* harmony import */ var _icons_CopyIcon__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../../icons/CopyIcon */ "./lib/icons/CopyIcon.js");
+/* harmony import */ var _icons_PlayButtonIcon__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../../../icons/PlayButtonIcon */ "./lib/icons/PlayButtonIcon.js");
+/* harmony import */ var _icons_AcceptIcon__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../../icons/AcceptIcon */ "./lib/icons/AcceptIcon.js");
+/* harmony import */ var _icons_RejectIcon__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../../icons/RejectIcon */ "./lib/icons/RejectIcon.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -1077,44 +1137,15 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
-
-
-
-const CodeBlock = ({ code, isCodeComplete, role, renderMimeRegistry, previewAICode, acceptAICode, rejectAICode, isLastAiMessage, codeReviewStatus, }) => {
-    const [isExpanded, setIsExpanded] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
-    if (role === 'user') {
-        const numCodePreviewLines = 5;
-        const isCodeExpandable = code.split('\n').length > numCodePreviewLines;
-        const previewCode = code.split('\n').slice(0, numCodePreviewLines).join('\n');
-        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'code-block-container active-cell-code-block', onClick: () => setIsExpanded(!isExpanded), style: { cursor: 'pointer' } },
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_PythonCode__WEBPACK_IMPORTED_MODULE_2__["default"], { code: isExpanded ? code : previewCode, renderMimeRegistry: renderMimeRegistry }),
-            isCodeExpandable && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'code-block-expand-button', title: isExpanded ? "Collapse" : "Expand", onClick: (e) => {
-                    e.stopPropagation();
-                    setIsExpanded(!isExpanded);
-                } },
-                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_ExpandIcon__WEBPACK_IMPORTED_MODULE_3__["default"], { isExpanded: isExpanded })))));
-    }
-    if (role === 'assistant') {
-        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'code-block-container' },
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement((react__WEBPACK_IMPORTED_MODULE_0___default().Fragment), null,
-                isLastAiMessage && isCodeComplete &&
-                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'code-block-toolbar' },
-                        codeReviewStatus === 'chatPreview' &&
-                            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_4__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_PlayButtonIcon__WEBPACK_IMPORTED_MODULE_5__["default"], null), title: "Overwrite Active Cell", onClick: () => { previewAICode(); } }),
-                        codeReviewStatus === 'codeCellPreview' &&
-                            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_4__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_AcceptIcon__WEBPACK_IMPORTED_MODULE_6__["default"], null), title: "Accept AI Generated Code", onClick: () => { acceptAICode(); }, style: { color: 'var(--green-700)' } }),
-                        codeReviewStatus === 'codeCellPreview' &&
-                            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_4__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_RejectIcon__WEBPACK_IMPORTED_MODULE_7__["default"], null), title: "Reject AI Generated Code", onClick: () => { rejectAICode(); }, style: { color: 'var(--red-700)' } }),
-                        codeReviewStatus !== 'codeCellPreview' &&
-                            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_4__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_CopyIcon__WEBPACK_IMPORTED_MODULE_8__["default"], null), title: "Copy", onClick: () => { void (0,_utils_copyToClipboard__WEBPACK_IMPORTED_MODULE_9__["default"])(code); } })),
-                !isLastAiMessage &&
-                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'code-block-toolbar' },
-                        react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_4__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_CopyIcon__WEBPACK_IMPORTED_MODULE_8__["default"], null), title: "Copy", onClick: () => { void (0,_utils_copyToClipboard__WEBPACK_IMPORTED_MODULE_9__["default"])(code); } }))),
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_PythonCode__WEBPACK_IMPORTED_MODULE_2__["default"], { code: code, renderMimeRegistry: renderMimeRegistry })));
-    }
-    return react__WEBPACK_IMPORTED_MODULE_0___default().createElement((react__WEBPACK_IMPORTED_MODULE_0___default().Fragment), null);
+const CodeBlockToolbar = ({ code, isLastAiMessage = false, codeReviewStatus, onPreview, onAccept, onReject }) => {
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'code-block-toolbar' },
+        isLastAiMessage && codeReviewStatus === 'chatPreview' && onPreview && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_1__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_PlayButtonIcon__WEBPACK_IMPORTED_MODULE_2__["default"], null), title: "Overwrite Active Cell", onClick: onPreview })),
+        isLastAiMessage && codeReviewStatus === 'codeCellPreview' && onAccept && onReject && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement((react__WEBPACK_IMPORTED_MODULE_0___default().Fragment), null,
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_1__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_AcceptIcon__WEBPACK_IMPORTED_MODULE_3__["default"], null), title: "Accept AI Generated Code", onClick: onAccept, style: { color: 'var(--green-700)' } }),
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_1__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_RejectIcon__WEBPACK_IMPORTED_MODULE_4__["default"], null), title: "Reject AI Generated Code", onClick: onReject, style: { color: 'var(--red-700)' } }))),
+        (!isLastAiMessage || codeReviewStatus !== 'codeCellPreview') && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_1__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_CopyIcon__WEBPACK_IMPORTED_MODULE_5__["default"], null), title: "Copy", onClick: () => { void (0,_utils_copyToClipboard__WEBPACK_IMPORTED_MODULE_6__["default"])(code); } }))));
 };
-/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (CodeBlock);
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (CodeBlockToolbar);
 
 
 /***/ }),
@@ -1447,6 +1478,47 @@ const ScrollableSuggestions = ({ onSelectSuggestion }) => {
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (ScrollableSuggestions);
 
 
+/***/ }),
+
+/***/ "./lib/Extensions/AiChat/ChatMessage/UserCodeBlock.js":
+/*!************************************************************!*\
+  !*** ./lib/Extensions/AiChat/ChatMessage/UserCodeBlock.js ***!
+  \************************************************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
+/* harmony import */ var _PythonCode__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ./PythonCode */ "./lib/Extensions/AiChat/ChatMessage/PythonCode.js");
+/* harmony import */ var _style_CodeBlock_css__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../../../style/CodeBlock.css */ "./style/CodeBlock.css");
+/* harmony import */ var _icons_ExpandIcon__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../../icons/ExpandIcon */ "./lib/icons/ExpandIcon.js");
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+
+
+
+const UserCodeBlock = ({ code, renderMimeRegistry, agentModeEnabled }) => {
+    const [isExpanded, setIsExpanded] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
+    const numCodePreviewLines = 5;
+    const isCodeExpandable = code.split('\n').length > numCodePreviewLines;
+    const previewCode = code.split('\n').slice(0, numCodePreviewLines).join('\n');
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: `code-block-container active-cell-code-block ${agentModeEnabled ? 'agent-mode' : ''}`, onClick: () => setIsExpanded(!isExpanded), style: { cursor: 'pointer' } },
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_PythonCode__WEBPACK_IMPORTED_MODULE_2__["default"], { code: isExpanded ? code : previewCode, renderMimeRegistry: renderMimeRegistry }),
+        isCodeExpandable && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'code-block-expand-button', title: isExpanded ? "Collapse" : "Expand", onClick: (e) => {
+                e.stopPropagation();
+                setIsExpanded(!isExpanded);
+            } },
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_ExpandIcon__WEBPACK_IMPORTED_MODULE_3__["default"], { isExpanded: isExpanded })))));
+};
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (UserCodeBlock);
+
+
 /***/ }),
 
 /***/ "./lib/Extensions/AiChat/ChatTaskpane.js":
@@ -1470,33 +1542,34 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony import */ var _style_TextButton_css__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../../style/TextButton.css */ "./style/TextButton.css");
 /* harmony import */ var _jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! @jupyterlab/ui-components */ "webpack/sharing/consume/default/@jupyterlab/ui-components");
 /* harmony import */ var _jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_6___default = /*#__PURE__*/__webpack_require__.n(_jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_6__);
-/* harmony import */ var _icons__WEBPACK_IMPORTED_MODULE_22__ = __webpack_require__(/*! ../../icons */ "./lib/icons/index.js");
-/* harmony import */ var _icons_MitoLogo__WEBPACK_IMPORTED_MODULE_23__ = __webpack_require__(/*! ../../icons/MitoLogo */ "./lib/icons/MitoLogo.js");
-/* harmony import */ var _ChatMessage_ChatInput__WEBPACK_IMPORTED_MODULE_31__ = __webpack_require__(/*! ./ChatMessage/ChatInput */ "./lib/Extensions/AiChat/ChatMessage/ChatInput.js");
-/* harmony import */ var _ChatMessage_ChatMessage__WEBPACK_IMPORTED_MODULE_25__ = __webpack_require__(/*! ./ChatMessage/ChatMessage */ "./lib/Extensions/AiChat/ChatMessage/ChatMessage.js");
-/* harmony import */ var _ChatMessage_ScrollableSuggestions__WEBPACK_IMPORTED_MODULE_29__ = __webpack_require__(/*! ./ChatMessage/ScrollableSuggestions */ "./lib/Extensions/AiChat/ChatMessage/ScrollableSuggestions.js");
+/* harmony import */ var _icons__WEBPACK_IMPORTED_MODULE_23__ = __webpack_require__(/*! ../../icons */ "./lib/icons/index.js");
+/* harmony import */ var _icons_MitoLogo__WEBPACK_IMPORTED_MODULE_24__ = __webpack_require__(/*! ../../icons/MitoLogo */ "./lib/icons/MitoLogo.js");
+/* harmony import */ var _ChatMessage_ChatInput__WEBPACK_IMPORTED_MODULE_32__ = __webpack_require__(/*! ./ChatMessage/ChatInput */ "./lib/Extensions/AiChat/ChatMessage/ChatInput.js");
+/* harmony import */ var _ChatMessage_ChatMessage__WEBPACK_IMPORTED_MODULE_26__ = __webpack_require__(/*! ./ChatMessage/ChatMessage */ "./lib/Extensions/AiChat/ChatMessage/ChatMessage.js");
+/* harmony import */ var _ChatMessage_ScrollableSuggestions__WEBPACK_IMPORTED_MODULE_30__ = __webpack_require__(/*! ./ChatMessage/ScrollableSuggestions */ "./lib/Extensions/AiChat/ChatMessage/ScrollableSuggestions.js");
 /* harmony import */ var _ChatHistoryManager__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ./ChatHistoryManager */ "./lib/Extensions/AiChat/ChatHistoryManager.js");
-/* harmony import */ var _CodeDiffDisplay__WEBPACK_IMPORTED_MODULE_18__ = __webpack_require__(/*! ./CodeDiffDisplay */ "./lib/Extensions/AiChat/CodeDiffDisplay.js");
-/* harmony import */ var _components_ToggleButton__WEBPACK_IMPORTED_MODULE_32__ = __webpack_require__(/*! ../../components/ToggleButton */ "./lib/components/ToggleButton.js");
-/* harmony import */ var _components_IconButton__WEBPACK_IMPORTED_MODULE_19__ = __webpack_require__(/*! ../../components/IconButton */ "./lib/components/IconButton.js");
-/* harmony import */ var _components_LoadingDots__WEBPACK_IMPORTED_MODULE_26__ = __webpack_require__(/*! ../../components/LoadingDots */ "./lib/components/LoadingDots.js");
-/* harmony import */ var _commands__WEBPACK_IMPORTED_MODULE_17__ = __webpack_require__(/*! ../../commands */ "./lib/commands.js");
-/* harmony import */ var _utils_codeDiff__WEBPACK_IMPORTED_MODULE_16__ = __webpack_require__(/*! ../../utils/codeDiff */ "./lib/utils/codeDiff.js");
-/* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_11__ = __webpack_require__(/*! ../../utils/notebook */ "./lib/utils/notebook.js");
-/* harmony import */ var _utils_strings__WEBPACK_IMPORTED_MODULE_13__ = __webpack_require__(/*! ../../utils/strings */ "./lib/utils/strings.js");
-/* harmony import */ var _utils_agentActions__WEBPACK_IMPORTED_MODULE_15__ = __webpack_require__(/*! ../../utils/agentActions */ "./lib/utils/agentActions.js");
-/* harmony import */ var _utils_scroll__WEBPACK_IMPORTED_MODULE_10__ = __webpack_require__(/*! ../../utils/scroll */ "./lib/utils/scroll.js");
-/* harmony import */ var _components_LoadingCircle__WEBPACK_IMPORTED_MODULE_33__ = __webpack_require__(/*! ../../components/LoadingCircle */ "./lib/components/LoadingCircle.js");
+/* harmony import */ var _CodeDiffDisplay__WEBPACK_IMPORTED_MODULE_19__ = __webpack_require__(/*! ./CodeDiffDisplay */ "./lib/Extensions/AiChat/CodeDiffDisplay.js");
+/* harmony import */ var _components_ToggleButton__WEBPACK_IMPORTED_MODULE_33__ = __webpack_require__(/*! ../../components/ToggleButton */ "./lib/components/ToggleButton.js");
+/* harmony import */ var _components_IconButton__WEBPACK_IMPORTED_MODULE_20__ = __webpack_require__(/*! ../../components/IconButton */ "./lib/components/IconButton.js");
+/* harmony import */ var _components_LoadingDots__WEBPACK_IMPORTED_MODULE_27__ = __webpack_require__(/*! ../../components/LoadingDots */ "./lib/components/LoadingDots.js");
+/* harmony import */ var _commands__WEBPACK_IMPORTED_MODULE_18__ = __webpack_require__(/*! ../../commands */ "./lib/commands.js");
+/* harmony import */ var _utils_codeDiff__WEBPACK_IMPORTED_MODULE_17__ = __webpack_require__(/*! ../../utils/codeDiff */ "./lib/utils/codeDiff.js");
+/* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_12__ = __webpack_require__(/*! ../../utils/notebook */ "./lib/utils/notebook.js");
+/* harmony import */ var _utils_strings__WEBPACK_IMPORTED_MODULE_14__ = __webpack_require__(/*! ../../utils/strings */ "./lib/utils/strings.js");
+/* harmony import */ var _utils_agentActions__WEBPACK_IMPORTED_MODULE_16__ = __webpack_require__(/*! ../../utils/agentActions */ "./lib/utils/agentActions.js");
+/* harmony import */ var _utils_scroll__WEBPACK_IMPORTED_MODULE_11__ = __webpack_require__(/*! ../../utils/scroll */ "./lib/utils/scroll.js");
+/* harmony import */ var _components_LoadingCircle__WEBPACK_IMPORTED_MODULE_34__ = __webpack_require__(/*! ../../components/LoadingCircle */ "./lib/components/LoadingCircle.js");
 /* harmony import */ var _components_ModelSelector__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ../../components/ModelSelector */ "./lib/components/ModelSelector.js");
-/* harmony import */ var _utils_blacklistedWords__WEBPACK_IMPORTED_MODULE_14__ = __webpack_require__(/*! ../../utils/blacklistedWords */ "./lib/utils/blacklistedWords.js");
-/* harmony import */ var _components_DropdownMenu__WEBPACK_IMPORTED_MODULE_21__ = __webpack_require__(/*! ../../components/DropdownMenu */ "./lib/components/DropdownMenu.js");
-/* harmony import */ var _SettingsManager_SettingsManagerPlugin__WEBPACK_IMPORTED_MODULE_20__ = __webpack_require__(/*! ../SettingsManager/SettingsManagerPlugin */ "./lib/Extensions/SettingsManager/SettingsManagerPlugin.js");
+/* harmony import */ var _utils_blacklistedWords__WEBPACK_IMPORTED_MODULE_15__ = __webpack_require__(/*! ../../utils/blacklistedWords */ "./lib/utils/blacklistedWords.js");
+/* harmony import */ var _components_DropdownMenu__WEBPACK_IMPORTED_MODULE_22__ = __webpack_require__(/*! ../../components/DropdownMenu */ "./lib/components/DropdownMenu.js");
+/* harmony import */ var _SettingsManager_SettingsManagerPlugin__WEBPACK_IMPORTED_MODULE_21__ = __webpack_require__(/*! ../SettingsManager/SettingsManagerPlugin */ "./lib/Extensions/SettingsManager/SettingsManagerPlugin.js");
 /* harmony import */ var _FirstMessage__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ./FirstMessage */ "./lib/Extensions/AiChat/FirstMessage.js");
-/* harmony import */ var _CTACarousel__WEBPACK_IMPORTED_MODULE_24__ = __webpack_require__(/*! ./CTACarousel */ "./lib/Extensions/AiChat/CTACarousel.js");
-/* harmony import */ var _components_NextStepsPills__WEBPACK_IMPORTED_MODULE_30__ = __webpack_require__(/*! ../../components/NextStepsPills */ "./lib/components/NextStepsPills.js");
-/* harmony import */ var _icons_UndoIcon__WEBPACK_IMPORTED_MODULE_28__ = __webpack_require__(/*! ../../icons/UndoIcon */ "./lib/icons/UndoIcon.js");
-/* harmony import */ var _components_TextAndIconButton__WEBPACK_IMPORTED_MODULE_27__ = __webpack_require__(/*! ../../components/TextAndIconButton */ "./lib/components/TextAndIconButton.js");
-/* harmony import */ var _utils_checkpoint__WEBPACK_IMPORTED_MODULE_12__ = __webpack_require__(/*! ../../utils/checkpoint */ "./lib/utils/checkpoint.js");
+/* harmony import */ var _CTACarousel__WEBPACK_IMPORTED_MODULE_25__ = __webpack_require__(/*! ./CTACarousel */ "./lib/Extensions/AiChat/CTACarousel.js");
+/* harmony import */ var _components_NextStepsPills__WEBPACK_IMPORTED_MODULE_31__ = __webpack_require__(/*! ../../components/NextStepsPills */ "./lib/components/NextStepsPills.js");
+/* harmony import */ var _icons_UndoIcon__WEBPACK_IMPORTED_MODULE_29__ = __webpack_require__(/*! ../../icons/UndoIcon */ "./lib/icons/UndoIcon.js");
+/* harmony import */ var _components_TextAndIconButton__WEBPACK_IMPORTED_MODULE_28__ = __webpack_require__(/*! ../../components/TextAndIconButton */ "./lib/components/TextAndIconButton.js");
+/* harmony import */ var _utils_checkpoint__WEBPACK_IMPORTED_MODULE_13__ = __webpack_require__(/*! ../../utils/checkpoint */ "./lib/utils/checkpoint.js");
+/* harmony import */ var _utils_waitForNotebookReady__WEBPACK_IMPORTED_MODULE_10__ = __webpack_require__(/*! ../../utils/waitForNotebookReady */ "./lib/utils/waitForNotebookReady.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -1534,6 +1607,7 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+
 
 
 const AGENT_EXECUTION_DEPTH_LIMIT = 20;
@@ -1765,6 +1839,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
                 }
                 const firstMessage = (0,_FirstMessage__WEBPACK_IMPORTED_MODULE_9__.getFirstMessageFromCookie)();
                 if (firstMessage) {
+                    await (0,_utils_waitForNotebookReady__WEBPACK_IMPORTED_MODULE_10__.waitForNotebookReady)(notebookTracker);
                     await startAgentExecution(firstMessage);
                 }
             }
@@ -1802,7 +1877,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
     // Scroll to bottom whenever chat history updates, but only if in follow mode
     (0,react__WEBPACK_IMPORTED_MODULE_2__.useEffect)(() => {
         if (autoScrollFollowMode) {
-            (0,_utils_scroll__WEBPACK_IMPORTED_MODULE_10__.scrollToDiv)(chatMessagesRef);
+            (0,_utils_scroll__WEBPACK_IMPORTED_MODULE_11__.scrollToDiv)(chatMessagesRef);
         }
     }, [chatHistoryManager.getDisplayOptimizedHistory().length, chatHistoryManager, autoScrollFollowMode]);
     // Add scroll event handler to detect manual scrolling
@@ -1911,7 +1986,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
             agentExecutionMetadata.index = messageIndex;
         }
         if (sendActiveCellOutput) {
-            const activeCellOutput = await (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_11__.getActiveCellOutput)(notebookTracker);
+            const activeCellOutput = await (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_12__.getActiveCellOutput)(notebookTracker);
             if (activeCellOutput !== undefined) {
                 agentExecutionMetadata.base64EncodedActiveCellOutput = activeCellOutput;
             }
@@ -1951,7 +2026,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
         // Add the active cell output to the metadata afterwards setting the chatHistoryManager so that 
         // we don't have to wait on turning the output into a base64 image before we can add the user's message
         // to the chat.
-        const activeCellOutput = await (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_11__.getActiveCellOutput)(notebookTracker);
+        const activeCellOutput = await (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_12__.getActiveCellOutput)(notebookTracker);
         if (activeCellOutput !== undefined) {
             chatMessageMetadata.base64EncodedActiveCellOutput = activeCellOutput;
         }
@@ -2110,7 +2185,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
         setAgentExecutionStatus('idle');
     };
     const startAgentExecution = async (input, messageIndex, selectedRules) => {
-        await (0,_utils_checkpoint__WEBPACK_IMPORTED_MODULE_12__.createCheckpoint)(app, setHasCheckpoint);
+        await (0,_utils_checkpoint__WEBPACK_IMPORTED_MODULE_13__.createCheckpoint)(app, setHasCheckpoint);
         setAgentExecutionStatus('working');
         // Enable follow mode when user starts agent execution
         setAutoScrollFollowMode(true);
@@ -2142,9 +2217,9 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
             const aiDisplayOptimizedChatItem = chatHistoryManagerRef.current.getLastAIDisplayOptimizedChatItem();
             // # TODO: Make this is a helper function so we can also use it in the auto error fixup! 
             if (aiDisplayOptimizedChatItem) {
-                const aiGeneratedCode = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_13__.getCodeBlockFromMessage)(aiDisplayOptimizedChatItem.message);
+                const aiGeneratedCode = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_14__.getCodeBlockFromMessage)(aiDisplayOptimizedChatItem.message);
                 if (aiGeneratedCode) {
-                    const securityCheck = (0,_utils_blacklistedWords__WEBPACK_IMPORTED_MODULE_14__.checkForBlacklistedWords)(aiGeneratedCode);
+                    const securityCheck = (0,_utils_blacklistedWords__WEBPACK_IMPORTED_MODULE_15__.checkForBlacklistedWords)(aiGeneratedCode);
                     if (!securityCheck.safe) {
                         console.error('Security Warning:', securityCheck.reason);
                         addAIMessageFromResponseAndUpdateState(`I cannot execute this code without your approval because this code did not pass my security checks. ${securityCheck.reason}. For your safety, I am stopping execution of this plan.`, 'agent:execution', chatHistoryManager);
@@ -2171,8 +2246,8 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
             }
             if (agentResponse.type === 'cell_update' && agentResponse.cell_update) {
                 // Run the code and handle any errors
-                await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_15__.acceptAndRunCellUpdate)(agentResponse.cell_update, notebookTracker, app, previewAICodeToActiveCell, acceptAICode);
-                const status = await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_15__.retryIfExecutionError)(notebookTracker, app, getDuplicateChatHistoryManager, addAIMessageFromResponseAndUpdateState, sendAgentSmartDebugMessage, previewAICodeToActiveCell, acceptAICode, shouldContinueAgentExecution, finalizeAgentStop, chatHistoryManagerRef);
+                await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_16__.acceptAndRunCellUpdate)(agentResponse.cell_update, notebookTracker, app, previewAICodeToActiveCell, acceptAICode);
+                const status = await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_16__.retryIfExecutionError)(notebookTracker, app, getDuplicateChatHistoryManager, addAIMessageFromResponseAndUpdateState, sendAgentSmartDebugMessage, previewAICodeToActiveCell, acceptAICode, shouldContinueAgentExecution, finalizeAgentStop, chatHistoryManagerRef);
                 if (status === 'interupted') {
                     break;
                 }
@@ -2201,33 +2276,33 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
         if (!aiMessage) {
             return;
         }
-        const updateCellCode = (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_11__.getCellCodeByID)(notebookTracker, updateCellID);
+        const updateCellCode = (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_12__.getCellCodeByID)(notebookTracker, updateCellID);
         if (updateCellID === undefined || updateCellCode === undefined) {
             return;
         }
         // Extract the code from the AI's message and then calculate the code diffs
-        const aiGeneratedCode = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_13__.getCodeBlockFromMessage)(aiMessage);
-        const aiGeneratedCodeCleaned = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_13__.removeMarkdownCodeFormatting)(aiGeneratedCode || '');
-        const { unifiedCodeString, unifiedDiffs } = (0,_utils_codeDiff__WEBPACK_IMPORTED_MODULE_16__.getCodeDiffsAndUnifiedCodeString)(updateCellCode, aiGeneratedCodeCleaned);
+        const aiGeneratedCode = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_14__.getCodeBlockFromMessage)(aiMessage);
+        const aiGeneratedCodeCleaned = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_14__.removeMarkdownCodeFormatting)(aiGeneratedCode || '');
+        const { unifiedCodeString, unifiedDiffs } = (0,_utils_codeDiff__WEBPACK_IMPORTED_MODULE_17__.getCodeDiffsAndUnifiedCodeString)(updateCellCode, aiGeneratedCodeCleaned);
         // Store the code cell ID where we write the code diffs so that we can
         // accept or reject the code diffs to the correct cell
         cellStateBeforeDiff.current = { codeCellID: updateCellID, code: updateCellCode };
         // Temporarily write the unified code string to the active cell so we can display
         // the code diffs to the user
-        (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_11__.writeCodeToCellByID)(notebookTracker, unifiedCodeString, updateCellID);
+        (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_12__.writeCodeToCellByID)(notebookTracker, unifiedCodeString, updateCellID);
         updateCodeCellsExtensions(unifiedDiffs);
         // Briefly highlight the code cell to draw the user's attention to it
-        (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_11__.highlightCodeCell)(notebookTracker, updateCellID);
+        (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_12__.highlightCodeCell)(notebookTracker, updateCellID);
     };
     const displayOptimizedChatHistory = chatHistoryManager.getDisplayOptimizedHistory();
     const previewAICodeToActiveCell = () => {
         setCodeReviewStatus('codeCellPreview');
-        const activeCellID = (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_11__.getActiveCellID)(notebookTracker);
+        const activeCellID = (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_12__.getActiveCellID)(notebookTracker);
         const lastAIDisplayMessage = chatHistoryManagerRef.current.getLastAIDisplayOptimizedChatItem();
         if (activeCellID === undefined || lastAIDisplayMessage === undefined) {
             return;
         }
-        (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_11__.scrollToCell)(notebookTracker, activeCellID, undefined, 'end');
+        (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_12__.scrollToCell)(notebookTracker, activeCellID, undefined, 'end');
         updateCodeDiffStripes(lastAIDisplayMessage.message, activeCellID);
         updateCellToolbarButtons();
     };
@@ -2237,7 +2312,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
         if (!lastAIMessage || !cellStateBeforeDiff.current) {
             return;
         }
-        const aiGeneratedCode = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_13__.getCodeBlockFromMessage)(lastAIMessage.message);
+        const aiGeneratedCode = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_14__.getCodeBlockFromMessage)(lastAIMessage.message);
         if (!aiGeneratedCode) {
             return;
         }
@@ -2246,10 +2321,10 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
         // Write to the cell that has the code diffs
         writeCodeToCellAndTurnOffDiffs(aiGeneratedCode, targetCellID);
         // Focus on the active cell after the code is written
-        const targetCell = (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_11__.getCellByID)(notebookTracker, targetCellID);
+        const targetCell = (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_12__.getCellByID)(notebookTracker, targetCellID);
         if (targetCell) {
             // Make the target cell the active cell
-            (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_11__.setActiveCellByID)(notebookTracker, targetCellID);
+            (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_12__.setActiveCellByID)(notebookTracker, targetCellID);
             // Focus on the active cell
             targetCell.activate();
         }
@@ -2274,7 +2349,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
         updateCodeCellsExtensions(undefined);
         cellStateBeforeDiff.current = undefined;
         if (codeCellID !== undefined) {
-            (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_11__.writeCodeToCellByID)(notebookTracker, code, codeCellID);
+            (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_12__.writeCodeToCellByID)(notebookTracker, code, codeCellID);
             updateCellToolbarButtons();
         }
     };
@@ -2285,17 +2360,17 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
             the first time we create the chat. Registering the command when it is already created causes
             errors.
         */
-        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_17__.COMMAND_MITO_AI_PREVIEW_LATEST_CODE, {
+        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_PREVIEW_LATEST_CODE, {
             execute: () => {
                 previewAICodeToActiveCell();
             }
         });
-        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_17__.COMMAND_MITO_AI_APPLY_LATEST_CODE, {
+        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_APPLY_LATEST_CODE, {
             execute: () => {
                 acceptAICode();
             }
         });
-        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_17__.COMMAND_MITO_AI_REJECT_LATEST_CODE, {
+        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_REJECT_LATEST_CODE, {
             execute: () => {
                 rejectAICode();
             }
@@ -2304,19 +2379,19 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
             Add a new command to the JupyterLab command registry that sends the current chat message.
             We use this to automatically send the message when the user adds an error to the chat.
         */
-        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_17__.COMMAND_MITO_AI_SEND_DEBUG_ERROR_MESSAGE, {
+        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_SEND_DEBUG_ERROR_MESSAGE, {
             execute: async (args) => {
                 if (args === null || args === void 0 ? void 0 : args.input) {
                     await sendSmartDebugMessage(args.input.toString());
                 }
             }
         });
-        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_17__.COMMAND_MITO_AI_SEND_EXPLAIN_CODE_MESSAGE, {
+        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_SEND_EXPLAIN_CODE_MESSAGE, {
             execute: async () => {
                 await sendExplainCodeMessage();
             }
         });
-        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_17__.COMMAND_MITO_AI_SEND_AGENT_MESSAGE, {
+        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_SEND_AGENT_MESSAGE, {
             execute: async (args) => {
                 if (args === null || args === void 0 ? void 0 : args.input) {
                     // Make sure we're in agent mode 
@@ -2335,7 +2410,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
         /*
             Register the code cell toolbar buttons for accepting and rejecting code.
         */
-        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_17__.COMMAND_MITO_AI_CELL_TOOLBAR_ACCEPT_CODE, {
+        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_CELL_TOOLBAR_ACCEPT_CODE, {
             label: `Accept ${operatingSystem === 'mac' ? '⌘Y' : 'Ctrl+Y'}`,
             className: 'text-button-mito-ai button-base button-green',
             caption: 'Accept Code',
@@ -2353,7 +2428,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
                 }
             }
         });
-        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_17__.COMMAND_MITO_AI_CELL_TOOLBAR_REJECT_CODE, {
+        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_CELL_TOOLBAR_REJECT_CODE, {
             label: `Reject ${operatingSystem === 'mac' ? '⌘U' : 'Ctrl+U'}`,
             className: 'text-button-mito-ai button-base button-red',
             caption: 'Reject Code',
@@ -2374,13 +2449,13 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
         // Register keyboard shortcuts 
         const accelYDisposable = app.commands.addKeyBinding({
             command: codeReviewStatus === 'chatPreview' ?
-                _commands__WEBPACK_IMPORTED_MODULE_17__.COMMAND_MITO_AI_PREVIEW_LATEST_CODE :
-                _commands__WEBPACK_IMPORTED_MODULE_17__.COMMAND_MITO_AI_APPLY_LATEST_CODE,
+                _commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_PREVIEW_LATEST_CODE :
+                _commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_APPLY_LATEST_CODE,
             keys: ['Accel Y'],
             selector: 'body',
         });
         const accelDDisposable = app.commands.addKeyBinding({
-            command: _commands__WEBPACK_IMPORTED_MODULE_17__.COMMAND_MITO_AI_REJECT_LATEST_CODE,
+            command: _commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_REJECT_LATEST_CODE,
             keys: ['Accel U'],
             selector: 'body',
             preventDefault: true,
@@ -2397,8 +2472,8 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
         // Without this, the user needs to take some action, like switching to a different cell 
         // and then switching back in order for the Jupyter to re-evaluate if it should
         // show the toolbar buttons.
-        app.commands.notifyCommandChanged(_commands__WEBPACK_IMPORTED_MODULE_17__.COMMAND_MITO_AI_CELL_TOOLBAR_ACCEPT_CODE);
-        app.commands.notifyCommandChanged(_commands__WEBPACK_IMPORTED_MODULE_17__.COMMAND_MITO_AI_CELL_TOOLBAR_REJECT_CODE);
+        app.commands.notifyCommandChanged(_commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_CELL_TOOLBAR_ACCEPT_CODE);
+        app.commands.notifyCommandChanged(_commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_CELL_TOOLBAR_REJECT_CODE);
     };
     // Create a WeakMap to store compartments per code cell
     const codeDiffStripesCompartments = react__WEBPACK_IMPORTED_MODULE_2___default().useRef(new WeakMap());
@@ -2413,6 +2488,8 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
         notebook.widgets.forEach((cell, index) => {
             if (cell.model.type === 'code') {
                 const isActiveCodeCell = activeCellIndex === index;
+                // TODO: Instead of casting, we should rely on the type system to make 
+                // sure we're using the correct types!
                 const codeCell = cell;
                 const cmEditor = codeCell.editor;
                 const editorView = cmEditor === null || cmEditor === void 0 ? void 0 : cmEditor.editor;
@@ -2424,13 +2501,13 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
                         codeDiffStripesCompartments.current.set(codeCell, compartment);
                         // Apply the initial configuration
                         editorView.dispatch({
-                            effects: _codemirror_state__WEBPACK_IMPORTED_MODULE_1__.StateEffect.appendConfig.of(compartment.of(unifiedDiffLines !== undefined && isActiveCodeCell ? (0,_CodeDiffDisplay__WEBPACK_IMPORTED_MODULE_18__.codeDiffStripesExtension)({ unifiedDiffLines: unifiedDiffLines }) : [])),
+                            effects: _codemirror_state__WEBPACK_IMPORTED_MODULE_1__.StateEffect.appendConfig.of(compartment.of(unifiedDiffLines !== undefined && isActiveCodeCell ? (0,_CodeDiffDisplay__WEBPACK_IMPORTED_MODULE_19__.codeDiffStripesExtension)({ unifiedDiffLines: unifiedDiffLines }) : [])),
                         });
                     }
                     else {
                         // Reconfigure the compartment
                         editorView.dispatch({
-                            effects: compartment.reconfigure(unifiedDiffLines !== undefined && isActiveCodeCell ? (0,_CodeDiffDisplay__WEBPACK_IMPORTED_MODULE_18__.codeDiffStripesExtension)({ unifiedDiffLines: unifiedDiffLines }) : []),
+                            effects: compartment.reconfigure(unifiedDiffLines !== undefined && isActiveCodeCell ? (0,_CodeDiffDisplay__WEBPACK_IMPORTED_MODULE_19__.codeDiffStripesExtension)({ unifiedDiffLines: unifiedDiffLines }) : []),
                         });
                     }
                 }
@@ -2444,16 +2521,16 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
     return (react__WEBPACK_IMPORTED_MODULE_2___default().createElement("div", { className: "chat-taskpane" },
         react__WEBPACK_IMPORTED_MODULE_2___default().createElement("div", { className: "chat-taskpane-header" },
             react__WEBPACK_IMPORTED_MODULE_2___default().createElement("div", { className: "chat-taskpane-header-left" },
-                react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_19__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_6__.settingsIcon.react, null), title: "Mito AI Settings", onClick: () => {
-                        void app.commands.execute(_SettingsManager_SettingsManagerPlugin__WEBPACK_IMPORTED_MODULE_20__.COMMAND_MITO_AI_SETTINGS);
+                react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_20__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_6__.settingsIcon.react, null), title: "Mito AI Settings", onClick: () => {
+                        void app.commands.execute(_SettingsManager_SettingsManagerPlugin__WEBPACK_IMPORTED_MODULE_21__.COMMAND_MITO_AI_SETTINGS);
                     } })),
             react__WEBPACK_IMPORTED_MODULE_2___default().createElement("div", { className: "chat-taskpane-header-right" },
-                react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_19__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_6__.addIcon.react, null), title: "Start New Chat", onClick: async () => { await startNewChat(); } }),
-                react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_components_DropdownMenu__WEBPACK_IMPORTED_MODULE_21__["default"], { trigger: react__WEBPACK_IMPORTED_MODULE_2___default().createElement("button", { className: "icon-button", title: "Chat Threads", onClick: fetchChatThreads },
+                react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_20__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_6__.addIcon.react, null), title: "Start New Chat", onClick: async () => { await startNewChat(); } }),
+                react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_components_DropdownMenu__WEBPACK_IMPORTED_MODULE_22__["default"], { trigger: react__WEBPACK_IMPORTED_MODULE_2___default().createElement("button", { className: "icon-button", title: "Chat Threads", onClick: fetchChatThreads },
                         react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_6__.historyIcon.react, null)), items: chatThreads.length > 0
                         ? chatThreads.map(thread => ({
                             label: thread.name,
-                            primaryIcon: activeThreadIdRef.current === thread.thread_id ? _icons__WEBPACK_IMPORTED_MODULE_22__.OpenIndicatorLabIcon.react : undefined,
+                            primaryIcon: activeThreadIdRef.current === thread.thread_id ? _icons__WEBPACK_IMPORTED_MODULE_23__.OpenIndicatorLabIcon.react : undefined,
                             onClick: () => fetchChatHistoryAndSetActiveThread(thread.thread_id),
                             secondaryActions: [
                                 {
@@ -2472,24 +2549,24 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
             displayOptimizedChatHistory.length === 0 &&
                 react__WEBPACK_IMPORTED_MODULE_2___default().createElement("div", { className: "chat-empty-message" },
                     react__WEBPACK_IMPORTED_MODULE_2___default().createElement("div", { style: { margin: '0 auto 8px', display: 'block', textAlign: 'center' } },
-                        react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_icons_MitoLogo__WEBPACK_IMPORTED_MODULE_23__["default"], { width: "60", height: "30" })),
+                        react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_icons_MitoLogo__WEBPACK_IMPORTED_MODULE_24__["default"], { width: "60", height: "30" })),
                     react__WEBPACK_IMPORTED_MODULE_2___default().createElement("span", { style: { display: 'block', textAlign: 'center', fontWeight: 'bold', fontSize: '20px', marginBottom: '15px' } }, "Data Copilot"),
-                    react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_CTACarousel__WEBPACK_IMPORTED_MODULE_24__["default"], { app: app })),
+                    react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_CTACarousel__WEBPACK_IMPORTED_MODULE_25__["default"], { app: app })),
             displayOptimizedChatHistory.map((displayOptimizedChat, index) => {
-                return (react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_ChatMessage_ChatMessage__WEBPACK_IMPORTED_MODULE_25__["default"], { key: index, message: displayOptimizedChat.message, promptType: displayOptimizedChat.promptType, messageType: displayOptimizedChat.type, agentResponse: displayOptimizedChat.agentResponse, codeCellID: displayOptimizedChat.codeCellID, mitoAIConnectionError: displayOptimizedChat.type === 'connection error', mitoAIConnectionErrorType: displayOptimizedChat.mitoAIConnectionErrorType || null, messageIndex: index, notebookTracker: notebookTracker, renderMimeRegistry: renderMimeRegistry, app: app, isLastAiMessage: index === lastAIMessagesIndex, isLastMessage: index === displayOptimizedChatHistory.length - 1, operatingSystem: operatingSystem, previewAICode: previewAICodeToActiveCell, acceptAICode: acceptAICode, rejectAICode: rejectAICode, onUpdateMessage: handleUpdateMessage, contextManager: contextManager, codeReviewStatus: codeReviewStatus, setNextSteps: setNextSteps }));
+                return (react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_ChatMessage_ChatMessage__WEBPACK_IMPORTED_MODULE_26__["default"], { key: index, message: displayOptimizedChat.message, promptType: displayOptimizedChat.promptType, messageType: displayOptimizedChat.type, agentResponse: displayOptimizedChat.agentResponse, codeCellID: displayOptimizedChat.codeCellID, mitoAIConnectionError: displayOptimizedChat.type === 'connection error', mitoAIConnectionErrorType: displayOptimizedChat.mitoAIConnectionErrorType || null, messageIndex: index, notebookTracker: notebookTracker, renderMimeRegistry: renderMimeRegistry, app: app, isLastAiMessage: index === lastAIMessagesIndex, isLastMessage: index === displayOptimizedChatHistory.length - 1, operatingSystem: operatingSystem, previewAICode: previewAICodeToActiveCell, acceptAICode: acceptAICode, rejectAICode: rejectAICode, onUpdateMessage: handleUpdateMessage, contextManager: contextManager, codeReviewStatus: codeReviewStatus, setNextSteps: setNextSteps, agentModeEnabled: agentModeEnabled }));
             }).filter(message => message !== null),
             loadingAIResponse &&
                 react__WEBPACK_IMPORTED_MODULE_2___default().createElement("div", { className: "chat-loading-message" },
                     "Thinking ",
-                    react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_components_LoadingDots__WEBPACK_IMPORTED_MODULE_26__["default"], null)),
+                    react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_components_LoadingDots__WEBPACK_IMPORTED_MODULE_27__["default"], null)),
             hasCheckpoint &&
                 agentModeEnabled &&
                 agentExecutionStatus === 'idle' &&
                 displayOptimizedChatHistory.length > 0 && (react__WEBPACK_IMPORTED_MODULE_2___default().createElement("div", { className: 'message message-assistant-chat' },
-                react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_components_TextAndIconButton__WEBPACK_IMPORTED_MODULE_27__["default"], { text: "Revert changes", icon: _icons_UndoIcon__WEBPACK_IMPORTED_MODULE_28__["default"], title: "Revert changes", onClick: () => (0,_utils_checkpoint__WEBPACK_IMPORTED_MODULE_12__.restoreCheckpoint)(app, notebookTracker, setHasCheckpoint, getDuplicateChatHistoryManager, setChatHistoryManager), variant: "gray", width: "fit-contents", iconPosition: "left" }),
+                react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_components_TextAndIconButton__WEBPACK_IMPORTED_MODULE_28__["default"], { text: "Revert changes", icon: _icons_UndoIcon__WEBPACK_IMPORTED_MODULE_29__["default"], title: "Revert changes", onClick: () => (0,_utils_checkpoint__WEBPACK_IMPORTED_MODULE_13__.restoreCheckpoint)(app, notebookTracker, setHasCheckpoint, getDuplicateChatHistoryManager, setChatHistoryManager), variant: "gray", width: "fit-contents", iconPosition: "left" }),
                 react__WEBPACK_IMPORTED_MODULE_2___default().createElement("p", { className: "text-muted text-sm" }, "Undo the most recent changes made by the agent")))),
         displayOptimizedChatHistory.length === 0 && (react__WEBPACK_IMPORTED_MODULE_2___default().createElement("div", { className: "suggestions-container" },
-            react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_ChatMessage_ScrollableSuggestions__WEBPACK_IMPORTED_MODULE_29__["default"], { onSelectSuggestion: (prompt) => {
+            react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_ChatMessage_ScrollableSuggestions__WEBPACK_IMPORTED_MODULE_30__["default"], { onSelectSuggestion: (prompt) => {
                     if (agentModeEnabled) {
                         void startAgentExecution(prompt);
                     }
@@ -2498,15 +2575,15 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
                     }
                 } }))),
         react__WEBPACK_IMPORTED_MODULE_2___default().createElement("div", { className: `connected-input-container ${nextSteps.length > 0 ? 'has-next-steps' : ''}` },
-            nextSteps.length > 0 && (react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_components_NextStepsPills__WEBPACK_IMPORTED_MODULE_30__["default"], { nextSteps: nextSteps, onSelectNextStep: agentModeEnabled ? startAgentExecution : sendChatInputMessage, displayedNextStepsIfAvailable: displayedNextStepsIfAvailable, setDisplayedNextStepsIfAvailable: setDisplayedNextStepsIfAvailable })),
-            react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_ChatMessage_ChatInput__WEBPACK_IMPORTED_MODULE_31__["default"], { initialContent: '', placeholder: agentExecutionStatus === 'working' ? 'Agent is working...' :
+            nextSteps.length > 0 && (react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_components_NextStepsPills__WEBPACK_IMPORTED_MODULE_31__["default"], { nextSteps: nextSteps, onSelectNextStep: agentModeEnabled ? startAgentExecution : sendChatInputMessage, displayedNextStepsIfAvailable: displayedNextStepsIfAvailable, setDisplayedNextStepsIfAvailable: setDisplayedNextStepsIfAvailable })),
+            react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_ChatMessage_ChatInput__WEBPACK_IMPORTED_MODULE_32__["default"], { initialContent: '', placeholder: agentExecutionStatus === 'working' ? 'Agent is working...' :
                     agentExecutionStatus === 'stopping' ? 'Agent is stopping...' :
                         agentModeEnabled ? 'Ask agent to do anything' :
                             displayOptimizedChatHistory.length < 2 ? `Ask question (${operatingSystem === 'mac' ? '⌘' : 'Ctrl'}E), @ to mention`
                                 : `Ask followup (${operatingSystem === 'mac' ? '⌘' : 'Ctrl'}E), @ to mention`, onSave: agentModeEnabled ? startAgentExecution : sendChatInputMessage, onCancel: undefined, isEditing: false, contextManager: contextManager, notebookTracker: notebookTracker, renderMimeRegistry: renderMimeRegistry, agentModeEnabled: agentModeEnabled })),
         agentExecutionStatus !== 'working' && agentExecutionStatus !== 'stopping' && (react__WEBPACK_IMPORTED_MODULE_2___default().createElement("div", { className: "chat-controls" },
             react__WEBPACK_IMPORTED_MODULE_2___default().createElement("div", { className: "chat-controls-left" },
-                react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_components_ToggleButton__WEBPACK_IMPORTED_MODULE_32__["default"], { leftText: "Chat", leftTooltip: "Chat mode suggests an edit to the active cell and let's you decide to accept or reject it.", rightText: "Agent", rightTooltip: "Agent mode writes and executes code until it's finished your request.", isLeftSelected: !agentModeEnabled, onChange: async (isLeftSelected) => {
+                react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_components_ToggleButton__WEBPACK_IMPORTED_MODULE_33__["default"], { leftText: "Chat", leftTooltip: "Chat mode suggests an edit to the active cell and let's you decide to accept or reject it.", rightText: "Agent", rightTooltip: "Agent mode writes and executes code until it's finished your request.", isLeftSelected: !agentModeEnabled, onChange: async (isLeftSelected) => {
                         await startNewChat(); // TODO: delete thread instead of starting new chat
                         setAgentModeEnabled(!isLeftSelected);
                         // Clear agent checkpoint when switching modes
@@ -2541,7 +2618,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
                 " \u23CE"))),
         (agentExecutionStatus === 'working' || agentExecutionStatus === 'stopping') && (react__WEBPACK_IMPORTED_MODULE_2___default().createElement("button", { className: "button-base button-red stop-agent-button", onClick: markAgentForStopping, disabled: agentExecutionStatus === 'stopping', "data-testid": "stop-agent-button" }, agentExecutionStatus === 'stopping' ? (react__WEBPACK_IMPORTED_MODULE_2___default().createElement("div", { className: "stop-agent-button-content" },
             "Stopping",
-            react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_components_LoadingCircle__WEBPACK_IMPORTED_MODULE_33__["default"], null),
+            react__WEBPACK_IMPORTED_MODULE_2___default().createElement(_components_LoadingCircle__WEBPACK_IMPORTED_MODULE_34__["default"], null),
             " ")) : ('Stop Agent')))));
 };
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (ChatTaskpane);
@@ -6434,6 +6511,31 @@ const AcceptIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createEle
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (AcceptIcon);
 
 
+/***/ }),
+
+/***/ "./lib/icons/CodeIcon.js":
+/*!*******************************!*\
+  !*** ./lib/icons/CodeIcon.js ***!
+  \*******************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+const CodeIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "14", height: "14", viewBox: "0 0 20 14", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
+    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("g", { fill: "currentColor" },
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M12.3162 0.0513449C12.8027 0.213518 13.0817 0.713391 12.9792 1.20347L12.9487 1.31624L8.94868 13.3162C8.77403 13.8402 8.20773 14.1233 7.68378 13.9487C7.19726 13.7865 6.91835 13.2866 7.02082 12.7966L7.05134 12.6838L11.0513 0.683791C11.226 0.159848 11.7923 -0.123303 12.3162 0.0513449ZM5.79999 3.40001C6.10587 3.80785 6.05312 4.37331 5.69568 4.71852L5.59998 4.8L2.667 7.00001L5.59998 9.20003C6.00782 9.50591 6.1155 10.0635 5.86902 10.495L5.79999 10.6C5.49411 11.0079 4.9365 11.1155 4.50501 10.869L4.40003 10.8L0.40003 7.8C-0.0977483 7.42667 -0.130987 6.7049 0.30042 6.28511L0.40003 6.20003L4.40003 3.20003C4.84186 2.86866 5.46862 2.95818 5.79999 3.40001ZM15.495 3.131L15.6 3.20003L19.6 6.20003C20.0978 6.57336 20.131 7.29513 19.6996 7.71492L19.6 7.8L15.6 10.8C15.1582 11.1314 14.5314 11.0418 14.2 10.6C13.8941 10.1922 13.9469 9.62672 14.3043 9.28151L14.4 9.20003L17.333 7.00001L14.4 4.8C13.9922 4.49412 13.8845 3.93654 14.131 3.50505L14.2 3.40001C14.5059 2.99217 15.0635 2.88452 15.495 3.131Z" }))));
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (CodeIcon);
+
+
 /***/ }),
 
 /***/ "./lib/icons/CopyIcon.js":
@@ -8615,6 +8717,70 @@ async function checkForUpdates(serverSettings) {
 }
 
 
+/***/ }),
+
+/***/ "./lib/utils/waitForNotebookReady.js":
+/*!*******************************************!*\
+  !*** ./lib/utils/waitForNotebookReady.js ***!
+  \*******************************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   waitForNotebookReady: () => (/* binding */ waitForNotebookReady)
+/* harmony export */ });
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+/*
+    Wait for the notebook to be ready and attached.
+    This is used to wait for the notebook to be ready before sending the first message.
+*/
+const waitForNotebookReady = async (notebookTracker) => {
+    const notebook = notebookTracker.currentWidget;
+    if (!notebook) {
+        console.warn('No active notebook found');
+        return;
+    }
+    // Wait for notebook to be ready and attached
+    await notebook.context.ready;
+    // Wait for attachment
+    if (!notebook.content.isAttached) {
+        await new Promise(resolve => {
+            const checkAttached = () => {
+                if (notebook.content.isAttached) {
+                    resolve();
+                }
+                else {
+                    setTimeout(checkAttached, 100);
+                }
+            };
+            checkAttached();
+        });
+    }
+    // Wait for all cells to be created and ready
+    await new Promise(resolve => {
+        const checkCellsReady = () => {
+            const cells = notebook.content.widgets;
+            // In large notebooks, not all of the cells are attatched I think. 
+            // So instead we just wait for any cell to be ready and then give it 
+            // another 500ms to be ready.
+            const anyCellReady = cells.some(cell => cell.isAttached && cell.model);
+            if (anyCellReady && cells.length > 0) {
+                resolve();
+            }
+            else {
+                setTimeout(checkCellsReady, 100);
+            }
+        };
+        checkCellsReady();
+    });
+    // Small buffer for final initialization
+    await new Promise(resolve => setTimeout(resolve, 500));
+};
+
+
 /***/ }),
 
 /***/ "./lib/websockets/BaseWebsocketClient.js":
@@ -10066,9 +10232,36 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
   background-color: transparent;
 }
 
+.agent-mode-toggle {
+  cursor: pointer;
+  padding: 10px;
+  color: var(--jp-ui-font-color2);
+  display: flex;
+  justify-content: space-between;
+  align-items: center;
+  border: 1px solid var(--jp-ui-font-color2);
+  border-radius: 5px;
+}
+
+.agent-mode-toggle-content {
+  display: flex;
+  align-items: center;
+  gap: 10px;
+}
+
+.agent-mode-toggle.expanded {
+  border: none;
+  border-bottom: 1px solid var(--grey-300);
+  border-radius: 5px 5px 0 0;
+}
+
+.agent-mode-collapsed {
+  border: none;
+}
+
 
 
-`, "",{"version":3,"sources":["webpack://./style/CodeBlock.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;EACE,aAAa;EACb,sBAAsB;EACtB,kBAAkB;EAClB,cAAc;;EAEd,8CAA8C;EAC9C,kBAAkB;EAClB,iCAAiC;EACjC,gBAAgB;AAClB;;AAEA;EACE,WAAW;EACX,gBAAgB;AAClB;;AAEA;EACE,aAAa;EACb,mBAAmB;EACnB,mBAAmB;EACnB,oBAAoB;;EAEpB,WAAW;EACX,2DAA2D;EAC3D,wCAAwC;EACxC,gBAAgB;AAClB;;AAEA;EACE,YAAY;EACZ,gBAAgB;EAChB,0CAA0C;AAC5C;;AAEA;EACE,2DAA2D;EAC3D,YAAY;EACZ,sCAAsC;EACtC,kBAAkB;;EAElB,gBAAgB;EAChB,0CAA0C;AAC5C;;AAEA;EACE,8CAA8C;EAC9C,+CAA+C;AACjD;;AAEA;EACE,YAAY;AACd;;AAEA;EACE,UAAU;EACV,gBAAgB;AAClB;;AAEA;EACE,aAAa;EACb,uBAAuB;EACvB,mBAAmB;EACnB,cAAc;EACd,WAAW;EACX,8CAA8C;EAC9C,iCAAiC;EACjC,eAAe;AACjB;;AAEA;EACE,qCAAqC;AACvC;;AAEA;EACE,cAAc;EACd,SAAS;AACX;;AAEA;EACE,6BAA6B;AAC/B","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.code-block-container {\n  display: flex;\n  flex-direction: column;\n  position: relative;\n  margin: 10px 0;\n\n  background-color: var(--chat-background-color);\n  border-radius: 4px;\n  border: 1px solid var(--grey-300);\n  overflow: hidden;\n}\n\n.code-content {\n  width: 100%;\n  overflow-x: auto;\n}\n\n.code-block-toolbar {\n  display: flex;\n  flex-direction: row;\n  align-items: center;\n  justify-content: end;\n\n  width: 100%;\n  background-color: var(--chat-user-message-background-color);\n  border-bottom: 1px solid var(--grey-300);\n  font-size: 0.8em;\n}\n\n.code-location {\n  flex-grow: 1;\n  margin-left: 5px;\n  color: var(--chat-user-message-font-color);\n}\n\n.code-block-toolbar button {\n  background-color: var(--chat-user-message-background-color);\n  border: none;\n  border-left: 1px solid var(--grey-300);\n  border-radius: 0px;\n\n  font-size: 0.8em;\n  color: var(--chat-user-message-font-color);\n}\n\n.code-block-toolbar button:hover {\n  background-color: var(--chat-background-color);\n  color: var(--chat-assistant-message-font-color);\n}\n\n.active-cell-code-block {\n  opacity: 0.7;\n}\n\n.active-cell-code-block:hover {\n  opacity: 1;\n  transition: 0.3s;\n}\n\n.code-block-expand-button {\n  display: flex;\n  justify-content: center;\n  align-items: center;\n  padding: 4px 0;\n  width: 100%;\n  background-color: var(--chat-background-color);\n  transition: background-color 0.2s;\n  cursor: pointer;\n}\n\n.code-block-expand-button:hover {\n  background-color: rgba(0, 0, 0, 0.05);\n}\n\n.code-block-expand-button svg {\n  display: block;\n  margin: 0;\n}\n\n.code-block-expand-button .icon-button {\n  background-color: transparent;\n}\n\n\n\n"],"sourceRoot":""}]);
+`, "",{"version":3,"sources":["webpack://./style/CodeBlock.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;EACE,aAAa;EACb,sBAAsB;EACtB,kBAAkB;EAClB,cAAc;;EAEd,8CAA8C;EAC9C,kBAAkB;EAClB,iCAAiC;EACjC,gBAAgB;AAClB;;AAEA;EACE,WAAW;EACX,gBAAgB;AAClB;;AAEA;EACE,aAAa;EACb,mBAAmB;EACnB,mBAAmB;EACnB,oBAAoB;;EAEpB,WAAW;EACX,2DAA2D;EAC3D,wCAAwC;EACxC,gBAAgB;AAClB;;AAEA;EACE,YAAY;EACZ,gBAAgB;EAChB,0CAA0C;AAC5C;;AAEA;EACE,2DAA2D;EAC3D,YAAY;EACZ,sCAAsC;EACtC,kBAAkB;;EAElB,gBAAgB;EAChB,0CAA0C;AAC5C;;AAEA;EACE,8CAA8C;EAC9C,+CAA+C;AACjD;;AAEA;EACE,YAAY;AACd;;AAEA;EACE,UAAU;EACV,gBAAgB;AAClB;;AAEA;EACE,aAAa;EACb,uBAAuB;EACvB,mBAAmB;EACnB,cAAc;EACd,WAAW;EACX,8CAA8C;EAC9C,iCAAiC;EACjC,eAAe;AACjB;;AAEA;EACE,qCAAqC;AACvC;;AAEA;EACE,cAAc;EACd,SAAS;AACX;;AAEA;EACE,6BAA6B;AAC/B;;AAEA;EACE,eAAe;EACf,aAAa;EACb,+BAA+B;EAC/B,aAAa;EACb,8BAA8B;EAC9B,mBAAmB;EACnB,0CAA0C;EAC1C,kBAAkB;AACpB;;AAEA;EACE,aAAa;EACb,mBAAmB;EACnB,SAAS;AACX;;AAEA;EACE,YAAY;EACZ,wCAAwC;EACxC,0BAA0B;AAC5B;;AAEA;EACE,YAAY;AACd","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.code-block-container {\n  display: flex;\n  flex-direction: column;\n  position: relative;\n  margin: 10px 0;\n\n  background-color: var(--chat-background-color);\n  border-radius: 4px;\n  border: 1px solid var(--grey-300);\n  overflow: hidden;\n}\n\n.code-content {\n  width: 100%;\n  overflow-x: auto;\n}\n\n.code-block-toolbar {\n  display: flex;\n  flex-direction: row;\n  align-items: center;\n  justify-content: end;\n\n  width: 100%;\n  background-color: var(--chat-user-message-background-color);\n  border-bottom: 1px solid var(--grey-300);\n  font-size: 0.8em;\n}\n\n.code-location {\n  flex-grow: 1;\n  margin-left: 5px;\n  color: var(--chat-user-message-font-color);\n}\n\n.code-block-toolbar button {\n  background-color: var(--chat-user-message-background-color);\n  border: none;\n  border-left: 1px solid var(--grey-300);\n  border-radius: 0px;\n\n  font-size: 0.8em;\n  color: var(--chat-user-message-font-color);\n}\n\n.code-block-toolbar button:hover {\n  background-color: var(--chat-background-color);\n  color: var(--chat-assistant-message-font-color);\n}\n\n.active-cell-code-block {\n  opacity: 0.7;\n}\n\n.active-cell-code-block:hover {\n  opacity: 1;\n  transition: 0.3s;\n}\n\n.code-block-expand-button {\n  display: flex;\n  justify-content: center;\n  align-items: center;\n  padding: 4px 0;\n  width: 100%;\n  background-color: var(--chat-background-color);\n  transition: background-color 0.2s;\n  cursor: pointer;\n}\n\n.code-block-expand-button:hover {\n  background-color: rgba(0, 0, 0, 0.05);\n}\n\n.code-block-expand-button svg {\n  display: block;\n  margin: 0;\n}\n\n.code-block-expand-button .icon-button {\n  background-color: transparent;\n}\n\n.agent-mode-toggle {\n  cursor: pointer;\n  padding: 10px;\n  color: var(--jp-ui-font-color2);\n  display: flex;\n  justify-content: space-between;\n  align-items: center;\n  border: 1px solid var(--jp-ui-font-color2);\n  border-radius: 5px;\n}\n\n.agent-mode-toggle-content {\n  display: flex;\n  align-items: center;\n  gap: 10px;\n}\n\n.agent-mode-toggle.expanded {\n  border: none;\n  border-bottom: 1px solid var(--grey-300);\n  border-radius: 5px 5px 0 0;\n}\n\n.agent-mode-collapsed {\n  border: none;\n}\n\n\n\n"],"sourceRoot":""}]);
 // Exports
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
 
@@ -13845,4 +14038,4 @@ var update = _node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js
 /***/ })
 
 }]);
-//# sourceMappingURL=lib_index_js.43d516a828f1937a42bb.js.map
+//# sourceMappingURL=lib_index_js.8fc39671fbc9ba62e74b.js.map