Switch from Pydantic to msgspec (#65)

Author: renan-r-santos

pixi_kernel/env.py

@@ -1,7 +1,7 @@
 import os
 from pathlib import Path
 
-from pydantic import ValidationError
+import msgspec
 
 from .compatibility import run_pixi
 from .types import PixiInfo
@@ -20,8 +20,8 @@ async def envs_from_path(path: Path) -> list[str]:
         return [DEFAULT_ENVIRONMENT]
 
     try:
-        pixi_info = PixiInfo.model_validate_json(stdout, strict=True)
-    except ValidationError:
+        pixi_info = msgspec.json.decode(stdout, type=PixiInfo)
+    except msgspec.MsgspecError:
         return [DEFAULT_ENVIRONMENT]
 
     if len(pixi_info.environments) == 0:

pixi_kernel/readiness.py

@@ -2,7 +2,7 @@
 import logging
 from pathlib import Path
 
-from pydantic import ValidationError
+import msgspec
 from returns.result import Failure, Result, Success
 
 from .compatibility import has_compatible_pixi, run_pixi
@@ -48,8 +48,8 @@ async def verify_env_readiness(
         return Failure(f"Failed to run 'pixi info': {stderr}")
 
     try:
-        pixi_info = PixiInfo.model_validate_json(stdout, strict=True)
-    except ValidationError as exception:
+        pixi_info = msgspec.json.decode(stdout, type=PixiInfo)
+    except msgspec.MsgspecError as exception:
         return Failure(f"Failed to parse 'pixi info' output: {stdout}\n{exception}")
 
     if pixi_info.project is None:

pixi_kernel/types.py

@@ -2,20 +2,20 @@
 
 from typing import Optional
 
-from pydantic import BaseModel, Field
+import msgspec
 
 
-class PixiInfo(BaseModel):
-    environments: list[Environment] = Field(alias="environments_info")
-    project: Optional[Project] = Field(alias="project_info")
+class PixiInfo(msgspec.Struct, frozen=True, kw_only=True):
+    environments: list[Environment] = msgspec.field(name="environments_info")
+    project: Optional[Project] = msgspec.field(name="project_info")
 
 
-class Environment(BaseModel):
+class Environment(msgspec.Struct, frozen=True, kw_only=True):
     name: str
     dependencies: list[str]
     pypi_dependencies: list[str]
     prefix: str
 
 
-class Project(BaseModel):
+class Project(msgspec.Struct, frozen=True, kw_only=True):
     manifest_path: str

pyproject.toml

@@ -35,8 +35,8 @@ requires-python = ">=3.9,<4.0"
 dependencies = [
     "ipykernel>=6",
     "jupyter-client>=7",
-    "jupyter_server>=2.4.0",
-    "pydantic>=2",
+    "jupyter_server>=2.4",
+    "msgspec>=0.18",
     "returns>=0.23",
     "tomli>=2; python_version<'3.11'",
 ]
@@ -50,7 +50,7 @@ Repository = "https://github.com/renan-r-santos/pixi-kernel"
 dev-dependencies = [
     "jupyter-kernel-test>=0.7",
     "jupyterlab>=4",
-    "msgspec[toml]>=0.19.0",
+    "msgspec[toml]>=0.18",
     "mypy>=1",
     "pytest>=8",
     "pytest-asyncio>=1",