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

mito_ai/_version.py

@@ -1,4 +1,4 @@
 # This file is auto-generated by Hatchling. As such, do not:
 #   - modify
 #   - track in version control e.g. be sure to add to .gitignore
-__version__ = VERSION = '0.1.27'
+__version__ = VERSION = '0.1.28'

{mito_ai-0.1.27.data → mito_ai-0.1.28.data}/data/share/jupyter/labextensions/mito_ai/build_log.json

@@ -710,7 +710,7 @@
             "semver": {},
             "vscode-diff": {},
             "mito_ai": {
-              "version": "0.1.27",
+              "version": "0.1.28",
               "singleton": true,
               "import": "/home/runner/work/mito/mito/mito-ai/lib/index.js"
             }

{mito_ai-0.1.27.data → mito_ai-0.1.28.data}/data/share/jupyter/labextensions/mito_ai/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mito_ai",
-  "version": "0.1.27",
+  "version": "0.1.28",
   "description": "AI chat for JupyterLab",
   "keywords": [
     "jupyter",
@@ -138,7 +138,7 @@
     "outputDir": "mito_ai/labextension",
     "schemaDir": "schema",
     "_build": {
-      "load": "static/remoteEntry.806d64f6e8a03c8a76c7.js",
+      "load": "static/remoteEntry.92c6411fdc4075df549b.js",
       "extension": "./extension",
       "style": "./style"
     }

{mito_ai-0.1.27.data → mito_ai-0.1.28.data}/data/share/jupyter/labextensions/mito_ai/schemas/mito_ai/package.json.orig

@@ -1,6 +1,6 @@
 {
   "name": "mito_ai",
-  "version": "0.1.27",
+  "version": "0.1.28",
   "description": "AI chat for JupyterLab",
   "keywords": [
     "jupyter",

mito_ai-0.1.27.data/data/share/jupyter/labextensions/mito_ai/static/lib_index_js.43d516a828f1937a42bb.js → mito_ai-0.1.28.data/data/share/jupyter/labextensions/mito_ai/static/lib_index_js.114d2b34bc18a45df338.js

@@ -1470,33 +1470,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 +1535,7 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+
 
 
 const AGENT_EXECUTION_DEPTH_LIMIT = 20;
@@ -1765,6 +1767,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 +1805,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 +1914,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 +1954,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 +2113,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 +2145,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 +2174,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 +2204,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 +2240,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 +2249,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 +2277,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 +2288,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 +2307,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 +2338,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 +2356,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 +2377,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 +2400,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 +2416,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 +2429,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 +2449,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 +2477,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 }));
             }).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 +2503,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 +2546,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);
@@ -8615,6 +8620,47 @@ 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 */ });
+/*
+    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 a bit more for cells to be fully initialized
+    if (!notebook.content.isAttached) {
+        await new Promise(resolve => {
+            const checkAttached = () => {
+                if (notebook.content.isAttached) {
+                    resolve();
+                }
+                else {
+                    setTimeout(checkAttached, 100);
+                }
+            };
+            checkAttached();
+        });
+    }
+};
+
+
 /***/ }),
 
 /***/ "./lib/websockets/BaseWebsocketClient.js":
@@ -13845,4 +13891,4 @@ var update = _node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js
 /***/ })
 
 }]);
-//# sourceMappingURL=lib_index_js.43d516a828f1937a42bb.js.map
+//# sourceMappingURL=lib_index_js.114d2b34bc18a45df338.js.map