fix: msgspec Python 3.14 compatibility

marimo/__main__.py

@@ -1,4 +1,6 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations
+
 from marimo._cli.cli import main
 
 if __name__ == "__main__":

marimo/_ai/_tools/tools/notebooks.py

@@ -1,4 +1,6 @@
 # Copyright 2025 Marimo. All rights reserved.
+from __future__ import annotations
+
 from dataclasses import dataclass, field
 from typing import Optional
 

marimo/_ai/_tools/tools/tables_and_variables.py

@@ -1,4 +1,6 @@
 # Copyright 2025 Marimo. All rights reserved.
+from __future__ import annotations
+
 from dataclasses import dataclass, field
 from typing import Optional
 
@@ -63,7 +65,7 @@ def handle(self, args: TablesAndVariablesArgs) -> TablesAndVariablesOutput:
         return self._get_tables_and_variables(session, args.variable_names)
 
     def _get_tables_and_variables(
-        self, session: "Session", variable_names: list[str]
+        self, session: Session, variable_names: list[str]
     ) -> TablesAndVariablesOutput:
         session_view = session.session_view
         # convert to set for O(1) lookup

marimo/_ai/_tools/tools_registry.py

@@ -1,4 +1,6 @@
 # Copyright 2025 Marimo. All rights reserved.
+from __future__ import annotations
+
 from typing import Any
 
 from marimo._ai._tools.base import ToolBase

marimo/_ai/llm/__init__.py

@@ -1,4 +1,5 @@
 # Copyright 2025 Marimo. All rights reserved.
+
 from marimo._ai.llm._impl import anthropic, bedrock, google, groq, openai
 
 __all__ = ["openai", "anthropic", "google", "groq", "bedrock"]

marimo/_ast/__init__.py

@@ -1 +1,2 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations

marimo/_ast/app_config.py

@@ -1,4 +1,5 @@
 # Copyright 2025 Marimo. All rights reserved.
+from __future__ import annotations
 
 import os
 from dataclasses import asdict, dataclass, field
@@ -43,7 +44,7 @@ class _AppConfig:
     @staticmethod
     def from_untrusted_dict(
         updates: dict[str, Any], silent: bool = False
-    ) -> "_AppConfig":
+    ) -> _AppConfig:
         # Certain flags are useful to pass to App for construction, but
         # shouldn't make it into the config. (e.g. the _filename flag is
         # internal)
@@ -65,7 +66,7 @@ def asdict(self) -> dict[str, Any]:
             k: v for (k, v) in asdict(self).items() if not k.startswith("_")
         }
 
-    def update(self, updates: dict[str, Any]) -> "_AppConfig":
+    def update(self, updates: dict[str, Any]) -> _AppConfig:
         config_dict = asdict(self)
         for key in updates:
             if key in config_dict:

marimo/_ast/cell_id.py

@@ -1,4 +1,6 @@
 # Copyright 2025 Marimo. All rights reserved.
+from __future__ import annotations
+
 import random
 import string
 from uuid import UUID, uuid4

marimo/_ast/errors.py

@@ -1,4 +1,5 @@
 # Copyright 2025 Marimo. All rights reserved.
+from __future__ import annotations
 
 
 class SetupRootError(Exception):

marimo/_ast/names.py

@@ -1,4 +1,6 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations
+
 DEFAULT_CELL_NAME = "_"
 SETUP_CELL_NAME = "setup"
 # Intentionally an invalid name character

marimo/_ast/sql_utils.py

@@ -1,4 +1,5 @@
 # Copyright 2025 Marimo. All rights reserved.
+from __future__ import annotations
 
 from typing import Literal, Optional, Union
 

marimo/_cli/__init__.py

@@ -1 +1,2 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations

marimo/_cli/cli_validators.py

@@ -1,4 +1,6 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations
+
 from pathlib import Path
 from typing import Any, Optional
 

marimo/_cli/export/cloudflare.py

@@ -1,4 +1,6 @@
 # Copyright 2025 Marimo. All rights reserved.
+from __future__ import annotations
+
 from pathlib import Path
 
 from marimo._cli.print import bold, echo, green

marimo/_cli/utils.py

@@ -1,12 +1,16 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations
 
-from pathlib import Path
 from sys import stdout
+from typing import TYPE_CHECKING
 
 import click
 
 from marimo._config.settings import GLOBAL_SETTINGS
 
+if TYPE_CHECKING:
+    from pathlib import Path
+
 
 def prompt_to_overwrite(path: Path) -> bool:
     if GLOBAL_SETTINGS.YES:

marimo/_config/__init__.py

