mito-ai 0.1.36__py3-none-any.whl → 0.1.37__py3-none-any.whl

mito_ai-0.1.36.data/data/share/jupyter/labextensions/mito_ai/static/lib_index_js.a20772bc113422d0f505.js → mito_ai-0.1.37.data/data/share/jupyter/labextensions/mito_ai/static/lib_index_js.831f63b48760c7119b9b.js

@@ -17,12 +17,17 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony import */ var _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_1___default = /*#__PURE__*/__webpack_require__.n(_jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_1__);
 /* harmony import */ var _jupyterlab_notebook__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! @jupyterlab/notebook */ "webpack/sharing/consume/default/@jupyterlab/notebook");
 /* harmony import */ var _jupyterlab_notebook__WEBPACK_IMPORTED_MODULE_2___default = /*#__PURE__*/__webpack_require__.n(_jupyterlab_notebook__WEBPACK_IMPORTED_MODULE_2__);
-/* harmony import */ var _ChatWidget__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ./ChatWidget */ "./lib/Extensions/AiChat/ChatWidget.js");
+/* harmony import */ var _ChatWidget__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ./ChatWidget */ "./lib/Extensions/AiChat/ChatWidget.js");
 /* harmony import */ var _jupyterlab_rendermime__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! @jupyterlab/rendermime */ "webpack/sharing/consume/default/@jupyterlab/rendermime");
 /* harmony import */ var _jupyterlab_rendermime__WEBPACK_IMPORTED_MODULE_3___default = /*#__PURE__*/__webpack_require__.n(_jupyterlab_rendermime__WEBPACK_IMPORTED_MODULE_3__);
-/* harmony import */ var _commands__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ../../commands */ "./lib/commands.js");
-/* harmony import */ var _token__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ./token */ "./lib/Extensions/AiChat/token.js");
-/* harmony import */ var _ContextManager_ContextManagerPlugin__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../ContextManager/ContextManagerPlugin */ "./lib/Extensions/ContextManager/ContextManagerPlugin.js");
+/* harmony import */ var _commands__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ../../commands */ "./lib/commands.js");
+/* harmony import */ var _token__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ./token */ "./lib/Extensions/AiChat/token.js");
+/* harmony import */ var _ContextManager_ContextManagerPlugin__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../ContextManager/ContextManagerPlugin */ "./lib/Extensions/ContextManager/ContextManagerPlugin.js");
+/* harmony import */ var _jupyterlab_settingregistry__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! @jupyterlab/settingregistry */ "webpack/sharing/consume/default/@jupyterlab/settingregistry");
+/* harmony import */ var _jupyterlab_settingregistry__WEBPACK_IMPORTED_MODULE_4___default = /*#__PURE__*/__webpack_require__.n(_jupyterlab_settingregistry__WEBPACK_IMPORTED_MODULE_4__);
+/* harmony import */ var _jupyterSettingsManager__WEBPACK_IMPORTED_MODULE_10__ = __webpack_require__(/*! ./jupyterSettingsManager */ "./lib/Extensions/AiChat/jupyterSettingsManager.js");
+/* harmony import */ var _jupyterlab_docmanager__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! @jupyterlab/docmanager */ "webpack/sharing/consume/default/@jupyterlab/docmanager");
+/* harmony import */ var _jupyterlab_docmanager__WEBPACK_IMPORTED_MODULE_5___default = /*#__PURE__*/__webpack_require__.n(_jupyterlab_docmanager__WEBPACK_IMPORTED_MODULE_5__);
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -35,6 +40,9 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+
+
+
 // The Widget Rank determins where the ChatIcon is displayed
 // in the left hand toolbar
 const WIDGET_RANK = 2000;
@@ -50,21 +58,23 @@ const AiChatPlugin = {
         _jupyterlab_notebook__WEBPACK_IMPORTED_MODULE_2__.INotebookTracker,
         _jupyterlab_apputils__WEBPACK_IMPORTED_MODULE_1__.ICommandPalette,
         _jupyterlab_rendermime__WEBPACK_IMPORTED_MODULE_3__.IRenderMimeRegistry,
-        _ContextManager_ContextManagerPlugin__WEBPACK_IMPORTED_MODULE_4__.IContextManager
+        _ContextManager_ContextManagerPlugin__WEBPACK_IMPORTED_MODULE_6__.IContextManager,
+        _jupyterlab_settingregistry__WEBPACK_IMPORTED_MODULE_4__.ISettingRegistry,
+        _jupyterlab_docmanager__WEBPACK_IMPORTED_MODULE_5__.IDocumentManager
     ],
     optional: [_jupyterlab_application__WEBPACK_IMPORTED_MODULE_0__.ILayoutRestorer],
-    provides: _token__WEBPACK_IMPORTED_MODULE_5__.IChatTracker,
-    activate: (app, labShell, notebookTracker, palette, rendermime, contextManager, restorer) => {
+    provides: _token__WEBPACK_IMPORTED_MODULE_7__.IChatTracker,
+    activate: (app, labShell, notebookTracker, palette, rendermime, contextManager, settingRegistry, documentManager, restorer) => {
         // Define a widget creator function,
         // 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_6__.buildChatWidget)(app, notebookTracker, rendermime, contextManager);
+            const chatWidget = (0,_ChatWidget__WEBPACK_IMPORTED_MODULE_8__.buildChatWidget)(app, notebookTracker, rendermime, contextManager);
             return chatWidget;
         };
         let widget = newWidget();
         // Add an application command
-        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_7__.COMMAND_MITO_AI_OPEN_CHAT, {
+        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_9__.COMMAND_MITO_AI_OPEN_CHAT, {
             label: 'Your friendly Python Expert chat bot',
             execute: (args) => {
                 // In order for the widget to be accessible, the widget must be:
@@ -98,14 +108,14 @@ const AiChatPlugin = {
             }
         });
         app.commands.addKeyBinding({
-            command: _commands__WEBPACK_IMPORTED_MODULE_7__.COMMAND_MITO_AI_OPEN_CHAT,
+            command: _commands__WEBPACK_IMPORTED_MODULE_9__.COMMAND_MITO_AI_OPEN_CHAT,
             keys: ['Accel E'],
             selector: 'body'
         });
         app.shell.add(widget, 'left', { rank: WIDGET_RANK });
         // Add the command to the palette.
         palette.addItem({
-            command: _commands__WEBPACK_IMPORTED_MODULE_7__.COMMAND_MITO_AI_OPEN_CHAT,
+            command: _commands__WEBPACK_IMPORTED_MODULE_9__.COMMAND_MITO_AI_OPEN_CHAT,
             category: 'AI Chat'
         });
         // Track and restore the widget state
@@ -128,6 +138,8 @@ const AiChatPlugin = {
             // This will override the default file browser selection
             labShell.activateById(widget.id);
         });
+        // Update jupyter settings to work best with mito-ai
+        void (0,_jupyterSettingsManager__WEBPACK_IMPORTED_MODULE_10__.setRenameUntitledFileOnSave)(settingRegistry, documentManager);
         // By returning a tracker token, we can require the token in other 
         // plugins. This allows us to force plugin load order. For example, 
         // we can ensure that the COMMAND_MITO_AI_OPEN_CHAT is created 
@@ -265,7 +277,7 @@ class ChatHistoryManager {
             promptType: 'chat'
         });
     }
-    async addChatInputMessage(input, activeThreadId, messageIndex, selectedRules) {
+    async addChatInputMessage(input, activeThreadId, messageIndex, additionalContext) {
         const activeCellCode = (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_0__.getActiveCellCode)(this.notebookTracker) || '';
         const activeCellID = (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_0__.getActiveCellID)(this.notebookTracker) || '';
         const chatMessageMetadata = {
@@ -277,17 +289,18 @@ class ChatHistoryManager {
             input: input,
             threadId: activeThreadId,
             index: messageIndex,
-            selectedRules: selectedRules
+            additionalContext: additionalContext
         };
         this.displayOptimizedChatHistory.push({
             message: getDisplayedOptimizedUserMessage(input, activeCellCode),
             type: 'openai message',
             codeCellID: activeCellID,
-            promptType: 'chat'
+            promptType: 'chat',
+            additionalContext: additionalContext
         });
         return chatMessageMetadata;
     }
-    addAgentExecutionMessage(activeThreadId, input, selectedRules) {
+    addAgentExecutionMessage(activeThreadId, input, additionalContext) {
         const aiOptimizedCells = (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_0__.getAIOptimizedCells)(this.notebookTracker);
         const agentExecutionMetadata = {
             promptType: 'agent:execution',
@@ -297,7 +310,7 @@ class ChatHistoryManager {
             input: input || '',
             threadId: activeThreadId,
             isChromeBrowser: (0,_utils_user__WEBPACK_IMPORTED_MODULE_1__.isChromeBasedBrowser)(),
-            selectedRules: selectedRules
+            additionalContext: additionalContext
         };
         // We use this function in two ways: 
         // 1. When the user sends the original agent:execution message to start the agent
@@ -317,6 +330,7 @@ class ChatHistoryManager {
             message: userMessage,
             type: 'openai message',
             promptType: 'chat',
+            additionalContext: additionalContext
         });
         return agentExecutionMetadata;
     }
@@ -576,8 +590,8 @@ __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_2__ = __webpack_require__(/*! ../../../utils/classNames */ "./lib/utils/classNames.js");
 /* harmony import */ var _restAPI_RestAPI__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../../restAPI/RestAPI */ "./lib/restAPI/RestAPI.js");
+/* harmony import */ var _ChatDropdownItems__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ./ChatDropdownItems */ "./lib/Extensions/AiChat/ChatMessage/ChatDropdownItems.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -585,38 +599,100 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
-const ChatDropdown = ({ options, onSelect, filterText, maxDropdownItems = 10, }) => {
+const ChatDropdown = ({ options, onSelect, filterText, maxDropdownItems = 10, isDropdownFromButton = false, onFilterChange, onClose, }) => {
     const [selectedIndex, setSelectedIndex] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(0);
+    const [localFilterText, setLocalFilterText] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(filterText);
+    const searchInputRef = (0,react__WEBPACK_IMPORTED_MODULE_0__.useRef)(null);
     const [rules, setRules] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)([]);
+    const [databaseConnections, setDatabaseConnections] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)({});
     (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
         const fetchRules = async () => {
             const rules = await (0,_restAPI_RestAPI__WEBPACK_IMPORTED_MODULE_1__.getRules)();
             setRules(rules);
         };
+        const fetchDatabaseConnections = async () => {
+            const databaseConnections = await (0,_restAPI_RestAPI__WEBPACK_IMPORTED_MODULE_1__.getDatabaseConnections)();
+            setDatabaseConnections(databaseConnections);
+        };
         void fetchRules();
+        void fetchDatabaseConnections();
     }, []);
+    // Focus search input when dropdown opens with search input
+    (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
+        if (isDropdownFromButton && searchInputRef.current) {
+            searchInputRef.current.focus();
+        }
+    }, [isDropdownFromButton]);
+    // Use local filter text when search input is shown, otherwise use prop
+    const effectiveFilterText = isDropdownFromButton ? localFilterText : filterText;
     // Create a list of all options with the format 
     // ['type': 'variable', "expandedVariable": variable]
     // ['type': 'rule', "rule": rule]
+    // ['type': 'file', "file": file]
     const allOptions = [
-        ...options.map((variable) => ({
-            type: 'variable',
-            variable: variable
-        })),
+        // Rules first
         ...rules.map((rule) => ({
             type: 'rule',
             rule: rule
         })),
+        // Files second
+        ...options
+            .filter(variable => variable.file_name) // Only files
+            .map((file) => ({
+            type: 'file',
+            file: file
+        })),
+        // Databases third
+        ...Object.entries(databaseConnections).map(([connectionId, connection]) => ({
+            type: 'db',
+            variable: {
+                variable_name: connection.alias || connection.database,
+                type: "db",
+                value: connectionId + " - " + connection.type,
+                parent_df: connection.type
+            }
+        })),
+        // Dataframes fourth
+        ...options
+            .filter(variable => !variable.file_name && variable.type === "pd.DataFrame")
+            .map((variable) => ({
+            type: 'variable',
+            variable: variable
+        })),
+        // Columns fifth
+        ...options
+            .filter(variable => !variable.file_name && variable.parent_df && variable.type !== "pd.DataFrame")
+            .map((variable) => ({
+            type: 'variable',
+            variable: variable
+        })),
+        // Other variables last
+        ...options
+            .filter(variable => !variable.file_name && !variable.parent_df && variable.type !== "pd.DataFrame")
+            .map((variable) => ({
+            type: 'variable',
+            variable: variable
+        })),
     ];
-    const filteredOptions = allOptions.filter((option) => option.type === 'variable' ?
-        option.variable.variable_name.toLowerCase().includes(filterText.toLowerCase()) &&
-            option.variable.type !== "<class 'module'>" &&
-            option.variable.variable_name !== "FUNCTIONS" // This is default exported from mitosheet when you run from mitosheet import * as FUNCTIONS
-        :
-            option.rule.toLowerCase().includes(filterText.toLowerCase())).slice(0, maxDropdownItems);
+    const filteredOptions = allOptions.filter((option) => {
+        if (option.type === 'variable') {
+            return option.variable.variable_name.toLowerCase().includes(effectiveFilterText.toLowerCase()) &&
+                option.variable.type !== "<class 'module'>" &&
+                option.variable.variable_name !== "FUNCTIONS"; // This is default exported from mitosheet when you run from mitosheet import * as FUNCTIONS
+        }
+        else if (option.type === 'file') {
+            return option.file.variable_name.toLowerCase().includes(effectiveFilterText.toLowerCase());
+        }
+        else if (option.type === 'db') {
+            return option.variable.value.toLowerCase().includes(effectiveFilterText.toLowerCase());
+        }
+        else {
+            return option.rule.toLowerCase().includes(effectiveFilterText.toLowerCase());
+        }
+    }).slice(0, maxDropdownItems);
     (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
         setSelectedIndex(0);
-    }, [options, rules, filterText]);
+    }, [options, rules, effectiveFilterText]);
     const handleKeyDown = (event) => {
         switch (event.key) {
             case 'ArrowDown':
@@ -650,28 +726,96 @@ const ChatDropdown = ({ options, onSelect, filterText, maxDropdownItems = 10, })
         document.addEventListener('keydown', handleKeyDown);
         return () => document.removeEventListener('keydown', handleKeyDown);
     }, [filteredOptions, selectedIndex]);
+    // Handle click outside to close dropdown
+    (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
+        const handleClickOutside = (event) => {
+            const target = event.target;
+            const dropdownElement = document.querySelector('.chat-dropdown');
+            if (dropdownElement && !dropdownElement.contains(target)) {
+                if (onClose) {
+                    onClose();
+                }
+            }
+        };
+        document.addEventListener('mousedown', handleClickOutside);
+        return () => document.removeEventListener('mousedown', handleClickOutside);
+    }, [onClose]);
+    const handleSearchInputChange = (event) => {
+        const newFilterText = event.target.value;
+        setLocalFilterText(newFilterText);
+        if (onFilterChange) {
+            onFilterChange(newFilterText);
+        }
+    };
+    const handleSearchInputKeyDown = (event) => {
+        if (event.key === 'Escape') {
+            event.preventDefault();
+            event.stopPropagation();
+            if (onFilterChange) {
+                onFilterChange(''); // Clear the filter
+            }
+            if (onClose) {
+                onClose(); // Close the dropdown
+            }
+        }
+    };
     return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: `chat-dropdown`, "data-testid": "chat-dropdown" },
+        isDropdownFromButton && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "chat-dropdown-search" },
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("input", { ref: searchInputRef, type: "text", placeholder: "Search variables and rules...", value: localFilterText, onChange: handleSearchInputChange, onKeyDown: handleSearchInputKeyDown, className: "chat-dropdown-search-input" }))),
         react__WEBPACK_IMPORTED_MODULE_0___default().createElement("ul", { className: "chat-dropdown-list", "data-testid": "chat-dropdown-list" },
             filteredOptions.length === 0 && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("li", { className: "chat-dropdown-item", "data-testid": "chat-dropdown-empty-item" }, "No variables found")),
             filteredOptions.map((option, index) => {
-                let uniqueKey;
-                if (option.type === 'variable') {
-                    uniqueKey = option.variable.parent_df
-                        ? `${option.variable.parent_df}.${option.variable.variable_name}`
-                        : option.variable.variable_name;
-                }
-                else {
-                    uniqueKey = option.rule;
-                }
-                if (option.type === 'variable') {
-                    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(VariableDropdownItem, { key: uniqueKey, variable: option.variable, index: index, selectedIndex: selectedIndex, onSelect: () => onSelect(option) }));
-                }
-                else {
-                    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(RuleDropdownItem, { key: uniqueKey, rule: option.rule, index: index, selectedIndex: selectedIndex, onSelect: () => onSelect(option) }));
+                switch (option.type) {
+                    case 'variable': {
+                        const uniqueKey = option.variable.parent_df
+                            ? `${option.variable.parent_df}.${option.variable.variable_name}`
+                            : option.variable.variable_name;
+                        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatDropdownItems__WEBPACK_IMPORTED_MODULE_2__.VariableDropdownItem, { key: uniqueKey, variable: option.variable, index: index, selectedIndex: selectedIndex, onSelect: () => onSelect(option) }));
+                    }
+                    case 'file': {
+                        const uniqueKey = option.file.variable_name;
+                        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatDropdownItems__WEBPACK_IMPORTED_MODULE_2__.FileDropdownItem, { key: uniqueKey, file: option.file, index: index, selectedIndex: selectedIndex, onSelect: () => onSelect(option) }));
+                    }
+                    case 'rule': {
+                        const uniqueKey = option.rule;
+                        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatDropdownItems__WEBPACK_IMPORTED_MODULE_2__.RuleDropdownItem, { key: uniqueKey, rule: option.rule, index: index, selectedIndex: selectedIndex, onSelect: () => onSelect(option) }));
+                    }
+                    case 'db': {
+                        const uniqueKey = option.variable.variable_name;
+                        // You can replace VariableDropdownItem with a custom DBDropdownItem if you want
+                        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatDropdownItems__WEBPACK_IMPORTED_MODULE_2__.VariableDropdownItem, { key: uniqueKey, variable: option.variable, index: index, selectedIndex: selectedIndex, onSelect: () => onSelect(option) }));
+                    }
+                    default:
+                        return null;
                 }
             }))));
 };
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (ChatDropdown);
+
+
+/***/ }),
+
+/***/ "./lib/Extensions/AiChat/ChatMessage/ChatDropdownItems.js":
+/*!****************************************************************!*\
+  !*** ./lib/Extensions/AiChat/ChatMessage/ChatDropdownItems.js ***!
+  \****************************************************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   FileDropdownItem: () => (/* binding */ FileDropdownItem),
+/* harmony export */   RuleDropdownItem: () => (/* binding */ RuleDropdownItem),
+/* harmony export */   VariableDropdownItem: () => (/* binding */ VariableDropdownItem)
+/* 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_1__ = __webpack_require__(/*! ../../../utils/classNames */ "./lib/utils/classNames.js");
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+
 const VariableDropdownItem = ({ variable, index, selectedIndex, onSelect }) => {
     const getShortType = (type) => {
         var _a;
@@ -686,7 +830,7 @@ const VariableDropdownItem = ({ variable, index, selectedIndex, onSelect }) => {
         }
         return type;
     };
-    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("li", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_2__.classNames)("chat-dropdown-item", { selected: index === selectedIndex }), onClick: () => onSelect(variable.variable_name, variable.parent_df), "data-testid": `chat-dropdown-item-${variable.variable_name}` },
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("li", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_1__.classNames)("chat-dropdown-item", { selected: index === selectedIndex }), onClick: () => onSelect(variable.variable_name, variable.parent_df), "data-testid": `chat-dropdown-item-${variable.variable_name}` },
         react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "chat-dropdown-item-type", title: getShortType(variable.type), "data-testid": `chat-dropdown-item-type-${variable.variable_name}` }, getShortType(variable.type)),
         react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "chat-dropdown-item-name", title: variable.variable_name, "data-testid": `chat-dropdown-item-name-${variable.variable_name}`, ref: (el) => {
                 // Show full text on hover if the text is too long
@@ -696,8 +840,25 @@ const VariableDropdownItem = ({ variable, index, selectedIndex, onSelect }) => {
             } }, variable.variable_name),
         variable.parent_df && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "chat-dropdown-item-parent-df" }, variable.parent_df))));
 };
+const FileDropdownItem = ({ file, index, selectedIndex, onSelect }) => {
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("li", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_1__.classNames)("chat-dropdown-item", { selected: index === selectedIndex }), onClick: () => onSelect(file), "data-testid": `chat-dropdown-item-${file.variable_name}` },
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "chat-dropdown-item-type", title: file.type, "data-testid": `chat-dropdown-item-type-${file.variable_name}` }, file.type),
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "chat-dropdown-item-name", title: file.variable_name, "data-testid": `chat-dropdown-item-name-${file.variable_name}`, ref: (el) => {
+                // Show full text on hover if the text is too long
+                if (el) {
+                    el.title = el.scrollWidth > el.clientWidth ? file.variable_name : '';
+                }
+            } }, file.variable_name)));
+};
 const RuleDropdownItem = ({ rule, index, selectedIndex, onSelect }) => {
-    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("li", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_2__.classNames)("chat-dropdown-item", { selected: index === selectedIndex }), onClick: () => onSelect(rule), "data-testid": `chat-dropdown-item-${rule}` }, rule));
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("li", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_1__.classNames)("chat-dropdown-item", { selected: index === selectedIndex }), onClick: () => onSelect(rule), "data-testid": `chat-dropdown-item-${rule}` },
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "chat-dropdown-item-type", title: "rule", "data-testid": `chat-dropdown-item-type-${rule}` }, "rule"),
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "chat-dropdown-item-name", title: rule, "data-testid": `chat-dropdown-item-name-${rule}`, ref: (el) => {
+                // Show full text on hover if the text is too long
+                if (el) {
+                    el.title = el.scrollWidth > el.clientWidth ? rule : '';
+                }
+            } }, rule)));
 };
 
 
@@ -716,13 +877,14 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
 /* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
 /* harmony import */ var _utils_classNames__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../../utils/classNames */ "./lib/utils/classNames.js");
-/* harmony import */ var _ChatDropdown__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ./ChatDropdown */ "./lib/Extensions/AiChat/ChatMessage/ChatDropdown.js");
+/* harmony import */ var _ChatDropdown__WEBPACK_IMPORTED_MODULE_9__ = __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 _PythonCode__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ./PythonCode */ "./lib/Extensions/AiChat/ChatMessage/PythonCode.js");
+/* harmony import */ var _PythonCode__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ./PythonCode */ "./lib/Extensions/AiChat/ChatMessage/PythonCode.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_4__ = __webpack_require__(/*! ../../../hooks/useDebouncedFunction */ "./lib/hooks/useDebouncedFunction.js");
-/* harmony import */ var _components_SelectedContextContainer__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../../../components/SelectedContextContainer */ "./lib/components/SelectedContextContainer.js");
+/* harmony import */ var _components_SelectedContextContainer__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ../../../components/SelectedContextContainer */ "./lib/components/SelectedContextContainer.js");
+/* harmony import */ var _components_DatabaseButton__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ../../../components/DatabaseButton */ "./lib/components/DatabaseButton.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -736,7 +898,8 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
-const ChatInput = ({ initialContent, placeholder, onSave, onCancel, isEditing, contextManager, notebookTracker, renderMimeRegistry, displayActiveCellCode = true, agentModeEnabled = false, }) => {
+
+const ChatInput = ({ app, initialContent, placeholder, onSave, onCancel, isEditing, contextManager, notebookTracker, renderMimeRegistry, displayActiveCellCode = true, agentModeEnabled = false, }) => {
     var _a;
     const [input, setInput] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(initialContent);
     const [expandedVariables, setExpandedVariables] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)([]);
@@ -745,7 +908,8 @@ const ChatInput = ({ initialContent, placeholder, onSave, onCancel, isEditing, c
     const [activeCellID, setActiveCellID] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)((0,_utils_notebook__WEBPACK_IMPORTED_MODULE_3__.getActiveCellID)(notebookTracker));
     const [isDropdownVisible, setDropdownVisible] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
     const [dropdownFilter, setDropdownFilter] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)('');
-    const [selectedRules, setSelectedRules] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)([]);
+    const [additionalContext, setAdditionalContext] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)([]);
+    const [isDropdownFromButton, setIsDropdownFromButton] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(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_4__.useDebouncedFunction)((newID) => {
@@ -788,13 +952,49 @@ const ChatInput = ({ initialContent, placeholder, onSave, onCancel, isEditing, c
             const query = currentWord.slice(1);
             setDropdownFilter(query);
             setDropdownVisible(true);
+            setIsDropdownFromButton(false);
         }
         else {
             setDropdownVisible(false);
             setDropdownFilter('');
+            setIsDropdownFromButton(false);
         }
     };
     const handleOptionSelect = (option) => {
+        if (isDropdownFromButton) {
+            // When triggered by "Add Context" button, add to SelectedContextContainer
+            if (option.type === 'variable') {
+                // For variables, we'll add them as a special context type
+                const contextName = option.variable.parent_df
+                    ? `${option.variable.parent_df}.${option.variable.variable_name}`
+                    : option.variable.variable_name;
+                setAdditionalContext(prev => [...prev, { type: 'variable', value: contextName }]);
+            }
+            else if (option.type === 'file') {
+                setAdditionalContext(prev => [...prev, { type: 'file', value: option.file.variable_name }]);
+            }
+            else if (option.type === 'rule') {
+                setAdditionalContext(prev => [...prev, { type: 'rule', value: option.rule }]);
+            }
+            else if (option.type === 'db') {
+                setAdditionalContext(prev => [
+                    ...prev,
+                    {
+                        type: 'db',
+                        value: option.variable.value,
+                        display: option.variable.variable_name
+                    }
+                ]);
+            }
+            setDropdownVisible(false);
+            // Use setTimeout to ensure this happens after React's state update cycle
+            setTimeout(() => {
+                var _a;
+                (_a = textAreaRef.current) === null || _a === void 0 ? void 0 : _a.focus();
+            }, 0);
+            return;
+        }
+        // Original behavior for @ dropdown - add to text input
         const textarea = textAreaRef.current;
         if (!textarea)
             return;
@@ -811,13 +1011,26 @@ const ChatInput = ({ initialContent, placeholder, onSave, onCancel, isEditing, c
                 contextChatRepresentation = `\`${option.variable.variable_name}\``;
             }
         }
+        else if (option.type === 'file') {
+            // For files, add them as both back-ticked elements and the additional context container
+            contextChatRepresentation = `\`${option.file.variable_name}\``;
+            setAdditionalContext([...additionalContext, { type: 'file', value: option.file.variable_name }]);
+        }
         else if (option.type === 'rule') {
             // We don't add the rule as an back ticked element in the chat input, 
             // and instead just add it as plain text because we also add it as 
             // a context container above the chat input and we want the user to 
             // delete the context from there if they want to. 
             contextChatRepresentation = option.rule;
-            setSelectedRules([...selectedRules, option.rule]);
+            setAdditionalContext([...additionalContext, { type: 'rule', value: option.rule }]);
+        }
+        else if (option.type === 'db') {
+            // For databases, add them as back-ticked elements
+            contextChatRepresentation = `\`${option.variable.variable_name}\``;
+            setAdditionalContext([
+                ...additionalContext,
+                { type: 'db', value: option.variable.value, display: option.variable.variable_name }
+            ]);
         }
         const newValue = input.slice(0, atIndex) +
             contextChatRepresentation +
@@ -834,6 +1047,11 @@ const ChatInput = ({ initialContent, placeholder, onSave, onCancel, isEditing, c
             }
         }, 0);
     };
+    const handleDropdownClose = () => {
+        setDropdownVisible(false);
+        setDropdownFilter('');
+        setIsDropdownFromButton(false);
+    };
     // Update the expandedVariables arr when the variable manager changes
     (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
         const expandedVariables = [
@@ -867,12 +1085,21 @@ const ChatInput = ({ initialContent, placeholder, onSave, onCancel, isEditing, c
     return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_5__.classNames)("chat-input-container"), onFocus: () => setIsFocused(true), onBlur: () => {
             setIsFocused(false);
         } },
-        selectedRules.length > 0 && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'context-container' }, selectedRules.map((rule) => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_SelectedContextContainer__WEBPACK_IMPORTED_MODULE_6__["default"], { key: rule, title: rule, onRemove: () => setSelectedRules(selectedRules.filter((r) => r !== rule)) }))))),
         displayActiveCellCode && activeCellCodePreview.length > 0 && !agentModeEnabled
             && (isFocused || input.length > 0)
             && react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'active-cell-preview-container', "data-testid": 'active-cell-preview-container' },
                 react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'code-block-container' },
