Fixes typing errors

Author: 3coins

jupyter_ai_tools/toolkits/notebook.py

@@ -211,8 +211,8 @@ async def get_cell_id_from_index(file_path: str, cell_index: int) -> str:
 
 async def add_cell(
     file_path: str,
-    content: str | None = None,
-    cell_id: str | None = None,
+    content: Optional[str] = None,
+    cell_id: Optional[str] = None,
     add_above: bool = False,
     cell_type: Literal["code", "markdown", "raw"] = "code",
 ):
@@ -295,8 +295,8 @@ async def add_cell(
 
 async def insert_cell(
     file_path: str,
-    content: str | None = None,
-    insert_index: int | None = None,
+    content: Optional[str] = None,
+    insert_index: Optional[int] = None,
     cell_type: Literal["code", "markdown", "raw"] = "code",
 ):
     """Inserts a new cell to the Jupyter notebook at the specified cell index.
@@ -887,7 +887,7 @@ def read_cell_nbformat(file_path: str, cell_id: str) -> Dict[str, Any]:
         raise ValueError(f"Cell with {cell_id=} not found in notebook at {file_path=}")
 
 
-def _get_cell_index_from_id_json(notebook_json, cell_id: str) -> int | None:
+def _get_cell_index_from_id_json(notebook_json, cell_id: str) -> Optional[int]:
     """Get cell index from cell_id by notebook json dict.
 
     Args:
@@ -905,7 +905,7 @@ def _get_cell_index_from_id_json(notebook_json, cell_id: str) -> int | None:
     return None
 
 
-def _get_cell_index_from_id_ydoc(ydoc, cell_id: str) -> int | None:
+def _get_cell_index_from_id_ydoc(ydoc, cell_id: str) -> Optional[int]:
     """Get cell index from cell_id using YDoc interface.
 
     Args:
@@ -924,7 +924,7 @@ def _get_cell_index_from_id_ydoc(ydoc, cell_id: str) -> int | None:
         return None
 
 
-def _get_cell_index_from_id_nbformat(notebook, cell_id: str) -> int | None:
+def _get_cell_index_from_id_nbformat(notebook, cell_id: str) -> Optional[int]:
     """Get cell index from cell_id using nbformat interface.
 
     Args:

jupyter_ai_tools/utils.py

@@ -2,7 +2,7 @@
 import inspect
 import os
 from pathlib import Path
-from typing import Optional
+from typing import Dict, List, Optional
 from urllib.parse import unquote
 
 from jupyter_server.auth.identity import User
@@ -269,7 +269,7 @@ def notebook_json_to_md(notebook_json: dict, include_outputs: bool = True) -> st
     return "\n\n".join(md_parts)
 
 
-def metadata_to_md(metadata_json: dict) -> str:
+def metadata_to_md(metadata_json: Dict) -> str:
     """Converts notebook or cell metadata to markdown string in YAML format.
 
     Args:
@@ -339,7 +339,7 @@ def cell_to_md(cell_json: dict, index: int = 0, include_outputs: bool = True) ->
     return "\n\n".join(md_parts)
 
 
-def format_outputs(outputs: list) -> str:
+def format_outputs(outputs: List) -> str:
     """Formats cell outputs into markdown.
 
     Args: