mito-ai 0.1.55__py3-none-any.whl → 0.1.56__py3-none-any.whl

mito_ai-0.1.55.data/data/share/jupyter/labextensions/mito_ai/static/lib_index_js.49c79c62671528877c61.js → mito_ai-0.1.56.data/data/share/jupyter/labextensions/mito_ai/static/lib_index_js.dfd7975de75d64db80d6.js

@@ -173,7 +173,7 @@ const AiChatPlugin = {
         // then call it to make a new widget
         const newWidget = () => {
             // Create a blank content widget inside of a MainAreaWidget
-            const chatWidget = (0,_ChatWidget__WEBPACK_IMPORTED_MODULE_9__.buildChatWidget)(app, notebookTracker, rendermime, contextManager, streamlitPreviewManager);
+            const chatWidget = (0,_ChatWidget__WEBPACK_IMPORTED_MODULE_9__.buildChatWidget)(app, notebookTracker, rendermime, contextManager, streamlitPreviewManager, documentManager);
             return chatWidget;
         };
         let widget = newWidget();
@@ -1327,17 +1327,19 @@ __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 _utils_classNames__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ../../../utils/classNames */ "./lib/utils/classNames.js");
-/* harmony import */ var _ChatDropdown__WEBPACK_IMPORTED_MODULE_11__ = __webpack_require__(/*! ./ChatDropdown */ "./lib/Extensions/AiChat/ChatMessage/ChatDropdown.js");
+/* harmony import */ var _utils_classNames__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ../../../utils/classNames */ "./lib/utils/classNames.js");
+/* harmony import */ var _ChatDropdown__WEBPACK_IMPORTED_MODULE_13__ = __webpack_require__(/*! ./ChatDropdown */ "./lib/Extensions/AiChat/ChatMessage/ChatDropdown.js");
 /* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../../utils/notebook */ "./lib/utils/notebook.js");
-/* harmony import */ var _utils_cellReferences__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../../../utils/cellReferences */ "./lib/utils/cellReferences.js");
+/* harmony import */ var _utils_cellReferences__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ../../../utils/cellReferences */ "./lib/utils/cellReferences.js");
 /* harmony import */ var _style_ChatInput_css__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../../../style/ChatInput.css */ "./style/ChatInput.css");
 /* harmony import */ var _style_ChatDropdown_css__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../../../../style/ChatDropdown.css */ "./style/ChatDropdown.css");
-/* harmony import */ var _hooks_useDebouncedFunction__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../../hooks/useDebouncedFunction */ "./lib/hooks/useDebouncedFunction.js");
-/* harmony import */ var _components_SelectedContextContainer__WEBPACK_IMPORTED_MODULE_10__ = __webpack_require__(/*! ../../../components/SelectedContextContainer */ "./lib/components/SelectedContextContainer.js");
-/* harmony import */ var _components_AttachFileButton__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ../../../components/AttachFileButton */ "./lib/components/AttachFileButton.js");
-/* harmony import */ var _components_DatabaseButton__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ../../../components/DatabaseButton */ "./lib/components/DatabaseButton.js");
-/* harmony import */ var _utils_fileUpload__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../../utils/fileUpload */ "./lib/utils/fileUpload.js");
+/* harmony import */ var _hooks_useDebouncedFunction__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ../../../hooks/useDebouncedFunction */ "./lib/hooks/useDebouncedFunction.js");
+/* harmony import */ var _hooks_useCellOrder__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../../hooks/useCellOrder */ "./lib/hooks/useCellOrder.js");
+/* harmony import */ var _hooks_useLineSelection__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../../hooks/useLineSelection */ "./lib/hooks/useLineSelection.js");
+/* harmony import */ var _components_SelectedContextContainer__WEBPACK_IMPORTED_MODULE_12__ = __webpack_require__(/*! ../../../components/SelectedContextContainer */ "./lib/components/SelectedContextContainer.js");
+/* harmony import */ var _components_AttachFileButton__WEBPACK_IMPORTED_MODULE_11__ = __webpack_require__(/*! ../../../components/AttachFileButton */ "./lib/components/AttachFileButton.js");
+/* harmony import */ var _components_DatabaseButton__WEBPACK_IMPORTED_MODULE_10__ = __webpack_require__(/*! ../../../components/DatabaseButton */ "./lib/components/DatabaseButton.js");
+/* harmony import */ var _utils_fileUpload__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../../../utils/fileUpload */ "./lib/utils/fileUpload.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -1354,6 +1356,8 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+
+
 const ChatInput = ({ app, initialContent, handleSubmitUserMessage, onCancel, isEditing, contextManager, notebookTracker, agentModeEnabled = false, agentExecutionStatus = 'idle', operatingSystem = 'mac', displayOptimizedChatHistoryLength = 0, agentTargetNotebookPanelRef, isSignedUp = true, messageIndex, }) => {
     var _a;
     const [input, setInput] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(initialContent);
@@ -1366,6 +1370,9 @@ const ChatInput = ({ app, initialContent, handleSubmitUserMessage, onCancel, isE
     const [isDropdownFromButton, setIsDropdownFromButton] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
     const [isDragOver, setIsDragOver] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
     const [isUploading, setIsUploading] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
+    // Track cell order and line selection
+    const cellOrder = (0,_hooks_useCellOrder__WEBPACK_IMPORTED_MODULE_4__.useCellOrder)(notebookTracker);
+    const lineSelection = (0,_hooks_useLineSelection__WEBPACK_IMPORTED_MODULE_5__.useLineSelection)(notebookTracker, cellOrder);
     const handleFileUpload = (file) => {
         let uploadType;
         if (file.type.startsWith('image/')) {
@@ -1409,7 +1416,7 @@ const ChatInput = ({ app, initialContent, handleSubmitUserMessage, onCancel, isE
             setIsUploading(true);
             try {
                 // Upload all files to backend using the shared utility
-                const uploadPromises = Array.from(files).map(file => (0,_utils_fileUpload__WEBPACK_IMPORTED_MODULE_4__.uploadFileToBackend)(file, notebookTracker, handleFileUpload));
+                const uploadPromises = Array.from(files).map(file => (0,_utils_fileUpload__WEBPACK_IMPORTED_MODULE_6__.uploadFileToBackend)(file, notebookTracker, handleFileUpload));
                 // Use allSettled so we wait for all uploads to finish (success or failure)
                 // before clearing the uploading state. This avoids background uploads
                 // mutating state after isUploading becomes false.
@@ -1420,9 +1427,41 @@ const ChatInput = ({ app, initialContent, handleSubmitUserMessage, onCancel, isE
             }
         }
     };
+    const handlePaste = async (e) => {
+        const items = e.clipboardData.items;
+        const clipboardItem = items === null || items === void 0 ? void 0 : items[items.length - 1];
+        if (!clipboardItem)
+            return;
+        // Check if it's a file (images are also files)
+        if (clipboardItem.kind === 'file') {
+            e.preventDefault();
+            const blob = clipboardItem.getAsFile();
+            if (!blob)
+                return;
+            // Determine file name - use blob name if available, otherwise generate one
+            let fileName = blob.name;
+            if (!fileName) {
+                // Generate a filename based on the MIME type
+                const extension = clipboardItem.type.startsWith('image/')
+                    ? clipboardItem.type.split('/')[1]
+                    : 'file';
+                fileName = `pasted-${Date.now()}.${extension}`;
+            }
+            const file = new File([blob], fileName, { type: clipboardItem.type });
+            if (isUploading)
+                return; // Prevent multiple simultaneous uploads
+            setIsUploading(true);
+            try {
+                await (0,_utils_fileUpload__WEBPACK_IMPORTED_MODULE_6__.uploadFileToBackend)(file, notebookTracker, handleFileUpload);
+            }
+            finally {
+                setIsUploading(false);
+            }
+        }
+    };
     // Debounce the active cell ID change to avoid multiple rerenders. 
     // We use this to avoid a flickering screen when the active cell changes. 
-    const debouncedSetActiveCellID = (0,_hooks_useDebouncedFunction__WEBPACK_IMPORTED_MODULE_5__.useDebouncedFunction)((newID) => {
+    const debouncedSetActiveCellID = (0,_hooks_useDebouncedFunction__WEBPACK_IMPORTED_MODULE_7__.useDebouncedFunction)((newID) => {
         setActiveCellID(newID);
     }, 100);
     (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
@@ -1592,7 +1631,7 @@ const ChatInput = ({ app, initialContent, handleSubmitUserMessage, onCancel, isE
     };
     // Convert @Cell N references to [MITO_CELL_REF:cell_id] format before submitting
     const processMessageForSubmission = (messageText) => {
-        return (0,_utils_cellReferences__WEBPACK_IMPORTED_MODULE_6__.convertCellReferencesToStableFormat)(messageText, notebookTracker.currentWidget);
+        return (0,_utils_cellReferences__WEBPACK_IMPORTED_MODULE_8__.convertCellReferencesToStableFormat)(messageText, notebookTracker.currentWidget);
     };
     const getExpandedVarialbes = () => {
         const activeNotebookContext = contextManager === null || contextManager === void 0 ? void 0 : contextManager.getActiveNotebookContext();
@@ -1680,13 +1719,58 @@ const ChatInput = ({ app, initialContent, handleSubmitUserMessage, onCancel, isE
             }
         }
     }, [agentModeEnabled, additionalContext, activeCellCode]);
-    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_7__.classNames)("chat-input-container", {
+    // Manage line selection context - shows selected lines when user has text selected in a code cell
+    (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
+        const LINE_SELECTION_TYPE = 'line_selection';
+        // Build the new value data for comparison (lines are 0-indexed internally)
+        const newValueData = lineSelection.hasSelection
+            ? JSON.stringify({
+                cellId: lineSelection.cellId,
+                startLine: lineSelection.startLine,
+                endLine: lineSelection.endLine,
+                selectedCode: lineSelection.selectedCode
+            })
+            : null;
+        // Treat null and undefined as the same "no selection" state
+        const normalizedNewValue = newValueData !== null && newValueData !== void 0 ? newValueData : null;
+        setAdditionalContext(prev => {
+            var _a;
+            // Find existing line selection context in the latest state (avoid stale render closures)
+            const existingContext = prev.find(context => context.type === LINE_SELECTION_TYPE);
+            const normalizedExistingValue = (_a = existingContext === null || existingContext === void 0 ? void 0 : existingContext.value) !== null && _a !== void 0 ? _a : null;
+            // Only update if the selection has changed (null and undefined are equivalent)
+            if (normalizedNewValue === normalizedExistingValue) {
+                return prev;
+            }
+            // Remove old line selection context if it exists
+            let additionalContextCopy = prev.filter(context => context.type !== LINE_SELECTION_TYPE);
+            // Add new line selection context if there is text selected
+            if (normalizedNewValue) {
+                // Build display text with 1-indexed lines for user display
+                const displayStartLine = lineSelection.startLine + 1;
+                const displayEndLine = lineSelection.endLine + 1;
+                const displayText = displayStartLine === displayEndLine
+                    ? `Cell ${lineSelection.cellNumber} line ${displayStartLine}`
+                    : `Cell ${lineSelection.cellNumber} line ${displayStartLine}-${displayEndLine}`;
+                additionalContextCopy = [
+                    ...additionalContextCopy,
+                    {
+                        type: LINE_SELECTION_TYPE,
+                        value: normalizedNewValue,
+                        display: displayText
+                    }
+                ];
+            }
+            return additionalContextCopy;
+        });
+    }, [lineSelection]);
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_9__.classNames)("chat-input-container", {
             "editing": isEditing,
             "drag-over": isDragOver
         }), onDragOver: handleDragOver, onDragLeave: handleDragLeave, onDrop: handleDrop },
         react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'context-container' },
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_DatabaseButton__WEBPACK_IMPORTED_MODULE_8__["default"], { app: app }),
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AttachFileButton__WEBPACK_IMPORTED_MODULE_9__["default"], { onFileUploaded: handleFileUpload, notebookTracker: notebookTracker }),
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_DatabaseButton__WEBPACK_IMPORTED_MODULE_10__["default"], { app: app }),
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AttachFileButton__WEBPACK_IMPORTED_MODULE_11__["default"], { onFileUploaded: handleFileUpload, notebookTracker: notebookTracker }),
             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { className: "context-button", onClick: () => {
                     var _a;
                     setDropdownVisible(true);
@@ -1694,9 +1778,9 @@ const ChatInput = ({ app, initialContent, handleSubmitUserMessage, onCancel, isE
                     setIsDropdownFromButton(true);
                     (_a = textAreaRef.current) === null || _a === void 0 ? void 0 : _a.focus();
                 } }, "\uFF20 Add Context"),
-            additionalContext.map((context, index) => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_SelectedContextContainer__WEBPACK_IMPORTED_MODULE_10__["default"], { key: `${context.type}-${context.value}-${index}`, title: context.display || context.value, type: context.type, onRemove: () => setAdditionalContext(additionalContext.filter((_, i) => i !== index)), notebookTracker: notebookTracker, activeCellID: activeCellID, value: context.value })))),
+            additionalContext.map((context, index) => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_SelectedContextContainer__WEBPACK_IMPORTED_MODULE_12__["default"], { key: `${context.type}-${context.value}-${index}`, title: context.display || context.value, type: context.type, onRemove: () => setAdditionalContext(additionalContext.filter((_, i) => i !== index)), notebookTracker: notebookTracker, activeCellID: activeCellID, value: context.value })))),
         react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'chat-input-text-area-container' },
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("textarea", { ref: textAreaRef, className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_7__.classNames)("message", "message-user", 'chat-input', { "agent-mode": agentModeEnabled }), placeholder: getPlaceholderText(), value: input, disabled: agentExecutionStatus === 'working' || agentExecutionStatus === 'stopping', onChange: handleInputChange, onKeyDown: (e) => {
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("textarea", { ref: textAreaRef, className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_9__.classNames)("message", "message-user", 'chat-input', { "agent-mode": agentModeEnabled }), placeholder: getPlaceholderText(), value: input, disabled: agentExecutionStatus === 'working' || agentExecutionStatus === 'stopping', onChange: handleInputChange, onPaste: handlePaste, onKeyDown: (e) => {
                     // If dropdown is visible, only handle escape to close it
                     if (isDropdownVisible) {
                         if (e.key === 'Escape') {
@@ -1725,7 +1809,7 @@ const ChatInput = ({ app, initialContent, handleSubmitUserMessage, onCancel, isE
                         }
                     }
                 } }),
-            isDropdownVisible && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatDropdown__WEBPACK_IMPORTED_MODULE_11__["default"], { options: getExpandedVarialbes(), onSelect: handleOptionSelect, filterText: dropdownFilter, isDropdownFromButton: isDropdownFromButton, onFilterChange: setDropdownFilter, onClose: handleDropdownClose, notebookTracker: notebookTracker }))),
+            isDropdownVisible && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatDropdown__WEBPACK_IMPORTED_MODULE_13__["default"], { options: getExpandedVarialbes(), onSelect: handleOptionSelect, filterText: dropdownFilter, isDropdownFromButton: isDropdownFromButton, onFilterChange: setDropdownFilter, onClose: handleDropdownClose, notebookTracker: notebookTracker }))),
         isEditing &&
             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "message-edit-buttons" },
                 react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { onClick: () => {
@@ -1752,14 +1836,14 @@ __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 _utils_classNames__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../../../utils/classNames */ "./lib/utils/classNames.js");
-/* harmony import */ var _UserCodeBlock__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ./UserCodeBlock */ "./lib/Extensions/AiChat/ChatMessage/UserCodeBlock.js");
-/* harmony import */ var _AssistantCodeBlock__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ./AssistantCodeBlock */ "./lib/Extensions/AiChat/ChatMessage/AssistantCodeBlock.js");
-/* harmony import */ var _AlertBlock__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ./AlertBlock */ "./lib/Extensions/AiChat/ChatMessage/AlertBlock.js");
+/* harmony import */ var _utils_classNames__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../../utils/classNames */ "./lib/utils/classNames.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_16__ = __webpack_require__(/*! ./MarkdownBlock */ "./lib/Extensions/AiChat/ChatMessage/MarkdownBlock.js");
-/* harmony import */ var _utils_strings__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../../utils/strings */ "./lib/utils/strings.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_18__ = __webpack_require__(/*! ../../../icons/Pencil */ "./lib/icons/Pencil.js");
-/* harmony import */ var _ChatInput__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ./ChatInput */ "./lib/Extensions/AiChat/ChatMessage/ChatInput.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_10__ = __webpack_require__(/*! ../../../components/TextAndIconButton */ "./lib/components/TextAndIconButton.js");
 /* harmony import */ var _icons_PlayButtonIcon__WEBPACK_IMPORTED_MODULE_11__ = __webpack_require__(/*! ../../../icons/PlayButtonIcon */ "./lib/icons/PlayButtonIcon.js");
 /* harmony import */ var _icons_CopyIcon__WEBPACK_IMPORTED_MODULE_13__ = __webpack_require__(/*! ../../../icons/CopyIcon */ "./lib/icons/CopyIcon.js");