-                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_PythonCode__WEBPACK_IMPORTED_MODULE_7__["default"], { code: activeCellCodePreview, renderMimeRegistry: renderMimeRegistry }))),
+                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_PythonCode__WEBPACK_IMPORTED_MODULE_6__["default"], { code: activeCellCodePreview, renderMimeRegistry: renderMimeRegistry }))),
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'context-container' },
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_DatabaseButton__WEBPACK_IMPORTED_MODULE_7__["default"], { app: app }),
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { className: "context-button", onClick: () => {
+                    var _a;
+                    setDropdownVisible(true);
+                    setDropdownFilter('');
+                    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_8__["default"], { key: `${context.type}-${context.value}-${index}`, title: context.type === 'db' && context.display ? context.display : context.value, type: context.type, onRemove: () => setAdditionalContext(additionalContext.filter((_, i) => i !== index)) })))),
         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_5__.classNames)("message", "message-user", 'chat-input', { "agent-mode": agentModeEnabled }), placeholder: placeholder, value: input, onChange: handleInputChange, onKeyDown: (e) => {
                     // If dropdown is visible, only handle escape to close it
@@ -888,10 +1115,10 @@ const ChatInput = ({ initialContent, placeholder, onSave, onCancel, isEditing, c
                     if (e.key === 'Enter' && !e.shiftKey) {
                         e.preventDefault();
                         adjustHeight(true);
-                        onSave(input, undefined, selectedRules);
+                        onSave(input, undefined, additionalContext.map(ctx => ctx.type === 'db' ? { type: ctx.type, value: ctx.value } : ctx));
                         // Reset
                         setInput('');
-                        setSelectedRules([]);
+                        setAdditionalContext([]);
                         setIsFocused(false);
                     }
                     // Escape key cancels editing
@@ -902,10 +1129,10 @@ const ChatInput = ({ initialContent, placeholder, onSave, onCancel, isEditing, c
                         }
                     }
                 } }),
-            isDropdownVisible && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatDropdown__WEBPACK_IMPORTED_MODULE_8__["default"], { options: expandedVariables, onSelect: handleOptionSelect, filterText: dropdownFilter }))),
+            isDropdownVisible && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatDropdown__WEBPACK_IMPORTED_MODULE_9__["default"], { options: expandedVariables, onSelect: handleOptionSelect, filterText: dropdownFilter, isDropdownFromButton: isDropdownFromButton, onFilterChange: setDropdownFilter, onClose: handleDropdownClose }))),
         isEditing &&
             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "message-edit-buttons" },
-                react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { onClick: () => onSave(input, undefined, selectedRules) }, "Save"),
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { onClick: () => onSave(input, undefined, additionalContext.map(ctx => ctx.type === 'db' ? { type: ctx.type, value: ctx.value } : ctx)) }, "Save"),
                 react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { onClick: onCancel }, "Cancel"))));
 };
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (ChatInput);
@@ -929,9 +1156,9 @@ __webpack_require__.r(__webpack_exports__);
 /* 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_14__ = __webpack_require__(/*! ./MarkdownBlock */ "./lib/Extensions/AiChat/ChatMessage/MarkdownBlock.js");
+/* harmony import */ var _MarkdownBlock__WEBPACK_IMPORTED_MODULE_15__ = __webpack_require__(/*! ./MarkdownBlock */ "./lib/Extensions/AiChat/ChatMessage/MarkdownBlock.js");
 /* harmony import */ var _utils_strings__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../../../utils/strings */ "./lib/utils/strings.js");
-/* harmony import */ var _icons_Pencil__WEBPACK_IMPORTED_MODULE_15__ = __webpack_require__(/*! ../../../icons/Pencil */ "./lib/icons/Pencil.js");
+/* harmony import */ var _icons_Pencil__WEBPACK_IMPORTED_MODULE_17__ = __webpack_require__(/*! ../../../icons/Pencil */ "./lib/icons/Pencil.js");
 /* harmony import */ var _ChatInput__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ./ChatInput */ "./lib/Extensions/AiChat/ChatMessage/ChatInput.js");
 /* harmony import */ var _components_TextAndIconButton__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ../../../components/TextAndIconButton */ "./lib/components/TextAndIconButton.js");
 /* harmony import */ var _icons_PlayButtonIcon__WEBPACK_IMPORTED_MODULE_10__ = __webpack_require__(/*! ../../../icons/PlayButtonIcon */ "./lib/icons/PlayButtonIcon.js");
@@ -940,7 +1167,9 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony import */ var _components_TextButton__WEBPACK_IMPORTED_MODULE_13__ = __webpack_require__(/*! ../../../components/TextButton */ "./lib/components/TextButton.js");
 /* harmony import */ var _style_ChatMessage_css__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../../../style/ChatMessage.css */ "./style/ChatMessage.css");
 /* harmony import */ var _style_MarkdownMessage_css__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../../../../style/MarkdownMessage.css */ "./style/MarkdownMessage.css");
-/* harmony import */ var _components_AgentComponents_GetCellOutputToolUI__WEBPACK_IMPORTED_MODULE_16__ = __webpack_require__(/*! ../../../components/AgentComponents/GetCellOutputToolUI */ "./lib/components/AgentComponents/GetCellOutputToolUI.js");
+/* harmony import */ var _components_AgentComponents_GetCellOutputToolUI__WEBPACK_IMPORTED_MODULE_18__ = __webpack_require__(/*! ../../../components/AgentComponents/GetCellOutputToolUI */ "./lib/components/AgentComponents/GetCellOutputToolUI.js");
+/* harmony import */ var _components_AgentComponents_AssumptionToolUI__WEBPACK_IMPORTED_MODULE_16__ = __webpack_require__(/*! ../../../components/AgentComponents/AssumptionToolUI */ "./lib/components/AgentComponents/AssumptionToolUI.js");
+/* harmony import */ var _components_SelectedContextContainer__WEBPACK_IMPORTED_MODULE_14__ = __webpack_require__(/*! ../../../components/SelectedContextContainer */ "./lib/components/SelectedContextContainer.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -962,7 +1191,9 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
-const ChatMessage = ({ message, messageType, promptType, agentResponse, messageIndex, mitoAIConnectionError, mitoAIConnectionErrorType, notebookTracker, renderMimeRegistry, isLastAiMessage, isLastMessage, operatingSystem, previewAICode, acceptAICode, rejectAICode, onUpdateMessage, contextManager, codeReviewStatus, setNextSteps, agentModeEnabled, }) => {
+
+
+const ChatMessage = ({ app, message, messageType, promptType, agentResponse, messageIndex, mitoAIConnectionError, mitoAIConnectionErrorType, notebookTracker, renderMimeRegistry, isLastAiMessage, isLastMessage, operatingSystem, previewAICode, acceptAICode, rejectAICode, onUpdateMessage, contextManager, codeReviewStatus, setNextSteps, agentModeEnabled, additionalContext, }) => {
     const [isEditing, setIsEditing] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
     if (message.role !== 'user' && message.role !== 'assistant') {
         return null;
@@ -973,14 +1204,16 @@ const ChatMessage = ({ message, messageType, promptType, agentResponse, messageI
     const handleEditClick = () => {
         setIsEditing(true);
     };
-    const handleSave = (content) => {
+    const handleSave = (content, _index, _selectedRules, _additionalContext) => {
         onUpdateMessage(messageIndex, content, messageType);
         setIsEditing(false);
     };
     const handleCancel = () => {
         setIsEditing(false);
     };
-    if (isLastMessage && (agentResponse === null || agentResponse === void 0 ? void 0 : agentResponse.type) === 'finished_task' && agentResponse.next_steps && agentResponse.next_steps.length > 0) {
+    if (isLastMessage && (agentResponse === null || agentResponse === void 0 ? void 0 : agentResponse.type) === 'finished_task' &&
+        agentResponse.next_steps &&
+        agentResponse.next_steps.length > 0) {
         /*
         We only want to set the next steps if the message is the last message in the chat.
         This is because the next steps are only available after the agent has finished its task.
@@ -990,7 +1223,7 @@ const ChatMessage = ({ message, messageType, promptType, agentResponse, messageI
         setNextSteps(agentResponse.next_steps);
     }
     if (isEditing) {
-        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatInput__WEBPACK_IMPORTED_MODULE_4__["default"], { initialContent: message.content.replace(/```[\s\S]*?```/g, '').trim(), placeholder: "Edit your message", onSave: handleSave, onCancel: handleCancel, isEditing: isEditing, contextManager: contextManager, notebookTracker: notebookTracker, renderMimeRegistry: renderMimeRegistry, displayActiveCellCode: true, agentModeEnabled: false }));
+        return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_ChatInput__WEBPACK_IMPORTED_MODULE_4__["default"], { app: app, initialContent: message.content.replace(/```[\s\S]*?```/g, '').trim(), placeholder: "Edit your message", onSave: handleSave, onCancel: handleCancel, isEditing: isEditing, contextManager: contextManager, notebookTracker: notebookTracker, renderMimeRegistry: renderMimeRegistry, displayActiveCellCode: true, agentModeEnabled: false }));
     }
     if (mitoAIConnectionError) {
         return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_5__.classNames)("message") },
@@ -1004,6 +1237,7 @@ const ChatMessage = ({ message, messageType, promptType, agentResponse, messageI
     // Users end up applying the code in the middle of streaming and it gets very confusing
     // very quickly for users. 
     let isCodeComplete = false;
+    console.log(messageContentParts);
     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;
@@ -1021,7 +1255,9 @@ const ChatMessage = ({ message, messageType, promptType, agentResponse, messageI
                         isLastAiMessage && isCodeComplete && codeReviewStatus === 'codeCellPreview' &&
                             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: 'chat-message-buttons' },
                                 react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_TextButton__WEBPACK_IMPORTED_MODULE_13__["default"], { onClick: () => { acceptAICode(); }, text: `Accept code ${operatingSystem === 'mac' ? '⌘Y' : 'Ctrl+Y'}`, title: 'Accept the Ai generated code', variant: 'green', width: 'fit-contents' }),
-                                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_TextButton__WEBPACK_IMPORTED_MODULE_13__["default"], { onClick: () => { rejectAICode(); }, text: `Reject code ${operatingSystem === 'mac' ? '⌘U' : 'Ctrl+U'}`, title: 'Reject the Ai generated code and revert to the previous version of the code cell', variant: 'red', width: 'fit-contents' }))));
+                                react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_TextButton__WEBPACK_IMPORTED_MODULE_13__["default"], { onClick: () => { rejectAICode(); }, text: `Reject code ${operatingSystem === 'mac' ? '⌘U' : 'Ctrl+U'}`, title: 'Reject the Ai generated code and revert to the previous version of the code cell', variant: 'red', width: 'fit-contents' })),
+                        message.role === 'user' && additionalContext && additionalContext.length > 0 &&
+                            react__WEBPACK_IMPORTED_MODULE_0___default().createElement((react__WEBPACK_IMPORTED_MODULE_0___default().Fragment), null, additionalContext.map((context, index) => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_SelectedContextContainer__WEBPACK_IMPORTED_MODULE_14__["default"], { key: `${context.type}-${context.value}-${index}`, title: `${context.type.charAt(0).toUpperCase() + context.type.slice(1)}: ${context.value}`, type: context.type, onRemove: () => { } }))))));
                 }
                 else {
                     // Return null for empty code blocks
@@ -1037,15 +1273,17 @@ const ChatMessage = ({ message, messageType, promptType, agentResponse, messageI
                             }
                         } }, message.role === 'user' && promptType === 'smartDebug' ? (
                     /* Use a pre tag to preserve the newline and indentation of the error message */
-                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("pre", { className: "chat-taskpane-smart-debug-error-message" }, messagePart)) : (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_MarkdownBlock__WEBPACK_IMPORTED_MODULE_14__["default"], { markdown: messagePart, renderMimeRegistry: renderMimeRegistry, notebookTracker: notebookTracker })))));
+                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("pre", { className: "chat-taskpane-smart-debug-error-message" }, messagePart)) : (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_MarkdownBlock__WEBPACK_IMPORTED_MODULE_15__["default"], { markdown: messagePart, renderMimeRegistry: renderMimeRegistry, notebookTracker: notebookTracker }))),
+                    (agentResponse === null || agentResponse === void 0 ? void 0 : agentResponse.analysis_assumptions) &&
+                        react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AgentComponents_AssumptionToolUI__WEBPACK_IMPORTED_MODULE_16__["default"], { assumptions: agentResponse.analysis_assumptions })));
             }
         }),
         editable &&
             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "message-action-buttons" },
                 react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { className: "message-start-editing-button", onClick: handleEditClick, title: "Edit message" },
-                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_Pencil__WEBPACK_IMPORTED_MODULE_15__["default"], null))),
+                    react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_Pencil__WEBPACK_IMPORTED_MODULE_17__["default"], null))),
         (agentResponse === null || agentResponse === void 0 ? void 0 : agentResponse.type) === 'get_cell_output' &&
-            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AgentComponents_GetCellOutputToolUI__WEBPACK_IMPORTED_MODULE_16__["default"], null)));
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_components_AgentComponents_GetCellOutputToolUI__WEBPACK_IMPORTED_MODULE_18__["default"], null)));
 };
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (ChatMessage);
 
@@ -1531,39 +1769,40 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony import */ var _jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_2___default = /*#__PURE__*/__webpack_require__.n(_jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_2__);
 /* harmony import */ var _lumino_coreutils__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! @lumino/coreutils */ "webpack/sharing/consume/default/@lumino/coreutils");
 /* harmony import */ var _lumino_coreutils__WEBPACK_IMPORTED_MODULE_3___default = /*#__PURE__*/__webpack_require__.n(_lumino_coreutils__WEBPACK_IMPORTED_MODULE_3__);
-/* harmony import */ var _commands__WEBPACK_IMPORTED_MODULE_18__ = __webpack_require__(/*! ../../commands */ "./lib/commands.js");
-/* harmony import */ var _components_AgentComponents_ErrorFixupToolUI__WEBPACK_IMPORTED_MODULE_27__ = __webpack_require__(/*! ../../components/AgentComponents/ErrorFixupToolUI */ "./lib/components/AgentComponents/ErrorFixupToolUI.js");
-/* harmony import */ var _components_DropdownMenu__WEBPACK_IMPORTED_MODULE_23__ = __webpack_require__(/*! ../../components/DropdownMenu */ "./lib/components/DropdownMenu.js");
-/* harmony import */ var _components_IconButton__WEBPACK_IMPORTED_MODULE_21__ = __webpack_require__(/*! ../../components/IconButton */ "./lib/components/IconButton.js");
-/* harmony import */ var _components_LoadingCircle__WEBPACK_IMPORTED_MODULE_36__ = __webpack_require__(/*! ../../components/LoadingCircle */ "./lib/components/LoadingCircle.js");
-/* harmony import */ var _components_LoadingDots__WEBPACK_IMPORTED_MODULE_29__ = __webpack_require__(/*! ../../components/LoadingDots */ "./lib/components/LoadingDots.js");
+/* harmony import */ var _commands__WEBPACK_IMPORTED_MODULE_19__ = __webpack_require__(/*! ../../commands */ "./lib/commands.js");
+/* harmony import */ var _components_AgentComponents_ErrorFixupToolUI__WEBPACK_IMPORTED_MODULE_28__ = __webpack_require__(/*! ../../components/AgentComponents/ErrorFixupToolUI */ "./lib/components/AgentComponents/ErrorFixupToolUI.js");
+/* harmony import */ var _components_DropdownMenu__WEBPACK_IMPORTED_MODULE_24__ = __webpack_require__(/*! ../../components/DropdownMenu */ "./lib/components/DropdownMenu.js");
+/* harmony import */ var _components_IconButton__WEBPACK_IMPORTED_MODULE_22__ = __webpack_require__(/*! ../../components/IconButton */ "./lib/components/IconButton.js");
+/* harmony import */ var _components_LoadingCircle__WEBPACK_IMPORTED_MODULE_37__ = __webpack_require__(/*! ../../components/LoadingCircle */ "./lib/components/LoadingCircle.js");
+/* harmony import */ var _components_LoadingDots__WEBPACK_IMPORTED_MODULE_30__ = __webpack_require__(/*! ../../components/LoadingDots */ "./lib/components/LoadingDots.js");
 /* harmony import */ var _components_ModelSelector__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ../../components/ModelSelector */ "./lib/components/ModelSelector.js");
-/* harmony import */ var _components_NextStepsPills__WEBPACK_IMPORTED_MODULE_33__ = __webpack_require__(/*! ../../components/NextStepsPills */ "./lib/components/NextStepsPills.js");
-/* harmony import */ var _components_TextAndIconButton__WEBPACK_IMPORTED_MODULE_30__ = __webpack_require__(/*! ../../components/TextAndIconButton */ "./lib/components/TextAndIconButton.js");
-/* harmony import */ var _components_ToggleButton__WEBPACK_IMPORTED_MODULE_35__ = __webpack_require__(/*! ../../components/ToggleButton */ "./lib/components/ToggleButton.js");
-/* harmony import */ var _icons__WEBPACK_IMPORTED_MODULE_24__ = __webpack_require__(/*! ../../icons */ "./lib/icons/index.js");
-/* harmony import */ var _icons_MitoLogo__WEBPACK_IMPORTED_MODULE_25__ = __webpack_require__(/*! ../../icons/MitoLogo */ "./lib/icons/MitoLogo.js");
-/* harmony import */ var _icons_UndoIcon__WEBPACK_IMPORTED_MODULE_31__ = __webpack_require__(/*! ../../icons/UndoIcon */ "./lib/icons/UndoIcon.js");
-/* harmony import */ var _utils_agentActions__WEBPACK_IMPORTED_MODULE_16__ = __webpack_require__(/*! ../../utils/agentActions */ "./lib/utils/agentActions.js");
-/* harmony import */ var _utils_blacklistedWords__WEBPACK_IMPORTED_MODULE_15__ = __webpack_require__(/*! ../../utils/blacklistedWords */ "./lib/utils/blacklistedWords.js");
-/* harmony import */ var _utils_checkpoint__WEBPACK_IMPORTED_MODULE_13__ = __webpack_require__(/*! ../../utils/checkpoint */ "./lib/utils/checkpoint.js");
-/* harmony import */ var _utils_chatHistory__WEBPACK_IMPORTED_MODULE_20__ = __webpack_require__(/*! ../../utils/chatHistory */ "./lib/utils/chatHistory.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 _components_NextStepsPills__WEBPACK_IMPORTED_MODULE_34__ = __webpack_require__(/*! ../../components/NextStepsPills */ "./lib/components/NextStepsPills.js");
+/* harmony import */ var _components_TextAndIconButton__WEBPACK_IMPORTED_MODULE_31__ = __webpack_require__(/*! ../../components/TextAndIconButton */ "./lib/components/TextAndIconButton.js");
+/* harmony import */ var _components_ToggleButton__WEBPACK_IMPORTED_MODULE_36__ = __webpack_require__(/*! ../../components/ToggleButton */ "./lib/components/ToggleButton.js");
+/* harmony import */ var _icons__WEBPACK_IMPORTED_MODULE_25__ = __webpack_require__(/*! ../../icons */ "./lib/icons/index.js");
+/* harmony import */ var _icons_MitoLogo__WEBPACK_IMPORTED_MODULE_26__ = __webpack_require__(/*! ../../icons/MitoLogo */ "./lib/icons/MitoLogo.js");
+/* harmony import */ var _icons_UndoIcon__WEBPACK_IMPORTED_MODULE_32__ = __webpack_require__(/*! ../../icons/UndoIcon */ "./lib/icons/UndoIcon.js");
+/* harmony import */ var _utils_agentActions__WEBPACK_IMPORTED_MODULE_17__ = __webpack_require__(/*! ../../utils/agentActions */ "./lib/utils/agentActions.js");
+/* harmony import */ var _utils_blacklistedWords__WEBPACK_IMPORTED_MODULE_16__ = __webpack_require__(/*! ../../utils/blacklistedWords */ "./lib/utils/blacklistedWords.js");
+/* harmony import */ var _utils_checkpoint__WEBPACK_IMPORTED_MODULE_14__ = __webpack_require__(/*! ../../utils/checkpoint */ "./lib/utils/checkpoint.js");
+/* harmony import */ var _utils_chatHistory__WEBPACK_IMPORTED_MODULE_21__ = __webpack_require__(/*! ../../utils/chatHistory */ "./lib/utils/chatHistory.js");
+/* harmony import */ var _utils_codeDiff__WEBPACK_IMPORTED_MODULE_18__ = __webpack_require__(/*! ../../utils/codeDiff */ "./lib/utils/codeDiff.js");
+/* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_13__ = __webpack_require__(/*! ../../utils/notebook */ "./lib/utils/notebook.js");
 /* harmony import */ var _utils_scroll__WEBPACK_IMPORTED_MODULE_11__ = __webpack_require__(/*! ../../utils/scroll */ "./lib/utils/scroll.js");
-/* harmony import */ var _utils_strings__WEBPACK_IMPORTED_MODULE_14__ = __webpack_require__(/*! ../../utils/strings */ "./lib/utils/strings.js");
+/* harmony import */ var _utils_strings__WEBPACK_IMPORTED_MODULE_15__ = __webpack_require__(/*! ../../utils/strings */ "./lib/utils/strings.js");
 /* harmony import */ var _utils_waitForNotebookReady__WEBPACK_IMPORTED_MODULE_10__ = __webpack_require__(/*! ../../utils/waitForNotebookReady */ "./lib/utils/waitForNotebookReady.js");
