core: pre-commit

Author: simonprovost

README.md

@@ -31,12 +31,12 @@ This repository provides an **MCPStack tool that wraps the official Python Jupyt
 
 - Upstream project: [datalayer/jupyter-mcp-server](https://github.com/datalayer/jupyter-mcp-server)
 - We **reuse their MCP actions** and surface them through **MCPStack**.
-- As the upstream evolves, **some actions / endpoints may deprecate**. Our wrapper is intentionally lightweight, so updating to new upstream versions should be straightforward.  
+- As the upstream evolves, **some actions / endpoints may deprecate**. Our wrapper is intentionally lightweight, so updating to new upstream versions should be straightforward.
   If you hit an incompatibility, **please open an issue** and we’ll track an update to align with the Jupyter MCP Server.
 
 ### What is MCPStack, in layman’s terms?
 
-The **Model Context Protocol (MCP)** standardises how tools talk to LLMs.  
+The **Model Context Protocol (MCP)** standardises how tools talk to LLMs.
 `MCPStack` lets you **stack multiple MCP tools together** into a pipeline and expose them to an LLM host (e.g., Claude Desktop).
 
 Think **scikit-learn pipelines, but for LLM tooling**:
@@ -96,7 +96,7 @@ uv run jupyter lab \
 ```
 
 > [!NOTE]
-> Docs reference: https://jupyter-mcp-server.datalayer.tech/jupyter/  
+> Docs reference: https://jupyter-mcp-server.datalayer.tech/jupyter/
 
 Make sure to have a notebook open in Jupyter lab, e.g., `notebook.ipynb` or whatever you have defined in the configuration.
 
@@ -108,7 +108,7 @@ Use the tool’s CLI to create a small `MCPStack ToolConfig` JSON. **At minimum
 uv run mcpstack tools jupyter configure \
   --token MY_TOKEN \
   --output jupyter_config.json
-  
+
 # MY_TOKEN can for instance be: 1117bf468693444a5608e882ab3b55d511f354a1750df02 (must match the Jupyter server token)
 ```
 
@@ -170,7 +170,7 @@ Tokens remain **required** and are enforced upfront by MCPStack when building th
 
 ## 📖 Programmatic API
 
-Use the `Jupyter` tool class directly in a pipeline.  
+Use the `Jupyter` tool class directly in a pipeline.
 Tokens are taken from the environment (the pipeline config or your process env):
 
 ```python
@@ -195,33 +195,33 @@ pipeline = (
 
 >[!NOTE]
 > Common upstream actions you can expose (see `configuration/tools.yaml`):
-> 
-> - `append_markdown_cell`, 
-> - `insert_markdown_cell`, 
-> - `overwrite_cell_source`, 
+>
+> - `append_markdown_cell`,
+> - `insert_markdown_cell`,
+> - `overwrite_cell_source`,
 > - `delete_cell`
-> - `append_execute_code_cell`, 
+> - `append_execute_code_cell`,
 > - `insert_execute_code_cell`
-> - `execute_cell_with_progress`, 
-> - `execute_cell_simple_timeout`, 
+> - `execute_cell_with_progress`,
+> - `execute_cell_simple_timeout`,
 > - `execute_cell_streaming`
-> - `read_cell`, 
-> - `read_all_cells`, 
+> - `read_cell`,
+> - `read_all_cells`,
 > - `get_notebook_info`
 
 ---
 
 ## 🧰 Troubleshooting
 
-- **403 Forbidden / `_xsrf` missing / cannot create kernels**  
-  Ensure you ran Jupyter with a token and that your `ToolConfig` provides **both** `DOCUMENT_TOKEN` and `RUNTIME_TOKEN`.  
+- **403 Forbidden / `_xsrf` missing / cannot create kernels**
+  Ensure you ran Jupyter with a token and that your `ToolConfig` provides **both** `DOCUMENT_TOKEN` and `RUNTIME_TOKEN`.
   In most setups it’s the same token.
 
-- **404 on `notebook.ipynb`**  
+- **404 on `notebook.ipynb`**
   Update `--document-id` to the actual notebook path relative to Jupyter’s working directory (e.g., `Untitled.ipynb` or `notebooks/analysis.ipynb`).
 
-- **Nothing happens in Lab**  
-  Prefer `http://127.0.0.1:8888` over `http://localhost:8888`.  
+- **Nothing happens in Lab**
+  Prefer `http://127.0.0.1:8888` over `http://localhost:8888`.
   Confirm your pipeline is running and that the tool is listed in `mcpstack list-tools`.
 
 ---

src/mcpstack_jupyter/tools/jupyter/__init__.py

@@ -1,3 +1,3 @@
 from .jupyter import Jupyter
 
-__all__ = ["Jupyter"]
+__all__ = ["Jupyter"]

src/mcpstack_jupyter/tools/jupyter/cli.py

@@ -12,8 +12,8 @@
 from rich.table import Table
 
 from mcpstack_jupyter.tools.jupyter.utils.config_loader import (
-    load_env_defaults,
     load_cli_defaults,
+    load_env_defaults,
     load_known_tools,
 )
 
@@ -37,8 +37,12 @@ def get_app(cls) -> typer.Typer:
             pretty_exceptions_show_locals=False,
             rich_markup_mode="markdown",
         )