@@ -1767,7 +1851,7 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony import */ var _components_TextButton__WEBPACK_IMPORTED_MODULE_14__ = __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 _utils_notebook__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../../utils/notebook */ "./lib/utils/notebook.js");
+/* harmony import */ var _cellUpdateUtils__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ../cellUpdateUtils */ "./lib/Extensions/AiChat/cellUpdateUtils.js");
 /* harmony import */ var _components_AgentComponents_GetCellOutputToolUI__WEBPACK_IMPORTED_MODULE_19__ = __webpack_require__(/*! ../../../components/AgentComponents/GetCellOutputToolUI */ "./lib/components/AgentComponents/GetCellOutputToolUI.js");
 /* harmony import */ var _components_AgentComponents_AssumptionToolUI__WEBPACK_IMPORTED_MODULE_17__ = __webpack_require__(/*! ../../../components/AgentComponents/AssumptionToolUI */ "./lib/components/AgentComponents/AssumptionToolUI.js");
 /* harmony import */ var _components_SelectedContextContainer__WEBPACK_IMPORTED_MODULE_15__ = __webpack_require__(/*! ../../../components/SelectedContextContainer */ "./lib/components/SelectedContextContainer.js");
@@ -1801,24 +1885,14 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
-const getCellIdFromCellUpdate = (cellUpdate, notebookTracker) => {
-    if (!cellUpdate) {
-        return undefined;
-    }
-    if (cellUpdate.type === 'modification') {
-        return cellUpdate.id;
-    }
-    // For 'new' type, get the cell ID by index
-    return (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_3__.getCellIDByIndexInNotebookPanel)(notebookTracker.currentWidget, cellUpdate.index);
-};
 const ChatMessage = ({ app, message, promptType, agentResponse, messageIndex, mitoAIConnectionError, mitoAIConnectionErrorType, notebookTracker, renderMimeRegistry, isLastAiMessage, isLastMessage, operatingSystem, previewAICode, acceptAICode, rejectAICode, contextManager, codeReviewStatus, setNextSteps, agentModeEnabled, additionalContext, handleSubmitUserMessage, }) => {
     const [isEditing, setIsEditing] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
     if (message.role !== 'user' && message.role !== 'assistant') {
         return null;
     }
     const editable = message.role === 'user';
-    const messageContentParts = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_4__.splitStringWithCodeBlocks)(message);
-    const messageContent = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_4__.getContentStringFromMessage)(message);
+    const messageContentParts = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_3__.splitStringWithCodeBlocks)(message);
+    const messageContent = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_3__.getContentStringFromMessage)(message);
     const handleEditClick = () => {
         setIsEditing(true);
     };
