Fix serialization of pydantic.BaseModel fields with pathlib.Path dict keys

python/langsmith/_internal/_serde.py

@@ -10,9 +10,7 @@
 import pathlib
 import re
 import uuid
-from typing import (
-    Any,
-)
+from typing import Any
 
 import orjson
 
@@ -33,14 +31,8 @@ def _simple_default(obj):
         # https://github.com/ijl/orjson#serialize
         if isinstance(obj, datetime.datetime):
             return obj.isoformat()
-        if isinstance(obj, uuid.UUID):
+        elif isinstance(obj, uuid.UUID):
             return str(obj)
-        if hasattr(obj, "model_dump") and callable(obj.model_dump):
-            return obj.model_dump()
-        elif hasattr(obj, "dict") and callable(obj.dict):
-            return obj.dict()
-        elif hasattr(obj, "_asdict") and callable(obj._asdict):
-            return obj._asdict()
         elif isinstance(obj, BaseException):
             return {"error": type(obj).__name__, "message": str(obj)}
         elif isinstance(obj, (set, frozenset, collections.deque)):
@@ -86,21 +78,22 @@ def _serialize_json(obj: Any) -> Any:
             return list(obj)
 
         serialization_methods = [
-            ("model_dump", True),  # Pydantic V2 with non-serializable fields
-            ("dict", False),  # Pydantic V1 with non-serializable field
-            ("to_dict", False),  # dataclasses-json
+            (
+                "model_dump",
+                {"exclude_none": True, "mode": "json"},
+            ),  # Pydantic V2 with non-serializable fields
+            ("dict", {}),  # Pydantic V1 with non-serializable field
+            ("to_dict", {}),  # dataclasses-json
         ]
-        for attr, exclude_none in serialization_methods:
+        for attr, kwargs in serialization_methods:
             if (
                 hasattr(obj, attr)
                 and callable(getattr(obj, attr))
                 and not isinstance(obj, type)
             ):
                 try:
                     method = getattr(obj, attr)
-                    response = (
-                        method(exclude_none=exclude_none) if exclude_none else method()
-                    )
+                    response = method(**kwargs)
                     if not isinstance(response, dict):
                         return str(response)
                     return response

python/tests/unit_tests/test_client.py

@@ -7,6 +7,7 @@
 import json
 import logging
 import math
+import pathlib
 import sys
 import time
 import uuid
@@ -719,16 +720,24 @@ def test_pydantic_serialize() -> None:
 
     class ChildPydantic(BaseModel):
         uid: uuid.UUID
+        child_path_keys: Dict[pathlib.Path, pathlib.Path]
 
     class MyPydantic(BaseModel):
         foo: str
         uid: uuid.UUID
         tim: datetime
         ex: Optional[str] = None
         child: Optional[ChildPydantic] = None
+        path_keys: Dict[pathlib.Path, pathlib.Path]
 
     obj = MyPydantic(
-        foo="bar", uid=test_uuid, tim=test_time, child=ChildPydantic(uid=test_uuid)
+        foo="bar",
+        uid=test_uuid,
+        tim=test_time,
+        child=ChildPydantic(
+            uid=test_uuid, child_path_keys={pathlib.Path("foo"): pathlib.Path("bar")}
+        ),
+        path_keys={pathlib.Path("foo"): pathlib.Path("bar")},
     )
     res = json.loads(json.dumps(obj, default=_serialize_json))
     expected = {
@@ -737,7 +746,9 @@ class MyPydantic(BaseModel):
         "tim": test_time.isoformat(),
         "child": {
             "uid": str(test_uuid),
+            "child_path_keys": {"foo": "bar"},
         },
+        "path_keys": {"foo": "bar"},
     }
     assert res == expected
 
@@ -777,6 +788,7 @@ def __repr__(self):
     class MyPydantic(BaseModel):
         foo: str
         bar: int
+        path_keys: Dict[pathlib.Path, "MyPydantic"]
 
     @dataclasses.dataclass
     class MyDataclass:
@@ -816,7 +828,11 @@ class MyNamedTuple(NamedTuple):
         "class_with_tee": ClassWithTee(),
         "my_dataclass": MyDataclass("foo", 1),
         "my_enum": MyEnum.FOO,
-        "my_pydantic": MyPydantic(foo="foo", bar=1),
+        "my_pydantic": MyPydantic(
+            foo="foo",
+            bar=1,
+            path_keys={pathlib.Path("foo"): MyPydantic(foo="foo", bar=1, path_keys={})},
+        ),
         "my_pydantic_class": MyPydantic,
         "person": Person(name="foo_person"),
         "a_bool": True,
@@ -842,7 +858,11 @@ class MyNamedTuple(NamedTuple):
         "class_with_tee": "tee_a, tee_b",
         "my_dataclass": {"foo": "foo", "bar": 1},
         "my_enum": "foo",
-        "my_pydantic": {"foo": "foo", "bar": 1},
+        "my_pydantic": {
+            "foo": "foo",
+            "bar": 1,
+            "path_keys": {"foo": {"foo": "foo", "bar": 1, "path_keys": {}}},
+        },
         "my_pydantic_class": lambda x: "MyPydantic" in x,
         "person": {"name": "foo_person"},
         "a_bool": True,