all lines

langchain-ai · Dec 12, 2024 · a031f82 · a031f82
1 parent 7d940a4
commit a031f82
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/libs/checkpoint/langgraph/checkpoint/serde/jsonplus.py b/libs/checkpoint/langgraph/checkpoint/serde/jsonplus.py
@@ -438,28 +438,36 @@ def _msgpack_default(obj: Any) -> Union[str, msgpack.ExtType]:
 def _msgpack_ext_hook(code: int, data: bytes) -> Any:
     if code == EXT_CONSTRUCTOR_SINGLE_ARG:
         try:
-            tup = msgpack.unpackb(data, ext_hook=_msgpack_ext_hook)
+            tup = msgpack.unpackb(
+                data, ext_hook=_msgpack_ext_hook, strict_map_key=False
+            )
             # module, name, arg
             return getattr(importlib.import_module(tup[0]), tup[1])(tup[2])
         except Exception:
             return
     elif code == EXT_CONSTRUCTOR_POS_ARGS:
         try:
-            tup = msgpack.unpackb(data, ext_hook=_msgpack_ext_hook)
+            tup = msgpack.unpackb(
+                data, ext_hook=_msgpack_ext_hook, strict_map_key=False
+            )
             # module, name, args
             return getattr(importlib.import_module(tup[0]), tup[1])(*tup[2])
         except Exception:
             return
     elif code == EXT_CONSTRUCTOR_KW_ARGS:
         try:
-            tup = msgpack.unpackb(data, ext_hook=_msgpack_ext_hook)
+            tup = msgpack.unpackb(
+                data, ext_hook=_msgpack_ext_hook, strict_map_key=False
+            )
             # module, name, args
             return getattr(importlib.import_module(tup[0]), tup[1])(**tup[2])
         except Exception:
             return
     elif code == EXT_METHOD_SINGLE_ARG:
         try:
-            tup = msgpack.unpackb(data, ext_hook=_msgpack_ext_hook)
+            tup = msgpack.unpackb(
+                data, ext_hook=_msgpack_ext_hook, strict_map_key=False
+            )
             # module, name, arg, method
             return getattr(getattr(importlib.import_module(tup[0]), tup[1]), tup[3])(
                 tup[2]