@@ -1841,11 +1915,11 @@ const ChatMessage = ({ app, message, promptType, agentResponse, messageIndex, mi
         setNextSteps(agentResponse.next_steps);
     }
     if (isEditing) {
-        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatInput__WEBPACK_IMPORTED_MODULE_5__["default"], { app: app, initialContent: message.content.replace(/```[\s\S]*?```/g, '').trim(), onCancel: handleCancel, isEditing: isEditing, contextManager: contextManager, notebookTracker: notebookTracker, agentModeEnabled: false, handleSubmitUserMessage: handleSubmitUserMessageAndCloseEditing, messageIndex: messageIndex }));
+        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatInput__WEBPACK_IMPORTED_MODULE_4__["default"], { app: app, initialContent: message.content.replace(/```[\s\S]*?```/g, '').trim(), onCancel: handleCancel, isEditing: isEditing, contextManager: contextManager, notebookTracker: notebookTracker, agentModeEnabled: false, handleSubmitUserMessage: handleSubmitUserMessageAndCloseEditing, messageIndex: messageIndex }));
     }
     if (mitoAIConnectionError) {
-        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_6__.classNames)("message") },
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_AlertBlock__WEBPACK_IMPORTED_MODULE_7__["default"], { content: message.content, mitoAIConnectionErrorType: mitoAIConnectionErrorType })));
+        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_5__.classNames)("message") },
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_AlertBlock__WEBPACK_IMPORTED_MODULE_6__["default"], { content: message.content, mitoAIConnectionErrorType: mitoAIConnectionErrorType })));
     }
     // If the message is empty, don't render anything
     if (messageContent === undefined || messageContent === '') {
@@ -1855,16 +1929,16 @@ const ChatMessage = ({ app, message, promptType, agentResponse, messageIndex, mi
     // Users end up applying the code in the middle of streaming and it gets very confusing
     // very quickly for users. 
     let isCodeComplete = false;
-    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_6__.classNames)("message", { "message-user": message.role === 'user' }, { 'message-assistant-chat': message.role === 'assistant' }) },
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_5__.classNames)("message", { "message-user": message.role === 'user' }, { 'message-assistant-chat': message.role === 'assistant' }) },
         messageContentParts.map((messagePart, index) => {
             var _a, _b;
-            if (messagePart.startsWith(_utils_strings__WEBPACK_IMPORTED_MODULE_4__.PYTHON_CODE_BLOCK_START_WITHOUT_NEW_LINE)) {
+            if (messagePart.startsWith(_utils_strings__WEBPACK_IMPORTED_MODULE_3__.PYTHON_CODE_BLOCK_START_WITHOUT_NEW_LINE)) {
                 isCodeComplete = messagePart.endsWith('```');
                 // Make sure that there is actually code in the message. 
                 // 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,
-                        message.role === 'user' ? (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_UserCodeBlock__WEBPACK_IMPORTED_MODULE_8__["default"], { code: messagePart, renderMimeRegistry: renderMimeRegistry, agentModeEnabled: agentModeEnabled })) : (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_AssistantCodeBlock__WEBPACK_IMPORTED_MODULE_9__["default"], { code: messagePart, codeSummary: (_b = (_a = agentResponse === null || agentResponse === void 0 ? void 0 : agentResponse.cell_update) === null || _a === void 0 ? void 0 : _a.code_summary) !== null && _b !== void 0 ? _b : ((agentResponse === null || agentResponse === void 0 ? void 0 : agentResponse.type) === 'run_all_cells' ? 'Running all cells' : undefined), isCodeComplete: isCodeComplete, renderMimeRegistry: renderMimeRegistry, previewAICode: previewAICode, acceptAICode: acceptAICode, rejectAICode: rejectAICode, isLastAiMessage: isLastAiMessage, codeReviewStatus: codeReviewStatus, agentModeEnabled: agentModeEnabled, cellId: getCellIdFromCellUpdate(agentResponse === null || agentResponse === void 0 ? void 0 : agentResponse.cell_update, notebookTracker), notebookPanel: notebookTracker.currentWidget })),
+                        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, codeSummary: (_b = (_a = agentResponse === null || agentResponse === void 0 ? void 0 : agentResponse.cell_update) === null || _a === void 0 ? void 0 : _a.code_summary) !== null && _b !== void 0 ? _b : ((agentResponse === null || agentResponse === void 0 ? void 0 : agentResponse.type) === 'run_all_cells' ? 'Running all cells' : undefined), isCodeComplete: isCodeComplete, renderMimeRegistry: renderMimeRegistry, previewAICode: previewAICode, acceptAICode: acceptAICode, rejectAICode: rejectAICode, isLastAiMessage: isLastAiMessage, codeReviewStatus: codeReviewStatus, agentModeEnabled: agentModeEnabled, cellId: (0,_cellUpdateUtils__WEBPACK_IMPORTED_MODULE_9__.getCellIdFromCellUpdate)(agentResponse === null || agentResponse === void 0 ? void 0 : agentResponse.cell_update, notebookTracker), notebookPanel: notebookTracker.currentWidget })),
                         !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_10__["default"], { onClick: () => { previewAICode(); }, text: 'Overwrite Active Cell', icon: _icons_PlayButtonIcon__WEBPACK_IMPORTED_MODULE_11__["default"], title: 'Write the Ai generated code to the active cell in the jupyter notebook, replacing the current code.', variant: 'gray', width: 'fit-contents' }),
@@ -1884,7 +1958,7 @@ const ChatMessage = ({ app, message, promptType, agentResponse, messageIndex, mi
                 }
             }
             else {
-                return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { key: index + messagePart, className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_6__.classNames)('markdown-message-part') },
+                return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { key: index + messagePart, className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_5__.classNames)('markdown-message-part') },
                     react__WEBPACK_IMPORTED_MODULE_0___default().createElement("p", { onDoubleClick: () => {
                             // Only allow users to edit their own messages, not the AI responses
                             if (message.role === 'user') {
@@ -2526,8 +2600,8 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony import */ var _utils_chatHistory__WEBPACK_IMPORTED_MODULE_27__ = __webpack_require__(/*! ../../utils/chatHistory */ "./lib/utils/chatHistory.js");
 /* harmony import */ var _utils_codeDiff__WEBPACK_IMPORTED_MODULE_26__ = __webpack_require__(/*! ../../utils/codeDiff */ "./lib/utils/codeDiff.js");
 /* harmony import */ var _utils_cellOutput__WEBPACK_IMPORTED_MODULE_18__ = __webpack_require__(/*! ../../utils/cellOutput */ "./lib/utils/cellOutput.js");
-/* harmony import */ var _utils_waitForNotebookReady__WEBPACK_IMPORTED_MODULE_23__ = __webpack_require__(/*! ../../utils/waitForNotebookReady */ "./lib/utils/waitForNotebookReady.js");
 /* harmony import */ var _utils__WEBPACK_IMPORTED_MODULE_17__ = __webpack_require__(/*! ./utils */ "./lib/Extensions/AiChat/utils.js");
+/* harmony import */ var _utils_waitForNotebookReady__WEBPACK_IMPORTED_MODULE_23__ = __webpack_require__(/*! ../../utils/waitForNotebookReady */ "./lib/utils/waitForNotebookReady.js");
 /* harmony import */ var _restAPI_RestAPI__WEBPACK_IMPORTED_MODULE_24__ = __webpack_require__(/*! ../../restAPI/RestAPI */ "./lib/restAPI/RestAPI.js");
 /* harmony import */ var _utils_sound__WEBPACK_IMPORTED_MODULE_20__ = __webpack_require__(/*! ../../utils/sound */ "./lib/utils/sound.js");
 /* harmony import */ var _SettingsManager_SettingsManagerPlugin__WEBPACK_IMPORTED_MODULE_30__ = __webpack_require__(/*! ../SettingsManager/SettingsManagerPlugin */ "./lib/Extensions/SettingsManager/SettingsManagerPlugin.js");
@@ -2586,6 +2660,7 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+
 // Internal imports - Chat components
 
 
@@ -2617,7 +2692,7 @@ const getDefaultChatHistoryManager = (notebookTracker, contextManager, app) => {
     const chatHistoryManager = new _ChatHistoryManager__WEBPACK_IMPORTED_MODULE_6__.ChatHistoryManager(contextManager, notebookTracker, app);
     return chatHistoryManager;
 };
-const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, streamlitPreviewManager, app, operatingSystem, websocketClient, }) => {
+const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, streamlitPreviewManager, app, operatingSystem, websocketClient, documentManager, }) => {
     // User signup state
     const { isSignedUp, refreshUserSignupState } = (0,_hooks_useUserSignup__WEBPACK_IMPORTED_MODULE_7__.useUserSignup)();
     // Core chat state management
@@ -2779,21 +2854,17 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
     const sendAgentExecutionMessage = async (input, messageIndex, sendCellIDOutput = undefined, additionalContext) => {
         // Step 0: reset the state for a new message
         resetForNewMessage();
-        const agentTargetNotebookPanel = agentTargetNotebookPanelRef.current;
-        if (agentTargetNotebookPanel === null) {
-            return;
-        }
         // Step 1: Add the user's message to the chat history
         const newChatHistoryManager = getDuplicateChatHistoryManager();
         if (messageIndex !== undefined) {
             // Drop all of the messages starting at the message index
             newChatHistoryManager.dropMessagesStartingAtIndex(messageIndex);
         }
-        const agentExecutionMetadata = newChatHistoryManager.addAgentExecutionMessage(activeThreadIdRef.current, agentTargetNotebookPanel, input, additionalContext);
+        const agentExecutionMetadata = newChatHistoryManager.addAgentExecutionMessage(activeThreadIdRef.current, agentTargetNotebookPanelRef.current, input, additionalContext);
         if (messageIndex !== undefined) {
             agentExecutionMetadata.index = messageIndex;
         }
-        agentExecutionMetadata.base64EncodedActiveCellOutput = await (0,_utils__WEBPACK_IMPORTED_MODULE_17__.getBase64EncodedCellOutputInNotebook)(agentTargetNotebookPanel, sendCellIDOutput);
+        agentExecutionMetadata.base64EncodedActiveCellOutput = await (0,_utils__WEBPACK_IMPORTED_MODULE_17__.getBase64EncodedCellOutputInNotebook)(agentTargetNotebookPanelRef.current, sendCellIDOutput);
         setChatHistoryManager(newChatHistoryManager);
         setLoadingStatus('thinking');
         // Step 2: Send the message to the AI
@@ -2811,6 +2882,8 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
     const sendChatInputMessage = async (input, messageIndex, additionalContext) => {
         // Step 0: reset the state for a new message
         resetForNewMessage();
+        // Ensure a notebook exists before proceeding
+        await (0,_utils__WEBPACK_IMPORTED_MODULE_17__.ensureNotebookExists)(notebookTracker, documentManager);
         // Enable follow mode when user sends a new message
         setAutoScrollFollowMode(true);
         // Step 1: Add the user's message to the chat history
@@ -3004,6 +3077,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
         app,
         streamlitPreviewManager,
         websocketClient,
+        documentManager,
         chatHistoryManagerRef,
         activeThreadIdRef,
         activeRequestControllerRef,
@@ -3146,6 +3220,24 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
             execute: rejectAICode,
             isVisible: () => (0,_utils_codeDiff__WEBPACK_IMPORTED_MODULE_26__.shouldShowDiffToolbarButtons)(notebookTracker, cellStateBeforeDiff.current, agentReview.changedCellsRef.current)
         });
+        /*
+            Register global command for starting a new chat.
+            This can be triggered from anywhere with CMD+K (Mac) or Ctrl+K (Windows/Linux).
+        */
+        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_13__.COMMAND_MITO_AI_START_NEW_CHAT, {
+            label: 'Start New Chat',
+            execute: async () => {
+                await startNewChat();
+                // Focus the chat input after starting new chat
+                const chatInput = document.querySelector('.chat-input');
+                chatInput === null || chatInput === void 0 ? void 0 : chatInput.focus();
+            }
+        });
+        app.commands.addKeyBinding({
+            command: _commands__WEBPACK_IMPORTED_MODULE_13__.COMMAND_MITO_AI_START_NEW_CHAT,
+            keys: ['Accel K'],
+            selector: 'body',
+        });
     }, []);
     (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
         // Register keyboard shortcuts 
@@ -3217,7 +3309,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, str
                     } }),
                 react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_UsageBadge__WEBPACK_IMPORTED_MODULE_31__["default"], { app: app, ref: usageBadgeRef })),
             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "chat-taskpane-header-right" },
