mito-ai 0.1.31__py3-none-any.whl → 0.1.33__py3-none-any.whl

mito_ai/_version.py

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

{mito_ai-0.1.31.data → mito_ai-0.1.33.data}/data/share/jupyter/labextensions/mito_ai/build_log.json

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

{mito_ai-0.1.31.data → mito_ai-0.1.33.data}/data/share/jupyter/labextensions/mito_ai/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mito_ai",
-  "version": "0.1.31",
+  "version": "0.1.33",
   "description": "AI chat for JupyterLab",
   "keywords": [
     "jupyter",
@@ -138,7 +138,7 @@
     "outputDir": "mito_ai/labextension",
     "schemaDir": "schema",
     "_build": {
-      "load": "static/remoteEntry.1445025069f6d660a9b4.js",
+      "load": "static/remoteEntry.4f1d00fd0c58fcc05d8d.js",
       "extension": "./extension",
       "style": "./style"
     }

{mito_ai-0.1.31.data → mito_ai-0.1.33.data}/data/share/jupyter/labextensions/mito_ai/schemas/mito_ai/package.json.orig

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

mito_ai-0.1.31.data/data/share/jupyter/labextensions/mito_ai/static/lib_index_js.4a24ec8bdd6fde22593c.js → mito_ai-0.1.33.data/data/share/jupyter/labextensions/mito_ai/static/lib_index_js.281f4b9af60d620c6fb1.js

@@ -2475,8 +2475,7 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
         app.commands.notifyCommandChanged(_commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_CELL_TOOLBAR_ACCEPT_CODE);
         app.commands.notifyCommandChanged(_commands__WEBPACK_IMPORTED_MODULE_18__.COMMAND_MITO_AI_CELL_TOOLBAR_REJECT_CODE);
     };