@@ -1 +1,2 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations

marimo/_convert/notebook.py

@@ -1,4 +1,6 @@
 # Copyright 2025 Marimo. All rights reserved.
+from __future__ import annotations
+
 from marimo._schemas.notebook import (
     NotebookCell,
     NotebookCellConfig,

marimo/_entrypoints/ids.py

@@ -1,4 +1,6 @@
 # Copyright 2025 Marimo. All rights reserved.
+from __future__ import annotations
+
 from typing import Literal
 
 # Internal entrypoints. Not user-facing as the API is not stable.

marimo/_entrypoints/registry.py

@@ -1,4 +1,6 @@
 # Copyright 2025 Marimo. All rights reserved.
+from __future__ import annotations
+
 import os
 from importlib.metadata import entry_points
 from typing import TYPE_CHECKING, Generic, Optional, TypeVar, cast
@@ -150,7 +152,7 @@ def __repr__(self) -> str:
         return f"{type(self).__name__}(group={self.entry_point_group!r}, registered={self.names()!r})"
 
 
-def get_entry_points(group: KnownEntryPoint) -> "EntryPoints":
+def get_entry_points(group: KnownEntryPoint) -> EntryPoints:
     ep = entry_points()
     if hasattr(ep, "select"):
         return ep.select(group=group)

marimo/_islands/__init__.py

@@ -1,4 +1,5 @@
 # Copyright 2024 Marimo. All rights reserved.
+
 __all__ = [
     "MarimoIslandGenerator",
     "MarimoIslandStub",

marimo/_lint/formatters/__init__.py

@@ -1,6 +1,8 @@
 # Copyright 2025 Marimo. All rights reserved.
 """Formatters for diagnostic output."""
 
+from __future__ import annotations
+
 from marimo._lint.formatters.base import DiagnosticFormatter
 from marimo._lint.formatters.full import FullFormatter
 from marimo._lint.formatters.json import (

marimo/_lint/rules/__init__.py

@@ -1,4 +1,6 @@
 # Copyright 2025 Marimo. All rights reserved.
+from __future__ import annotations
+
 from marimo._lint.rules.base import LintRule
 from marimo._lint.rules.breaking import BREAKING_RULE_CODES
 from marimo._lint.rules.formatting import FORMATTING_RULE_CODES

marimo/_lint/rules/breaking/__init__.py

@@ -1,4 +1,6 @@
 # Copyright 2025 Marimo. All rights reserved.
+from __future__ import annotations
+
 from marimo._lint.rules.base import LintRule
 from marimo._lint.rules.breaking.graph import (
     CycleDependenciesRule,

marimo/_lint/rules/formatting/__init__.py

@@ -1,4 +1,6 @@
 # Copyright 2025 Marimo. All rights reserved.
+from __future__ import annotations
+
 from marimo._lint.rules.base import LintRule, UnsafeFixRule
 from marimo._lint.rules.formatting.empty_cells import EmptyCellRule
 from marimo._lint.rules.formatting.general import GeneralFormattingRule

marimo/_lint/rules/runtime/__init__.py

@@ -1,4 +1,6 @@
 # Copyright 2025 Marimo. All rights reserved.
+from __future__ import annotations
+
 from marimo._lint.rules.base import LintRule
 from marimo._lint.rules.runtime.self_import import SelfImportRule
 

marimo/_lint/visitors.py

@@ -1,6 +1,8 @@
 # Copyright 2025 Marimo. All rights reserved.
 """AST visitors for linting purposes."""
 
+from __future__ import annotations
+
 import ast
 from typing import Optional
 

marimo/_mcp/server/lifespan.py

@@ -1,18 +1,21 @@
 # Copyright 2025 Marimo. All rights reserved.
+from __future__ import annotations
+
 import contextlib
-from collections.abc import AsyncIterator
 from typing import TYPE_CHECKING
 
 from marimo._loggers import marimo_logger
 
 LOGGER = marimo_logger()
 
 if TYPE_CHECKING:
+    from collections.abc import AsyncIterator
+
     from starlette.applications import Starlette
 
 
 @contextlib.asynccontextmanager
-async def mcp_server_lifespan(app: "Starlette") -> AsyncIterator[None]:
+async def mcp_server_lifespan(app: Starlette) -> AsyncIterator[None]:
     """Lifespan for MCP server functionality (exposing marimo as MCP server)."""
 
     try:

marimo/_mcp/server/main.py

@@ -6,6 +6,8 @@
 notebook context and functionality.
 """
 
+from __future__ import annotations
+
 from typing import TYPE_CHECKING
 
 from marimo._ai._tools.base import ToolContext
@@ -17,9 +19,10 @@
 
 if TYPE_CHECKING:
     from starlette.applications import Starlette
+    from starlette.types import Receive, Scope, Send
 
 
-def setup_mcp_server(app: "Starlette") -> None:
+def setup_mcp_server(app: Starlette) -> None:
     """Create and configure MCP server for marimo integration.
 
     Args:
@@ -34,7 +37,6 @@ def setup_mcp_server(app: "Starlette") -> None:
     from starlette.middleware.base import BaseHTTPMiddleware
     from starlette.responses import JSONResponse
     from starlette.routing import Mount
-    from starlette.types import Receive, Scope, Send
 
     mcp = FastMCP(
         "marimo-mcp-server",

marimo/_messaging/__init__.py

@@ -1 +1,2 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations

marimo/_output/formatters/__init__.py

@@ -1 +1,2 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations

marimo/_output/justify.py

@@ -1,4 +1,6 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations
+
 from marimo._output.formatting import as_html
 from marimo._output.hypertext import Html
 from marimo._output.rich_help import mddoc

marimo/_output/md_extensions/breakless_lists.py

@@ -1,7 +1,8 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations
 
 import re
-from xml.etree.ElementTree import Element
+from typing import TYPE_CHECKING
 
 from markdown import (  # type: ignore
     Extension,
@@ -10,6 +11,9 @@
     treeprocessors,
 )
 
+if TYPE_CHECKING:
+    from xml.etree.ElementTree import Element
+
 
 class BreaklessListsPreprocessor(preprocessors.Preprocessor):  # type: ignore[misc]
     """

marimo/_output/md_extensions/external_links.py

@@ -1,9 +1,14 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
 from urllib.parse import urlparse
-from xml.etree.ElementTree import Element
 
 from markdown import Extension, Markdown, treeprocessors  # type: ignore
 
+if TYPE_CHECKING:
+    from xml.etree.ElementTree import Element
+
 # Adapted from https://github.com/squidfunk/mkdocs-material/discussions/3660#discussioncomment-6725823  # noqa: E501
 
 

marimo/_output/md_extensions/flexible_indent.py

@@ -1,4 +1,5 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations
 
 import re
 

marimo/_output/superjson.py

@@ -1,5 +1,7 @@
 # Copyright 2025 Marimo. All rights reserved.
 # This data serializes
+from __future__ import annotations
+
 from typing import Any
 
 

marimo/_plugins/__init__.py

@@ -1 +1,2 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations

marimo/_plugins/stateless/__init__.py

@@ -1 +1,2 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations

marimo/_plugins/ui/_core/__init__.py

@@ -1 +1,2 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations

marimo/_plugins/ui/_impl/__init__.py

@@ -1 +1,2 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations

marimo/_plugins/ui/_impl/chat/__init__.py

@@ -1 +1,2 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations

marimo/_plugins/ui/_impl/dataframes/__init__.py

@@ -1 +1,2 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations

marimo/_pyodide/restartable_task.py

@@ -1,4 +1,6 @@
 # Copyright 2025 Marimo. All rights reserved.
+from __future__ import annotations
+
 import asyncio
 from typing import Any, Callable, Optional
 

marimo/_runtime/__init__.py

@@ -1 +1,2 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations

marimo/_runtime/context/__init__.py

@@ -1,4 +1,6 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations
+
 __all__ = [
     "get_context",
     "safe_get_context",

marimo/_runtime/handlers.py

@@ -1,4 +1,6 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations
+
 import os
 from dataclasses import dataclass
 from typing import TYPE_CHECKING, Any, Callable
@@ -34,7 +36,7 @@ def interrupt_handler(signum: int, frame: Any) -> None:
     return interrupt_handler
 
 
-def construct_sigterm_handler(kernel: "Kernel") -> Callable[[int, Any], None]:
+def construct_sigterm_handler(kernel: Kernel) -> Callable[[int, Any], None]:
     del kernel
 
     @dataclass

marimo/_runtime/input_override.py

@@ -1,4 +1,6 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations
+
 import functools
 import sys
 

marimo/_runtime/packages/package_managers.py

@@ -1,4 +1,6 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations
+
 from marimo._runtime.packages.conda_package_manager import PixiPackageManager
 from marimo._runtime.packages.package_manager import PackageManager
 from marimo._runtime.packages.pypi_package_manager import (

marimo/_runtime/reload/__init__.py

@@ -1 +1,2 @@
 # Copyright 2024 Marimo. All rights reserved.
+from __future__ import annotations