-/* harmony import */ var _SettingsManager_SettingsManagerPlugin__WEBPACK_IMPORTED_MODULE_22__ = __webpack_require__(/*! ../SettingsManager/SettingsManagerPlugin */ "./lib/Extensions/SettingsManager/SettingsManagerPlugin.js");
-/* harmony import */ var _CTACarousel__WEBPACK_IMPORTED_MODULE_26__ = __webpack_require__(/*! ./CTACarousel */ "./lib/Extensions/AiChat/CTACarousel.js");
-/* harmony import */ var _CodeDiffDisplay__WEBPACK_IMPORTED_MODULE_19__ = __webpack_require__(/*! ./CodeDiffDisplay */ "./lib/Extensions/AiChat/CodeDiffDisplay.js");
+/* harmony import */ var _SettingsManager_SettingsManagerPlugin__WEBPACK_IMPORTED_MODULE_23__ = __webpack_require__(/*! ../SettingsManager/SettingsManagerPlugin */ "./lib/Extensions/SettingsManager/SettingsManagerPlugin.js");
+/* harmony import */ var _CTACarousel__WEBPACK_IMPORTED_MODULE_27__ = __webpack_require__(/*! ./CTACarousel */ "./lib/Extensions/AiChat/CTACarousel.js");
+/* harmony import */ var _CodeDiffDisplay__WEBPACK_IMPORTED_MODULE_20__ = __webpack_require__(/*! ./CodeDiffDisplay */ "./lib/Extensions/AiChat/CodeDiffDisplay.js");
 /* harmony import */ var _FirstMessage__WEBPACK_IMPORTED_MODULE_9__ = __webpack_require__(/*! ./FirstMessage */ "./lib/Extensions/AiChat/FirstMessage.js");
-/* harmony import */ var _ChatMessage_ChatInput__WEBPACK_IMPORTED_MODULE_34__ = __webpack_require__(/*! ./ChatMessage/ChatInput */ "./lib/Extensions/AiChat/ChatMessage/ChatInput.js");
-/* harmony import */ var _ChatMessage_ChatMessage__WEBPACK_IMPORTED_MODULE_28__ = __webpack_require__(/*! ./ChatMessage/ChatMessage */ "./lib/Extensions/AiChat/ChatMessage/ChatMessage.js");
-/* harmony import */ var _ChatMessage_ScrollableSuggestions__WEBPACK_IMPORTED_MODULE_32__ = __webpack_require__(/*! ./ChatMessage/ScrollableSuggestions */ "./lib/Extensions/AiChat/ChatMessage/ScrollableSuggestions.js");
+/* harmony import */ var _ChatMessage_ChatInput__WEBPACK_IMPORTED_MODULE_35__ = __webpack_require__(/*! ./ChatMessage/ChatInput */ "./lib/Extensions/AiChat/ChatMessage/ChatInput.js");
+/* harmony import */ var _ChatMessage_ChatMessage__WEBPACK_IMPORTED_MODULE_29__ = __webpack_require__(/*! ./ChatMessage/ChatMessage */ "./lib/Extensions/AiChat/ChatMessage/ChatMessage.js");
+/* harmony import */ var _ChatMessage_ScrollableSuggestions__WEBPACK_IMPORTED_MODULE_33__ = __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 _style_button_css__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../../../style/button.css */ "./style/button.css");
 /* harmony import */ var _style_ChatTaskpane_css__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../../style/ChatTaskpane.css */ "./style/ChatTaskpane.css");
 /* harmony import */ var _style_TextButton_css__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ../../../style/TextButton.css */ "./style/TextButton.css");
+/* harmony import */ var _utils__WEBPACK_IMPORTED_MODULE_12__ = __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.
@@ -1613,6 +1852,7 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
+
 const AGENT_EXECUTION_DEPTH_LIMIT = 20;
 const getDefaultChatHistoryManager = (notebookTracker, contextManager) => {
     const chatHistoryManager = new _ChatHistoryManager__WEBPACK_IMPORTED_MODULE_7__.ChatHistoryManager(contextManager, notebookTracker);
@@ -1982,7 +2222,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
         await _sendMessageAndSaveResponse(explainCompletionRequest, newChatHistoryManager);
         // Step 3: No post processing step needed for explaining code. 
     };
-    const sendAgentExecutionMessage = async (input, messageIndex, sendActiveCellOutput = false, selectedRules) => {
+    const sendAgentExecutionMessage = async (input, messageIndex, sendCellIDOutput = undefined, additionalContext) => {
         // Step 0: reset the state for a new message
         resetForNewMessage();
         // Step 1: Add the user's message to the chat history
@@ -1991,16 +2231,11 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
             // Drop all of the messages starting at the message index
             newChatHistoryManager.dropMessagesStartingAtIndex(messageIndex);
         }
-        const agentExecutionMetadata = newChatHistoryManager.addAgentExecutionMessage(activeThreadIdRef.current, input, selectedRules);
+        const agentExecutionMetadata = newChatHistoryManager.addAgentExecutionMessage(activeThreadIdRef.current, input, additionalContext);
         if (messageIndex !== undefined) {
             agentExecutionMetadata.index = messageIndex;
         }
-        if (sendActiveCellOutput) {
-            const activeCellOutput = await (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_12__.getActiveCellOutput)(notebookTracker);
-            if (activeCellOutput !== undefined) {
-                agentExecutionMetadata.base64EncodedActiveCellOutput = activeCellOutput;
-            }
-        }
+        agentExecutionMetadata.base64EncodedActiveCellOutput = await (0,_utils__WEBPACK_IMPORTED_MODULE_12__.getBase64EncodedCellOutput)(notebookTracker, sendCellIDOutput);
         setChatHistoryManager(newChatHistoryManager);
         setLoadingAIResponse(true);
         // Step 2: Send the message to the AI
@@ -2015,7 +2250,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
     /*
         Send whatever message is currently in the chat input
     */
-    const sendChatInputMessage = async (input, messageIndex, selectedRules) => {
+    const sendChatInputMessage = async (input, messageIndex, additionalContext) => {
         // Step 0: reset the state for a new message
         resetForNewMessage();
         // Enable follow mode when user sends a new message
@@ -2026,7 +2261,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
             // Drop all of the messages starting at the message index
             newChatHistoryManager.dropMessagesStartingAtIndex(messageIndex);
         }
-        const chatMessageMetadata = await newChatHistoryManager.addChatInputMessage(input, activeThreadIdRef.current, messageIndex, selectedRules);
+        const chatMessageMetadata = await newChatHistoryManager.addChatInputMessage(input, activeThreadIdRef.current, messageIndex, additionalContext);
         setChatHistoryManager(newChatHistoryManager);
         setLoadingAIResponse(true);
         // Yield control briefly to allow React to re-render the UI
@@ -2036,7 +2271,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_12__.getActiveCellOutput)(notebookTracker);
+        const activeCellOutput = await (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_13__.getActiveCellOutput)(notebookTracker);
         if (activeCellOutput !== undefined) {
             chatMessageMetadata.base64EncodedActiveCellOutput = activeCellOutput;
         }
@@ -2194,8 +2429,8 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
         // Reset agent to idle state
         setAgentExecutionStatus('idle');
     };
-    const startAgentExecution = async (input, messageIndex, selectedRules) => {
-        await (0,_utils_checkpoint__WEBPACK_IMPORTED_MODULE_13__.createCheckpoint)(app, setHasCheckpoint);
+    const startAgentExecution = async (input, messageIndex, additionalContext) => {
+        await (0,_utils_checkpoint__WEBPACK_IMPORTED_MODULE_14__.createCheckpoint)(app, setHasCheckpoint);
         setAgentExecutionStatus('working');
         // Enable follow mode when user starts agent execution
         setAutoScrollFollowMode(true);
@@ -2203,7 +2438,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
         shouldContinueAgentExecution.current = true;
         let isAgentFinished = false;
         let agentExecutionDepth = 1;
-        let sendActiveCellOutput = false;
+        let sendCellIDOutput = undefined;
         // Loop through each message in the plan and send it to the AI
         while (!isAgentFinished && agentExecutionDepth <= AGENT_EXECUTION_DEPTH_LIMIT) {
             // Check if we should continue execution
@@ -2214,12 +2449,12 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
             // Only the first message sent to the Agent should contain the user's input.
             // All other messages only contain updated information about the state of the notebook.
             if (agentExecutionDepth === 1) {
-                await sendAgentExecutionMessage(input, messageIndex, false, selectedRules);
+                await sendAgentExecutionMessage(input, messageIndex, undefined, additionalContext);
             }
             else {
-                await sendAgentExecutionMessage('', undefined, sendActiveCellOutput);
+                await sendAgentExecutionMessage('', undefined, sendCellIDOutput);
                 // Reset flag back to false until the agent requests the active cell output again
-                sendActiveCellOutput = false;
+                sendCellIDOutput = undefined;
             }
             // Iterate the agent execution depth
             agentExecutionDepth++;
@@ -2227,9 +2462,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_14__.getCodeBlockFromMessage)(aiDisplayOptimizedChatItem.message);
+                const aiGeneratedCode = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_15__.getCodeBlockFromMessage)(aiDisplayOptimizedChatItem.message);
                 if (aiGeneratedCode) {
-                    const securityCheck = (0,_utils_blacklistedWords__WEBPACK_IMPORTED_MODULE_15__.checkForBlacklistedWords)(aiGeneratedCode);
+                    const securityCheck = (0,_utils_blacklistedWords__WEBPACK_IMPORTED_MODULE_16__.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);
@@ -2256,8 +2491,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_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);
+                await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_17__.acceptAndRunCellUpdate)(agentResponse.cell_update, notebookTracker, app, previewAICodeToActiveCell, acceptAICode);
+                const status = await (0,_utils_agentActions__WEBPACK_IMPORTED_MODULE_17__.retryIfExecutionError)(notebookTracker, app, getDuplicateChatHistoryManager, addAIMessageFromResponseAndUpdateState, sendAgentSmartDebugMessage, previewAICodeToActiveCell, acceptAICode, shouldContinueAgentExecution, finalizeAgentStop, chatHistoryManagerRef);
                 if (status === 'interupted') {
                     break;
                 }
@@ -2272,9 +2507,9 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
                 }
             }
             if (agentResponse.type === 'get_cell_output') {
-                // Mark that we should send the active cell output to the agent 
+                // Mark that we should send the cell output to the agent 
                 // in the next loop iteration
-                sendActiveCellOutput = true;
+                sendCellIDOutput = agentResponse.cell_id;
             }
         }
         if (agentExecutionDepth > AGENT_EXECUTION_DEPTH_LIMIT) {
@@ -2286,33 +2521,33 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
         if (!aiMessage) {
             return;
         }
-        const updateCellCode = (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_12__.getCellCodeByID)(notebookTracker, updateCellID);
+        const updateCellCode = (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_13__.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_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);
+        const aiGeneratedCode = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_15__.getCodeBlockFromMessage)(aiMessage);
+        const aiGeneratedCodeCleaned = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_15__.removeMarkdownCodeFormatting)(aiGeneratedCode || '');
+        const { unifiedCodeString, unifiedDiffs } = (0,_utils_codeDiff__WEBPACK_IMPORTED_MODULE_18__.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_12__.writeCodeToCellByID)(notebookTracker, unifiedCodeString, updateCellID);
+        (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_13__.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_12__.highlightCodeCell)(notebookTracker, updateCellID);
+        (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_13__.highlightCodeCell)(notebookTracker, updateCellID);
     };
     const displayOptimizedChatHistory = chatHistoryManager.getDisplayOptimizedHistory();
     const previewAICodeToActiveCell = () => {
         setCodeReviewStatus('codeCellPreview');
-        const activeCellID = (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_12__.getActiveCellID)(notebookTracker);
+        const activeCellID = (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_13__.getActiveCellID)(notebookTracker);
         const lastAIDisplayMessage = chatHistoryManagerRef.current.getLastAIDisplayOptimizedChatItem();
         if (activeCellID === undefined || lastAIDisplayMessage === undefined) {
             return;
         }
-        (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_12__.scrollToCell)(notebookTracker, activeCellID, undefined, 'end');
+        (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_13__.scrollToCell)(notebookTracker, activeCellID, undefined, 'end');
         updateCodeDiffStripes(lastAIDisplayMessage.message, activeCellID);
         updateCellToolbarButtons();
     };
@@ -2322,7 +2557,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
         if (!lastAIMessage || !cellStateBeforeDiff.current) {
             return;
         }
-        const aiGeneratedCode = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_14__.getCodeBlockFromMessage)(lastAIMessage.message);
+        const aiGeneratedCode = (0,_utils_strings__WEBPACK_IMPORTED_MODULE_15__.getCodeBlockFromMessage)(lastAIMessage.message);
         if (!aiGeneratedCode) {
             return;
         }
@@ -2331,10 +2566,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_12__.getCellByID)(notebookTracker, targetCellID);
+        const targetCell = (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_13__.getCellByID)(notebookTracker, targetCellID);
         if (targetCell) {
             // Make the target cell the active cell
-            (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_12__.setActiveCellByID)(notebookTracker, targetCellID);
+            (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_13__.setActiveCellByID)(notebookTracker, targetCellID);
             // Focus on the active cell
             targetCell.activate();
         }
@@ -2359,7 +2594,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
         updateCodeCellsExtensions(undefined);
         cellStateBeforeDiff.current = undefined;
         if (codeCellID !== undefined) {
-            (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_12__.writeCodeToCellByID)(notebookTracker, code, codeCellID);
+            (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_13__.writeCodeToCellByID)(notebookTracker, code, codeCellID);
             updateCellToolbarButtons();
         }
     };
@@ -2370,17 +2605,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_18__.COMMAND_MITO_AI_PREVIEW_LATEST_CODE, {
+        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_19__.COMMAND_MITO_AI_PREVIEW_LATEST_CODE, {
             execute: () => {
                 previewAICodeToActiveCell();
             }
         });
-        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_APPLY_LATEST_CODE, {
+        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_19__.COMMAND_MITO_AI_APPLY_LATEST_CODE, {
             execute: () => {
                 acceptAICode();
             }
         });
-        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_REJECT_LATEST_CODE, {
+        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_19__.COMMAND_MITO_AI_REJECT_LATEST_CODE, {
             execute: () => {
                 rejectAICode();
             }
@@ -2389,19 +2624,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_18__.COMMAND_MITO_AI_SEND_DEBUG_ERROR_MESSAGE, {
+        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_19__.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_18__.COMMAND_MITO_AI_SEND_EXPLAIN_CODE_MESSAGE, {
+        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_19__.COMMAND_MITO_AI_SEND_EXPLAIN_CODE_MESSAGE, {
             execute: async () => {
                 await sendExplainCodeMessage();
             }
         });
-        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_SEND_AGENT_MESSAGE, {
+        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_19__.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 
@@ -2420,7 +2655,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_18__.COMMAND_MITO_AI_CELL_TOOLBAR_ACCEPT_CODE, {
+        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_19__.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',
@@ -2438,7 +2673,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
                 }
             }
         });
-        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_CELL_TOOLBAR_REJECT_CODE, {
+        app.commands.addCommand(_commands__WEBPACK_IMPORTED_MODULE_19__.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',
@@ -2459,13 +2694,13 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
         // Register keyboard shortcuts 
         const accelYDisposable = app.commands.addKeyBinding({
             command: codeReviewStatus === 'chatPreview' ?
-                _commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_PREVIEW_LATEST_CODE :
-                _commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_APPLY_LATEST_CODE,
+                _commands__WEBPACK_IMPORTED_MODULE_19__.COMMAND_MITO_AI_PREVIEW_LATEST_CODE :
+                _commands__WEBPACK_IMPORTED_MODULE_19__.COMMAND_MITO_AI_APPLY_LATEST_CODE,
             keys: ['Accel Y'],
             selector: 'body',
         });
         const accelDDisposable = app.commands.addKeyBinding({
-            command: _commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_REJECT_LATEST_CODE,
+            command: _commands__WEBPACK_IMPORTED_MODULE_19__.COMMAND_MITO_AI_REJECT_LATEST_CODE,
             keys: ['Accel U'],
             selector: 'body',
             preventDefault: true,
@@ -2482,8 +2717,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_18__.COMMAND_MITO_AI_CELL_TOOLBAR_ACCEPT_CODE);
-        app.commands.notifyCommandChanged(_commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_CELL_TOOLBAR_REJECT_CODE);
+        app.commands.notifyCommandChanged(_commands__WEBPACK_IMPORTED_MODULE_19__.COMMAND_MITO_AI_CELL_TOOLBAR_ACCEPT_CODE);
+        app.commands.notifyCommandChanged(_commands__WEBPACK_IMPORTED_MODULE_19__.COMMAND_MITO_AI_CELL_TOOLBAR_REJECT_CODE);
     };
     const codeDiffStripesCompartments = react__WEBPACK_IMPORTED_MODULE_1___default().useRef(new Map());
     // Function to update the extensions of code cells
@@ -2511,13 +2746,13 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
                         codeDiffStripesCompartments.current.set(cellId, compartment);
                         // Apply the initial configuration
                         editorView.dispatch({
-                            effects: _codemirror_state__WEBPACK_IMPORTED_MODULE_0__.StateEffect.appendConfig.of(compartment.of(unifiedDiffLines !== undefined && isActiveCodeCell ? (0,_CodeDiffDisplay__WEBPACK_IMPORTED_MODULE_19__.codeDiffStripesExtension)({ unifiedDiffLines: unifiedDiffLines }) : [])),
+                            effects: _codemirror_state__WEBPACK_IMPORTED_MODULE_0__.StateEffect.appendConfig.of(compartment.of(unifiedDiffLines !== undefined && isActiveCodeCell ? (0,_CodeDiffDisplay__WEBPACK_IMPORTED_MODULE_20__.codeDiffStripesExtension)({ unifiedDiffLines: unifiedDiffLines }) : [])),
                         });
                     }
                     else {
                         // Reconfigure the compartment
                         editorView.dispatch({
-                            effects: compartment.reconfigure(unifiedDiffLines !== undefined && isActiveCodeCell ? (0,_CodeDiffDisplay__WEBPACK_IMPORTED_MODULE_19__.codeDiffStripesExtension)({ unifiedDiffLines: unifiedDiffLines }) : []),
+                            effects: compartment.reconfigure(unifiedDiffLines !== undefined && isActiveCodeCell ? (0,_CodeDiffDisplay__WEBPACK_IMPORTED_MODULE_20__.codeDiffStripesExtension)({ unifiedDiffLines: unifiedDiffLines }) : []),
                         });
                     }
                 }
@@ -2532,7 +2767,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
     // In agent mode, we group consecutive error messages together. 
     // In chat mode, we display messages individually as they were sent
     if (agentModeEnabled) {
-        processedDisplayOptimizedChatHistory = (0,_utils_chatHistory__WEBPACK_IMPORTED_MODULE_20__.processChatHistoryForErrorGrouping)(chatHistoryManager.getDisplayOptimizedHistory());
+        processedDisplayOptimizedChatHistory = (0,_utils_chatHistory__WEBPACK_IMPORTED_MODULE_21__.processChatHistoryForErrorGrouping)(chatHistoryManager.getDisplayOptimizedHistory());
     }
     else {
         processedDisplayOptimizedChatHistory = chatHistoryManager.getDisplayOptimizedHistory();
@@ -2544,16 +2779,16 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
     return (react__WEBPACK_IMPORTED_MODULE_1___default().createElement("div", { className: "chat-taskpane" },
         react__WEBPACK_IMPORTED_MODULE_1___default().createElement("div", { className: "chat-taskpane-header" },
             react__WEBPACK_IMPORTED_MODULE_1___default().createElement("div", { className: "chat-taskpane-header-left" },
-                react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_21__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_2__.settingsIcon.react, null), title: "Mito AI Settings", onClick: () => {
-                        void app.commands.execute(_SettingsManager_SettingsManagerPlugin__WEBPACK_IMPORTED_MODULE_22__.COMMAND_MITO_AI_SETTINGS);
+                react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_22__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_2__.settingsIcon.react, null), title: "Mito AI Settings", onClick: () => {
+                        void app.commands.execute(_SettingsManager_SettingsManagerPlugin__WEBPACK_IMPORTED_MODULE_23__.COMMAND_MITO_AI_SETTINGS);
                     } })),
             react__WEBPACK_IMPORTED_MODULE_1___default().createElement("div", { className: "chat-taskpane-header-right" },
-                react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_21__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_2__.addIcon.react, null), title: "Start New Chat", onClick: async () => { await startNewChat(); } }),
-                react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_components_DropdownMenu__WEBPACK_IMPORTED_MODULE_23__["default"], { trigger: react__WEBPACK_IMPORTED_MODULE_1___default().createElement("button", { className: "icon-button", title: "Chat Threads", onClick: fetchChatThreads },
+                react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_components_IconButton__WEBPACK_IMPORTED_MODULE_22__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_2__.addIcon.react, null), title: "Start New Chat", onClick: async () => { await startNewChat(); } }),
+                react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_components_DropdownMenu__WEBPACK_IMPORTED_MODULE_24__["default"], { trigger: react__WEBPACK_IMPORTED_MODULE_1___default().createElement("button", { className: "icon-button", title: "Chat Threads", onClick: fetchChatThreads },
                         react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_jupyterlab_ui_components__WEBPACK_IMPORTED_MODULE_2__.historyIcon.react, null)), items: chatThreads.length > 0
                         ? chatThreads.map(thread => ({
                             label: thread.name,
-                            primaryIcon: activeThreadIdRef.current === thread.thread_id ? _icons__WEBPACK_IMPORTED_MODULE_24__.OpenIndicatorLabIcon.react : undefined,
+                            primaryIcon: activeThreadIdRef.current === thread.thread_id ? _icons__WEBPACK_IMPORTED_MODULE_25__.OpenIndicatorLabIcon.react : undefined,
                             onClick: () => fetchChatHistoryAndSetActiveThread(thread.thread_id),
                             secondaryActions: [
                                 {
@@ -2572,29 +2807,29 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
             displayOptimizedChatHistory.length === 0 &&
                 react__WEBPACK_IMPORTED_MODULE_1___default().createElement("div", { className: "chat-empty-message" },
                     react__WEBPACK_IMPORTED_MODULE_1___default().createElement("div", { style: { margin: '0 auto 8px', display: 'block', textAlign: 'center' } },
-                        react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_icons_MitoLogo__WEBPACK_IMPORTED_MODULE_25__["default"], { width: "60", height: "30" })),
+                        react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_icons_MitoLogo__WEBPACK_IMPORTED_MODULE_26__["default"], { width: "60", height: "30" })),
                     react__WEBPACK_IMPORTED_MODULE_1___default().createElement("span", { style: { display: 'block', textAlign: 'center', fontWeight: 'bold', fontSize: '20px', marginBottom: '15px' } }, "Data Copilot"),
-                    react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_CTACarousel__WEBPACK_IMPORTED_MODULE_26__["default"], { app: app })),
+                    react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_CTACarousel__WEBPACK_IMPORTED_MODULE_27__["default"], { app: app })),
             processedDisplayOptimizedChatHistory.map((displayOptimizedChat, index) => {
                 if (isGroupedErrorMessages(displayOptimizedChat)) {
-                    return (react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_components_AgentComponents_ErrorFixupToolUI__WEBPACK_IMPORTED_MODULE_27__["default"], { key: index, messages: displayOptimizedChat, renderMimeRegistry: renderMimeRegistry }));
+                    return (react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_components_AgentComponents_ErrorFixupToolUI__WEBPACK_IMPORTED_MODULE_28__["default"], { key: index, messages: displayOptimizedChat, renderMimeRegistry: renderMimeRegistry }));
                 }
                 else {
-                    return (react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_ChatMessage_ChatMessage__WEBPACK_IMPORTED_MODULE_28__["default"], { key: index, message: displayOptimizedChat.message, promptType: displayOptimizedChat.promptType, messageType: displayOptimizedChat.type, agentResponse: displayOptimizedChat.agentResponse, codeCellID: displayOptimizedChat.codeCellID, mitoAIConnectionError: displayOptimizedChat.type === 'connection error', mitoAIConnectionErrorType: displayOptimizedChat.mitoAIConnectionErrorType || null, messageIndex: index, notebookTracker: notebookTracker, renderMimeRegistry: renderMimeRegistry, app: app, isLastAiMessage: index === lastAIMessagesIndex, isLastMessage: index === displayOptimizedChatHistory.length - 1, operatingSystem: operatingSystem, previewAICode: previewAICodeToActiveCell, acceptAICode: acceptAICode, rejectAICode: rejectAICode, onUpdateMessage: handleUpdateMessage, contextManager: contextManager, codeReviewStatus: codeReviewStatus, setNextSteps: setNextSteps, agentModeEnabled: agentModeEnabled }));
+                    return (react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_ChatMessage_ChatMessage__WEBPACK_IMPORTED_MODULE_29__["default"], { key: index, message: displayOptimizedChat.message, promptType: displayOptimizedChat.promptType, messageType: displayOptimizedChat.type, agentResponse: displayOptimizedChat.agentResponse, codeCellID: displayOptimizedChat.codeCellID, mitoAIConnectionError: displayOptimizedChat.type === 'connection error', mitoAIConnectionErrorType: displayOptimizedChat.mitoAIConnectionErrorType || null, messageIndex: index, notebookTracker: notebookTracker, renderMimeRegistry: renderMimeRegistry, app: app, isLastAiMessage: index === lastAIMessagesIndex, isLastMessage: index === displayOptimizedChatHistory.length - 1, operatingSystem: operatingSystem, previewAICode: previewAICodeToActiveCell, acceptAICode: acceptAICode, rejectAICode: rejectAICode, onUpdateMessage: handleUpdateMessage, contextManager: contextManager, codeReviewStatus: codeReviewStatus, setNextSteps: setNextSteps, agentModeEnabled: agentModeEnabled, additionalContext: displayOptimizedChat.additionalContext }));
                 }
             }).filter(message => message !== null),
             loadingAIResponse &&
                 react__WEBPACK_IMPORTED_MODULE_1___default().createElement("div", { className: "chat-loading-message" },
                     "Thinking ",
-                    react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_components_LoadingDots__WEBPACK_IMPORTED_MODULE_29__["default"], null)),
+                    react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_components_LoadingDots__WEBPACK_IMPORTED_MODULE_30__["default"], null)),
             hasCheckpoint &&
                 agentModeEnabled &&
                 agentExecutionStatus === 'idle' &&
                 displayOptimizedChatHistory.length > 0 && (react__WEBPACK_IMPORTED_MODULE_1___default().createElement("div", { className: 'message message-assistant-chat' },
-                react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_components_TextAndIconButton__WEBPACK_IMPORTED_MODULE_30__["default"], { text: "Revert changes", icon: _icons_UndoIcon__WEBPACK_IMPORTED_MODULE_31__["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_1___default().createElement(_components_TextAndIconButton__WEBPACK_IMPORTED_MODULE_31__["default"], { text: "Revert changes", icon: _icons_UndoIcon__WEBPACK_IMPORTED_MODULE_32__["default"], title: "Revert changes", onClick: () => (0,_utils_checkpoint__WEBPACK_IMPORTED_MODULE_14__.restoreCheckpoint)(app, notebookTracker, setHasCheckpoint, getDuplicateChatHistoryManager, setChatHistoryManager), variant: "gray", width: "fit-contents", iconPosition: "left" }),
                 react__WEBPACK_IMPORTED_MODULE_1___default().createElement("p", { className: "text-muted text-sm" }, "Undo the most recent changes made by the agent")))),
         displayOptimizedChatHistory.length === 0 && (react__WEBPACK_IMPORTED_MODULE_1___default().createElement("div", { className: "suggestions-container" },
-            react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_ChatMessage_ScrollableSuggestions__WEBPACK_IMPORTED_MODULE_32__["default"], { onSelectSuggestion: (prompt) => {
+            react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_ChatMessage_ScrollableSuggestions__WEBPACK_IMPORTED_MODULE_33__["default"], { onSelectSuggestion: (prompt) => {
                     if (agentModeEnabled) {
                         void startAgentExecution(prompt);
                     }
@@ -2603,15 +2838,15 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
                     }
                 } }))),
         react__WEBPACK_IMPORTED_MODULE_1___default().createElement("div", { className: `connected-input-container ${nextSteps.length > 0 ? 'has-next-steps' : ''}` },
-            nextSteps.length > 0 && (react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_components_NextStepsPills__WEBPACK_IMPORTED_MODULE_33__["default"], { nextSteps: nextSteps, onSelectNextStep: agentModeEnabled ? startAgentExecution : sendChatInputMessage, displayedNextStepsIfAvailable: displayedNextStepsIfAvailable, setDisplayedNextStepsIfAvailable: setDisplayedNextStepsIfAvailable })),
-            react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_ChatMessage_ChatInput__WEBPACK_IMPORTED_MODULE_34__["default"], { initialContent: '', placeholder: agentExecutionStatus === 'working' ? 'Agent is working...' :
+            nextSteps.length > 0 && (react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_components_NextStepsPills__WEBPACK_IMPORTED_MODULE_34__["default"], { nextSteps: nextSteps, onSelectNextStep: agentModeEnabled ? startAgentExecution : sendChatInputMessage, displayedNextStepsIfAvailable: displayedNextStepsIfAvailable, setDisplayedNextStepsIfAvailable: setDisplayedNextStepsIfAvailable })),
+            react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_ChatMessage_ChatInput__WEBPACK_IMPORTED_MODULE_35__["default"], { app: app, 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_1___default().createElement("div", { className: "chat-controls" },
             react__WEBPACK_IMPORTED_MODULE_1___default().createElement("div", { className: "chat-controls-left" },
-                react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_components_ToggleButton__WEBPACK_IMPORTED_MODULE_35__["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_1___default().createElement(_components_ToggleButton__WEBPACK_IMPORTED_MODULE_36__["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
@@ -2646,7 +2881,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
                 " \u23CE"))),
         (agentExecutionStatus === 'working' || agentExecutionStatus === 'stopping') && (react__WEBPACK_IMPORTED_MODULE_1___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_1___default().createElement("div", { className: "stop-agent-button-content" },
             "Stopping",
-            react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_components_LoadingCircle__WEBPACK_IMPORTED_MODULE_36__["default"], null),
+            react__WEBPACK_IMPORTED_MODULE_1___default().createElement(_components_LoadingCircle__WEBPACK_IMPORTED_MODULE_37__["default"], null),
             " ")) : ('Stop Agent')))));
 };
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (ChatTaskpane);
@@ -2920,6 +3155,37 @@ const getFirstMessageFromCookie = () => {
 };
 
 