-                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_29__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_1__.addIcon.react, null), title: "Start New Chat", onClick: async () => { await startNewChat(); } }),
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_29__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_1__.addIcon.react, null), title: `Start New Chat (${operatingSystem === 'mac' ? '⌘' : 'Ctrl'}K)`, onClick: async () => { await startNewChat(); } }),
                 react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_DropdownMenu__WEBPACK_IMPORTED_MODULE_32__["default"], { trigger: react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { className: "icon-button", title: "Chat Threads", onClick: fetchChatThreads },
                         react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_1__.historyIcon.react, null)), items: chatThreads.length > 0
                         ? chatThreads.map(thread => ({
@@ -3398,7 +3490,7 @@ class ChatWidget extends _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_1__.React
         super.dispose();
     }
     render() {
-        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatTaskpane__WEBPACK_IMPORTED_MODULE_6__["default"], { app: this.options.app, notebookTracker: this.options.notebookTracker, renderMimeRegistry: this.options.renderMimeRegistry, contextManager: this.options.contextManager, streamlitPreviewManager: this.options.streamlitPreviewManager, operatingSystem: this.options.operatingSystem, websocketClient: this.websocketClient }));
+        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatTaskpane__WEBPACK_IMPORTED_MODULE_6__["default"], { app: this.options.app, notebookTracker: this.options.notebookTracker, renderMimeRegistry: this.options.renderMimeRegistry, contextManager: this.options.contextManager, streamlitPreviewManager: this.options.streamlitPreviewManager, operatingSystem: this.options.operatingSystem, websocketClient: this.websocketClient, documentManager: this.options.documentManager }));
     }
     onMessage(client, message) {
         switch (message.type) {
@@ -3411,7 +3503,7 @@ class ChatWidget extends _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_1__.React
         }
     }
 }
-function buildChatWidget(app, notebookTracker, renderMimeRegistry, contextManager, streamlitPreviewManager) {
+function buildChatWidget(app, notebookTracker, renderMimeRegistry, contextManager, streamlitPreviewManager, documentManager) {
     // Get the operating system here so we don't have to do it each time the chat changes.
     // The operating system won't change, duh.
     const operatingSystem = (0,_utils_user__WEBPACK_IMPORTED_MODULE_7__.getOperatingSystem)();
@@ -3421,7 +3513,8 @@ function buildChatWidget(app, notebookTracker, renderMimeRegistry, contextManage
         renderMimeRegistry,
         contextManager,
         streamlitPreviewManager: streamlitPreviewManager,
-        operatingSystem
+        operatingSystem,
+        documentManager
     });
     chatWidget.id = 'mito_ai';
     return chatWidget;
@@ -3760,6 +3853,50 @@ UsageBadge.displayName = 'UsageBadge';
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (UsageBadge);
 
 
+/***/ }),
+
+/***/ "./lib/Extensions/AiChat/cellUpdateUtils.js":
+/*!**************************************************!*\
+  !*** ./lib/Extensions/AiChat/cellUpdateUtils.js ***!
+  \**************************************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   getCellIdFromCellUpdate: () => (/* binding */ getCellIdFromCellUpdate)
+/* harmony export */ });
+/* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! ../../utils/notebook */ "./lib/utils/notebook.js");
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+const getCellIdFromCellUpdate = (cellUpdate, notebookTracker) => {
+    if (!cellUpdate) {
+        return undefined;
+    }
+    if (cellUpdate.type === 'modification') {
+        return cellUpdate.id;
+    }
+    // For 'new' type, find the cell that was inserted after after_cell_id
+    const notebookPanel = notebookTracker.currentWidget;
+    if (!notebookPanel) {
+        return undefined;
+    }
+    if (cellUpdate.after_cell_id === 'new cell') {
+        // New cell was inserted at the top, so it's at index 0
+        return (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_0__.getCellIDByIndexInNotebookPanel)(notebookPanel, 0);
+    }
+    // Get the index of the cell we inserted after, then get the next cell's ID
+    const afterCellIndex = (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_0__.getCellIndexByIDInNotebookPanel)(notebookPanel, cellUpdate.after_cell_id);
+    if (afterCellIndex === undefined) {
+        return undefined;
+    }
+    // The new cell is at index afterCellIndex + 1
+    return (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_0__.getCellIDByIndexInNotebookPanel)(notebookPanel, afterCellIndex + 1);
+};
+
+
 /***/ }),
 
 /***/ "./lib/Extensions/AiChat/components/AgentReviewPanel.js":
@@ -3814,12 +3951,13 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
 /* harmony import */ var _lumino_coreutils__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! @lumino/coreutils */ "webpack/sharing/consume/default/@lumino/coreutils");
 /* harmony import */ var _lumino_coreutils__WEBPACK_IMPORTED_MODULE_1___default = /*#__PURE__*/__webpack_require__.n(_lumino_coreutils__WEBPACK_IMPORTED_MODULE_1__);
-/* harmony import */ var _utils_checkpoint__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../../utils/checkpoint */ "./lib/utils/checkpoint.js");
-/* harmony import */ var _utils_agentActions__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ../../../utils/agentActions */ "./lib/utils/agentActions.js");
-/* harmony import */ var _utils_blacklistedWords__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../../../utils/blacklistedWords */ "./lib/utils/blacklistedWords.js");
+/* harmony import */ var _utils_checkpoint__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../../utils/checkpoint */ "./lib/utils/checkpoint.js");
+/* harmony import */ var _utils_agentActions__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ../../../utils/agentActions */ "./lib/utils/agentActions.js");
+/* harmony import */ var _utils_blacklistedWords__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ../../../utils/blacklistedWords */ "./lib/utils/blacklistedWords.js");
 /* harmony import */ var _utils_sound__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../../../utils/sound */ "./lib/utils/sound.js");
-/* harmony import */ var _utils_strings__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../../utils/strings */ "./lib/utils/strings.js");
-/* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../../utils/notebook */ "./lib/utils/notebook.js");
+/* harmony import */ var _utils_strings__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../../../utils/strings */ "./lib/utils/strings.js");
+/* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../../utils/notebook */ "./lib/utils/notebook.js");
+/* harmony import */ var _utils__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../utils */ "./lib/Extensions/AiChat/utils.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -3832,8 +3970,9 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+
 const AGENT_EXECUTION_DEPTH_LIMIT = 20;
