Merge pull request #2140 from mito-ds/dev

Release Jan 7, 2026

Author: ngafar

evals/agent_evals/agent_testing_utils.py

@@ -15,8 +15,21 @@ def process_notebook_update(notebook, cell_update):
         source=code)
 
     if update_type == "new":
-        index = cell_update["index"]
-        notebook.cells.insert(index, new_cell)
+        after_cell_id = cell_update.get("after_cell_id")
+        if after_cell_id == "new cell":
+            # Insert at the very top of the notebook
+            notebook.cells.insert(0, new_cell)
+        else:
+            # Find the cell by ID and get its index
+            target_cell_index = None
+            for i, cell in enumerate(notebook.cells):
+                if cell.get("id") == after_cell_id:
+                    target_cell_index = i
+                    break
+            if target_cell_index is None:
+                raise ValueError(f"Cell with ID {after_cell_id} not found in notebook")
+            # Insert after the target cell
+            notebook.cells.insert(target_cell_index + 1, new_cell)
     elif update_type == "modification":
         target_cell_id = cell_update["id"]
         for cell in notebook.cells:

evals/agent_evals/eval_metrics.py

@@ -214,7 +214,18 @@ def check_cell_addition_index(self, input_params):
         response = self.immediate_next_response
         expected_cell_index = input_params["index_expecting_change"]
         if response["type"] == "cell_update" and response["cell_update"]["type"] == "new":
-            return response["cell_update"]["index"] == expected_cell_index
+            after_cell_id = response["cell_update"].get("after_cell_id")
+            
+            # If expecting cell at index 0, after_cell_id should be 'new cell'
+            if expected_cell_index == 0:
+                return after_cell_id == "new cell"
+            
+            # Otherwise, find the cell ID at (expected_cell_index - 1) in the input notebook
+            if expected_cell_index > 0 and expected_cell_index - 1 < len(self.input_nb.cells):
+                expected_after_cell_id = self.input_nb.cells[expected_cell_index - 1].get("id")
+                return after_cell_id == expected_after_cell_id
+            
+            return False
         return False
 
     def test_correct_cell_edit(self, input_params):

mito-ai/.cursorrules

@@ -0,0 +1,17 @@
+# Cursor Rules for mito-ai Development
+
+## Jupyter Lab Access Token
+
+When testing Jupyter extension changes, use the token from `.cursor-jupyter-token` file.
+
+The Jupyter Lab URL format is:
+```
+http://localhost:8888/lab?token={token}
+```
+
+To get the token, read it from `.cursor-jupyter-token` in the repo root.
+
+**Important**: 
+1. This token configuration is development-only and will not affect production installations.
+The dev config file (`jupyter-config/jupyter_server_config.d/mito_ai_dev.json`) is gitignored.
+2. If the token is not there, tell the user to run dev/setup_jupyter_dev_token.py and relaunch Jupyter Lab.

mito-ai/.gitignore

@@ -127,3 +127,7 @@ dmypy.json
 # Secrets
 .env
 secrets.tsx
+
+# Development-only Jupyter config (contains fixed token for Cursor testing)
+jupyter-config/jupyter_server_config.d/mito_ai_dev.py
+.cursor-jupyter-token

mito-ai/CLAUDE.md

@@ -24,6 +24,10 @@ jlpm build
 jupyter labextension develop . --overwrite
 jupyter server extension enable --py mito_ai
 
+# (Optional) Setup fixed Jupyter token for Cursor agent testing
+# This creates a development-only config that allows Cursor to automatically test changes
+python dev/setup_jupyter_dev_token.py
+
 # Start development environment
 jlpm watch  # Terminal 1: auto-rebuild frontend
 jupyter lab --autoreload  # Terminal 2: run JupyterLab with backend hot reload

mito-ai/README.md

@@ -77,6 +77,10 @@ jupyter labextension develop . --overwrite
 # Start the jupyter server extension for development
 jupyter server extension enable --py mito_ai
 
+# (Optional) Setup fixed Jupyter token for Cursor agent testing
+# This creates a development-only config that allows Cursor to automatically test changes
+python dev/setup_jupyter_dev_token.py
+
 # Watch the source directory in one terminal, automatically rebuilding when needed
 # In case of Error: If this command fails because the lib directory was not created (the error will say something like
 # unable to find main entry point) then run `jlpm run clean:lib` first to get rid of the old buildcache 

mito-ai/dev/README.md

@@ -1,5 +1,13 @@
 # Mito AI Dev Scripts
 
+### View Chat History
+
+A Streamlit app that helps you explore the chat history of a Mito AI chat file.
+
+```bash
+streamlit run dev/chat_history_viewer.py
+```
+
 ### Generate AI Chats
 
 A simple script that creates a bunch of fake Mito AI chat files so you can quickly generate a fully populated chat history.