+/***/ }),
+
+/***/ "./lib/Extensions/AiChat/jupyterSettingsManager.js":
+/*!*********************************************************!*\
+  !*** ./lib/Extensions/AiChat/jupyterSettingsManager.js ***!
+  \*********************************************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   setRenameUntitledFileOnSave: () => (/* binding */ setRenameUntitledFileOnSave)
+/* harmony export */ });
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+// Document manager plugin ID
+const DOCMANAGER_PLUGIN_ID = '@jupyterlab/docmanager-extension:plugin';
+// Set renameUntitledFileOnSave to false when the extension loads
+const setRenameUntitledFileOnSave = async (settingRegistry, _documentManager) => {
+    // Note we don't use the documentManager, but we require it as a parameter to make sure
+    // that we leave the documentManager as a required token in the AiChatPlugin.ts file.
+    try {
+        await settingRegistry.set(DOCMANAGER_PLUGIN_ID, 'renameUntitledFileOnSave', false);
+    }
+    catch (error) {
+        console.error('Failed to set renameUntitledFileOnSave setting:', error);
+    }
+};
+
+
 /***/ }),
 
 /***/ "./lib/Extensions/AiChat/token.js":
@@ -2952,6 +3218,47 @@ This makes sure that the chat plugin is loaded before these other extensions.
 const IChatTracker = new _lumino_coreutils__WEBPACK_IMPORTED_MODULE_0__.Token('mito-ai/IChatTracker', 'Widget tracker for the chat sidebar.');
 
 
+/***/ }),
+
+/***/ "./lib/Extensions/AiChat/utils.js":
+/*!****************************************!*\
+  !*** ./lib/Extensions/AiChat/utils.js ***!
+  \****************************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   getBase64EncodedCellOutput: () => (/* binding */ getBase64EncodedCellOutput)
+/* harmony export */ });
+/* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! ../../utils/notebook */ "./lib/utils/notebook.js");
+/* harmony import */ var _restAPI_RestAPI__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../restAPI/RestAPI */ "./lib/restAPI/RestAPI.js");
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+
+const getBase64EncodedCellOutput = async (notebookTracker, cellID) => {
+    if (cellID === undefined) {
+        return undefined;
+    }
+    // Check if the cell id is part of the notebook
+    const cellIndex = (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_0__.getCellIndexByID)(notebookTracker, cellID);
+    if (cellIndex === undefined) {
+        // Log that the cell id is not part of the notebook
+        console.log(`Cell with id ${cellID} not found in notebook`);
+        void (0,_restAPI_RestAPI__WEBPACK_IMPORTED_MODULE_1__.logEvent)('get_cell_output_requested_non_existent_cell');
+        return undefined;
+    }
+    (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_0__.scrollToCell)(notebookTracker, cellID, 0);
+    const activeCellOutput = await (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_0__.getActiveCellOutput)(notebookTracker);
+    if (activeCellOutput !== undefined) {
+        return activeCellOutput;
+    }
+    return undefined;
+};
+
+
 /***/ }),
 
 /***/ "./lib/Extensions/AppBuilder/AppBuilderPlugin.js":
@@ -3088,18 +3395,14 @@ __webpack_require__.r(__webpack_exports__);
 /* harmony export */ __webpack_require__.d(__webpack_exports__, {
 /* harmony export */   convertNotebookToStreamlit: () => (/* binding */ convertNotebookToStreamlit)
 /* harmony export */ });