-const useAgentExecution = ({ notebookTracker, app, streamlitPreviewManager, websocketClient, chatHistoryManagerRef, activeThreadIdRef, activeRequestControllerRef, setLoadingStatus, setAutoScrollFollowMode, setHasCheckpoint, addAIMessageFromResponseAndUpdateState, getDuplicateChatHistoryManager, sendAgentExecutionMessage, sendAgentSmartDebugMessage, agentReview, agentTargetNotebookPanelRef, audioContextRef, }) => {
+const useAgentExecution = ({ notebookTracker, app, streamlitPreviewManager, websocketClient, documentManager, chatHistoryManagerRef, activeThreadIdRef, activeRequestControllerRef, setLoadingStatus, setAutoScrollFollowMode, setHasCheckpoint, addAIMessageFromResponseAndUpdateState, getDuplicateChatHistoryManager, sendAgentExecutionMessage, sendAgentSmartDebugMessage, agentReview, agentTargetNotebookPanelRef, audioContextRef, }) => {
     // Agent execution state
     const [agentExecutionStatus, setAgentExecutionStatus] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)('idle');
     const shouldContinueAgentExecution = (0,react__WEBPACK_IMPORTED_MODULE_0__.useRef)(true);
@@ -3866,10 +4005,12 @@ const useAgentExecution = ({ notebookTracker, app, streamlitPreviewManager, webs
         (0,_utils_sound__WEBPACK_IMPORTED_MODULE_2__.playCompletionSound)(audioContextRef.current);
     };
     const startAgentExecution = async (input, setAgentReviewStatus, messageIndex, additionalContext) => {
-        agentTargetNotebookPanelRef.current = notebookTracker.currentWidget;
+        // Ensure a notebook exists before proceeding with agent execution
+        const agentTargetNotebookPanel = await (0,_utils__WEBPACK_IMPORTED_MODULE_3__.ensureNotebookExists)(notebookTracker, documentManager);
+        agentTargetNotebookPanelRef.current = agentTargetNotebookPanel;
         agentReview.acceptAllAICode();
-        agentReview.setNotebookSnapshotPreAgentExecution((0,_utils_notebook__WEBPACK_IMPORTED_MODULE_3__.getAIOptimizedCellsInNotebookPanel)(agentTargetNotebookPanelRef.current));
-        await (0,_utils_checkpoint__WEBPACK_IMPORTED_MODULE_4__.createCheckpoint)(app, setHasCheckpoint);
+        agentReview.setNotebookSnapshotPreAgentExecution((0,_utils_notebook__WEBPACK_IMPORTED_MODULE_4__.getAIOptimizedCellsInNotebookPanel)(agentTargetNotebookPanelRef.current));
+        await (0,_utils_checkpoint__WEBPACK_IMPORTED_MODULE_5__.createCheckpoint)(app, setHasCheckpoint);
         setAgentExecutionStatus('working');
         setAgentReviewStatus('pre-agent-code-review');
         // Enable follow mode when user starts agent execution
@@ -3910,9 +4051,9 @@ const useAgentExecution = ({ notebookTracker, app, streamlitPreviewManager, webs
             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_5__.getCodeBlockFromMessage)(aiDisplayOptimizedChatItem.message);
+                const aiGeneratedCode = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_6__.getCodeBlockFromMessage)(aiDisplayOptimizedChatItem.message);
                 if (aiGeneratedCode) {
-                    const securityCheck = (0,_utils_blacklistedWords__WEBPACK_IMPORTED_MODULE_6__.checkForBlacklistedWords)(aiGeneratedCode);
+                    const securityCheck = (0,_utils_blacklistedWords__WEBPACK_IMPORTED_MODULE_7__.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', chatHistoryManagerRef.current);
@@ -3956,12 +4097,12 @@ const useAgentExecution = ({ notebookTracker, app, streamlitPreviewManager, webs
                 // Run the code and handle any errors
                 setLoadingStatus('running-code');
                 try {
-                    await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_7__.acceptAndRunCellUpdate)(agentResponse.cell_update, agentTargetNotebookPanelRef.current);
+                    await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_8__.acceptAndRunCellUpdate)(agentResponse.cell_update, agentTargetNotebookPanelRef.current);
                 }
                 finally {
                     setLoadingStatus(undefined);
                 }
-                const status = await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_7__.retryIfExecutionError)(agentTargetNotebookPanelRef.current, app, sendAgentSmartDebugMessage, shouldContinueAgentExecution, markAgentForStopping, chatHistoryManagerRef, setLoadingStatus);
+                const status = await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_8__.retryIfExecutionError)(agentTargetNotebookPanelRef.current, app, sendAgentSmartDebugMessage, shouldContinueAgentExecution, markAgentForStopping, chatHistoryManagerRef, setLoadingStatus);
                 if (status === 'interupted') {
                     break;
                 }
@@ -3984,7 +4125,7 @@ const useAgentExecution = ({ notebookTracker, app, streamlitPreviewManager, webs
                 setLoadingStatus('running-code');
                 let result;
                 try {
-                    result = await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_7__.runAllCells)(app, agentTargetNotebookPanelRef.current);
+                    result = await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_8__.runAllCells)(app, agentTargetNotebookPanelRef.current);
                 }
                 finally {
                     setLoadingStatus(undefined);
@@ -3992,8 +4133,8 @@ const useAgentExecution = ({ notebookTracker, app, streamlitPreviewManager, webs
                 // If run_all_cells resulted in an error, handle it through the error fixup process
                 if (!result.success && result.errorMessage && result.errorCellId) {
                     // Set the error cell as active so the error retry logic can work with it
-                    (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_3__.setActiveCellByIDInNotebookPanel)(agentTargetNotebookPanelRef.current, result.errorCellId);
-                    const status = await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_7__.retryIfExecutionError)(agentTargetNotebookPanelRef.current, app, sendAgentSmartDebugMessage, shouldContinueAgentExecution, markAgentForStopping, chatHistoryManagerRef, setLoadingStatus);
+                    (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_4__.setActiveCellByIDInNotebookPanel)(agentTargetNotebookPanelRef.current, result.errorCellId);
+                    const status = await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_8__.retryIfExecutionError)(agentTargetNotebookPanelRef.current, app, sendAgentSmartDebugMessage, shouldContinueAgentExecution, markAgentForStopping, chatHistoryManagerRef, setLoadingStatus);
                     if (status === 'interupted') {
                         break;
                     }
@@ -5098,12 +5239,15 @@ const IChatTracker = new _lumino_coreutils__WEBPACK_IMPORTED_MODULE_0__.Token('m
 
 __webpack_require__.r(__webpack_exports__);
 /* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   ensureNotebookExists: () => (/* binding */ ensureNotebookExists),
 /* harmony export */   getBase64EncodedCellOutput: () => (/* binding */ getBase64EncodedCellOutput),
 /* harmony export */   getBase64EncodedCellOutputInNotebook: () => (/* binding */ getBase64EncodedCellOutputInNotebook)
 /* harmony export */ });
 /* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! ../../utils/notebook */ "./lib/utils/notebook.js");
 /* harmony import */ var _utils_cellOutput__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../../utils/cellOutput */ "./lib/utils/cellOutput.js");
 /* harmony import */ var _restAPI_RestAPI__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../restAPI/RestAPI */ "./lib/restAPI/RestAPI.js");
+/* harmony import */ var _utils_waitForNotebookReady__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../utils/waitForNotebookReady */ "./lib/utils/waitForNotebookReady.js");
+/* harmony import */ var _utils_notebookMetadata__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../utils/notebookMetadata */ "./lib/utils/notebookMetadata.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -5111,6 +5255,8 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+
+
 const getBase64EncodedCellOutput = async (notebookTracker, cellID) => {
     const notebookPanel = notebookTracker.currentWidget;
     return getBase64EncodedCellOutputInNotebook(notebookPanel, cellID);
@@ -5132,6 +5278,38 @@ const getBase64EncodedCellOutputInNotebook = async (notebookPanel, cellID) => {
     }
     return undefined;
 };
+/*
+    Ensure a notebook exists. If no notebook is open, create a new one.
+    Returns the notebook panel.
+*/
+const ensureNotebookExists = async (notebookTracker, documentManager) => {
+    // Check if a notebook already exists
+    let notebookPanel = notebookTracker.currentWidget;
+    if (notebookPanel !== null) {
+        return notebookPanel;
+    }
+    // No notebook exists, create a new one
+    try {
+        // Create a new notebook model (Contents.IModel has a path property)
+        const model = await documentManager.newUntitled({ type: 'notebook' });
+        // Open the notebook using the path from the model
+        await documentManager.open(model.path);
+        // Wait for the notebook to appear in the tracker and be ready
+        await (0,_utils_waitForNotebookReady__WEBPACK_IMPORTED_MODULE_3__.waitForNotebookReady)(notebookTracker);
+        // Get the notebook panel from the tracker
+        notebookPanel = notebookTracker.currentWidget;
+        if (notebookPanel === null) {
+            throw new Error('Failed to get notebook panel after creation');
+        }
+        // Set the notebook ID if it doesn't exist
+        (0,_utils_notebookMetadata__WEBPACK_IMPORTED_MODULE_4__.setNotebookID)(notebookPanel);
+        return notebookPanel;
+    }
+    catch (error) {
+        console.error('Error creating new notebook:', error);
+        throw error;
+    }
+};
 
 
 /***/ }),
@@ -7996,6 +8174,11 @@ const plugin = {
                 removeButtonsFromAllNotebooks();
             }
         });
+        // Set Mito Light as default theme if user hasn't explicitly chosen a different theme
+        // This runs after registration so the theme is available
+        if (manager.theme === 'JupyterLab Light' || !manager.theme) {
+            void manager.setTheme('Mito Light');
+        }
     },
     autoStart: true
 };
