chore: improve subclass handling (#6948)

If subclass doesn't support the new parameter, we the current
implementation would create an error.

libs/checkpoint/langgraph/checkpoint/serde/jsonplus.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import copy
 import dataclasses
 import decimal
 import importlib
@@ -106,14 +107,15 @@ class JsonPlusSerializer(SerializerProtocol):
             allowed_msgpack_modules = tuple(self._allowed_msgpack_modules)
         else:
             allowed_msgpack_modules = self._allowed_msgpack_modules
-        return self.__class__(
-            pickle_fallback=self.pickle_fallback,
-            allowed_json_modules=self._allowed_json_modules,
-            allowed_msgpack_modules=allowed_msgpack_modules,
-            __unpack_ext_hook__=(
-                self._unpack_ext_hook if self._custom_unpack_ext_hook else None
-            ),
-        )
+
+        clone = copy.copy(self)
+        clone._allowed_json_modules = _normalize_allowlist(self._allowed_json_modules)
+        clone._allowed_msgpack_modules = _normalize_allowlist(allowed_msgpack_modules)
+        if not clone._custom_unpack_ext_hook:
+            clone._unpack_ext_hook = _create_msgpack_ext_hook(
+                clone._allowed_msgpack_modules
+            )
+        return clone
 
     def _encode_constructor_args(
         self,

libs/checkpoint/tests/test_jsonplus.py

@@ -777,6 +777,43 @@ def test_with_msgpack_allowlist_noop_returns_same_instance() -> None:
     assert result is serde
 
 
+def test_with_msgpack_allowlist_supports_subclass_without_init_kwargs() -> None:
+    class CustomSerializer(JsonPlusSerializer):
+        def __init__(self) -> None:
+            super().__init__(allowed_msgpack_modules=None)
+
+    serde = CustomSerializer()
+    result = serde.with_msgpack_allowlist([MyDataclass])
+
+    assert isinstance(result, CustomSerializer)
+    assert result is not serde
+    assert serde._allowed_msgpack_modules is None
+    assert result._allowed_msgpack_modules == {
+        (MyDataclass.__module__, MyDataclass.__name__)
+    }
+
+
+def test_with_msgpack_allowlist_rebuilds_default_unpack_hook() -> None:
+    serde = JsonPlusSerializer(allowed_msgpack_modules=None)
+    original_hook = serde._unpack_ext_hook
+
+    result = serde.with_msgpack_allowlist([MyDataclass])
+
+    assert result._unpack_ext_hook is not original_hook
+
+
+def test_with_msgpack_allowlist_preserves_custom_unpack_hook() -> None:
+    def custom_hook(code: int, data: bytes) -> None:
+        return None
+
+    serde = JsonPlusSerializer(
+        allowed_msgpack_modules=None, __unpack_ext_hook__=custom_hook
+    )
+    result = serde.with_msgpack_allowlist([MyDataclass])
+
+    assert result._unpack_ext_hook is custom_hook
+
+
 @pytest.mark.skipif(sys.version_info >= (3, 14), reason="pydantic v1 not on 3.14+")
 def test_msgpack_pydantic_v1_allowlist(caplog: pytest.LogCaptureFixture) -> None:
     """Pydantic v1 models in allowlist should deserialize without warnings."""