-/* harmony import */ var _jupyterlab_cells__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! @jupyterlab/cells */ "webpack/sharing/consume/default/@jupyterlab/cells");
-/* harmony import */ var _jupyterlab_cells__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(_jupyterlab_cells__WEBPACK_IMPORTED_MODULE_0__);
-/* harmony import */ var _jupyterlab_coreutils__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! @jupyterlab/coreutils */ "webpack/sharing/consume/default/@jupyterlab/coreutils");
-/* harmony import */ var _jupyterlab_coreutils__WEBPACK_IMPORTED_MODULE_1___default = /*#__PURE__*/__webpack_require__.n(_jupyterlab_coreutils__WEBPACK_IMPORTED_MODULE_1__);
-/* harmony import */ var _utils_notebook__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ../../utils/notebook */ "./lib/utils/notebook.js");
-/* harmony import */ var _cellConversionUtils__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ./cellConversionUtils */ "./lib/Extensions/AppBuilder/cellConversionUtils.js");
-/* harmony import */ var _requirementsUtils__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! ./requirementsUtils */ "./lib/Extensions/AppBuilder/requirementsUtils.js");
-/* harmony import */ var _fileUtils__WEBPACK_IMPORTED_MODULE_7__ = __webpack_require__(/*! ./fileUtils */ "./lib/Extensions/AppBuilder/fileUtils.js");
-/* harmony import */ var _visualizationConversionUtils__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ./visualizationConversionUtils */ "./lib/Extensions/AppBuilder/visualizationConversionUtils.js");
-/* harmony import */ var _lumino_coreutils__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! @lumino/coreutils */ "webpack/sharing/consume/default/@lumino/coreutils");
-/* harmony import */ var _lumino_coreutils__WEBPACK_IMPORTED_MODULE_2___default = /*#__PURE__*/__webpack_require__.n(_lumino_coreutils__WEBPACK_IMPORTED_MODULE_2__);
-/* harmony import */ var _DeployAppNotification__WEBPACK_IMPORTED_MODULE_8__ = __webpack_require__(/*! ./DeployAppNotification */ "./lib/Extensions/AppBuilder/DeployAppNotification.js");
+/* harmony import */ var _jupyterlab_coreutils__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! @jupyterlab/coreutils */ "webpack/sharing/consume/default/@jupyterlab/coreutils");
+/* harmony import */ var _jupyterlab_coreutils__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(_jupyterlab_coreutils__WEBPACK_IMPORTED_MODULE_0__);
+/* harmony import */ var _requirementsUtils__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ./requirementsUtils */ "./lib/Extensions/AppBuilder/requirementsUtils.js");
+/* harmony import */ var _fileUtils__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ./fileUtils */ "./lib/Extensions/AppBuilder/fileUtils.js");
+/* 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 _DeployAppNotification__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! ./DeployAppNotification */ "./lib/Extensions/AppBuilder/DeployAppNotification.js");
+/* harmony import */ var _auth__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ./auth */ "./lib/Extensions/AppBuilder/auth.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -3110,113 +3413,49 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
-
-
-
 /*
-This function converts a notebook into a streamlit app. It processes each cell one by one,
-and makes sure to convert specific functionality from the jupyter to streamlit equivalents.
-For example, in Jupyter, you can display a streamlit graph by calling plt.show(), whereas in Streamlit,
-you need to call st.plotly_chart().
-
-This function returns:
-1. A <notebook-name>-streamlit-app.py file
-2. A requirements.txt file that lists the dependencies for the streamlit app
+This function generates a requirements.txt file that lists the dependencies for the streamlit app
 */
 const convertNotebookToStreamlit = async (notebookTracker, appBuilderService) => {
+    // Check authentication before proceeding with deployment
+    const isAuthenticated = await (0,_auth__WEBPACK_IMPORTED_MODULE_2__.checkAuthenticationAndRedirect)();
+    if (!isAuthenticated) {
+        console.log('User not authenticated, redirected to signup');
+        return;
+    }
     const notebookPanel = notebookTracker.currentWidget;
     if (!notebookPanel) {
         console.error('No notebook is currently active');
         return;
     }
     const notebookPath = notebookPanel.context.path;
-    const notebookName = _jupyterlab_coreutils__WEBPACK_IMPORTED_MODULE_1__.PathExt.basename(notebookPath, '.ipynb');
-    const appFilePath = `./app.py`;
-    // Initialize Streamlit code with imports
-    let streamlitCode = [
-        "import streamlit as st",
-        "",
-        `st.title('${notebookName}')`,
-        (0,_visualizationConversionUtils__WEBPACK_IMPORTED_MODULE_3__.generateDisplayVizFunction)(),
-        ""
-    ];
-    // TODO: we can set the app favicon https://docs.streamlit.io/develop/api-reference/configuration/st.set_page_config
-    console.log("Converting notebook code to streamlit app");
-    // Process each cell
-    notebookPanel.content.widgets.forEach((cellWidget) => {
-        const cellModel = cellWidget.model;
-        let cellContent = (0,_cellConversionUtils__WEBPACK_IMPORTED_MODULE_4__.getCellContent)(cellModel);
-        // Check if the cell is marked to skip.
-        const includeCellInApp = (0,_utils_notebook__WEBPACK_IMPORTED_MODULE_5__.getIncludeCellInApp)(notebookTracker, cellModel.id);
-        if (!includeCellInApp) {
-            return;
-        }
-        // Check if it's an empty code cell
-        if (cellModel.type === 'code' && !cellModel.sharedModel.source.trim()) {
-            return;
-        }
-        if (cellWidget instanceof _jupyterlab_cells__WEBPACK_IMPORTED_MODULE_0__.MarkdownCell) {
-            streamlitCode.push("\n# Converting Markdown Cell");
-            // Convert markdown cells to st.markdown
-            // TODO: The single # Heading markdown in Streamlit is as big as the title, maybe larger.
-            // So we might want to downsize them all by one # to make it look nicer.
-            const escapedContent = cellContent.replace(/"""/g, '\\"\\"\\"');
-            streamlitCode.push(`st.markdown("""${escapedContent}""")`);
-            streamlitCode.push("");
-        }
-        else if (cellWidget instanceof _jupyterlab_cells__WEBPACK_IMPORTED_MODULE_0__.CodeCell) {
-            streamlitCode.push("\n# Converting Code Cell");
-            // Convert the Mito App Input into Streamlit components
-            // Temporarily replace \\n with a placeholder
-            const placeholder = '__ESCAPED_NEWLINE_PLACEHOLDER__';
-            cellContent = cellContent.replace(/\\n/g, placeholder);
-            cellContent = (0,_cellConversionUtils__WEBPACK_IMPORTED_MODULE_4__.removeInvalidLines)(cellContent);
-            // Now split on actual newlines safely
-            cellContent = cellContent.split('\n').map(line => {
-                return (0,_cellConversionUtils__WEBPACK_IMPORTED_MODULE_4__.transformMitoAppInput)(line);
-            }).join('\n');
-            // Transform the cell for visualizations using our new unified approach
-            cellContent = (0,_visualizationConversionUtils__WEBPACK_IMPORTED_MODULE_3__.transformVisualizationCell)(cellContent);
-            // Restore the \\n sequences
-            cellContent = cellContent.replace(new RegExp(placeholder, 'g'), '\\n');
-            streamlitCode = streamlitCode.concat(cellContent);
-            /*
-            Displaying dataframes:
-            
-            Streamlit automatically renders variables that appear alone on a line,
-            similar to Jupyter's behavior with the last line of a cell.
-            
-            Benefits:
-            - Default Mito dataframe output works without modification
-            
-            TODO: Edge cases to handle:
-            1. User/AI-generated display(df) calls. We can tell the AI to not generate this. Users almost never do.
-            2. mitosheet.sheet() calls - consider converting to Mito spreadsheet component or just dataframe output
-            */
-        }
-    });
-    // Create the streamlit app.py file
-    const streamlitSourceCode = streamlitCode.join('\n');
+    const notebookName = _jupyterlab_coreutils__WEBPACK_IMPORTED_MODULE_0__.PathExt.basename(notebookPath, '.ipynb');
+    // Get full path to folder
+    const pathToFolder = _jupyterlab_coreutils__WEBPACK_IMPORTED_MODULE_0__.PathExt.dirname(notebookPath);
+    console.log('Notebook path:', notebookPath);
+    console.log('Notebook name:', notebookName);
+    console.log('Current working directory info:', notebookPanel.context);
     // Build the requirements.txt file
     console.debug("Building requirements.txt file");
-    const requirementsContent = await (0,_requirementsUtils__WEBPACK_IMPORTED_MODULE_6__.generateRequirementsTxt)(notebookTracker);
+    const requirementsContent = await (0,_requirementsUtils__WEBPACK_IMPORTED_MODULE_3__.generateRequirementsTxt)(notebookTracker);
     // Save the files to the current directory
-    await (0,_fileUtils__WEBPACK_IMPORTED_MODULE_7__.saveFileWithKernel)(notebookTracker, './requirements.txt', requirementsContent);
-    await (0,_fileUtils__WEBPACK_IMPORTED_MODULE_7__.saveFileWithKernel)(notebookTracker, appFilePath, streamlitSourceCode);
-    // Get the full path to the folder
-    const pathToFolder = _jupyterlab_coreutils__WEBPACK_IMPORTED_MODULE_1__.PathExt.dirname(notebookPath);
+    await (0,_fileUtils__WEBPACK_IMPORTED_MODULE_4__.saveFileWithKernel)(notebookTracker, './requirements.txt', requirementsContent);
     // After building the files, we need to send a request to the backend to deploy the app
     if (appBuilderService) {
         try {
             console.log("Sending request to deploy the app");
+            // Get JWT token for authentication
+            const jwtToken = (0,_auth__WEBPACK_IMPORTED_MODULE_2__.getJWTToken)();
             const response = await appBuilderService.client.sendMessage({
                 type: 'build-app',
-                message_id: _lumino_coreutils__WEBPACK_IMPORTED_MODULE_2__.UUID.uuid4(),
-                path: pathToFolder
+                message_id: _lumino_coreutils__WEBPACK_IMPORTED_MODULE_1__.UUID.uuid4(),
+                notebook_path: notebookPath,
+                app_path: pathToFolder,
+                jwt_token: jwtToken || appBuilderService.client.serverSettings.token
             });
             console.log("App deployment response:", response);
             const url = response.url;
-            (0,_DeployAppNotification__WEBPACK_IMPORTED_MODULE_8__.deployAppNotification)(url);
+            (0,_DeployAppNotification__WEBPACK_IMPORTED_MODULE_5__.deployAppNotification)(url);
         }
         catch (error) {
             // TODO: Do something with the error
@@ -3231,78 +3470,171 @@ const convertNotebookToStreamlit = async (notebookTracker, appBuilderService) =>
 
 /***/ }),
 
-/***/ "./lib/Extensions/AppBuilder/cellConversionUtils.js":
-/*!**********************************************************!*\
-  !*** ./lib/Extensions/AppBuilder/cellConversionUtils.js ***!
-  \**********************************************************/
+/***/ "./lib/Extensions/AppBuilder/auth.js":
+/*!*******************************************!*\
+  !*** ./lib/Extensions/AppBuilder/auth.js ***!
+  \*******************************************/
 /***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
 
 __webpack_require__.r(__webpack_exports__);
 /* harmony export */ __webpack_require__.d(__webpack_exports__, {
-/* harmony export */   getCellContent: () => (/* binding */ getCellContent),
-/* harmony export */   getCellType: () => (/* binding */ getCellType),
-/* harmony export */   removeInvalidLines: () => (/* binding */ removeInvalidLines),
-/* harmony export */   transformMitoAppInput: () => (/* binding */ transformMitoAppInput)
+/* harmony export */   checkAuthenticationAndRedirect: () => (/* binding */ checkAuthenticationAndRedirect),
+/* harmony export */   exchangeCodeForTokens: () => (/* binding */ exchangeCodeForTokens),
+/* harmony export */   getAuthHeaders: () => (/* binding */ getAuthHeaders),
+/* harmony export */   getJWTToken: () => (/* binding */ getJWTToken),
+/* harmony export */   isUserAuthenticated: () => (/* binding */ isUserAuthenticated),
+/* harmony export */   redirectToSignin: () => (/* binding */ redirectToSignin),
+/* harmony export */   redirectToSignup: () => (/* binding */ redirectToSignup),
+/* harmony export */   removeJWTToken: () => (/* binding */ removeJWTToken),
+/* harmony export */   setJWTToken: () => (/* binding */ setJWTToken)
 /* harmony export */ });
+/* harmony import */ var _restAPI_utils__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! ../../restAPI/utils */ "./lib/restAPI/utils.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
  */
-// Helper function to get cell content
-const getCellContent = (cell) => {
-    return cell.sharedModel.source;
+
+const domain_dev = 'https://mito-app-auth.auth.us-east-1.amazoncognito.com';
+const client_id = '6ara3u3l8sss738hrhbq1qtiqf';
+// Change this to domain_dev for dev deployments
+const active_domain = domain_dev;
+const currentUrl = window.location.href;
+console.log("currentUrl: ", currentUrl);
+// AWS Cognito configuration
+const COGNITO_CONFIG = {
+    SIGNUP_URL: `${active_domain}/signup?client_id=${client_id}&response_type=code&scope=email+openid+profile&redirect_uri=${currentUrl}`,
+    SIGNIN_URL: `${active_domain}/login?client_id=${client_id}&response_type=code&scope=email+openid+profile&redirect_uri=${currentUrl}`,
+    JWT_COOKIE_NAME: 'mito-auth-token',
+    JWT_COOKIE_EXPIRY_HOURS: 1
 };
-// Helper function to get cell type
-const getCellType = (cell) => {
-    return cell.type;
+/**
+ * Check if user is authenticated by looking for JWT token in cookies
+ */
+const isUserAuthenticated = () => {
+    const token = getJWTToken();
+    return token !== null && token !== undefined && token !== '';
 };
-const getVariableNameDefaultAndLabel = (line, identifier) => {
-    var _a, _b;
-    // Split on the equal sign to get the variable name. We must use this full
-    // name because its what the python script uses. 
-    const variableName = ((_a = line.split(' ')[0]) === null || _a === void 0 ? void 0 : _a.trim()) || '';
-    // Split on the identifier to get the unique label for this variable
-    let variableLabel = (variableName === null || variableName === void 0 ? void 0 : variableName.split(identifier)[1]) || '';
-    if (variableLabel.startsWith("_")) {
-        variableLabel = variableLabel.slice(1);
+/**
+ * Get JWT token from cookies
+ */
+const getJWTToken = () => {
+    const cookies = document.cookie.split(';');
+    const jwtCookie = cookies.find(cookie => cookie.trim().startsWith(`${COGNITO_CONFIG.JWT_COOKIE_NAME}=`));
+    if (jwtCookie) {
+        const token = jwtCookie.split('=')[1];
+        return token || null;
     }
-    // Get the value after the equal sign to get the default value for the variable
-    const defaultValue = ((_b = line.split('=')[1]) === null || _b === void 0 ? void 0 : _b.trim()) || '';
-    return [variableName, variableLabel, defaultValue];
+    return null;
 };
-const transformMitoAppInput = (line) => {
-    const textInputIdentifer = 'mito_app_text_input';
-    if (line.startsWith(textInputIdentifer)) {
-        const [variableName, variableLabel, defaultValue] = getVariableNameDefaultAndLabel(line, textInputIdentifer);
-        return `${variableName} = st.text_input('${variableLabel}', ${defaultValue})`;
+/**
+ * Set JWT token as a cookie
+ */
+const setJWTToken = (token) => {
+    const date = new Date();
+    date.setTime(date.getTime() + (COGNITO_CONFIG.JWT_COOKIE_EXPIRY_HOURS * 60 * 60 * 1000));
+    const expires = "expires=" + date.toUTCString();
+    // Set cookie for all subdomains of trymito.io (production)
+    document.cookie = `${COGNITO_CONFIG.JWT_COOKIE_NAME}=${token}; ${expires}; path=/; domain=.trymito.io; SameSite=Lax; Secure`;
+    // For localhost development
+    if (window.location.hostname === 'localhost') {
+        document.cookie = `${COGNITO_CONFIG.JWT_COOKIE_NAME}=${token}; ${expires}; path=/`;
     }
-    const numberInputIdentifier = 'mito_app_number_input';
-    if (line.startsWith(numberInputIdentifier)) {
-        const [variableName, variableLabel, defaultValue] = getVariableNameDefaultAndLabel(line, numberInputIdentifier);
-        return `${variableName} = st.number_input('${variableLabel}', ${defaultValue})`;
+};
+/**
+ * Remove JWT token from cookies
+ */
+const removeJWTToken = () => {
+    // Delete for current domain
+    document.cookie = `${COGNITO_CONFIG.JWT_COOKIE_NAME}=; expires=Thu, 01 Jan 1970 00:00:00 UTC; path=/;`;
+    // Delete for all subdomains
+    document.cookie = `${COGNITO_CONFIG.JWT_COOKIE_NAME}=; expires=Thu, 01 Jan 1970 00:00:00 UTC; path=/; domain=.trymito.io;`;
+};
+/**
+ * Exchange authorization code for JWT tokens
+ */
+const exchangeCodeForTokens = async (code) => {
+    try {
+        const response = await (0,_restAPI_utils__WEBPACK_IMPORTED_MODULE_0__.requestAPI)('auth/token', {
+            method: 'POST',
+            body: JSON.stringify({ code })
+        });
+        if (response.error) {
+            console.error('Failed to exchange code for tokens:', response.error);
+            return false;
+        }
+        if (response.data) {
+            const data = response.data;
+            const { access_token } = data;
+            // Store the access token as the JWT token
+            if (access_token) {
+                setJWTToken(access_token);
+                return true;
+            }
+        }
+        return false;
     }
-    const dateInputIdentifier = 'mito_app_date_input';
-    if (line.startsWith(dateInputIdentifier)) {
-        const [variableName, variableLabel, defaultValue] = getVariableNameDefaultAndLabel(line, dateInputIdentifier);
-        // The user is responsible for making sure the right hand side is a valid option:
-        // "today", datetime.date, datetime.datetime, "YYYY-MM-DD". 
-        return `${variableName} = st.date_input('${variableLabel}', ${defaultValue})`;
+    catch (error) {
+        console.error('Error exchanging code for tokens:', error);
+        return false;
     }
-    const booleanInputIdentifier = 'mito_app_boolean_input';
-    if (line.startsWith(booleanInputIdentifier)) {
-        const [variableName, variableLabel, defaultValue] = getVariableNameDefaultAndLabel(line, booleanInputIdentifier);
-        return `${variableName} = st.checkbox('${variableLabel}', ${defaultValue})`;
+};
+/**
+ * Redirect user to AWS Cognito signup page
+ */
+const redirectToSignup = () => {
+    window.location.href = COGNITO_CONFIG.SIGNUP_URL;
+};
+/**
+ * Redirect user to AWS Cognito signin page
+ */
+const redirectToSignin = () => {
+    window.location.href = COGNITO_CONFIG.SIGNIN_URL;
+};
+/**
+ * Handle authentication check and redirect if needed
+ * Returns true if user is authenticated, false if redirected
+ */
+const checkAuthenticationAndRedirect = async () => {
+    if (!isUserAuthenticated()) {
+        // Check if there's a code parameter in the URL (indicating return from Cognito)
+        const urlParams = new URLSearchParams(window.location.search);
+        const code = urlParams.get('code');
+        if (code) {
+            // User has returned from Cognito with authorization code
+            console.log('Received authorization code from Cognito:', code);
+            // Exchange the code for JWT tokens
+            const success = await exchangeCodeForTokens(code);
+            if (success) {
+                // Clean up URL
+                const newUrl = window.location.pathname;
+                window.history.replaceState({}, document.title, newUrl);
+                return true;
+            }
+            else {
+                // Failed to exchange code, redirect to signup
+                redirectToSignup();
+                return false;
+            }
+        }
+        else {
+            // No token and no code, redirect to signup
+            redirectToSignup();
+            return false;
+        }
     }
-    // If there was no text_input to create, then just return the original line.
-    return line;
+    return true;
 };
-const removeInvalidLines = (cellContent) => {
-    // Remove lines that are special to ipython and won't work in streamlit. These lines start with:
-    // ! -> like !pip install pandas
-    // % -> like %pip install pandas
-    // %% -> like %%pip install pandas
-    const invalidLinesPrefixes = ['!', '%', '%%'];
-    return cellContent.split('\n').filter(line => !invalidLinesPrefixes.some(prefix => line.trim().startsWith(prefix))).join('\n');
+/**
+ * Get authentication headers for API requests
+ */
+const getAuthHeaders = () => {
+    const token = getJWTToken();
+    if (token) {
+        return {
+            'Authorization': `Bearer ${token}`
+        };
+    }
+    return {};
 };
 
 
@@ -3409,7 +3741,15 @@ const generateRequirementsTxt = async (notebookTracker) => {
             // Gather all code cells content
             notebook.widgets.forEach(cell => {
                 if (cell.model.type === 'code') {
-                    codeContent += cell.model.sharedModel.source + '\n\n';
+                    const source = cell.model.sharedModel.source;
+                    // Filter out lines that start with shell commands
+                    const filteredLines = source.split('\n').filter(line => {
+                        const trimmed = line.trim();
+                        return !trimmed.startsWith('!') && !trimmed.startsWith('%');
+                    });
+                    if (filteredLines.length > 0) {
+                        codeContent += filteredLines.join('\n') + '\n\n';
+                    }
                 }
             });
             // Create Python code to run pipreqs on a temporary directory
@@ -3508,154 +3848,6 @@ with tempfile.TemporaryDirectory() as temp_dir:
 };
 
 
-/***/ }),
-
-/***/ "./lib/Extensions/AppBuilder/visualizationConversionUtils.js":
-/*!*******************************************************************!*\
-  !*** ./lib/Extensions/AppBuilder/visualizationConversionUtils.js ***!
-  \*******************************************************************/
-/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
-
-__webpack_require__.r(__webpack_exports__);
-/* harmony export */ __webpack_require__.d(__webpack_exports__, {
-/* harmony export */   extractVisualizationVariables: () => (/* binding */ extractVisualizationVariables),
-/* harmony export */   generateDisplayVizFunction: () => (/* binding */ generateDisplayVizFunction),
-/* harmony export */   transformVisualizationCell: () => (/* binding */ transformVisualizationCell)
-/* harmony export */ });
-/*
- * Copyright (c) Saga Inc.
- * Distributed under the terms of the GNU Affero General Public License v3.0 License.
- */
-// Generate the display_viz helper function for Streamlit
-const generateDisplayVizFunction = () => {
-    return `
-def display_viz(fig):
-    """Display a visualization in Streamlit based on its type."""
-
-    # Check for SymPy plot objects first (most specific)
-    if hasattr(fig, '_backend'):
-        fig._backend.process_series()
-        matplotlib_fig = fig._backend.plt.gcf()
-        st.pyplot(matplotlib_fig)
-        return
-
-    # Check for Plotly figure
-    if hasattr(fig, 'update_layout') or str(type(fig)).find('plotly') >= 0:
-        st.plotly_chart(fig)
-        return
-
-    # Check for Matplotlib figure (be more specific to avoid SymPy conflicts)
-    if hasattr(fig, 'add_subplot') and hasattr(fig, 'savefig'):
-        st.pyplot(fig)
-        return
-
-    # Fallback - try pyplot as it's most common
-    try:
-        st.pyplot(fig)
-    except Exception as e:
-        st.error(f"Couldn't display visualization of type: {type(fig)}. Error: {str(e)}")
-        st.write(fig)  # Show the object for debugging
-`;
-};
-// Extract all potential visualization variable names from a code cell
-const extractVisualizationVariables = (cellContent) => {
-    // Clean up the content and trim whitespace
-    const trimmedContent = cellContent.trim();
-    const vizVariables = [];
-    // Check for common visualization patterns for various libraries
-    const vizPatterns = [
-        // Plotly patterns
-        /([a-zA-Z_][a-zA-Z0-9_]*)\s*=\s*px\./,
-        /([a-zA-Z_][a-zA-Z0-9_]*)\s*=\s*go\.Figure/,
-        /([a-zA-Z_][a-zA-Z0-9_]*)\s*=\s*ff\./,
-        /([a-zA-Z_][a-zA-Z0-9_]*)\s*=\s*plotly\./,
-        // Matplotlib patterns
-        /([a-zA-Z_][a-zA-Z0-9_]*)\s*=\s*plt\.figure/,
-        /([a-zA-Z_][a-zA-Z0-9_]*)\s*,\s*([a-zA-Z_][a-zA-Z0-9_]*)\s*=\s*plt\.subplots/,
-        // Generic patterns that might be visualizations
-        /([a-zA-Z_][a-zA-Z0-9_]*)\s*=\s*.*\.(plot|scatter|bar|hist|pie|boxplot|violin)/,
-        /([a-zA-Z_][a-zA-Z0-9_]*)\s*=\s*.*\.(figure|Figure)/,
-        // Any variable involved in a show() call
-        /([a-zA-Z_][a-zA-Z0-9_]*)\.show\(/
-    ];
-    const lines = trimmedContent.split('\n');
-    // Find all variable assignments that match visualization patterns
-    for (const line of lines) {
-        for (const pattern of vizPatterns) {
-            const matches = [...line.matchAll(new RegExp(pattern, 'g'))];
-            for (const match of matches) {
-                // For plt.subplots(), capture both fig and ax variables
-                if (pattern.toString().includes('subplots') && match[2]) {
-                    if (match[1] && !vizVariables.includes(match[1])) {
-                        vizVariables.push(match[1]); // fig variable
-                    }
-                    if (match[2] && !vizVariables.includes(match[2])) {
-                        vizVariables.push(match[2]); // ax variable
-                    }
-                }
-                else if (match[1] && !vizVariables.includes(match[1])) {
-                    // Check if it's not a Python keyword
-                    const pythonKeywords = ['if', 'else', 'elif', 'for', 'while', 'def', 'class', 'return', 'import', 'from', 'print'];
-                    if (!pythonKeywords.includes(match[1])) {
-                        vizVariables.push(match[1]);
-                    }
-                }
-            }
-        }
-    }
-    return vizVariables;
-};
-// Transform visualization code for Streamlit using runtime detection
-const transformVisualizationCell = (cellContent) => {
-    var _a;
-    const lines = cellContent.split('\n');
-    const transformedLines = [];
-    // Extract figure variable names
-    const figVariables = extractVisualizationVariables(cellContent);
-    for (let i = 0; i < lines.length; i++) {
-        const line = (_a = lines[i]) !== null && _a !== void 0 ? _a : '';
-        let replacedLine = false;
-        // Check for plt.show() calls We need to replace these with display_viz(plt.gcf())
-        // because we need to make sure we display the current figure.
-        if (line.trim().match(/plt\.show\(/)) {
-            transformedLines.push("display_viz(plt.gcf())");
-            replacedLine = true;
-        }
-        // Check for SymPy plot calls
-        else if (line.trim().match(/(sp|sym|sy|sm|sympy)\.plot\s*\(.*show\s*=\s*True.*\)/)) {
-            const modifiedLine = line.replace(/(.*?)(sp|sym|sy|sm|sympy)\.plot\s*\(/, '$1display_viz($2.plot(') + ')';
-            transformedLines.push(modifiedLine);
-            replacedLine = true;
-        }
-        // If show=False, do not call display_viz, just keep the line as is
-        else if (line.trim().match(/((sp|sym|sy|sm|sympy)\.)?plot\s*\(.*show\s*=\s*False.*\)/)) {
-            // Do nothing, just let it fall through to the original line
-        }
-        // Check for direct plot(...) calls
-        else if (line.trim().match(/^plot\s*\(/)) {
-            const modifiedLine = line.replace(/^plot\s*\(/, 'display_viz(plot(') + ')';
-            transformedLines.push(modifiedLine);
-            replacedLine = true;
-        }
-        // Check for figure.show() calls for any detected figure variables
-        else {
-            for (const figVar of figVariables) {
-                if (line.trim().startsWith(`${figVar}.show`)) {
-                    transformedLines.push(`display_viz(${figVar})`);
-                    replacedLine = true;
-                    break;
-                }
-            }
-        }
-        // If we didn't replace the line, keep the original
-        if (!replacedLine) {
-            transformedLines.push(line);
-        }
-    }
-    return transformedLines.join('\n');
-};
-
-
 /***/ }),
 
 /***/ "./lib/Extensions/ContextManager/ContextManagerPlugin.js":
@@ -4717,11 +4909,15 @@ const ConnectionList = ({ connections, loading, error, onDelete }) => {
         var _a;
         return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { key: id, className: "connection-card" },
             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "connection-card-header" },
-                react__WEBPACK_IMPORTED_MODULE_0___default().createElement("h3", null, connection.type)),
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement("h3", { className: "connection-alias" }, connection.alias ? connection.alias : connection.database),
+                react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "connection-type" }, connection.type.toUpperCase())),
+            react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "connection-divider" }),
             react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "connection-details" }, (_a = _model__WEBPACK_IMPORTED_MODULE_1__.databaseConfigs[connection.type]) === null || _a === void 0 ? void 0 : _a.fields.map(field => {
-                // Skip password fields for security
+                // Skip specific fields
                 if (field.type === 'password')
                     return null;
+                if (field.name === 'alias')
+                    return null;
                 const value = connection[field.name];
                 // Only show fields that have values
                 if (value === undefined || value === '')
@@ -4872,12 +5068,20 @@ __webpack_require__.r(__webpack_exports__);
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
  */
+const ALIAS_PLACEHOLDER = 'Enter a nickname for this database';
 const databaseConfigs = {
     mssql: {
         type: 'mssql',
         displayName: 'Microsoft SQL Server',
         alertText: 'Microsoft SQL Server requires an additional driver. If you\'ve already installed it, you can safely ignore this message. For more info, consult the <a href="https://docs.trymito.io/mito-ai/database-connectors/microsoft-sql-server" target="_blank">Mito docs</a>.',
         fields: [
+            {
+                name: 'alias',
+                type: 'text',
+                label: 'Alias',
+                placeholder: ALIAS_PLACEHOLDER,
+                required: true
+            },
             {
                 name: 'username',
                 type: 'text',
@@ -4926,6 +5130,13 @@ const databaseConfigs = {
         type: 'mysql',
         displayName: 'MySQL',
         fields: [
+            {
+                name: 'alias',
+                type: 'text',
+                label: 'Alias',
+                placeholder: ALIAS_PLACEHOLDER,
+                required: true
+            },
             {
                 name: 'username',
                 type: 'text',
@@ -4967,6 +5178,13 @@ const databaseConfigs = {
         type: 'oracle',
         displayName: 'Oracle',
         fields: [
+            {
+                name: 'alias',
+                type: 'text',
+                label: 'Alias',
+                placeholder: ALIAS_PLACEHOLDER,
+                required: true
+            },
             {
                 name: 'username',
                 type: 'text',
@@ -5008,6 +5226,13 @@ const databaseConfigs = {
         type: 'postgres',
         displayName: 'PostgreSQL',
         fields: [
+            {
+                name: 'alias',
+                type: 'text',
+                label: 'Alias',
+                placeholder: ALIAS_PLACEHOLDER,
+                required: true
+            },
             {
                 name: 'username',
                 type: 'text',
@@ -5049,6 +5274,13 @@ const databaseConfigs = {
         type: 'snowflake',
         displayName: 'Snowflake',
         fields: [
+            {
+                name: 'alias',
+                type: 'text',
+                label: 'Alias',
+                placeholder: ALIAS_PLACEHOLDER,
+                required: true
+            },
             {
                 name: 'username',
                 type: 'text',
@@ -5083,6 +5315,13 @@ const databaseConfigs = {
         type: 'sqlite',
         displayName: 'SQLite',
         fields: [
+            {
+                name: 'alias',
+                type: 'text',
+                label: 'Alias',
+                placeholder: ALIAS_PLACEHOLDER,
+                required: true
+            },
             {
                 name: 'database',
                 type: 'text',
@@ -5996,6 +6235,39 @@ const AgentComponentHeader = ({ icon, text, onClick, isExpanded, displayBorder =
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (AgentComponentHeader);
 
 
+/***/ }),
+
+/***/ "./lib/components/AgentComponents/AssumptionToolUI.js":
+/*!************************************************************!*\
+  !*** ./lib/components/AgentComponents/AssumptionToolUI.js ***!
+  \************************************************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
+/* harmony import */ var _utils_classNames__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../../utils/classNames */ "./lib/utils/classNames.js");
+/* harmony import */ var _style_AssumptionTool_css__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../../style/AssumptionTool.css */ "./style/AssumptionTool.css");
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+
+
+const AssumptionToolUI = ({ assumptions, }) => {
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_2__.classNames)('assumption-tool-container') },
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "assumption-header-label" }, "Assumption"),
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("p", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_2__.classNames)('assumption-content') },
+            assumptions.length > 1 && (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("ul", null, assumptions.map((assumption, index) => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("li", { key: index }, assumption))))),
+            assumptions.length === 1 && (assumptions[0]))));
+};
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (AssumptionToolUI);
+
+
 /***/ }),
 
 /***/ "./lib/components/AgentComponents/ErrorFixupToolUI.js":
@@ -6106,6 +6378,62 @@ const GetCellOutputToolUI = () => {
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (GetCellOutputToolUI);
 
 
+/***/ }),
+
+/***/ "./lib/components/DatabaseButton.js":
+/*!******************************************!*\
+  !*** ./lib/components/DatabaseButton.js ***!
+  \******************************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
+/* harmony import */ var _IconButton__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ./IconButton */ "./lib/components/IconButton.js");
+/* harmony import */ var _icons_DatabaseOutlineIcon__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../icons/DatabaseOutlineIcon */ "./lib/icons/DatabaseOutlineIcon.js");
+/* harmony import */ var _Extensions_SettingsManager_SettingsManagerPlugin__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../Extensions/SettingsManager/SettingsManagerPlugin */ "./lib/Extensions/SettingsManager/SettingsManagerPlugin.js");
+/* harmony import */ var _restAPI_RestAPI__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../restAPI/RestAPI */ "./lib/restAPI/RestAPI.js");
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+
+
+
+
+const DatabaseButton = ({ app }) => {
+    const [databaseConnections, setDatabaseConnections] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)({});
+    const [isIconVisible, setIsIconVisible] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(true);
+    // Fetch database connections
+    const fetchDatabaseConnections = async () => {
+        const databaseConnections = await (0,_restAPI_RestAPI__WEBPACK_IMPORTED_MODULE_1__.getDatabaseConnections)();
+        setDatabaseConnections(databaseConnections);
+    };
+    // Fetch database connections
+    (0,react__WEBPACK_IMPORTED_MODULE_0__.useEffect)(() => {
+        void fetchDatabaseConnections();
+    }, []);
+    // Determine notification dot type based on connections
+    const getNotificationDotType = () => {
+        if (!isIconVisible) {
+            return null;
+        }
+        return Object.keys(databaseConnections).length > 0 ? 'success' : 'warning';
+    };
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_IconButton__WEBPACK_IMPORTED_MODULE_2__["default"], { icon: react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_DatabaseOutlineIcon__WEBPACK_IMPORTED_MODULE_3__["default"], null), title: 'Add Database', onClick: () => {
+            void app.commands.execute(_Extensions_SettingsManager_SettingsManagerPlugin__WEBPACK_IMPORTED_MODULE_4__.COMMAND_MITO_AI_SETTINGS);
+            setIsIconVisible(false);
+        }, notificationDotType: getNotificationDotType(), className: 'icon-button-hover', style: {
+            height: 'var(--chat-context-button-height)'
+        } }));
+};
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (DatabaseButton);
+
+
 /***/ }),
 
 /***/ "./lib/components/DropdownMenu.js":
@@ -6197,8 +6525,10 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
-const IconButton = ({ icon, onClick, title, style }) => {
-    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_3__.classNames)("icon-button"), onClick: onClick, title: title, style: style }, icon));
+const IconButton = ({ icon, onClick, title, style, notificationDotType = null, className }) => {
+    return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { className: (0,_utils_classNames__WEBPACK_IMPORTED_MODULE_3__.classNames)("icon-button", className), onClick: onClick, title: title, style: style },
+        icon,
+        notificationDotType && react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: `notification-dot notification-dot-${notificationDotType}` })));
 };
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (IconButton);
 
@@ -6459,6 +6789,8 @@ __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 _style_SelectedContextContainer_css__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../../style/SelectedContextContainer.css */ "./style/SelectedContextContainer.css");
 /* harmony import */ var _icons_RuleIcon__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! ../icons/RuleIcon */ "./lib/icons/RuleIcon.js");
+/* harmony import */ var _icons_CodeIcon__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! ../icons/CodeIcon */ "./lib/icons/CodeIcon.js");
+/* harmony import */ var _icons_DatabaseIcon__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! ../icons/DatabaseIcon */ "./lib/icons/DatabaseIcon.js");
 /*
  * Copyright (c) Saga Inc.
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
@@ -6466,10 +6798,24 @@ __webpack_require__.r(__webpack_exports__);
 
 
 
-const SelectedContextContainer = ({ title, onRemove }) => {
+
+
+const SelectedContextContainer = ({ title, type, onRemove }) => {
     const [isHovered, setIsHovered] = (0,react__WEBPACK_IMPORTED_MODULE_0__.useState)(false);
+    let icon = react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_RuleIcon__WEBPACK_IMPORTED_MODULE_2__["default"], null);
+    switch (type) {
+        case 'rule':
+            icon = react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_RuleIcon__WEBPACK_IMPORTED_MODULE_2__["default"], null);
+            break;
+        case 'variable':
+            icon = react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_CodeIcon__WEBPACK_IMPORTED_MODULE_3__["default"], null);
+            break;
+        case 'db':
+            icon = react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_DatabaseIcon__WEBPACK_IMPORTED_MODULE_4__["default"], null);
+            break;
+    }
     return (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("button", { className: "selected-context-container", onMouseEnter: () => setIsHovered(true), onMouseLeave: () => setIsHovered(false), "data-testid": "selected-context-container" },
-        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: `icon`, onClick: () => onRemove(), title: isHovered ? "Remove rule" : "Selected rule" }, isHovered ? (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("span", { className: "icon" }, "\u2715")) : (react__WEBPACK_IMPORTED_MODULE_0___default().createElement(_icons_RuleIcon__WEBPACK_IMPORTED_MODULE_2__["default"], null))),
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: `icon`, onClick: () => onRemove(), title: isHovered ? "Remove rule" : "Selected rule" }, isHovered ? (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);
@@ -6675,16 +7021,69 @@ __webpack_require__.r(__webpack_exports__);
 
 const AlertIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "14", height: "14", viewBox: "0 0 96 84", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
     react__WEBPACK_IMPORTED_MODULE_0___default().createElement("g", { fill: "currentColor" },
-        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M94.0805 67.8053L57.025 5.49763C55.1179 2.28364 51.657 0.313801 47.9194 0.313801C44.1818 0.313801 40.7209 2.28355 38.8137 5.49763L1.91856 67.8053C0.0350518 71.0665 0.0267897 75.0839 1.89611 78.3535C3.76543 81.6232 7.23119 83.6541 10.9969 83.6862H85.1079C88.8549 83.6177 92.2873 81.5723 94.1307 78.3088C95.974 75.0452 95.9546 71.0513 94.0805 67.8053ZM48.0531 73.0989C46.6493 73.0989 45.3023 72.5412 44.3097 71.5486C43.3172 70.5561 42.7594 69.209 42.7594 67.8053C42.7594 66.4015 43.3172 65.0545 44.3097 64.0619C45.3023 63.0694 46.6493 62.5116 48.0531 62.5116C49.4568 62.5116 50.8039 63.0694 51.7964 64.0619C52.789 65.0545 53.3467 66.4015 53.3467 67.8053C53.3467 69.209 52.789 70.5561 51.7964 71.5486C50.8039 72.5412 49.4568 73.0989 48.0531 73.0989ZM53.3467 49.2775C53.3467 51.1693 52.3376 52.9168 50.6999 53.8621C49.0622 54.8074 47.044 54.8074 45.4063 53.8621C43.7686 52.9168 42.7594 51.1692 42.7594 49.2775V28.103C42.7594 26.2112 43.7685 24.4637 45.4063 23.5184C47.044 22.5731 49.0622 22.5731 50.6999 23.5184C52.3376 24.4637 53.3467 26.2113 53.3467 28.103V49.2775Z" }))));
-/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (AlertIcon);
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M94.0805 67.8053L57.025 5.49763C55.1179 2.28364 51.657 0.313801 47.9194 0.313801C44.1818 0.313801 40.7209 2.28355 38.8137 5.49763L1.91856 67.8053C0.0350518 71.0665 0.0267897 75.0839 1.89611 78.3535C3.76543 81.6232 7.23119 83.6541 10.9969 83.6862H85.1079C88.8549 83.6177 92.2873 81.5723 94.1307 78.3088C95.974 75.0452 95.9546 71.0513 94.0805 67.8053ZM48.0531 73.0989C46.6493 73.0989 45.3023 72.5412 44.3097 71.5486C43.3172 70.5561 42.7594 69.209 42.7594 67.8053C42.7594 66.4015 43.3172 65.0545 44.3097 64.0619C45.3023 63.0694 46.6493 62.5116 48.0531 62.5116C49.4568 62.5116 50.8039 63.0694 51.7964 64.0619C52.789 65.0545 53.3467 66.4015 53.3467 67.8053C53.3467 69.209 52.789 70.5561 51.7964 71.5486C50.8039 72.5412 49.4568 73.0989 48.0531 73.0989ZM53.3467 49.2775C53.3467 51.1693 52.3376 52.9168 50.6999 53.8621C49.0622 54.8074 47.044 54.8074 45.4063 53.8621C43.7686 52.9168 42.7594 51.1692 42.7594 49.2775V28.103C42.7594 26.2112 43.7685 24.4637 45.4063 23.5184C47.044 22.5731 49.0622 22.5731 50.6999 23.5184C52.3376 24.4637 53.3467 26.2113 53.3467 28.103V49.2775Z" }))));
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (AlertIcon);
+
+
+/***/ }),
+
+/***/ "./lib/icons/CodeIcon.js":
+/*!*******************************!*\
+  !*** ./lib/icons/CodeIcon.js ***!
+  \*******************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+const CodeIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "14", height: "14", viewBox: "0 0 20 14", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
+    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("g", { fill: "currentColor" },
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M12.3162 0.0513449C12.8027 0.213518 13.0817 0.713391 12.9792 1.20347L12.9487 1.31624L8.94868 13.3162C8.77403 13.8402 8.20773 14.1233 7.68378 13.9487C7.19726 13.7865 6.91835 13.2866 7.02082 12.7966L7.05134 12.6838L11.0513 0.683791C11.226 0.159848 11.7923 -0.123303 12.3162 0.0513449ZM5.79999 3.40001C6.10587 3.80785 6.05312 4.37331 5.69568 4.71852L5.59998 4.8L2.667 7.00001L5.59998 9.20003C6.00782 9.50591 6.1155 10.0635 5.86902 10.495L5.79999 10.6C5.49411 11.0079 4.9365 11.1155 4.50501 10.869L4.40003 10.8L0.40003 7.8C-0.0977483 7.42667 -0.130987 6.7049 0.30042 6.28511L0.40003 6.20003L4.40003 3.20003C4.84186 2.86866 5.46862 2.95818 5.79999 3.40001ZM15.495 3.131L15.6 3.20003L19.6 6.20003C20.0978 6.57336 20.131 7.29513 19.6996 7.71492L19.6 7.8L15.6 10.8C15.1582 11.1314 14.5314 11.0418 14.2 10.6C13.8941 10.1922 13.9469 9.62672 14.3043 9.28151L14.4 9.20003L17.333 7.00001L14.4 4.8C13.9922 4.49412 13.8845 3.93654 14.131 3.50505L14.2 3.40001C14.5059 2.99217 15.0635 2.88452 15.495 3.131Z" }))));
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (CodeIcon);
+
+
+/***/ }),
+
+/***/ "./lib/icons/CopyIcon.js":
+/*!*******************************!*\
+  !*** ./lib/icons/CopyIcon.js ***!
+  \*******************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! react */ "webpack/sharing/consume/default/react");
+/* harmony import */ var react__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(react__WEBPACK_IMPORTED_MODULE_0__);
+/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+const CopyIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "1em", height: "1em", viewBox: "0 0 10 11", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
+    react__WEBPACK_IMPORTED_MODULE_0___default().createElement("g", { fill: "currentColor" },
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M3.12502 4.03283H4.27968C4.35507 4.02072 4.4082 3.95276 4.40156 3.87658V2.59542C4.25976 2.64425 4.13359 2.72979 4.03593 2.84385L3.32343 3.65323C3.22929 3.76299 3.16172 3.89307 3.125 4.03291L3.12502 4.03283Z" }),
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M7.75784 2.55308H4.70626V3.87342C4.71524 4.12263 4.5211 4.33201 4.27188 4.34217H3.0813V9.2C3.08208 9.56993 3.38012 9.8707 3.75005 9.875H7.75788C7.93757 9.87578 8.11022 9.80508 8.23757 9.67852C8.3653 9.55195 8.43717 9.37969 8.43757 9.2V3.22808C8.43718 3.0484 8.3653 2.87613 8.23757 2.74957C8.11022 2.623 7.93757 2.5523 7.75788 2.55308H7.75784Z" }),
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M3.80163 2.62967C4.01452 2.38553 4.32155 2.24373 4.64538 2.2406H6.91879V1.79998C6.91801 1.43005 6.61997 1.12928 6.25004 1.12498H3.19221V2.44531C3.20119 2.69453 3.00705 2.9039 2.75783 2.91406H1.5625V7.77189C1.56328 8.14416 1.86523 8.44611 2.2375 8.44689H2.76407V4.30631C2.76172 3.98795 2.87578 3.67974 3.08438 3.43914L3.80163 2.62967Z" }),
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M2.75783 2.60467C2.83322 2.59256 2.88635 2.52459 2.87971 2.44842V1.16725C2.74182 1.21569 2.61916 1.29928 2.52346 1.40943L1.80783 2.21881C1.71096 2.33013 1.64065 2.46217 1.60315 2.60474L2.75783 2.60467Z" }))));
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (CopyIcon);
 
 
 /***/ }),
 
-/***/ "./lib/icons/CodeIcon.js":
-/*!*******************************!*\
-  !*** ./lib/icons/CodeIcon.js ***!
-  \*******************************/
+/***/ "./lib/icons/DatabaseIcon.js":
+/*!***********************************!*\
+  !*** ./lib/icons/DatabaseIcon.js ***!
+  \***********************************/
 /***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
 
 __webpack_require__.r(__webpack_exports__);
@@ -6698,18 +7097,18 @@ __webpack_require__.r(__webpack_exports__);
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
  */
 
-const CodeIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "14", height: "14", viewBox: "0 0 20 14", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
+const DatabaseIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "10", height: "10", viewBox: "0 0 100 100", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
     react__WEBPACK_IMPORTED_MODULE_0___default().createElement("g", { fill: "currentColor" },
-        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M12.3162 0.0513449C12.8027 0.213518 13.0817 0.713391 12.9792 1.20347L12.9487 1.31624L8.94868 13.3162C8.77403 13.8402 8.20773 14.1233 7.68378 13.9487C7.19726 13.7865 6.91835 13.2866 7.02082 12.7966L7.05134 12.6838L11.0513 0.683791C11.226 0.159848 11.7923 -0.123303 12.3162 0.0513449ZM5.79999 3.40001C6.10587 3.80785 6.05312 4.37331 5.69568 4.71852L5.59998 4.8L2.667 7.00001L5.59998 9.20003C6.00782 9.50591 6.1155 10.0635 5.86902 10.495L5.79999 10.6C5.49411 11.0079 4.9365 11.1155 4.50501 10.869L4.40003 10.8L0.40003 7.8C-0.0977483 7.42667 -0.130987 6.7049 0.30042 6.28511L0.40003 6.20003L4.40003 3.20003C4.84186 2.86866 5.46862 2.95818 5.79999 3.40001ZM15.495 3.131L15.6 3.20003L19.6 6.20003C20.0978 6.57336 20.131 7.29513 19.6996 7.71492L19.6 7.8L15.6 10.8C15.1582 11.1314 14.5314 11.0418 14.2 10.6C13.8941 10.1922 13.9469 9.62672 14.3043 9.28151L14.4 9.20003L17.333 7.00001L14.4 4.8C13.9922 4.49412 13.8845 3.93654 14.131 3.50505L14.2 3.40001C14.5059 2.99217 15.0635 2.88452 15.495 3.131Z" }))));
-/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (CodeIcon);
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M50 43.839C71.447 43.839 89.737 41.166 96.839 37.411C98.881 38.49 100 39.659 100 40.88V59.12C100 64.586 77.614 69.017 50 69.017C22.386 69.017 0 64.585 0 59.12V40.88C0 39.659 1.12 38.49 3.161 37.411C10.263 41.166 28.553 43.839 50 43.839ZM3.161 68.393C1.12 69.473 0 70.642 0 71.862V90.102C0 95.568 22.386 100 50 100C77.614 100 100 95.568 100 90.103V71.863C100 70.642 98.881 69.473 96.839 68.393C89.737 72.147 71.447 74.821 50 74.821C28.553 74.821 10.263 72.147 3.161 68.393ZM50 0C22.386 0 0 4.432 0 9.898V28.137C0 33.604 22.386 38.035 50 38.035C77.614 38.035 100 33.604 100 28.137V9.898C100 4.432 77.614 0 50 0Z" }))));
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (DatabaseIcon);
 
 
 /***/ }),
 
-/***/ "./lib/icons/CopyIcon.js":
-/*!*******************************!*\
-  !*** ./lib/icons/CopyIcon.js ***!
-  \*******************************/
+/***/ "./lib/icons/DatabaseOutlineIcon.js":
+/*!******************************************!*\
+  !*** ./lib/icons/DatabaseOutlineIcon.js ***!
+  \******************************************/
 /***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
 
 __webpack_require__.r(__webpack_exports__);
@@ -6723,13 +7122,10 @@ __webpack_require__.r(__webpack_exports__);
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
  */
 
-const CopyIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "1em", height: "1em", viewBox: "0 0 10 11", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
+const DatabaseOutlineIcon = () => (react__WEBPACK_IMPORTED_MODULE_0___default().createElement("svg", { width: "14", height: "14", viewBox: "0 0 82 96", fill: "none", xmlns: "http://www.w3.org/2000/svg" },
     react__WEBPACK_IMPORTED_MODULE_0___default().createElement("g", { fill: "currentColor" },
-        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M3.12502 4.03283H4.27968C4.35507 4.02072 4.4082 3.95276 4.40156 3.87658V2.59542C4.25976 2.64425 4.13359 2.72979 4.03593 2.84385L3.32343 3.65323C3.22929 3.76299 3.16172 3.89307 3.125 4.03291L3.12502 4.03283Z" }),
-        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M7.75784 2.55308H4.70626V3.87342C4.71524 4.12263 4.5211 4.33201 4.27188 4.34217H3.0813V9.2C3.08208 9.56993 3.38012 9.8707 3.75005 9.875H7.75788C7.93757 9.87578 8.11022 9.80508 8.23757 9.67852C8.3653 9.55195 8.43717 9.37969 8.43757 9.2V3.22808C8.43718 3.0484 8.3653 2.87613 8.23757 2.74957C8.11022 2.623 7.93757 2.5523 7.75788 2.55308H7.75784Z" }),
-        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M3.80163 2.62967C4.01452 2.38553 4.32155 2.24373 4.64538 2.2406H6.91879V1.79998C6.91801 1.43005 6.61997 1.12928 6.25004 1.12498H3.19221V2.44531C3.20119 2.69453 3.00705 2.9039 2.75783 2.91406H1.5625V7.77189C1.56328 8.14416 1.86523 8.44611 2.2375 8.44689H2.76407V4.30631C2.76172 3.98795 2.87578 3.67974 3.08438 3.43914L3.80163 2.62967Z" }),
-        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M2.75783 2.60467C2.83322 2.59256 2.88635 2.52459 2.87971 2.44842V1.16725C2.74182 1.21569 2.61916 1.29928 2.52346 1.40943L1.80783 2.21881C1.71096 2.33013 1.64065 2.46217 1.60315 2.60474L2.75783 2.60467Z" }))));
-/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (CopyIcon);
+        react__WEBPACK_IMPORTED_MODULE_0___default().createElement("path", { d: "M41 95.5C61.0983 95.5 81.4446 90.198 81.4446 80.0651V15.9349C81.4403 5.80206 61.0963 0.5 41 0.5C20.9037 0.5 0.559864 5.80206 0.555621 15.9349L0.55542 80.0651C0.55542 90.198 20.9017 95.5 41 95.5ZM41 6.83333C63.0713 6.83333 75.1112 12.8476 75.1112 15.938C75.1112 19.0284 63.0713 25.0432 41 25.0432C18.9287 25.0432 6.88875 19.0284 6.88875 15.938C6.88875 12.8476 18.9287 6.83333 41 6.83333ZM6.88875 24.6315C14.4639 29.0832 27.7848 31.3765 41 31.3765C54.2152 31.3765 67.5361 29.0832 75.1112 24.6315V37.3079C75.1112 40.4003 63.0713 46.4198 41 46.4198C18.9287 46.4198 6.88875 40.4003 6.88875 37.3079L6.88875 24.6315ZM6.88875 46.0052C14.4639 50.4589 27.7848 52.7531 41 52.7531C54.2152 52.7531 67.5361 50.4589 75.1112 46.0052V58.6916C75.1112 61.7805 63.0713 67.7932 41 67.7932C18.9287 67.7932 6.88875 61.7805 6.88875 58.6916L6.88875 46.0052ZM6.88875 67.383C14.4639 71.8337 27.7848 74.1265 41 74.1265C54.2152 74.1265 67.5361 71.8337 75.1112 67.383V80.0651C75.1112 83.1544 63.0713 89.1667 41 89.1667C18.9287 89.1667 6.88875 83.1544 6.88875 80.0651L6.88875 67.383Z" }))));
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (DatabaseOutlineIcon);
 
 
 /***/ }),
@@ -7166,6 +7562,7 @@ __webpack_require__.r(__webpack_exports__);
 
 __webpack_require__.r(__webpack_exports__);
 /* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   getDatabaseConnections: () => (/* binding */ getDatabaseConnections),
 /* harmony export */   getRule: () => (/* binding */ getRule),
 /* harmony export */   getRules: () => (/* binding */ getRules),
 /* harmony export */   getSetting: () => (/* binding */ getSetting),
@@ -7249,6 +7646,18 @@ const getRules = async () => {
     }
     return resp.data || [];
 };
+/************************************
+
+DATABASE ENDPOINTS
+
+************************************/
+const getDatabaseConnections = async () => {
+    const resp = await (0,_utils__WEBPACK_IMPORTED_MODULE_0__.requestAPI)('db/connections');
+    if (resp.error) {
+        throw new Error(resp.error.message);
+    }
+    return resp.data || {};
+};
 
 
 /***/ }),
@@ -9522,13 +9931,13 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
 
 .agent-component-header {
     cursor: pointer;
-    padding: 10px;
+    padding: 10px var(--chat-taskpane-tool-call-horizontal-padding);
     color: var(--jp-ui-font-color2);
     display: flex;
     justify-content: space-between;
     align-items: center;
     border: 1px solid var(--jp-cell-editor-border-color);
-    border-radius: 5px;
+    border-radius: var(--chat-taskpane-item-border-radius);
     transition: background-color 0.2s ease, border-color 0.2s ease;
 }
 
@@ -9573,7 +9982,78 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
 
 .agent-component-header-expand-icon.expanded {
     transform: rotate(90deg);
-} `, "",{"version":3,"sources":["webpack://./style/AgentComponentHeader.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;IACI,eAAe;IACf,aAAa;IACb,+BAA+B;IAC/B,aAAa;IACb,8BAA8B;IAC9B,mBAAmB;IACnB,oDAAoD;IACpD,kBAAkB;IAClB,8DAA8D;AAClE;;AAEA;IACI,yCAAyC;AAC7C;;AAEA;IACI,YAAY;AAChB;;AAEA;IACI,YAAY;IACZ,0BAA0B;IAC1B,yCAAyC;AAC7C;;AAEA;IACI,YAAY;AAChB;;AAEA;IACI,yCAAyC;AAC7C;;AAEA;IACI,aAAa;IACb,mBAAmB;IACnB,SAAS;IACT,OAAO;IACP,YAAY;AAChB;;AAEA;IACI,cAAc;AAClB;;AAEA;IACI,cAAc;IACd,+BAA+B;AACnC;;AAEA;IACI,wBAAwB;AAC5B","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.agent-component-header {\n    cursor: pointer;\n    padding: 10px;\n    color: var(--jp-ui-font-color2);\n    display: flex;\n    justify-content: space-between;\n    align-items: center;\n    border: 1px solid var(--jp-cell-editor-border-color);\n    border-radius: 5px;\n    transition: background-color 0.2s ease, border-color 0.2s ease;\n}\n\n.agent-component-header:hover {\n    background-color: var(--jp-layout-color3);\n}\n\n.agent-component-header.no-border {\n    border: none;\n}\n\n.agent-component-header.expanded {\n    border: none;\n    border-radius: 5px 5px 0 0;\n    background-color: var(--jp-layout-color2);\n}\n\n.agent-component-header.error-fixup {\n    border: none;\n}\n\n.agent-component-header.error-fixup.expanded {\n    background-color: var(--jp-layout-color2);\n}\n\n.agent-component-header-content {\n    display: flex;\n    align-items: center;\n    gap: 10px;\n    flex: 1;\n    min-width: 0;\n}\n\n.agent-component-header-content svg {\n    flex-shrink: 0;\n}\n\n.agent-component-header-expand-icon {\n    flex-shrink: 0;\n    transition: transform 0.2s ease;\n}\n\n.agent-component-header-expand-icon.expanded {\n    transform: rotate(90deg);\n} "],"sourceRoot":""}]);
+} `, "",{"version":3,"sources":["webpack://./style/AgentComponentHeader.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;IACI,eAAe;IACf,+DAA+D;IAC/D,+BAA+B;IAC/B,aAAa;IACb,8BAA8B;IAC9B,mBAAmB;IACnB,oDAAoD;IACpD,sDAAsD;IACtD,8DAA8D;AAClE;;AAEA;IACI,yCAAyC;AAC7C;;AAEA;IACI,YAAY;AAChB;;AAEA;IACI,YAAY;IACZ,0BAA0B;IAC1B,yCAAyC;AAC7C;;AAEA;IACI,YAAY;AAChB;;AAEA;IACI,yCAAyC;AAC7C;;AAEA;IACI,aAAa;IACb,mBAAmB;IACnB,SAAS;IACT,OAAO;IACP,YAAY;AAChB;;AAEA;IACI,cAAc;AAClB;;AAEA;IACI,cAAc;IACd,+BAA+B;AACnC;;AAEA;IACI,wBAAwB;AAC5B","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.agent-component-header {\n    cursor: pointer;\n    padding: 10px var(--chat-taskpane-tool-call-horizontal-padding);\n    color: var(--jp-ui-font-color2);\n    display: flex;\n    justify-content: space-between;\n    align-items: center;\n    border: 1px solid var(--jp-cell-editor-border-color);\n    border-radius: var(--chat-taskpane-item-border-radius);\n    transition: background-color 0.2s ease, border-color 0.2s ease;\n}\n\n.agent-component-header:hover {\n    background-color: var(--jp-layout-color3);\n}\n\n.agent-component-header.no-border {\n    border: none;\n}\n\n.agent-component-header.expanded {\n    border: none;\n    border-radius: 5px 5px 0 0;\n    background-color: var(--jp-layout-color2);\n}\n\n.agent-component-header.error-fixup {\n    border: none;\n}\n\n.agent-component-header.error-fixup.expanded {\n    background-color: var(--jp-layout-color2);\n}\n\n.agent-component-header-content {\n    display: flex;\n    align-items: center;\n    gap: 10px;\n    flex: 1;\n    min-width: 0;\n}\n\n.agent-component-header-content svg {\n    flex-shrink: 0;\n}\n\n.agent-component-header-expand-icon {\n    flex-shrink: 0;\n    transition: transform 0.2s ease;\n}\n\n.agent-component-header-expand-icon.expanded {\n    transform: rotate(90deg);\n} "],"sourceRoot":""}]);
+// Exports
+/* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
+
+
+/***/ }),
+
+/***/ "./node_modules/css-loader/dist/cjs.js!./style/AssumptionTool.css":
+/*!************************************************************************!*\
+  !*** ./node_modules/css-loader/dist/cjs.js!./style/AssumptionTool.css ***!
+  \************************************************************************/
+/***/ ((module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var _node_modules_css_loader_dist_runtime_sourceMaps_js__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! ../node_modules/css-loader/dist/runtime/sourceMaps.js */ "./node_modules/css-loader/dist/runtime/sourceMaps.js");
+/* harmony import */ var _node_modules_css_loader_dist_runtime_sourceMaps_js__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(_node_modules_css_loader_dist_runtime_sourceMaps_js__WEBPACK_IMPORTED_MODULE_0__);
+/* harmony import */ var _node_modules_css_loader_dist_runtime_api_js__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! ../node_modules/css-loader/dist/runtime/api.js */ "./node_modules/css-loader/dist/runtime/api.js");
+/* harmony import */ var _node_modules_css_loader_dist_runtime_api_js__WEBPACK_IMPORTED_MODULE_1___default = /*#__PURE__*/__webpack_require__.n(_node_modules_css_loader_dist_runtime_api_js__WEBPACK_IMPORTED_MODULE_1__);
+// Imports
+
+
+var ___CSS_LOADER_EXPORT___ = _node_modules_css_loader_dist_runtime_api_js__WEBPACK_IMPORTED_MODULE_1___default()((_node_modules_css_loader_dist_runtime_sourceMaps_js__WEBPACK_IMPORTED_MODULE_0___default()));
+// Module
+___CSS_LOADER_EXPORT___.push([module.id, `/*
+ * Copyright (c) Saga Inc.
+ * Distributed under the terms of the GNU Affero General Public License v3.0 License.
+ */
+
+.assumption-tool-container {
+    color: var(--jp-ui-font-color2);
+    display: flex;
+    justify-content: space-between;
+    align-items: center;
+    margin-top: 10px;
+    margin-bottom: 10px;
+    position: relative;
+    padding: 8px var(--chat-taskpane-tool-call-horizontal-padding);
+    border-radius: var(--chat-taskpane-item-border-radius);
+    border: 1px solid var(--jp-cell-editor-border-color);
+    cursor: default;
+}
+
+.assumption-header-label {
+    position: absolute;
+    top: -10px;
+    left: 12px;
+    background-color: var(--chat-background-color);
+    padding: 2px 8px;
+    border-radius: 4px;
+    font-size: 12px;
+    color: var(--jp-ui-font-color2);
+    font-style: italic;
+}
+
+.assumption-content {
+    display: flex;
+    align-items: flex-start;
+    gap: 8px;
+    color: var(--jp-ui-font-color2);
+    font-size: 14px;
+    line-height: 1.4;
+    margin: 0;
+    padding: 10px 10px 5px 10px;
+}
+
+.assumption-content ul {
+    margin: 0;
+    padding-left: 10px;
+}`, "",{"version":3,"sources":["webpack://./style/AssumptionTool.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;IACI,+BAA+B;IAC/B,aAAa;IACb,8BAA8B;IAC9B,mBAAmB;IACnB,gBAAgB;IAChB,mBAAmB;IACnB,kBAAkB;IAClB,8DAA8D;IAC9D,sDAAsD;IACtD,oDAAoD;IACpD,eAAe;AACnB;;AAEA;IACI,kBAAkB;IAClB,UAAU;IACV,UAAU;IACV,8CAA8C;IAC9C,gBAAgB;IAChB,kBAAkB;IAClB,eAAe;IACf,+BAA+B;IAC/B,kBAAkB;AACtB;;AAEA;IACI,aAAa;IACb,uBAAuB;IACvB,QAAQ;IACR,+BAA+B;IAC/B,eAAe;IACf,gBAAgB;IAChB,SAAS;IACT,2BAA2B;AAC/B;;AAEA;IACI,SAAS;IACT,kBAAkB;AACtB","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.assumption-tool-container {\n    color: var(--jp-ui-font-color2);\n    display: flex;\n    justify-content: space-between;\n    align-items: center;\n    margin-top: 10px;\n    margin-bottom: 10px;\n    position: relative;\n    padding: 8px var(--chat-taskpane-tool-call-horizontal-padding);\n    border-radius: var(--chat-taskpane-item-border-radius);\n    border: 1px solid var(--jp-cell-editor-border-color);\n    cursor: default;\n}\n\n.assumption-header-label {\n    position: absolute;\n    top: -10px;\n    left: 12px;\n    background-color: var(--chat-background-color);\n    padding: 2px 8px;\n    border-radius: 4px;\n    font-size: 12px;\n    color: var(--jp-ui-font-color2);\n    font-style: italic;\n}\n\n.assumption-content {\n    display: flex;\n    align-items: flex-start;\n    gap: 8px;\n    color: var(--jp-ui-font-color2);\n    font-size: 14px;\n    line-height: 1.4;\n    margin: 0;\n    padding: 10px 10px 5px 10px;\n}\n\n.assumption-content ul {\n    margin: 0;\n    padding-left: 10px;\n}"],"sourceRoot":""}]);
 // Exports
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
 
@@ -9692,9 +10172,10 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
 
   /* Style the dropdown */
   display: flex;
-  justify-content: space-evenly;
-  align-items: baseline;
+  flex-direction: column;
   width: 100%;
+  border: 1px solid var(--jp-border-color1);
+  border-radius: 5px;
 }
 
 .chat-dropdown-list {
@@ -9716,6 +10197,9 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
   align-items: center;
   gap: 8px;
   color: var(--jp-content-font-color1);
+  overflow: hidden;
+  text-overflow: ellipsis;
+  white-space: nowrap;
 }
 
 .chat-dropdown-item:hover,
@@ -9723,18 +10207,6 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
   background-color: var(--jp-layout-color3);
 }
 
-.chat-dropdown-item:first-child {
-  /* Add rounded corners to the top of the first selected dropdown item */
-  border-top-left-radius: 5px;
-  border-top-right-radius: 5px;
-}
-
-.chat-dropdown-item:last-child {
-  /* Add rounded corners to the bottom of the last selected dropdown item */
-  border-bottom-left-radius: 5px;
-  border-bottom-right-radius: 5px;
-}
-
 .chat-dropdown-item-type {
   color: var(--jp-brand-color1);
   font-family: var(--jp-code-font-family);
@@ -9771,7 +10243,36 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
 .dropdown-item-disabled:hover {
   background-color: inherit;
 }
-`, "",{"version":3,"sources":["webpack://./style/ChatDropdown.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;EACE,4CAA4C;EAC5C,kBAAkB;EAClB,aAAa;EACb,YAAY;EACZ,kBAAkB;;EAElB,uBAAuB;EACvB,aAAa;EACb,6BAA6B;EAC7B,qBAAqB;EACrB,WAAW;AACb;;AAEA;EACE,kBAAkB;EAClB,WAAW;EACX,kBAAkB;EAClB,yCAAyC;EACzC,qBAAqB;EACrB,UAAU;EACV,SAAS;EACT,WAAW;EACX,wCAAwC;AAC1C;;AAEA;EACE,YAAY;EACZ,eAAe;EACf,aAAa;EACb,mBAAmB;EACnB,QAAQ;EACR,oCAAoC;AACtC;;AAEA;;EAEE,yCAAyC;AAC3C;;AAEA;EACE,uEAAuE;EACvE,2BAA2B;EAC3B,4BAA4B;AAC9B;;AAEA;EACE,yEAAyE;EACzE,8BAA8B;EAC9B,+BAA+B;AACjC;;AAEA;EACE,6BAA6B;EAC7B,uCAAuC;EACvC,eAAe;EACf,WAAW;EACX,mBAAmB;EACnB,gBAAgB;EAChB,uBAAuB;AACzB;;AAEA;EACE,eAAe;EACf,aAAa;EACb,mBAAmB;EACnB,mBAAmB;EACnB,gBAAgB;EAChB,uBAAuB;AACzB;;AAEA;EACE,8BAA8B;EAC9B,eAAe;EACf,gBAAgB;EAChB,mBAAmB;EACnB,gBAAgB;EAChB,uBAAuB;AACzB;;AAEA;EACE,YAAY;EACZ,mBAAmB;AACrB;;AAEA;EACE,yBAAyB;AAC3B","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.chat-dropdown {\n  /* Position the dropdown in the right place*/\n  position: absolute;\n  z-index: 9999;\n  bottom: 100%;\n  margin-bottom: 5px;\n\n  /* Style the dropdown */\n  display: flex;\n  justify-content: space-evenly;\n  align-items: baseline;\n  width: 100%;\n}\n\n.chat-dropdown-list {\n  position: relative;\n  border: 0px;\n  border-radius: 5px;\n  background-color: var(--jp-layout-color1);\n  list-style-type: none;\n  padding: 0;\n  margin: 0;\n  width: 100%;\n  box-shadow: 0 2px 4px rgba(0, 0, 0, 0.2);\n}\n\n.chat-dropdown-item {\n  padding: 8px;\n  cursor: pointer;\n  display: flex;\n  align-items: center;\n  gap: 8px;\n  color: var(--jp-content-font-color1);\n}\n\n.chat-dropdown-item:hover,\n.chat-dropdown-item.selected {\n  background-color: var(--jp-layout-color3);\n}\n\n.chat-dropdown-item:first-child {\n  /* Add rounded corners to the top of the first selected dropdown item */\n  border-top-left-radius: 5px;\n  border-top-right-radius: 5px;\n}\n\n.chat-dropdown-item:last-child {\n  /* Add rounded corners to the bottom of the last selected dropdown item */\n  border-bottom-left-radius: 5px;\n  border-bottom-right-radius: 5px;\n}\n\n.chat-dropdown-item-type {\n  color: var(--jp-brand-color1);\n  font-family: var(--jp-code-font-family);\n  min-width: 35px;\n  width: 35px;\n  white-space: nowrap;\n  overflow: hidden;\n  text-overflow: ellipsis;\n}\n\n.chat-dropdown-item-name {\n  font-size: 15px;\n  display: flex;\n  align-items: center;\n  white-space: nowrap;\n  overflow: hidden;\n  text-overflow: ellipsis;\n}\n\n.chat-dropdown-item-parent-df {\n  color: var(--muted-text-color);\n  font-size: 12px;\n  margin-left: 4px;\n  white-space: nowrap;\n  overflow: hidden;\n  text-overflow: ellipsis;\n}\n\n.dropdown-item-disabled {\n  opacity: 0.5;\n  cursor: not-allowed;\n}\n\n.dropdown-item-disabled:hover {\n  background-color: inherit;\n}\n"],"sourceRoot":""}]);
+
+.chat-dropdown-search {
+  padding: 8px;
+  background-color: var(--jp-layout-color1);
+  border-top-left-radius: 5px;
+  border-top-right-radius: 5px;
+}
+
+.chat-dropdown-search-input {
+  width: 100%;
+  padding: 6px 8px;
+  border-radius: 3px;
+  font-size: 12px;
+  font-family: var(--jp-ui-font-family);
+  background-color: var(--jp-layout-color1);
+  color: var(--jp-content-font-color1);
+  outline: none;
+  border: none;
+  box-sizing: border-box;
+}
+
+.chat-dropdown-search-input:focus {
+  outline: none;
+  border: none;
+}
+
+.chat-dropdown-search-input::placeholder {
+  color: var(--jp-content-font-color3);
+}
+`, "",{"version":3,"sources":["webpack://./style/ChatDropdown.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;EACE,4CAA4C;EAC5C,kBAAkB;EAClB,aAAa;EACb,YAAY;EACZ,kBAAkB;;EAElB,uBAAuB;EACvB,aAAa;EACb,sBAAsB;EACtB,WAAW;EACX,yCAAyC;EACzC,kBAAkB;AACpB;;AAEA;EACE,kBAAkB;EAClB,WAAW;EACX,kBAAkB;EAClB,yCAAyC;EACzC,qBAAqB;EACrB,UAAU;EACV,SAAS;EACT,WAAW;EACX,wCAAwC;AAC1C;;AAEA;EACE,YAAY;EACZ,eAAe;EACf,aAAa;EACb,mBAAmB;EACnB,QAAQ;EACR,oCAAoC;EACpC,gBAAgB;EAChB,uBAAuB;EACvB,mBAAmB;AACrB;;AAEA;;EAEE,yCAAyC;AAC3C;;AAEA;EACE,6BAA6B;EAC7B,uCAAuC;EACvC,eAAe;EACf,WAAW;EACX,mBAAmB;EACnB,gBAAgB;EAChB,uBAAuB;AACzB;;AAEA;EACE,eAAe;EACf,aAAa;EACb,mBAAmB;EACnB,mBAAmB;EACnB,gBAAgB;EAChB,uBAAuB;AACzB;;AAEA;EACE,8BAA8B;EAC9B,eAAe;EACf,gBAAgB;EAChB,mBAAmB;EACnB,gBAAgB;EAChB,uBAAuB;AACzB;;AAEA;EACE,YAAY;EACZ,mBAAmB;AACrB;;AAEA;EACE,yBAAyB;AAC3B;;AAEA;EACE,YAAY;EACZ,yCAAyC;EACzC,2BAA2B;EAC3B,4BAA4B;AAC9B;;AAEA;EACE,WAAW;EACX,gBAAgB;EAChB,kBAAkB;EAClB,eAAe;EACf,qCAAqC;EACrC,yCAAyC;EACzC,oCAAoC;EACpC,aAAa;EACb,YAAY;EACZ,sBAAsB;AACxB;;AAEA;EACE,aAAa;EACb,YAAY;AACd;;AAEA;EACE,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.chat-dropdown {\n  /* Position the dropdown in the right place*/\n  position: absolute;\n  z-index: 9999;\n  bottom: 100%;\n  margin-bottom: 5px;\n\n  /* Style the dropdown */\n  display: flex;\n  flex-direction: column;\n  width: 100%;\n  border: 1px solid var(--jp-border-color1);\n  border-radius: 5px;\n}\n\n.chat-dropdown-list {\n  position: relative;\n  border: 0px;\n  border-radius: 5px;\n  background-color: var(--jp-layout-color1);\n  list-style-type: none;\n  padding: 0;\n  margin: 0;\n  width: 100%;\n  box-shadow: 0 2px 4px rgba(0, 0, 0, 0.2);\n}\n\n.chat-dropdown-item {\n  padding: 8px;\n  cursor: pointer;\n  display: flex;\n  align-items: center;\n  gap: 8px;\n  color: var(--jp-content-font-color1);\n  overflow: hidden;\n  text-overflow: ellipsis;\n  white-space: nowrap;\n}\n\n.chat-dropdown-item:hover,\n.chat-dropdown-item.selected {\n  background-color: var(--jp-layout-color3);\n}\n\n.chat-dropdown-item-type {\n  color: var(--jp-brand-color1);\n  font-family: var(--jp-code-font-family);\n  min-width: 35px;\n  width: 35px;\n  white-space: nowrap;\n  overflow: hidden;\n  text-overflow: ellipsis;\n}\n\n.chat-dropdown-item-name {\n  font-size: 15px;\n  display: flex;\n  align-items: center;\n  white-space: nowrap;\n  overflow: hidden;\n  text-overflow: ellipsis;\n}\n\n.chat-dropdown-item-parent-df {\n  color: var(--muted-text-color);\n  font-size: 12px;\n  margin-left: 4px;\n  white-space: nowrap;\n  overflow: hidden;\n  text-overflow: ellipsis;\n}\n\n.dropdown-item-disabled {\n  opacity: 0.5;\n  cursor: not-allowed;\n}\n\n.dropdown-item-disabled:hover {\n  background-color: inherit;\n}\n\n.chat-dropdown-search {\n  padding: 8px;\n  background-color: var(--jp-layout-color1);\n  border-top-left-radius: 5px;\n  border-top-right-radius: 5px;\n}\n\n.chat-dropdown-search-input {\n  width: 100%;\n  padding: 6px 8px;\n  border-radius: 3px;\n  font-size: 12px;\n  font-family: var(--jp-ui-font-family);\n  background-color: var(--jp-layout-color1);\n  color: var(--jp-content-font-color1);\n  outline: none;\n  border: none;\n  box-sizing: border-box;\n}\n\n.chat-dropdown-search-input:focus {\n  outline: none;\n  border: none;\n}\n\n.chat-dropdown-search-input::placeholder {\n  color: var(--jp-content-font-color3);\n}\n"],"sourceRoot":""}]);
 // Exports
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
 
@@ -9841,7 +10342,6 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
 
 .active-cell-preview-container {
   padding: 10px;
-  background-color: var(--chat-user-message-background-color);
 }
 
 .chat-input-container:focus-within {
@@ -9864,7 +10364,24 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
 .chat-input:focus::placeholder {
   color: var(--jp-ui-font-color3);
 }
-`, "",{"version":3,"sources":["webpack://./style/ChatInput.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;EACE,gBAAgB;EAChB,SAAS;EACT,cAAc;EACd,gBAAgB;EAChB,2DAA2D;EAC3D;;;iCAG+B;EAC/B,WAAW;EACX,mBAAmB;EACnB,iDAAiD;AACnD;;AAEA;EACE,kBAAkB;EAClB,mBAAmB;AACrB;;AAEA;EACE,aAAa;EACb,YAAY;EACZ,YAAY;EACZ,WAAW;EACX,aAAa;EACb,kBAAkB;EAClB,sBAAsB;EACtB,yBAAyB;EACzB,2DAA2D;EAC3D,eAAe;EACf,gBAAgB;EAChB,qCAAqC;EACrC,gBAAgB;EAChB,iDAAiD;AACnD;;AAEA;EACE,aAAa;EACb,2DAA2D;AAC7D;;AAEA;EACE,2BAA2B;EAC3B;;;iCAG+B;AACjC;;AAEA;EACE,+BAA+B;AACjC;;AAEA;EACE,+BAA+B;EAC/B,2BAA2B;AAC7B;;AAEA;EACE,+BAA+B;AACjC","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.chat-input-container {\n  position: sticky;\n  bottom: 0;\n  margin: 10px 0;\n  margin-top: auto;\n  background-color: var(--chat-user-message-background-color);\n  box-shadow: \n    0 4px 12px rgba(0, 0, 0, 0.08),\n    0 2px 4px rgba(0, 0, 0, 0.06),\n    0 0 0 1px rgba(0, 0, 0, 0.02);\n  width: 100%;\n  border-radius: 12px;\n  transition: all 0.3s cubic-bezier(0.4, 0, 0.2, 1);\n}\n\n.chat-input-text-area-container {\n  position: relative;\n  height: min-content;\n}\n\n.chat-input {\n  outline: none;\n  border: none;\n  resize: none;\n  width: 100%;\n  padding: 10px;\n  overflow-y: hidden;\n  box-sizing: border-box;\n  flex-shrink: 0 !important;\n  background-color: var(--chat-user-message-background-color);\n  font-size: 12px;\n  font-weight: 400;\n  font-family: var(--jp-ui-font-family);\n  line-height: 1.2;\n  transition: all 0.2s cubic-bezier(0.4, 0, 0.2, 1);\n}\n\n.active-cell-preview-container {\n  padding: 10px;\n  background-color: var(--chat-user-message-background-color);\n}\n\n.chat-input-container:focus-within {\n  transform: translateY(-1px);\n  box-shadow: \n    0 2px 8px rgba(0, 0, 0, 0.12),\n    0 8px 16px rgba(0, 0, 0, 0.08),\n    0 0 0 1px rgba(0, 0, 0, 0.04);\n}\n\n.chat-input:focus {\n  color: var(--jp-ui-font-color1);\n}\n\n.chat-input::placeholder {\n  color: var(--jp-ui-font-color2);\n  transition: color 0.2s ease;\n}\n\n.chat-input:focus::placeholder {\n  color: var(--jp-ui-font-color3);\n}\n"],"sourceRoot":""}]);
+
+.context-button {
+  display: flex;
+  align-items: center;
+  background-color: var(--jp-layout-color2);
+  border: 1px solid var(--jp-border-color1);
+  border-radius: 3px;
+  padding: 4px 8px;
+  font-size: 12px;
+  height: var(--chat-context-button-height);
+  cursor: pointer;
+  color: var(--jp-content-font-color1);
+  transition: background-color 0.2s ease;
+}
+
+.context-button:hover {
+  background-color: var(--jp-layout-color3);
+}`, "",{"version":3,"sources":["webpack://./style/ChatInput.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;EACE,gBAAgB;EAChB,SAAS;EACT,cAAc;EACd,gBAAgB;EAChB,2DAA2D;EAC3D;;;iCAG+B;EAC/B,WAAW;EACX,mBAAmB;EACnB,iDAAiD;AACnD;;AAEA;EACE,kBAAkB;EAClB,mBAAmB;AACrB;;AAEA;EACE,aAAa;EACb,YAAY;EACZ,YAAY;EACZ,WAAW;EACX,aAAa;EACb,kBAAkB;EAClB,sBAAsB;EACtB,yBAAyB;EACzB,2DAA2D;EAC3D,eAAe;EACf,gBAAgB;EAChB,qCAAqC;EACrC,gBAAgB;EAChB,iDAAiD;AACnD;;AAEA;EACE,aAAa;AACf;;AAEA;EACE,2BAA2B;EAC3B;;;iCAG+B;AACjC;;AAEA;EACE,+BAA+B;AACjC;;AAEA;EACE,+BAA+B;EAC/B,2BAA2B;AAC7B;;AAEA;EACE,+BAA+B;AACjC;;AAEA;EACE,aAAa;EACb,mBAAmB;EACnB,yCAAyC;EACzC,yCAAyC;EACzC,kBAAkB;EAClB,gBAAgB;EAChB,eAAe;EACf,yCAAyC;EACzC,eAAe;EACf,oCAAoC;EACpC,sCAAsC;AACxC;;AAEA;EACE,yCAAyC;AAC3C","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.chat-input-container {\n  position: sticky;\n  bottom: 0;\n  margin: 10px 0;\n  margin-top: auto;\n  background-color: var(--chat-user-message-background-color);\n  box-shadow: \n    0 4px 12px rgba(0, 0, 0, 0.08),\n    0 2px 4px rgba(0, 0, 0, 0.06),\n    0 0 0 1px rgba(0, 0, 0, 0.02);\n  width: 100%;\n  border-radius: 12px;\n  transition: all 0.3s cubic-bezier(0.4, 0, 0.2, 1);\n}\n\n.chat-input-text-area-container {\n  position: relative;\n  height: min-content;\n}\n\n.chat-input {\n  outline: none;\n  border: none;\n  resize: none;\n  width: 100%;\n  padding: 10px;\n  overflow-y: hidden;\n  box-sizing: border-box;\n  flex-shrink: 0 !important;\n  background-color: var(--chat-user-message-background-color);\n  font-size: 12px;\n  font-weight: 400;\n  font-family: var(--jp-ui-font-family);\n  line-height: 1.2;\n  transition: all 0.2s cubic-bezier(0.4, 0, 0.2, 1);\n}\n\n.active-cell-preview-container {\n  padding: 10px;\n}\n\n.chat-input-container:focus-within {\n  transform: translateY(-1px);\n  box-shadow: \n    0 2px 8px rgba(0, 0, 0, 0.12),\n    0 8px 16px rgba(0, 0, 0, 0.08),\n    0 0 0 1px rgba(0, 0, 0, 0.04);\n}\n\n.chat-input:focus {\n  color: var(--jp-ui-font-color1);\n}\n\n.chat-input::placeholder {\n  color: var(--jp-ui-font-color2);\n  transition: color 0.2s ease;\n}\n\n.chat-input:focus::placeholder {\n  color: var(--jp-ui-font-color3);\n}\n\n.context-button {\n  display: flex;\n  align-items: center;\n  background-color: var(--jp-layout-color2);\n  border: 1px solid var(--jp-border-color1);\n  border-radius: 3px;\n  padding: 4px 8px;\n  font-size: 12px;\n  height: var(--chat-context-button-height);\n  cursor: pointer;\n  color: var(--jp-content-font-color1);\n  transition: background-color 0.2s ease;\n}\n\n.context-button:hover {\n  background-color: var(--jp-layout-color3);\n}"],"sourceRoot":""}]);
 // Exports
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
 
@@ -9901,7 +10418,7 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
   box-sizing: border-box;
   width: 100%;
   font-size: 14px;
-  border-radius: 5px;
+  border-radius: var(--chat-taskpane-item-border-radius);
   padding: 0 var(--chat-taskpane-item-indent);
 }
 
@@ -9928,7 +10445,7 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
   flex-direction: row;
   gap: 8px;
   margin-top: 10px;
-  margin-bottom: 10px;
+  margin-bottom: 4px;
   flex-wrap: wrap;
 }
 
@@ -10043,7 +10560,7 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
   color: var(--purple-700);
   text-decoration: underline;
 }
-`, "",{"version":3,"sources":["webpack://./style/ChatMessage.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;EACE,kBAAkB;EAClB,mBAAmB;EACnB,sBAAsB;EACtB,WAAW;EACX,eAAe;EACf,kBAAkB;EAClB,2CAA2C;AAC7C;;AAEA;EACE,mBAAmB;AACrB;;AAEA;EACE,qBAAqB;AACvB;;AAEA;EACE,2DAA2D;EAC3D,0CAA0C;EAC1C,aAAa;AACf;;AAEA;EACE,+CAA+C;AACjD;;AAEA;EACE,aAAa;EACb,mBAAmB;EACnB,QAAQ;EACR,gBAAgB;EAChB,mBAAmB;EACnB,eAAe;AACjB;;AAEA;EACE,gBAAgB;EAChB,sBAAsB;EACtB,SAAS;EACT,gBAAgB;AAClB;;AAEA;EACE,mBAAmB;AACrB;;AAEA,qCAAqC;AACrC;EACE,kBAAkB;EAClB,WAAW;EACX,UAAU;EACV,aAAa;EACb,kBAAkB;EAClB,gBAAgB,EAAE,0DAA0D;AAC9E;;AAEA,qCAAqC;AACrC;;EAEE,eAAe;EACf,UAAU;EACV;;qCAEmC;EACnC,yCAAyC;EACzC,YAAY;EACZ,aAAa;EACb,mBAAmB;EACnB,uBAAuB;EACvB,oCAAoC;AACtC;;AAEA,gCAAgC;AAChC;EACE,WAAW;EACX,YAAY;EACZ,UAAU;EACV,gBAAgB,EAAE,oCAAoC;AACxD;;AAEA,yCAAyC;AACzC;EACE,+CAA+C;AACjD;;AAEA,8FAA8F;AAC9F;;EAEE,gBAAgB;AAClB;;AAEA,2BAA2B;AAC3B;EACE,kBAAkB;EAClB,YAAY;EACZ,kBAAkB;AACpB;;AAEA,mBAAmB;AACnB;EACE,oCAAoC;AACtC;;AAEA,iBAAiB;AACjB;;EAEE,yCAAyC;AAC3C;;AAEA,kCAAkC;AAClC;;EAEE,UAAU;AACZ;;AAEA;EACE,aAAa;EACb,QAAQ;EACR,YAAY;AACd;;AAEA;EACE,iBAAiB;EACjB,kBAAkB;EAClB,yCAAyC;EACzC,oCAAoC;EACpC,yCAAyC;EACzC,eAAe;AACjB;;AAEA;EACE,yCAAyC;AAC3C;;AAEA;EACE,mCAAmC;EACnC,kBAAkB;EAClB,aAAa;EACb,mCAAmC;EACnC,sBAAsB;AACxB;;AAEA;EACE,wBAAwB;EACxB,0BAA0B;AAC5B","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.message {\n  position: relative;\n  height: min-content;\n  box-sizing: border-box;\n  width: 100%;\n  font-size: 14px;\n  border-radius: 5px;\n  padding: 0 var(--chat-taskpane-item-indent);\n}\n\n.message:not(.chat-input) {\n  margin-bottom: 20px;\n}\n\n.message:hover .message-start-editing-button {\n  opacity: 1 !important;\n}\n\n.message-user {\n  background-color: var(--chat-user-message-background-color);\n  color: var(--chat-user-message-font-color);\n  padding: 10px;\n}\n\n.message-assistant-chat {\n  color: var(--chat-assistant-message-font-color);\n}\n\n.chat-message-buttons {\n  display: flex;\n  flex-direction: row;\n  gap: 8px;\n  margin-top: 10px;\n  margin-bottom: 10px;\n  flex-wrap: wrap;\n}\n\n.chat-taskpane-smart-debug-error-message {\n  white-space: pre;\n  font-family: monospace;\n  margin: 0;\n  overflow-x: auto;\n}\n\n.message-text {\n  align-items: center;\n}\n\n/* Message Action Buttons Container */\n.message-action-buttons {\n  position: absolute;\n  bottom: 8px;\n  right: 8px;\n  display: flex;\n  border-radius: 4px;\n  overflow: hidden; /* Ensures inner buttons don't break the rounded corners */\n}\n\n/* Common styles for action buttons */\n.message-action-buttons button,\n.message-start-editing-button {\n  cursor: pointer;\n  opacity: 0;\n  transition:\n    opacity 0.2s ease-in-out,\n    background-color 0.2s ease-in-out;\n  background-color: var(--jp-layout-color3);\n  border: none;\n  display: flex;\n  align-items: center;\n  justify-content: center;\n  color: var(--jp-content-font-color1);\n}\n\n/* Action buttons in the group */\n.message-action-buttons button {\n  width: 28px;\n  height: 28px;\n  padding: 0;\n  border-radius: 0; /* Remove individual button radius */\n}\n\n/* Add subtle separator between buttons */\n.message-action-buttons button:not(:last-child) {\n  border-right: 1px solid var(--jp-border-color1);\n}\n\n/* Remove the individual button radius styles since we're handling it at the container level */\n.message-action-buttons button:first-child,\n.message-action-buttons button:last-child {\n  border-radius: 0;\n}\n\n/* Standalone edit button */\n.message-start-editing-button {\n  position: relative;\n  padding: 4px;\n  border-radius: 4px;\n}\n\n/* SVG icon color */\n.message-action-buttons button svg {\n  color: var(--jp-content-font-color1);\n}\n\n/* Hover states */\n.message-action-buttons button:hover,\n.message-start-editing-button:hover {\n  background-color: var(--jp-layout-color4);\n}\n\n/* Show buttons on message hover */\n.message:hover .message-action-buttons button,\n.message:hover .message-start-editing-button {\n  opacity: 1;\n}\n\n.message-edit-buttons {\n  display: flex;\n  gap: 8px;\n  padding: 5px;\n}\n\n.message-edit-buttons button {\n  padding: 4px 12px;\n  border-radius: 4px;\n  background-color: var(--jp-layout-color1);\n  color: var(--jp-content-font-color1);\n  border: 1px solid var(--jp-border-color1);\n  cursor: pointer;\n}\n\n.message-edit-buttons button:hover {\n  background-color: var(--jp-layout-color2);\n}\n\n.chat-message-alert {\n  background-color: var(--purple-300);\n  border-radius: 5px;\n  padding: 10px;\n  border: 1px solid var(--purple-500);\n  color: var(--grey-900);\n}\n\n.chat-message-alert a {\n  color: var(--purple-700);\n  text-decoration: underline;\n}\n"],"sourceRoot":""}]);
+`, "",{"version":3,"sources":["webpack://./style/ChatMessage.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;EACE,kBAAkB;EAClB,mBAAmB;EACnB,sBAAsB;EACtB,WAAW;EACX,eAAe;EACf,sDAAsD;EACtD,2CAA2C;AAC7C;;AAEA;EACE,mBAAmB;AACrB;;AAEA;EACE,qBAAqB;AACvB;;AAEA;EACE,2DAA2D;EAC3D,0CAA0C;EAC1C,aAAa;AACf;;AAEA;EACE,+CAA+C;AACjD;;AAEA;EACE,aAAa;EACb,mBAAmB;EACnB,QAAQ;EACR,gBAAgB;EAChB,kBAAkB;EAClB,eAAe;AACjB;;AAEA;EACE,gBAAgB;EAChB,sBAAsB;EACtB,SAAS;EACT,gBAAgB;AAClB;;AAEA;EACE,mBAAmB;AACrB;;AAEA,qCAAqC;AACrC;EACE,kBAAkB;EAClB,WAAW;EACX,UAAU;EACV,aAAa;EACb,kBAAkB;EAClB,gBAAgB,EAAE,0DAA0D;AAC9E;;AAEA,qCAAqC;AACrC;;EAEE,eAAe;EACf,UAAU;EACV;;qCAEmC;EACnC,yCAAyC;EACzC,YAAY;EACZ,aAAa;EACb,mBAAmB;EACnB,uBAAuB;EACvB,oCAAoC;AACtC;;AAEA,gCAAgC;AAChC;EACE,WAAW;EACX,YAAY;EACZ,UAAU;EACV,gBAAgB,EAAE,oCAAoC;AACxD;;AAEA,yCAAyC;AACzC;EACE,+CAA+C;AACjD;;AAEA,8FAA8F;AAC9F;;EAEE,gBAAgB;AAClB;;AAEA,2BAA2B;AAC3B;EACE,kBAAkB;EAClB,YAAY;EACZ,kBAAkB;AACpB;;AAEA,mBAAmB;AACnB;EACE,oCAAoC;AACtC;;AAEA,iBAAiB;AACjB;;EAEE,yCAAyC;AAC3C;;AAEA,kCAAkC;AAClC;;EAEE,UAAU;AACZ;;AAEA;EACE,aAAa;EACb,QAAQ;EACR,YAAY;AACd;;AAEA;EACE,iBAAiB;EACjB,kBAAkB;EAClB,yCAAyC;EACzC,oCAAoC;EACpC,yCAAyC;EACzC,eAAe;AACjB;;AAEA;EACE,yCAAyC;AAC3C;;AAEA;EACE,mCAAmC;EACnC,kBAAkB;EAClB,aAAa;EACb,mCAAmC;EACnC,sBAAsB;AACxB;;AAEA;EACE,wBAAwB;EACxB,0BAA0B;AAC5B","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.message {\n  position: relative;\n  height: min-content;\n  box-sizing: border-box;\n  width: 100%;\n  font-size: 14px;\n  border-radius: var(--chat-taskpane-item-border-radius);\n  padding: 0 var(--chat-taskpane-item-indent);\n}\n\n.message:not(.chat-input) {\n  margin-bottom: 20px;\n}\n\n.message:hover .message-start-editing-button {\n  opacity: 1 !important;\n}\n\n.message-user {\n  background-color: var(--chat-user-message-background-color);\n  color: var(--chat-user-message-font-color);\n  padding: 10px;\n}\n\n.message-assistant-chat {\n  color: var(--chat-assistant-message-font-color);\n}\n\n.chat-message-buttons {\n  display: flex;\n  flex-direction: row;\n  gap: 8px;\n  margin-top: 10px;\n  margin-bottom: 4px;\n  flex-wrap: wrap;\n}\n\n.chat-taskpane-smart-debug-error-message {\n  white-space: pre;\n  font-family: monospace;\n  margin: 0;\n  overflow-x: auto;\n}\n\n.message-text {\n  align-items: center;\n}\n\n/* Message Action Buttons Container */\n.message-action-buttons {\n  position: absolute;\n  bottom: 8px;\n  right: 8px;\n  display: flex;\n  border-radius: 4px;\n  overflow: hidden; /* Ensures inner buttons don't break the rounded corners */\n}\n\n/* Common styles for action buttons */\n.message-action-buttons button,\n.message-start-editing-button {\n  cursor: pointer;\n  opacity: 0;\n  transition:\n    opacity 0.2s ease-in-out,\n    background-color 0.2s ease-in-out;\n  background-color: var(--jp-layout-color3);\n  border: none;\n  display: flex;\n  align-items: center;\n  justify-content: center;\n  color: var(--jp-content-font-color1);\n}\n\n/* Action buttons in the group */\n.message-action-buttons button {\n  width: 28px;\n  height: 28px;\n  padding: 0;\n  border-radius: 0; /* Remove individual button radius */\n}\n\n/* Add subtle separator between buttons */\n.message-action-buttons button:not(:last-child) {\n  border-right: 1px solid var(--jp-border-color1);\n}\n\n/* Remove the individual button radius styles since we're handling it at the container level */\n.message-action-buttons button:first-child,\n.message-action-buttons button:last-child {\n  border-radius: 0;\n}\n\n/* Standalone edit button */\n.message-start-editing-button {\n  position: relative;\n  padding: 4px;\n  border-radius: 4px;\n}\n\n/* SVG icon color */\n.message-action-buttons button svg {\n  color: var(--jp-content-font-color1);\n}\n\n/* Hover states */\n.message-action-buttons button:hover,\n.message-start-editing-button:hover {\n  background-color: var(--jp-layout-color4);\n}\n\n/* Show buttons on message hover */\n.message:hover .message-action-buttons button,\n.message:hover .message-start-editing-button {\n  opacity: 1;\n}\n\n.message-edit-buttons {\n  display: flex;\n  gap: 8px;\n  padding: 5px;\n}\n\n.message-edit-buttons button {\n  padding: 4px 12px;\n  border-radius: 4px;\n  background-color: var(--jp-layout-color1);\n  color: var(--jp-content-font-color1);\n  border: 1px solid var(--jp-border-color1);\n  cursor: pointer;\n}\n\n.message-edit-buttons button:hover {\n  background-color: var(--jp-layout-color2);\n}\n\n.chat-message-alert {\n  background-color: var(--purple-300);\n  border-radius: 5px;\n  padding: 10px;\n  border: 1px solid var(--purple-500);\n  color: var(--grey-900);\n}\n\n.chat-message-alert a {\n  color: var(--purple-700);\n  text-decoration: underline;\n}\n"],"sourceRoot":""}]);
 // Exports
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
 
@@ -10741,20 +11258,32 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
     display: flex;
     justify-content: space-between;
     align-items: center;
-    margin-bottom: 12px;
+    margin-bottom: 8px;
 }
 
-.connection-card-header h3 {
-    margin: 0;
-    font-size: 1.1em;
+.connection-alias {
+    font-size: 1.2em;
+    font-weight: bold;
     color: var(--jp-ui-font-color1);
+    margin: 0;
 }
 
 .connection-type {
-    font-size: 0.9em;
-    color: var(--jp-ui-font-color2);
+    background: var(--purple-100);
+    color: var(--purple-700);
+    border-radius: 12px;
+    padding: 2px 10px;
+    font-size: 0.85em;
+    font-weight: 500;
     text-transform: capitalize;
-    margin-bottom: 8px;
+    margin-left: 8px;
+}
+
+.connection-divider {
+    height: 1px;
+    background: var(--jp-border-color2);
+    margin: 8px 0 12px 0;
+    width: 100%;
 }
 
 .connection-details {
@@ -10794,7 +11323,7 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
 .confirmation-buttons {
     display: flex;
     gap: 8px;
-} `, "",{"version":3,"sources":["webpack://./style/DatabasePage.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;IACI,aAAa;IACb,4DAA4D;IAC5D,SAAS;IACT,gBAAgB;AACpB;;AAEA;IACI,mCAAmC;IACnC,yCAAyC;IACzC,kBAAkB;IAClB,aAAa;IACb,yBAAyB;IACzB,aAAa;IACb,sBAAsB;AAC1B;;AAEA;IACI,mCAAmC;AACvC;;AAEA;IACI,aAAa;IACb,8BAA8B;IAC9B,mBAAmB;IACnB,mBAAmB;AACvB;;AAEA;IACI,SAAS;IACT,gBAAgB;IAChB,+BAA+B;AACnC;;AAEA;IACI,gBAAgB;IAChB,+BAA+B;IAC/B,0BAA0B;IAC1B,kBAAkB;AACtB;;AAEA;IACI,YAAY;AAChB;;AAEA;IACI,aAAa;IACb,gBAAgB;IAChB,+BAA+B;AACnC;;AAEA;IACI,aAAa;IACb,gBAAgB;IAChB,QAAQ;AACZ;;AAEA;IACI,gBAAgB;IAChB,gBAAgB;AACpB;;AAEA;IACI,6BAA6B;AACjC;;AAEA;IACI,wCAAwC;IACxC,YAAY;AAChB;;AAEA;IACI,kBAAkB;AACtB;;AAEA;IACI,aAAa;IACb,QAAQ;AACZ","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.connections-grid {\n    display: grid;\n    grid-template-columns: repeat(auto-fill, minmax(300px, 1fr));\n    gap: 20px;\n    margin-top: 20px;\n}\n\n.connection-card {\n    background: var(--jp-layout-color0);\n    border: 1px solid var(--jp-border-color1);\n    border-radius: 8px;\n    padding: 16px;\n    transition: all 0.2s ease;\n    display: flex;\n    flex-direction: column;\n}\n\n.connection-card:hover {\n    border: 1px solid var(--purple-500);\n}\n\n.connection-card-header {\n    display: flex;\n    justify-content: space-between;\n    align-items: center;\n    margin-bottom: 12px;\n}\n\n.connection-card-header h3 {\n    margin: 0;\n    font-size: 1.1em;\n    color: var(--jp-ui-font-color1);\n}\n\n.connection-type {\n    font-size: 0.9em;\n    color: var(--jp-ui-font-color2);\n    text-transform: capitalize;\n    margin-bottom: 8px;\n}\n\n.connection-details {\n    flex-grow: 1;\n}\n\n.connection-details p {\n    margin: 4px 0;\n    font-size: 0.9em;\n    color: var(--jp-ui-font-color2);\n}\n\n.connection-actions {\n    display: flex;\n    margin-top: 12px;\n    gap: 8px;\n}\n\n.connection-actions button {\n    padding: 4px 8px;\n    font-size: 0.9em;\n}\n\n.connection-actions .delete-button {\n    color: var(--jp-error-color1);\n}\n\n.connection-actions .delete-button:hover {\n    background-color: var(--jp-error-color2);\n    color: white;\n}\n\n.delete-container {\n    position: relative;\n}\n\n.confirmation-buttons {\n    display: flex;\n    gap: 8px;\n} "],"sourceRoot":""}]);
+} `, "",{"version":3,"sources":["webpack://./style/DatabasePage.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;IACI,aAAa;IACb,4DAA4D;IAC5D,SAAS;IACT,gBAAgB;AACpB;;AAEA;IACI,mCAAmC;IACnC,yCAAyC;IACzC,kBAAkB;IAClB,aAAa;IACb,yBAAyB;IACzB,aAAa;IACb,sBAAsB;AAC1B;;AAEA;IACI,mCAAmC;AACvC;;AAEA;IACI,aAAa;IACb,8BAA8B;IAC9B,mBAAmB;IACnB,kBAAkB;AACtB;;AAEA;IACI,gBAAgB;IAChB,iBAAiB;IACjB,+BAA+B;IAC/B,SAAS;AACb;;AAEA;IACI,6BAA6B;IAC7B,wBAAwB;IACxB,mBAAmB;IACnB,iBAAiB;IACjB,iBAAiB;IACjB,gBAAgB;IAChB,0BAA0B;IAC1B,gBAAgB;AACpB;;AAEA;IACI,WAAW;IACX,mCAAmC;IACnC,oBAAoB;IACpB,WAAW;AACf;;AAEA;IACI,YAAY;AAChB;;AAEA;IACI,aAAa;IACb,gBAAgB;IAChB,+BAA+B;AACnC;;AAEA;IACI,aAAa;IACb,gBAAgB;IAChB,QAAQ;AACZ;;AAEA;IACI,gBAAgB;IAChB,gBAAgB;AACpB;;AAEA;IACI,6BAA6B;AACjC;;AAEA;IACI,wCAAwC;IACxC,YAAY;AAChB;;AAEA;IACI,kBAAkB;AACtB;;AAEA;IACI,aAAa;IACb,QAAQ;AACZ","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.connections-grid {\n    display: grid;\n    grid-template-columns: repeat(auto-fill, minmax(300px, 1fr));\n    gap: 20px;\n    margin-top: 20px;\n}\n\n.connection-card {\n    background: var(--jp-layout-color0);\n    border: 1px solid var(--jp-border-color1);\n    border-radius: 8px;\n    padding: 16px;\n    transition: all 0.2s ease;\n    display: flex;\n    flex-direction: column;\n}\n\n.connection-card:hover {\n    border: 1px solid var(--purple-500);\n}\n\n.connection-card-header {\n    display: flex;\n    justify-content: space-between;\n    align-items: center;\n    margin-bottom: 8px;\n}\n\n.connection-alias {\n    font-size: 1.2em;\n    font-weight: bold;\n    color: var(--jp-ui-font-color1);\n    margin: 0;\n}\n\n.connection-type {\n    background: var(--purple-100);\n    color: var(--purple-700);\n    border-radius: 12px;\n    padding: 2px 10px;\n    font-size: 0.85em;\n    font-weight: 500;\n    text-transform: capitalize;\n    margin-left: 8px;\n}\n\n.connection-divider {\n    height: 1px;\n    background: var(--jp-border-color2);\n    margin: 8px 0 12px 0;\n    width: 100%;\n}\n\n.connection-details {\n    flex-grow: 1;\n}\n\n.connection-details p {\n    margin: 4px 0;\n    font-size: 0.9em;\n    color: var(--jp-ui-font-color2);\n}\n\n.connection-actions {\n    display: flex;\n    margin-top: 12px;\n    gap: 8px;\n}\n\n.connection-actions button {\n    padding: 4px 8px;\n    font-size: 0.9em;\n}\n\n.connection-actions .delete-button {\n    color: var(--jp-error-color1);\n}\n\n.connection-actions .delete-button:hover {\n    background-color: var(--jp-error-color2);\n    color: white;\n}\n\n.delete-container {\n    position: relative;\n}\n\n.confirmation-buttons {\n    display: flex;\n    gap: 8px;\n} "],"sourceRoot":""}]);
 // Exports
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
 
@@ -11163,12 +11692,35 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
 .icon-button {
   display: flex;
   align-items: center;
+  text-align: center;
   justify-content: center;
   background-color: transparent;
   border: none;
   cursor: pointer;
+  position: relative;
+}
+
+.icon-button-hover:hover {
+  background-color: var(--jp-layout-color3);
+}
+
+.notification-dot {
+  position: absolute;
+  top: -1px;
+  right: 1px;
+  width: 8px;
+  height: 8px;
+  border-radius: 50%;
+  border: 1px solid white;
 }
-`, "",{"version":3,"sources":["webpack://./style/IconButton.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;EACE,aAAa;EACb,mBAAmB;EACnB,uBAAuB;EACvB,6BAA6B;EAC7B,YAAY;EACZ,eAAe;AACjB","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.icon-button {\n  display: flex;\n  align-items: center;\n  justify-content: center;\n  background-color: transparent;\n  border: none;\n  cursor: pointer;\n}\n"],"sourceRoot":""}]);
+
+.notification-dot-success {
+  background-color: var(--green-600);
+}
+
+.notification-dot-warning {
+  background-color: var(--yellow-600);
+}`, "",{"version":3,"sources":["webpack://./style/IconButton.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;EACE,aAAa;EACb,mBAAmB;EACnB,kBAAkB;EAClB,uBAAuB;EACvB,6BAA6B;EAC7B,YAAY;EACZ,eAAe;EACf,kBAAkB;AACpB;;AAEA;EACE,yCAAyC;AAC3C;;AAEA;EACE,kBAAkB;EAClB,SAAS;EACT,UAAU;EACV,UAAU;EACV,WAAW;EACX,kBAAkB;EAClB,uBAAuB;AACzB;;AAEA;EACE,kCAAkC;AACpC;;AAEA;EACE,mCAAmC;AACrC","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\n */\n\n.icon-button {\n  display: flex;\n  align-items: center;\n  text-align: center;\n  justify-content: center;\n  background-color: transparent;\n  border: none;\n  cursor: pointer;\n  position: relative;\n}\n\n.icon-button-hover:hover {\n  background-color: var(--jp-layout-color3);\n}\n\n.notification-dot {\n  position: absolute;\n  top: -1px;\n  right: 1px;\n  width: 8px;\n  height: 8px;\n  border-radius: 50%;\n  border: 1px solid white;\n}\n\n.notification-dot-success {\n  background-color: var(--green-600);\n}\n\n.notification-dot-warning {\n  background-color: var(--yellow-600);\n}"],"sourceRoot":""}]);
 // Exports
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
 
@@ -12126,36 +12678,56 @@ ___CSS_LOADER_EXPORT___.push([module.id, `/*
  * Distributed under the terms of the GNU Affero General Public License v3.0 License.
  */
 
+.context-container {
+  display: flex;
+  flex-wrap: wrap;
+  align-items: center;
+  gap: 4px;
+  padding: 8px 10px;
+}
+
+/* Consistent spacing for all children of context container */
+.context-container>* {
+  margin: 2px;
+}
+
 .selected-context-container {
-    display: flex;
-    align-items: center;
-    text-align: center;
-    background-color: var(--jp-layout-color2);
-    border-radius: 3px;
-    padding: 4px 8px;
-    margin: 4px;
+  display: flex;
+  align-items: center;
+  text-align: center;
+  background-color: var(--jp-layout-color2);
+  border-radius: 3px;
+  padding: 4px 8px;
+  margin: 2px;
 
-    font-size: 12px;
-    height: 20px;
-    width: fit-content;
+  font-size: 12px;
+  height: 20px;
+  width: fit-content;
 
-    border: 1px solid var(--jp-border-color1);
-    cursor: pointer;
+  border: 1px solid var(--jp-border-color1);
+  cursor: pointer;
 
-    &:hover {
-        background-color: var(--jp-layout-color3);
-    }
+  &:hover {
+    background-color: var(--jp-layout-color3);
+  }
 }
 
 .selected-context-container .rule-name {
-    margin-left: 6px;
-    color: var(--jp-content-font-color1);
+  margin-left: 6px;
+  color: var(--jp-content-font-color1);
+  white-space: nowrap;
+  overflow: hidden;
+  text-overflow: ellipsis;
+  max-width: 200px;
 }
 
 .selected-context-container .icon {
-    width: 12px;
-    height: 12px;
-}`, "",{"version":3,"sources":["webpack://./style/SelectedContextContainer.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;IACI,aAAa;IACb,mBAAmB;IACnB,kBAAkB;IAClB,yCAAyC;IACzC,kBAAkB;IAClB,gBAAgB;IAChB,WAAW;;IAEX,eAAe;IACf,YAAY;IACZ,kBAAkB;;IAElB,yCAAyC;IACzC,eAAe;;IAEf;QACI,yCAAyC;IAC7C;AACJ;;AAEA;IACI,gBAAgB;IAChB,oCAAoC;AACxC;;AAEA;IACI,WAAW;IACX,YAAY;AAChB","sourcesContent":["/*\n * Copyright (c) Saga Inc.\n * Distributed under the terms of the GNU Affero General Public License v3.0 License.\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: 4px;\n\n    font-size: 12px;\n    height: 20px;\n    width: fit-content;\n\n    border: 1px solid var(--jp-border-color1);\n    cursor: pointer;\n\n    &:hover {\n        background-color: var(--jp-layout-color3);\n    }\n}\n\n.selected-context-container .rule-name {\n    margin-left: 6px;\n    color: var(--jp-content-font-color1);\n}\n\n.selected-context-container .icon {\n    width: 12px;\n    height: 12px;\n}"],"sourceRoot":""}]);
+  display: flex;
+  flex-direction: column;
+  justify-content: center;
+  width: 12px;
+  max-height: 14px;
+}`, "",{"version":3,"sources":["webpack://./style/SelectedContextContainer.css"],"names":[],"mappings":"AAAA;;;EAGE;;AAEF;EACE,aAAa;EACb,eAAe;EACf,mBAAmB;EACnB,QAAQ;EACR,iBAAiB;AACnB;;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,kBAAkB;;EAElB,yCAAyC;EACzC,eAAe;;EAEf;IACE,yCAAyC;EAC3C;AACF;;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;AAClB","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}\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  width: fit-content;\n\n  border: 1px solid var(--jp-border-color1);\n  cursor: pointer;\n\n  &:hover {\n    background-color: var(--jp-layout-color3);\n  }\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}"],"sourceRoot":""}]);
 // Exports
 /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (___CSS_LOADER_EXPORT___);
 
@@ -13065,6 +13637,60 @@ var update = _node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js
        /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (_node_modules_css_loader_dist_cjs_js_AgentComponentHeader_css__WEBPACK_IMPORTED_MODULE_6__["default"] && _node_modules_css_loader_dist_cjs_js_AgentComponentHeader_css__WEBPACK_IMPORTED_MODULE_6__["default"].locals ? _node_modules_css_loader_dist_cjs_js_AgentComponentHeader_css__WEBPACK_IMPORTED_MODULE_6__["default"].locals : undefined);
 
 
+/***/ }),
+
+/***/ "./style/AssumptionTool.css":
+/*!**********************************!*\
+  !*** ./style/AssumptionTool.css ***!
+  \**********************************/
+/***/ ((__unused_webpack_module, __webpack_exports__, __webpack_require__) => {
+
+__webpack_require__.r(__webpack_exports__);
+/* harmony export */ __webpack_require__.d(__webpack_exports__, {
+/* harmony export */   "default": () => (__WEBPACK_DEFAULT_EXPORT__)
+/* harmony export */ });
+/* harmony import */ var _node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js__WEBPACK_IMPORTED_MODULE_0__ = __webpack_require__(/*! !../node_modules/style-loader/dist/runtime/injectStylesIntoStyleTag.js */ "./node_modules/style-loader/dist/runtime/injectStylesIntoStyleTag.js");
+/* harmony import */ var _node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js__WEBPACK_IMPORTED_MODULE_0___default = /*#__PURE__*/__webpack_require__.n(_node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js__WEBPACK_IMPORTED_MODULE_0__);
+/* harmony import */ var _node_modules_style_loader_dist_runtime_styleDomAPI_js__WEBPACK_IMPORTED_MODULE_1__ = __webpack_require__(/*! !../node_modules/style-loader/dist/runtime/styleDomAPI.js */ "./node_modules/style-loader/dist/runtime/styleDomAPI.js");
+/* harmony import */ var _node_modules_style_loader_dist_runtime_styleDomAPI_js__WEBPACK_IMPORTED_MODULE_1___default = /*#__PURE__*/__webpack_require__.n(_node_modules_style_loader_dist_runtime_styleDomAPI_js__WEBPACK_IMPORTED_MODULE_1__);
+/* harmony import */ var _node_modules_style_loader_dist_runtime_insertBySelector_js__WEBPACK_IMPORTED_MODULE_2__ = __webpack_require__(/*! !../node_modules/style-loader/dist/runtime/insertBySelector.js */ "./node_modules/style-loader/dist/runtime/insertBySelector.js");
+/* harmony import */ var _node_modules_style_loader_dist_runtime_insertBySelector_js__WEBPACK_IMPORTED_MODULE_2___default = /*#__PURE__*/__webpack_require__.n(_node_modules_style_loader_dist_runtime_insertBySelector_js__WEBPACK_IMPORTED_MODULE_2__);
+/* harmony import */ var _node_modules_style_loader_dist_runtime_setAttributesWithoutAttributes_js__WEBPACK_IMPORTED_MODULE_3__ = __webpack_require__(/*! !../node_modules/style-loader/dist/runtime/setAttributesWithoutAttributes.js */ "./node_modules/style-loader/dist/runtime/setAttributesWithoutAttributes.js");
+/* harmony import */ var _node_modules_style_loader_dist_runtime_setAttributesWithoutAttributes_js__WEBPACK_IMPORTED_MODULE_3___default = /*#__PURE__*/__webpack_require__.n(_node_modules_style_loader_dist_runtime_setAttributesWithoutAttributes_js__WEBPACK_IMPORTED_MODULE_3__);
+/* harmony import */ var _node_modules_style_loader_dist_runtime_insertStyleElement_js__WEBPACK_IMPORTED_MODULE_4__ = __webpack_require__(/*! !../node_modules/style-loader/dist/runtime/insertStyleElement.js */ "./node_modules/style-loader/dist/runtime/insertStyleElement.js");
+/* harmony import */ var _node_modules_style_loader_dist_runtime_insertStyleElement_js__WEBPACK_IMPORTED_MODULE_4___default = /*#__PURE__*/__webpack_require__.n(_node_modules_style_loader_dist_runtime_insertStyleElement_js__WEBPACK_IMPORTED_MODULE_4__);
+/* harmony import */ var _node_modules_style_loader_dist_runtime_styleTagTransform_js__WEBPACK_IMPORTED_MODULE_5__ = __webpack_require__(/*! !../node_modules/style-loader/dist/runtime/styleTagTransform.js */ "./node_modules/style-loader/dist/runtime/styleTagTransform.js");
+/* harmony import */ var _node_modules_style_loader_dist_runtime_styleTagTransform_js__WEBPACK_IMPORTED_MODULE_5___default = /*#__PURE__*/__webpack_require__.n(_node_modules_style_loader_dist_runtime_styleTagTransform_js__WEBPACK_IMPORTED_MODULE_5__);
+/* harmony import */ var _node_modules_css_loader_dist_cjs_js_AssumptionTool_css__WEBPACK_IMPORTED_MODULE_6__ = __webpack_require__(/*! !!../node_modules/css-loader/dist/cjs.js!./AssumptionTool.css */ "./node_modules/css-loader/dist/cjs.js!./style/AssumptionTool.css");
+
+      
+      
+      
+      
+      
+      
+      
+      
+      
+
+var options = {};
+
+options.styleTagTransform = (_node_modules_style_loader_dist_runtime_styleTagTransform_js__WEBPACK_IMPORTED_MODULE_5___default());
+options.setAttributes = (_node_modules_style_loader_dist_runtime_setAttributesWithoutAttributes_js__WEBPACK_IMPORTED_MODULE_3___default());
+
+      options.insert = _node_modules_style_loader_dist_runtime_insertBySelector_js__WEBPACK_IMPORTED_MODULE_2___default().bind(null, "head");
+    
+options.domAPI = (_node_modules_style_loader_dist_runtime_styleDomAPI_js__WEBPACK_IMPORTED_MODULE_1___default());
+options.insertStyleElement = (_node_modules_style_loader_dist_runtime_insertStyleElement_js__WEBPACK_IMPORTED_MODULE_4___default());
+
+var update = _node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js__WEBPACK_IMPORTED_MODULE_0___default()(_node_modules_css_loader_dist_cjs_js_AssumptionTool_css__WEBPACK_IMPORTED_MODULE_6__["default"], options);
+
+
+
+
+       /* harmony default export */ const __WEBPACK_DEFAULT_EXPORT__ = (_node_modules_css_loader_dist_cjs_js_AssumptionTool_css__WEBPACK_IMPORTED_MODULE_6__["default"] && _node_modules_css_loader_dist_cjs_js_AssumptionTool_css__WEBPACK_IMPORTED_MODULE_6__["default"].locals ? _node_modules_css_loader_dist_cjs_js_AssumptionTool_css__WEBPACK_IMPORTED_MODULE_6__["default"].locals : undefined);
+
+
 /***/ }),
 
 /***/ "./style/CTACarousel.css":
@@ -14472,4 +15098,4 @@ var update = _node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js
 /***/ })
 
 }]);
-//# sourceMappingURL=lib_index_js.a20772bc113422d0f505.js.map
+//# sourceMappingURL=lib_index_js.831f63b48760c7119b9b.js.map