Add notebook language metadata to types

scverse · Nov 29, 2023 · 3c8a32b · 3c8a32b
1 parent 9729de3
commit 3c8a32b
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 2 deletions.
diff --git a/.vscode/settings.json b/.vscode/settings.json
@@ -12,5 +12,6 @@
     "[jsonc][yaml]": {
         "editor.defaultFormatter": "esbenp.prettier-vscode",
     },
-    "python.analysis.typeCheckingMode": "basic",
+    "python.analysis.typeCheckingMode": "strict",
+    "python.testing.pytestEnabled": true,
 }
diff --git a/src/prelude_runner/types.py b/src/prelude_runner/types.py
@@ -39,17 +39,37 @@ class Stream(Protocol):
     text: str
 
 
-class CodeCell(Cell):
+class CodeCell(Cell, Protocol):
     """A code cell."""
 
     cell_type: Literal["code"]
     # TODO: add other output types  # noqa: TD003
     outputs: list[Stream | Any]
 
 
+class KernelSpec(TypedDict):
+    """Kernel specification metadata."""
+
+    name: str
+    language: str
+    display_name: str
+
+
+class LanguageInfo(TypedDict):
+    """Language metadata."""
+
+    name: str
+    version: str
+    mimetype: str
+    file_extension: str
+    codemirror_mode: dict[str, str]
+    pygments_lexer: str
+
+
 class NotebookMetadata(TypedDict):
     """Common metadata for all notebooks."""
 
+    kernelspec: KernelSpec
     language_info: dict[str, str]