@@ -9890,11 +10073,12 @@ const emptyCellPlaceholder = {
     // so that the keybinding can be properly resolved.
     tracker, extensions) => {
         var _a;
-        const keyBindings = app.commands.keyBindings.find(b => b.command === _commands__WEBPACK_IMPORTED_MODULE_4__.COMMAND_MITO_AI_OPEN_CHAT);
-        const pythonAdvice = `Start writing python or Press ${_lumino_commands__WEBPACK_IMPORTED_MODULE_2__.CommandRegistry.formatKeystroke((_a = keyBindings === null || keyBindings === void 0 ? void 0 : keyBindings.keys[0]) !== null && _a !== void 0 ? _a : 'Accel E')
+        const openChatKeyBindings = app.commands.keyBindings.find(b => b.command === _commands__WEBPACK_IMPORTED_MODULE_4__.COMMAND_MITO_AI_OPEN_CHAT);
+        const openChatKeystroke = _lumino_commands__WEBPACK_IMPORTED_MODULE_2__.CommandRegistry.formatKeystroke((_a = openChatKeyBindings === null || openChatKeyBindings === void 0 ? void 0 : openChatKeyBindings.keys[0]) !== null && _a !== void 0 ? _a : 'Accel E')
             .split(/[+\s]/)
             .map(s => `<kbd>${s}</kbd>`)
-            .join(' + ')} to ask Mito AI to write code for you.`;
+            .join(' + ');
+        const pythonAdvice = `Write Python or Press ${openChatKeystroke} to open Mito AI`;
         extensions.addExtension({
             name: 'mito-ai:empty-editor-advice',
             factory: options => {
@@ -10228,7 +10412,8 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony export */   COMMAND_MITO_AI_SEND_AGENT_MESSAGE: () => (/* binding */ COMMAND_MITO_AI_SEND_AGENT_MESSAGE),
 /* harmony export */   COMMAND_MITO_AI_SEND_DEBUG_ERROR_MESSAGE: () => (/* binding */ COMMAND_MITO_AI_SEND_DEBUG_ERROR_MESSAGE),
 /* harmony export */   COMMAND_MITO_AI_SEND_EXPLAIN_CODE_MESSAGE: () => (/* binding */ COMMAND_MITO_AI_SEND_EXPLAIN_CODE_MESSAGE),
-/* harmony export */   COMMAND_MITO_AI_SEND_MESSAGE: () => (/* binding */ COMMAND_MITO_AI_SEND_MESSAGE)
+/* harmony export */   COMMAND_MITO_AI_SEND_MESSAGE: () => (/* binding */ COMMAND_MITO_AI_SEND_MESSAGE),
+/* harmony export */   COMMAND_MITO_AI_START_NEW_CHAT: () => (/* binding */ COMMAND_MITO_AI_START_NEW_CHAT)
 /* harmony export */ });
 /*
  * Copyright (c) Saga Inc.
@@ -10245,6 +10430,7 @@ const COMMAND_MITO_AI_SEND_MESSAGE = `${MITO_AI}:send-message`;
 const COMMAND_MITO_AI_SEND_EXPLAIN_CODE_MESSAGE = `${MITO_AI}:send-explain-code-message`;
 const COMMAND_MITO_AI_SEND_DEBUG_ERROR_MESSAGE = `${MITO_AI}:send-debug-error-message`;
 const COMMAND_MITO_AI_SEND_AGENT_MESSAGE = `${MITO_AI}:send-agent-message`;
+const COMMAND_MITO_AI_START_NEW_CHAT = `${MITO_AI}:start-new-chat`;
 const COMMAND_MITO_AI_CELL_TOOLBAR_ACCEPT_CODE = `toolbar-button:accept-code`;
 const COMMAND_MITO_AI_CELL_TOOLBAR_REJECT_CODE = `toolbar-button:reject-code`;
 // Beta mode commands
@@ -10409,16 +10595,16 @@ __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 _utils_classNames__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../utils/classNames */ "./lib/utils/classNames.js");
-/* harmony import */ var _utils_strings__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../utils/strings */ "./lib/utils/strings.js");
-/* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../utils/notebook */ "./lib/utils/notebook.js");
-/* harmony import */ var _Extensions_AiChat_ChatMessage_PythonCode__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ../../Extensions/AiChat/ChatMessage/PythonCode */ "./lib/Extensions/AiChat/ChatMessage/PythonCode.js");
-/* harmony import */ var _Extensions_AiChat_ChatMessage_AssistantCodeBlock__WEBPACK_IMPORTED_MODULE_10__ = __webpack_require__(/*! ../../Extensions/AiChat/ChatMessage/AssistantCodeBlock */ "./lib/Extensions/AiChat/ChatMessage/AssistantCodeBlock.js");
-/* harmony import */ var _icons_AlertIcon__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ../../icons/AlertIcon */ "./lib/icons/AlertIcon.js");
-/* harmony import */ var _AgentComponentHeader__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ./AgentComponentHeader */ "./lib/components/AgentComponents/AgentComponentHeader.js");
+/* harmony import */ var _utils_classNames__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../utils/classNames */ "./lib/utils/classNames.js");
+/* harmony import */ var _utils_strings__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../utils/strings */ "./lib/utils/strings.js");
+/* harmony import */ var _Extensions_AiChat_cellUpdateUtils__WEBPACK_IMPORTED_MODULE_10__ = __webpack_require__(/*! ../../Extensions/AiChat/cellUpdateUtils */ "./lib/Extensions/AiChat/cellUpdateUtils.js");
+/* harmony import */ var _Extensions_AiChat_ChatMessage_PythonCode__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ../../Extensions/AiChat/ChatMessage/PythonCode */ "./lib/Extensions/AiChat/ChatMessage/PythonCode.js");
+/* harmony import */ var _Extensions_AiChat_ChatMessage_AssistantCodeBlock__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ../../Extensions/AiChat/ChatMessage/AssistantCodeBlock */ "./lib/Extensions/AiChat/ChatMessage/AssistantCodeBlock.js");
+/* harmony import */ var _icons_AlertIcon__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../../icons/AlertIcon */ "./lib/icons/AlertIcon.js");
+/* harmony import */ var _AgentComponentHeader__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ./AgentComponentHeader */ "./lib/components/AgentComponents/AgentComponentHeader.js");
 /* harmony import */ var _style_ErrorFixupToolUI_css__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../../style/ErrorFixupToolUI.css */ "./style/ErrorFixupToolUI.css");
 /* harmony import */ var _style_AgentComponentHeader_css__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../../../style/AgentComponentHeader.css */ "./style/AgentComponentHeader.css");
-/* harmony import */ var _RunAllCellsToolUI__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ./RunAllCellsToolUI */ "./lib/components/AgentComponents/RunAllCellsToolUI.js");
+/* harmony import */ var _RunAllCellsToolUI__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ./RunAllCellsToolUI */ "./lib/components/AgentComponents/RunAllCellsToolUI.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -10434,16 +10620,6 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
-const getCellIdFromCellUpdate = (cellUpdate, notebookTracker) => {
-    if (!cellUpdate) {
-        return undefined;
-    }
-    if (cellUpdate.type === 'modification') {
-        return cellUpdate.id;
-    }
-    // For 'new' type, get the cell ID by index
-    return (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_3__.getCellIDByIndexInNotebookPanel)(notebookTracker.currentWidget, cellUpdate.index);
-};
 const parsePythonErrorType = (content) => {
     if (!content)
         return 'Error';
@@ -10460,16 +10636,16 @@ const processErrorContent = (content) => {
 };
 const ErrorDetectedBlock = ({ errorMessage, renderMimeRegistry, }) => {
     const [expandedError, setExpandedError] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
-    const rawErrorContent = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_4__.getContentStringFromMessage)(errorMessage);
+    const rawErrorContent = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_3__.getContentStringFromMessage)(errorMessage);
     const errorContent = processErrorContent(rawErrorContent);
     const errorType = parsePythonErrorType(errorContent);
     const toggleError = () => {
         setExpandedError(!expandedError);
     };
-    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_5__.classNames)('error-fixup-container', { expanded: expandedError }) },
-        react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_AgentComponentHeader__WEBPACK_IMPORTED_MODULE_6__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_AlertIcon__WEBPACK_IMPORTED_MODULE_7__["default"], null), text: `${errorType} Detected`, onClick: toggleError, isExpanded: expandedError, displayBorder: false }),
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_4__.classNames)('error-fixup-container', { expanded: expandedError }) },
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_AgentComponentHeader__WEBPACK_IMPORTED_MODULE_5__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_AlertIcon__WEBPACK_IMPORTED_MODULE_6__["default"], null), text: `${errorType} Detected`, onClick: toggleError, isExpanded: expandedError, displayBorder: false }),
         expandedError && errorContent && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "error-fixup-expanded" },
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_Extensions_AiChat_ChatMessage_PythonCode__WEBPACK_IMPORTED_MODULE_8__["default"], { code: errorContent, renderMimeRegistry: renderMimeRegistry })))));
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_Extensions_AiChat_ChatMessage_PythonCode__WEBPACK_IMPORTED_MODULE_7__["default"], { code: errorContent, renderMimeRegistry: renderMimeRegistry })))));
 };
 const GroupedErrorsAndFixes = ({ messages, renderMimeRegistry, notebookTracker, }) => {
     return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "error-fixup-root" },
@@ -10482,9 +10658,9 @@ const GroupedErrorsAndFixes = ({ messages, renderMimeRegistry, notebookTracker,
                 return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(ErrorDetectedBlock, { key: `error-${index}`, errorMessage: messageItem.message, renderMimeRegistry: renderMimeRegistry }));
             }
             if (((_a = messageItem.agentResponse) === null || _a === void 0 ? void 0 : _a.type) === 'run_all_cells') {
-                return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_RunAllCellsToolUI__WEBPACK_IMPORTED_MODULE_9__["default"], { key: `run-all-cells-${index}`, inErrorFixup: true }));
+                return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_RunAllCellsToolUI__WEBPACK_IMPORTED_MODULE_8__["default"], { key: `run-all-cells-${index}`, inErrorFixup: true }));
             }
-            return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_Extensions_AiChat_ChatMessage_AssistantCodeBlock__WEBPACK_IMPORTED_MODULE_10__["default"], { key: `assistant-${index}`, code: messageItem.message.content, codeSummary: (_d = (_c = (_b = messageItem.agentResponse) === null || _b === void 0 ? void 0 : _b.cell_update) === null || _c === void 0 ? void 0 : _c.code_summary) !== null && _d !== void 0 ? _d : 'Fixing error', isCodeComplete: true, renderMimeRegistry: renderMimeRegistry, previewAICode: () => { }, acceptAICode: () => { }, rejectAICode: () => { }, isLastAiMessage: false, codeReviewStatus: "chatPreview", agentModeEnabled: true, isErrorFixup: true, cellId: getCellIdFromCellUpdate((_e = messageItem.agentResponse) === null || _e === void 0 ? void 0 : _e.cell_update, notebookTracker), notebookPanel: notebookTracker.currentWidget }));
+            return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_Extensions_AiChat_ChatMessage_AssistantCodeBlock__WEBPACK_IMPORTED_MODULE_9__["default"], { key: `assistant-${index}`, code: messageItem.message.content, codeSummary: (_d = (_c = (_b = messageItem.agentResponse) === null || _b === void 0 ? void 0 : _b.cell_update) === null || _c === void 0 ? void 0 : _c.code_summary) !== null && _d !== void 0 ? _d : 'Fixing error', isCodeComplete: true, renderMimeRegistry: renderMimeRegistry, previewAICode: () => { }, acceptAICode: () => { }, rejectAICode: () => { }, isLastAiMessage: false, codeReviewStatus: "chatPreview", agentModeEnabled: true, isErrorFixup: true, cellId: (0,_Extensions_AiChat_cellUpdateUtils__WEBPACK_IMPORTED_MODULE_10__.getCellIdFromCellUpdate)((_e = messageItem.agentResponse) === null || _e === void 0 ? void 0 : _e.cell_update, notebookTracker), notebookPanel: notebookTracker.currentWidget }));
         }))));
 };
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (GroupedErrorsAndFixes);
@@ -11001,7 +11177,7 @@ const MODEL_MAPPINGS = [
 ];
 const ALL_MODEL_DISPLAY_NAMES = MODEL_MAPPINGS.map(mapping => mapping.displayName);
 // Maximum length for displayed model name before truncating