-        app.command(help="Configure the Jupyter tool (Token,Provider,Kernel, etc.).")(cls.configure)
-        app.command(help="Display the current status of the Jupyter tool (Validate with Original Library, etc.")(cls.status)
+        app.command(help="Configure the Jupyter tool (Token,Provider,Kernel, etc.).")(
+            cls.configure
+        )
+        app.command(
+            help="Display the current status of the Jupyter tool (Validate with Original Library, etc."
+        )(cls.status)
         return app
 
     @classmethod
@@ -242,4 +246,4 @@ def status(cls, verbose: bool = False) -> None:
 
         except Exception as e:
             console.print(f"[red]❌ Error getting status: {e}[/red]")
-            logger.error("Status failed: %s", e, exc_info=True)
+            logger.error("Status failed: %s", e, exc_info=True)

src/mcpstack_jupyter/tools/jupyter/configuration/cli_defaults.yaml

@@ -4,4 +4,4 @@ output_filename: "jupyter_config.json"
 prompts:
   document_url: "Document URL"
   document_id: "Notebook path"
-  runtime_url: "Runtime URL (Enter to reuse Document URL)"
+  runtime_url: "Runtime URL (Enter to reuse Document URL)"

src/mcpstack_jupyter/tools/jupyter/configuration/env_defaults.yaml

@@ -3,4 +3,4 @@ provider: "jupyter"
 document_url: "http://127.0.0.1:8888"
 document_id: "notebook.ipynb"
 runtime_url: null
-require_tokens: true
+require_tokens: true

src/mcpstack_jupyter/tools/jupyter/configuration/tools.yaml

@@ -18,4 +18,4 @@ known_tools:
 read_only:
   - read_cell
   - read_all_cells
-  - get_notebook_info
+  - get_notebook_info

src/mcpstack_jupyter/tools/jupyter/jupyter.py

@@ -1,21 +1,26 @@
 import importlib
 import os
-from typing import Any, Dict, List, Optional, Callable
+from collections.abc import Callable
+from typing import Any, ClassVar
 
 from beartype import beartype
 from MCPStack.core.tool.base import BaseTool
-from mcpstack_jupyter.tools.jupyter.utils.config_loader import load_env_defaults, load_known_tools
+
+from mcpstack_jupyter.tools.jupyter.utils.config_loader import (
+    load_env_defaults,
+    load_known_tools,
+)
 
 
 @beartype
 class Jupyter(BaseTool):
-    KNOWN_TOOLS: List[str] = []
+    KNOWN_TOOLS: ClassVar[list[str]] = []
 
-    def __init__(self, include: Optional[List[str]] = None) -> None:
+    def __init__(self, include: list[str] | None = None) -> None:
         super().__init__()
         self.include = include
         self._server = None
-        self._bound: List[Callable[..., Any]] = []
+        self._bound: list[Callable[..., Any]] = []
 
         env_cfg = load_env_defaults()
         self._env_defaults = env_cfg
@@ -42,14 +47,14 @@ def _teardown(self) -> None:
     def _post_load(self) -> None:
         pass
 
-    def actions(self) -> List[Callable[..., Any]]:
+    def actions(self) -> list[Callable[..., Any]]:
         return list(self._bound)
 
-    def to_dict(self) -> Dict[str, Any]:
+    def to_dict(self) -> dict[str, Any]:
         return {"include": self.include}
 
     @classmethod
-    def from_dict(cls, params: Dict[str, Any]) -> "Jupyter":
+    def from_dict(cls, params: dict[str, Any]) -> "Jupyter":
         return cls(include=params.get("include"))
 
     def _import_and_bind(self) -> None:
@@ -98,4 +103,4 @@ def _import_and_bind(self) -> None:
                     f"'{name}' not found in jupyter_mcp_server.server. "
                     "Install a compatible version that provides this tool."
                 ) from e
-            self._bound.append(fn)
+            self._bound.append(fn)

src/mcpstack_jupyter/tools/jupyter/utils/config_loader.py

@@ -1,27 +1,29 @@
 from __future__ import annotations
 
 from importlib.resources import files
-from typing import Any, Dict, List
+from typing import Any
 
 import yaml
 
 PKG = "mcpstack_jupyter.tools.jupyter.configuration"
 
+
 def _load_yaml(name: str) -> dict:
     with files(PKG).joinpath(name).open("rb") as f:
         data = yaml.safe_load(f)
     return data or {}
 
 
-def load_env_defaults() -> Dict[str, Any]:
+def load_env_defaults() -> dict[str, Any]:
     return _load_yaml("env_defaults.yaml")
 
-def load_cli_defaults() -> Dict[str, Any]:
+
+def load_cli_defaults() -> dict[str, Any]:
     return _load_yaml("cli_defaults.yaml")
 
 
-def load_known_tools() -> Dict[str, List[str]]:
+def load_known_tools() -> dict[str, list[str]]:
     data = _load_yaml("tools.yaml")
     data.setdefault("known_tools", [])
     data.setdefault("read_only", [])
-    return data
+    return data