-    // Create a WeakMap to store compartments per code cell
-    const codeDiffStripesCompartments = react__WEBPACK_IMPORTED_MODULE_2___default().useRef(new WeakMap());
+    const codeDiffStripesCompartments = react__WEBPACK_IMPORTED_MODULE_2___default().useRef(new Map());
     // Function to update the extensions of code cells
     const updateCodeCellsExtensions = (unifiedDiffLines) => {
         var _a;
@@ -2485,42 +2484,38 @@ const ChatTaskpane = ({ notebookTracker, renderMimeRegistry, contextManager, app
             return;
         }
         const activeCellIndex = notebook.activeCellIndex;
-        try {
-            notebook.widgets.forEach((cell, index) => {
-                if (cell.model.type === 'code') {
-                    const isActiveCodeCell = activeCellIndex === index;
-                    // TODO: Instead of casting, we should rely on the type system to make 
-                    // sure we're using the correct types!
-                    const codeCell = cell;
-                    const cmEditor = codeCell.editor;
-                    const editorView = cmEditor === null || cmEditor === void 0 ? void 0 : cmEditor.editor;
-                    if (editorView) {
-                        let compartment = codeDiffStripesCompartments.current.get(codeCell);
-                        if (!compartment) {
-                            // Create a new compartment and store it
-                            compartment = new _codemirror_state__WEBPACK_IMPORTED_MODULE_1__.Compartment();
-                            codeDiffStripesCompartments.current.set(codeCell, compartment);
-                            // Apply the initial configuration
-                            editorView.dispatch({
-                                effects: _codemirror_state__WEBPACK_IMPORTED_MODULE_1__.StateEffect.appendConfig.of(compartment.of(unifiedDiffLines !== undefined && isActiveCodeCell ? (0,_CodeDiffDisplay__WEBPACK_IMPORTED_MODULE_19__.codeDiffStripesExtension)({ unifiedDiffLines: unifiedDiffLines }) : [])),
-                            });
-                        }
-                        else {
-                            // Reconfigure the compartment
-                            editorView.dispatch({
-                                effects: compartment.reconfigure(unifiedDiffLines !== undefined && isActiveCodeCell ? (0,_CodeDiffDisplay__WEBPACK_IMPORTED_MODULE_19__.codeDiffStripesExtension)({ unifiedDiffLines: unifiedDiffLines }) : []),
-                            });
-                        }
+        notebook.widgets.forEach((cell, index) => {
+            if (cell.model.type === 'code') {
+                const isActiveCodeCell = activeCellIndex === index;
+                // TODO: Instead of casting, we should rely on the type system to make 
+                // sure we're using the correct types!
+                const codeCell = cell;
+                const cmEditor = codeCell.editor;
+                const editorView = cmEditor === null || cmEditor === void 0 ? void 0 : cmEditor.editor;
+                if (editorView) {
+                    const cellId = codeCell.model.id;
+                    let compartment = codeDiffStripesCompartments.current.get(cellId);
+                    if (!compartment) {
+                        // Create a new compartment and store it
+                        compartment = new _codemirror_state__WEBPACK_IMPORTED_MODULE_1__.Compartment();
+                        codeDiffStripesCompartments.current.set(cellId, compartment);
+                        // Apply the initial configuration
+                        editorView.dispatch({
+                            effects: _codemirror_state__WEBPACK_IMPORTED_MODULE_1__.StateEffect.appendConfig.of(compartment.of(unifiedDiffLines !== undefined && isActiveCodeCell ? (0,_CodeDiffDisplay__WEBPACK_IMPORTED_MODULE_19__.codeDiffStripesExtension)({ unifiedDiffLines: unifiedDiffLines }) : [])),
+                        });
                     }
                     else {
-                        console.log('Mito AI: editor view not found when applying code diff stripes');
+                        // Reconfigure the compartment
+                        editorView.dispatch({
+                            effects: compartment.reconfigure(unifiedDiffLines !== undefined && isActiveCodeCell ? (0,_CodeDiffDisplay__WEBPACK_IMPORTED_MODULE_19__.codeDiffStripesExtension)({ unifiedDiffLines: unifiedDiffLines }) : []),
+                        });
                     }
                 }
-            });
-        }
-        catch (error) {
-            console.error('Error updating code cells extensions', error);
-        }
+                else {
+                    console.log('Mito AI: editor view not found when applying code diff stripes');
+                }
+            }
+        });
     };
     const lastAIMessagesIndex = chatHistoryManager.getLastAIMessageIndex();
     return (react__WEBPACK_IMPORTED_MODULE_2___default().createElement("div", { className: "chat-taskpane" },
@@ -8743,46 +8738,62 @@ __webpack_require__.r(__webpack_exports__);
     This is used to wait for the notebook to be ready before sending the first message.
 */
 const waitForNotebookReady = async (notebookTracker) => {
-    const notebook = notebookTracker.currentWidget;
-    if (!notebook) {
-        console.warn('No active notebook found');
-        return;
-    }
-    // Wait for notebook to be ready and attached
-    await notebook.context.ready;
-    // Wait for attachment
-    if (!notebook.content.isAttached) {
+    // This is not critical and we never want it to break the app.
+    try {
+        // First, wait for a notebook to exist with retry pattern
+        let notebook = notebookTracker.currentWidget;
+        let attempts = 0;
+        const maxAttempts = 30; // Wait up to 30 seconds (30 attempts * 1 second each)
+        while (!notebook && attempts < maxAttempts) {
+            console.log(`🔍 Waiting for notebook to appear (attempt ${attempts + 1}/${maxAttempts})...`);
+            await new Promise(resolve => setTimeout(resolve, 1000)); // Wait 1 second
+            notebook = notebookTracker.currentWidget;
+            attempts++;
+        }
+        if (!notebook) {
+            throw new Error(`No active notebook found after ${maxAttempts} seconds`);
+        }
+        console.log('🔍 Notebook found, waiting for it to be ready...');
+        // Wait for notebook to be ready and attached
+        await notebook.context.ready;
+        // Wait for attachment
+        if (!notebook.content.isAttached) {
+            await new Promise(resolve => {
+                const checkAttached = () => {
+                    if (notebook.content.isAttached) {
+                        resolve();
+                    }
+                    else {
+                        setTimeout(checkAttached, 100);
+                    }
+                };
+                checkAttached();
+            });
+        }
+        // Wait for all cells to be created and ready
         await new Promise(resolve => {
-            const checkAttached = () => {
-                if (notebook.content.isAttached) {
+            const checkCellsReady = () => {
+                const cells = notebook.content.widgets;
+                // In large notebooks, not all of the cells are attatched I think. 
+                // So instead we just wait for any cell to be ready and then give it 
+                // another 500ms to be ready.
+                const anyCellReady = cells.some(cell => cell.isAttached && cell.model);
+                if (anyCellReady && cells.length > 0) {
                     resolve();
                 }
                 else {
-                    setTimeout(checkAttached, 100);
+                    setTimeout(checkCellsReady, 100);
                 }
             };
-            checkAttached();
+            checkCellsReady();
         });
+        // Small buffer for final initialization
+        await new Promise(resolve => setTimeout(resolve, 500));
+    }
+    catch (error) {
+        console.error('Error waiting for notebook to be ready:', error);
+        throw error;
     }
-    // Wait for all cells to be created and ready
-    await new Promise(resolve => {
-        const checkCellsReady = () => {
-            const cells = notebook.content.widgets;
-            // In large notebooks, not all of the cells are attatched I think. 
-            // So instead we just wait for any cell to be ready and then give it 
-            // another 500ms to be ready.
-            const anyCellReady = cells.some(cell => cell.isAttached && cell.model);
-            if (anyCellReady && cells.length > 0) {
-                resolve();
-            }
-            else {
-                setTimeout(checkCellsReady, 100);
-            }
-        };
-        checkCellsReady();
-    });
-    // Small buffer for final initialization
-    await new Promise(resolve => setTimeout(resolve, 500));
 };
 
 
@@ -14043,4 +14054,4 @@ var update = _node_modules_style_loader_dist_runtime_injectStylesIntoStyleTag_js
 /***/ })
 
 }]);
-//# sourceMappingURL=lib_index_js.4a24ec8bdd6fde22593c.js.map
+//# sourceMappingURL=lib_index_js.281f4b9af60d620c6fb1.js.map