-const DEFAULT_MODEL = _utils_models__WEBPACK_IMPORTED_MODULE_3__.CLAUDE_SONNET_DISPLAY_NAME;
+const DEFAULT_MODEL = _utils_models__WEBPACK_IMPORTED_MODULE_3__.CLAUDE_HAIKU_DISPLAY_NAME;
 const ModelSelector = ({ onConfigChange }) => {
     const [selectedModel, setSelectedModel] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(DEFAULT_MODEL);
     const [isOpen, setIsOpen] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
@@ -11513,6 +11689,9 @@ const SelectedContextContainer = ({ title, type, onRemove, onClick, notebookTrac
     else if (type === 'cell') {
         icon = react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_CodeIcon__WEBPACK_IMPORTED_MODULE_4__["default"], null);
     }
+    else if (type === 'line_selection') {
+        icon = react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_CodeIcon__WEBPACK_IMPORTED_MODULE_4__["default"], null);
+    }
     const handleClick = () => {
         var _a;
         if (type === 'active_cell') {
@@ -11541,16 +11720,66 @@ const SelectedContextContainer = ({ title, type, onRemove, onClick, notebookTrac
                 (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_6__.highlightCodeCell)(notebookTracker, value);
             }, 500);
         }
+        else if (type === 'line_selection' && notebookTracker && value) {
+            // Handle line selection context click - scroll to and highlight selected lines
+            try {
+                const selectionInfo = JSON.parse(value);
+                const currentWidget = notebookTracker.currentWidget;
+                if (currentWidget) {
+                    // Scroll to the cell, positioning based on the start line
+                    // Lines are stored 0-indexed, matching the citation format
+                    (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_6__.scrollToCell)(currentWidget, selectionInfo.cellId, selectionInfo.startLine, 'center');
+                    // Highlight the selected lines
+                    setTimeout(() => {
+                        // Re-check currentWidget inside the callback since it may have changed
+                        const widget = notebookTracker.currentWidget;
+                        if (widget) {
+                            (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_6__.highlightLinesOfCodeInCodeCell)(widget, selectionInfo.cellId, selectionInfo.startLine, selectionInfo.endLine);
+                        }
+                    }, 500);
+                }
+            }
+            catch (_b) {
+                // Ignore JSON parse errors
+            }
+        }
         else if (onClick) {
             // Call the custom onClick handler for other context types
             onClick();
         }
     };
-    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { className: "selected-context-container", onMouseEnter: () => setIsHovered(true), onMouseLeave: () => setIsHovered(false), onClick: handleClick, "data-testid": "selected-context-container" },
+    const getTooltipText = () => {
+        if (type.startsWith('image/')) {
+            return `The AI will be able to view the ${title} image before deciding how to respond`;
+        }
+        else if (type === 'file') {
+            return `The path ${title} will be shared with the AI`;
+        }
+        else if (type === 'notebook') {
+            return "The AI will be able to read all of the code and markdown in your notebook. It is included by default in Agent mode.";
+        }
+        else if (type === 'active_cell') {
+            return "The AI will write its code based on the currently active cell. It is included by default in Chat mode.";
+        }
+        else if (type === 'cell') {
+            return `The AI will be able to see the code in ${title}`;
+        }
+        else if (type === 'variable') {
+            return `The AI will receive a summary of the ${title} variable`;
+        }
+        else if (type === 'rule') {
+            return `The AI will be guided by the ${title} rule`;
+        }
+        else if (type === 'db') {
+            return `The AI will be able to access the ${title} database connection`;
+        }
+        return "This context will be included in your message to help the AI understand what you're working with";
+    };
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { className: "selected-context-container", onMouseEnter: () => setIsHovered(true), onMouseLeave: () => setIsHovered(false), onClick: handleClick, "data-testid": "selected-context-container", "data-type": type, title: getTooltipText() },
         react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: `icon`, onClick: (e) => {
                 e.stopPropagation(); // Prevent triggering the button's onClick
                 onRemove();
-            }, title: isHovered ? "Remove rule" : "Selected rule" }, isHovered && type !== 'active_cell' && type !== 'notebook' ? (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "remove-icon" }, "X")) : (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "icon" }, icon))),
+            }, title: isHovered ? "Remove rule" : "Selected rule" }, isHovered && type !== 'active_cell' && type !== 'notebook' && type !== 'line_selection' ? (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "remove-icon" }, "X")) : (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "icon" }, icon))),
         react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "rule-name" }, title)));
 };
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (SelectedContextContainer);
@@ -11790,6 +12019,125 @@ const useDebouncedFunction = (callback, delay) => {
 };
 
 
+/***/ }),
+
+/***/ "./lib/hooks/useLineSelection.js":
+/*!***************************************!*\
+  !*** ./lib/hooks/useLineSelection.js ***!
+  \***************************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   useLineSelection: () => (/* binding */ useLineSelection)
+/* 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 _utils_notebook__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../utils/notebook */ "./lib/utils/notebook.js");
+/* harmony import */ var _useDebouncedFunction__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ./useDebouncedFunction */ "./lib/hooks/useDebouncedFunction.js");
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+
+
+const NO_SELECTION = {
+    cellId: '',
+    cellNumber: 0,
+    startLine: 0,
+    endLine: 0,
+    selectedCode: '',
+    hasSelection: false
+};
+/**
+ * Hook that tracks text selection in the active code cell.
+ * Returns selection info when user has text selected (not just cursor position).
+ *
+ * @param notebookTracker - The notebook tracker to monitor
+ * @param cellOrder - Map of cellId to cellNumber (1-indexed)
+ * @returns LineSelectionInfo with selection details, or hasSelection: false if no selection
+ */
+const useLineSelection = (notebookTracker, cellOrder) => {
+    const [selection, setSelection] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(NO_SELECTION);
+    // Debounce selection updates to avoid excessive re-renders during text selection
+    // `setSelection` is stable, so this keeps the debounced function stable too.
+    const debouncedSetSelection = (0,_useDebouncedFunction__WEBPACK_IMPORTED_MODULE_1__.useDebouncedFunction)(setSelection, 100);
+    // Get current selection from active cell
+    const checkSelection = (0,react__WEBPACK_IMPORTED_MODULE_0__.useCallback)(() => {
+        const activeCell = (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_2__.getActiveCell)(notebookTracker);
+        if (!activeCell || activeCell.model.type !== 'code') {
+            debouncedSetSelection(NO_SELECTION);
+            return;
+        }
+        const cellId = activeCell.model.id;
+        const codeCell = activeCell;
+        const cmEditor = codeCell.editor;
+        const editorView = cmEditor === null || cmEditor === void 0 ? void 0 : cmEditor.editor;
+        if (!editorView) {
+            debouncedSetSelection(NO_SELECTION);
+            return;
+        }
+        const state = editorView.state;
+        const mainSelection = state.selection.main;
+        // Check if there's actual text selected (not just cursor)
+        if (mainSelection.from === mainSelection.to) {
+            debouncedSetSelection(NO_SELECTION);
+            return;
+        }
+        // Get 0-indexed line numbers (matching citation format)
+        // CodeMirror's lineAt().number is 1-indexed, so subtract 1
+        const startLineInfo = state.doc.lineAt(mainSelection.from);
+        const endLineInfo = state.doc.lineAt(mainSelection.to);
+        const startLine = startLineInfo.number - 1;
+        const endLine = endLineInfo.number - 1;
+        const cellNumber = cellOrder.get(cellId) || 0;
+        // Extract the full lines that contain the selection
+        const selectedLines = [];
+        for (let lineNum = startLineInfo.number; lineNum <= endLineInfo.number; lineNum++) {
+            const line = state.doc.line(lineNum);
+            selectedLines.push(line.text);
+        }
+        const selectedCode = selectedLines.join('\n');
+        debouncedSetSelection({
+            cellId,
+            cellNumber,
+            startLine,
+            endLine,
+            selectedCode,
+            hasSelection: true
+        });
+    }, [notebookTracker, cellOrder, debouncedSetSelection]);
+    // Poll for selection changes using document-level events
+    (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
+        // Check selection on various events that might indicate selection change
+        const handleSelectionChange = () => {
+            checkSelection();
+        };
+        // Listen to selection changes at document level
+        document.addEventListener('selectionchange', handleSelectionChange);
+        // Also listen to mouseup for drag selections
+        document.addEventListener('mouseup', handleSelectionChange);
+        // Listen to keyup for keyboard selections (shift+arrow keys)
+        document.addEventListener('keyup', handleSelectionChange);
+        // Check on active cell change
+        const handleActiveCellChanged = () => {
+            checkSelection();
+        };
+        notebookTracker.activeCellChanged.connect(handleActiveCellChanged);
+        // Initial check
+        checkSelection();
+        return () => {
+            document.removeEventListener('selectionchange', handleSelectionChange);
+            document.removeEventListener('mouseup', handleSelectionChange);
+            document.removeEventListener('keyup', handleSelectionChange);
+            notebookTracker.activeCellChanged.disconnect(handleActiveCellChanged);
+        };
+    }, [notebookTracker, checkSelection]);
+    return selection;
+};
+
+
 /***/ }),
 
 /***/ "./lib/icons/AcceptIcon.js":
@@ -13058,7 +13406,11 @@ const acceptAndRunCellUpdate = async (cellUpdate, notebookPanel) => {
     // in other notebooks.
     if (cellUpdate.type === 'new') {
         // makes the cell the active cell
-        (0,_notebook__WEBPACK_IMPORTED_MODULE_1__.createCodeCellAtIndexAndActivate)(notebookPanel, cellUpdate.index);
+        if (cellUpdate.after_cell_id === undefined || cellUpdate.after_cell_id === null) {
+            console.error('after_cell_id is required for new cell creation');
+            return;
+        }
+        (0,_notebook__WEBPACK_IMPORTED_MODULE_1__.createCodeCellAfterCellIDAndActivate)(notebookPanel, cellUpdate.after_cell_id);
     }
     else {
         (0,_notebook__WEBPACK_IMPORTED_MODULE_1__.setActiveCellByIDInNotebookPanel)(notebookPanel, cellUpdate.id);
@@ -14377,7 +14729,7 @@ const GEMINI_3_PRO_MODEL_NAME = 'gemini-3-pro-preview';
 __webpack_require__.r(__webpack_exports__);
 /* harmony export */ __webpack_require__.d(__webpack_exports__, {
 /* harmony export */   applyCellEditorExtension: () => (/* binding */ applyCellEditorExtension),
-/* harmony export */   createCodeCellAtIndexAndActivate: () => (/* binding */ createCodeCellAtIndexAndActivate),
+/* harmony export */   createCodeCellAfterCellIDAndActivate: () => (/* binding */ createCodeCellAfterCellIDAndActivate),
 /* harmony export */   didCellExecutionError: () => (/* binding */ didCellExecutionError),
 /* harmony export */   getAIOptimizedCells: () => (/* binding */ getAIOptimizedCells),
 /* harmony export */   getAIOptimizedCellsInNotebookPanel: () => (/* binding */ getAIOptimizedCellsInNotebookPanel),
@@ -14538,24 +14890,33 @@ const getAIOptimizedCellsInNotebookPanel = (notebookPanel) => {
     }
     return cells;
 };
-function createCodeCellAtIndexAndActivate(notebookPanel, index) {
+function createCodeCellAfterCellIDAndActivate(notebookPanel, afterCellID) {
     /*
-        Create a new code cell at index and make it the active cell.
+        Create a new code cell after the cell with the given ID and make it the active cell.
+        If afterCellID is 'new cell', insert at the very top of the notebook.
     */
     const notebook = notebookPanel.content;
     if (notebook === undefined) {
         return;
     }
-    if (index > 0) {
-        notebook.activeCellIndex = index - 1;
-        // insertBelow makes the new cell the active cell
-        _jupyterlab_notebook__WEBPACK_IMPORTED_MODULE_0__.NotebookActions.insertBelow(notebook);
-    }
-    else {
+    if (afterCellID === 'new cell') {
+        // Insert at the very top of the notebook
         notebook.activeCellIndex = 0;
         // insertAbove makes the new cell the active cell
         _jupyterlab_notebook__WEBPACK_IMPORTED_MODULE_0__.NotebookActions.insertAbove(notebook);
     }
+    else {
+        // Find the cell by ID and get its index
+        const cellIndex = getCellIndexByIDInNotebookPanel(notebookPanel, afterCellID);
+        if (cellIndex === undefined) {
+            console.error(`Cell with ID ${afterCellID} not found`);
+            return;
+        }
+        // Set the active cell to the cell we want to insert after
+        notebook.activeCellIndex = cellIndex;
+        // insertBelow makes the new cell the active cell
+        _jupyterlab_notebook__WEBPACK_IMPORTED_MODULE_0__.NotebookActions.insertBelow(notebook);
+    }
 }
 const didCellExecutionError = (cell) => {
     var _a;
@@ -20942,7 +21303,7 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
   display: flex;
   align-items: center;
   text-align: center;
-  background-color: var(--jp-layout-color2);
+  background-color: var(--purple-300);
   border-radius: 3px;
   padding: 4px 8px;
   margin: 2px;
@@ -20952,17 +21313,23 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
   max-width: 100%;
   min-width: 0;
 
-  border: 1px solid var(--jp-border-color1);
+  border: none;
   cursor: pointer;
 }
 
 .selected-context-container:hover {
-  background-color: var(--jp-layout-color3);
+  background-color: var(--purple-400);
+}
+
+/* Disable hover effect for notebook and active_cell types since they don't have user actions */
+.selected-context-container[data-type="notebook"]:hover,
+.selected-context-container[data-type="active_cell"]:hover {
+  background-color: var(--purple-300);
 }
 
 .selected-context-container .rule-name {
   margin-left: 6px;
-  color: var(--jp-content-font-color1);
+  color: var(--purple-700);
   white-space: nowrap;
   overflow: hidden;
   text-overflow: ellipsis;
@@ -20975,8 +21342,8 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
   justify-content: center;
   width: 12px;
   max-height: 14px;
-  color: var(--jp-content-font-color3);
-}`, "",{"version":3,"sources":["webpack://./style/SelectedContextContainer.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;EACE,aAAa;EACb,eAAe;EACf,mBAAmB;EACnB,QAAQ;EACR,iBAAiB;EACjB,kBAAkB;AACpB;;AAEA,6DAA6D;AAC7D;EACE,WAAW;AACb;;AAEA;EACE,aAAa;EACb,mBAAmB;EACnB,kBAAkB;EAClB,yCAAyC;EACzC,kBAAkB;EAClB,gBAAgB;EAChB,WAAW;;EAEX,eAAe;EACf,YAAY;EACZ,eAAe;EACf,YAAY;;EAEZ,yCAAyC;EACzC,eAAe;AACjB;;AAEA;EACE,yCAAyC;AAC3C;;AAEA;EACE,gBAAgB;EAChB,oCAAoC;EACpC,mBAAmB;EACnB,gBAAgB;EAChB,uBAAuB;EACvB,gBAAgB;AAClB;;AAEA;EACE,aAAa;EACb,sBAAsB;EACtB,uBAAuB;EACvB,WAAW;EACX,gBAAgB;EAChB,oCAAoC;AACtC","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.context-container {\n  display: flex;\n  flex-wrap: wrap;\n  align-items: center;\n  gap: 4px;\n  padding: 8px 10px;\n  overflow-x: hidden;\n}\n\n/* Consistent spacing for all children of context container */\n.context-container>* {\n  margin: 2px;\n}\n\n.selected-context-container {\n  display: flex;\n  align-items: center;\n  text-align: center;\n  background-color: var(--jp-layout-color2);\n  border-radius: 3px;\n  padding: 4px 8px;\n  margin: 2px;\n\n  font-size: 12px;\n  height: 20px;\n  max-width: 100%;\n  min-width: 0;\n\n  border: 1px solid var(--jp-border-color1);\n  cursor: pointer;\n}\n\n.selected-context-container:hover {\n  background-color: var(--jp-layout-color3);\n}\n\n.selected-context-container .rule-name {\n  margin-left: 6px;\n  color: var(--jp-content-font-color1);\n  white-space: nowrap;\n  overflow: hidden;\n  text-overflow: ellipsis;\n  max-width: 200px;\n}\n\n.selected-context-container .icon {\n  display: flex;\n  flex-direction: column;\n  justify-content: center;\n  width: 12px;\n  max-height: 14px;\n  color: var(--jp-content-font-color3);\n}"],"sourceRoot":""}]);
+  color: var(--purple-700);
+}`, "",{"version":3,"sources":["webpack://./style/SelectedContextContainer.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;EACE,aAAa;EACb,eAAe;EACf,mBAAmB;EACnB,QAAQ;EACR,iBAAiB;EACjB,kBAAkB;AACpB;;AAEA,6DAA6D;AAC7D;EACE,WAAW;AACb;;AAEA;EACE,aAAa;EACb,mBAAmB;EACnB,kBAAkB;EAClB,mCAAmC;EACnC,kBAAkB;EAClB,gBAAgB;EAChB,WAAW;;EAEX,eAAe;EACf,YAAY;EACZ,eAAe;EACf,YAAY;;EAEZ,YAAY;EACZ,eAAe;AACjB;;AAEA;EACE,mCAAmC;AACrC;;AAEA,+FAA+F;AAC/F;;EAEE,mCAAmC;AACrC;;AAEA;EACE,gBAAgB;EAChB,wBAAwB;EACxB,mBAAmB;EACnB,gBAAgB;EAChB,uBAAuB;EACvB,gBAAgB;AAClB;;AAEA;EACE,aAAa;EACb,sBAAsB;EACtB,uBAAuB;EACvB,WAAW;EACX,gBAAgB;EAChB,wBAAwB;AAC1B","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.context-container {\n  display: flex;\n  flex-wrap: wrap;\n  align-items: center;\n  gap: 4px;\n  padding: 8px 10px;\n  overflow-x: hidden;\n}\n\n/* Consistent spacing for all children of context container */\n.context-container>* {\n  margin: 2px;\n}\n\n.selected-context-container {\n  display: flex;\n  align-items: center;\n  text-align: center;\n  background-color: var(--purple-300);\n  border-radius: 3px;\n  padding: 4px 8px;\n  margin: 2px;\n\n  font-size: 12px;\n  height: 20px;\n  max-width: 100%;\n  min-width: 0;\n\n  border: none;\n  cursor: pointer;\n}\n\n.selected-context-container:hover {\n  background-color: var(--purple-400);\n}\n\n/* Disable hover effect for notebook and active_cell types since they don't have user actions */\n.selected-context-container[data-type=\"notebook\"]:hover,\n.selected-context-container[data-type=\"active_cell\"]:hover {\n  background-color: var(--purple-300);\n}\n\n.selected-context-container .rule-name {\n  margin-left: 6px;\n  color: var(--purple-700);\n  white-space: nowrap;\n  overflow: hidden;\n  text-overflow: ellipsis;\n  max-width: 200px;\n}\n\n.selected-context-container .icon {\n  display: flex;\n  flex-direction: column;\n  justify-content: center;\n  width: 12px;\n  max-height: 14px;\n  color: var(--purple-700);\n}"],"sourceRoot":""}]);
 // Exports
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
 
@@ -24213,4 +24580,4 @@ var update = _node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js
 /***/ })
 
 }]);
-//# sourceMappingURL=lib_index_js.49c79c62671528877c61.js.map
+//# sourceMappingURL=lib_index_js.dfd7975de75d64db80d6.js.map