feat: simple serialization defaults

Author: edis-uipath

src/uipath/core/tracing/_utils.py

@@ -3,14 +3,11 @@
 import inspect
 import json
 from collections.abc import Callable
-from dataclasses import asdict, is_dataclass
-from datetime import datetime, timezone
-from enum import Enum
-from typing import Any, Mapping, Optional, cast
-from zoneinfo import ZoneInfo
+from typing import Any, Mapping, Optional
 
 from opentelemetry.trace import Span
-from pydantic import BaseModel
+
+from uipath.core.utils.serialization import simple_serialize_defaults
 
 
 def get_supported_params(
@@ -31,64 +28,19 @@ def get_supported_params(
     return supported
 
 
-def _simple_serialize_defaults(
-    obj: Any,
-) -> dict[str, Any] | list[Any] | str | int | float | bool | None:
-    # Handle Pydantic BaseModel instances
-    if hasattr(obj, "model_dump") and not isinstance(obj, type):
-        return obj.model_dump(exclude_none=True, mode="json")
-
-    # Handle classes - convert to schema representation
-    if isinstance(obj, type) and issubclass(obj, BaseModel):
-        return {
-            "__class__": obj.__name__,
-            "__module__": obj.__module__,
-            "schema": obj.model_json_schema(),
-        }
-    if hasattr(obj, "dict") and not isinstance(obj, type):
-        return obj.dict()
-    if hasattr(obj, "to_dict") and not isinstance(obj, type):
-        return obj.to_dict()
-
-    # Handle dataclasses
-    if is_dataclass(obj) and not isinstance(obj, type):
-        return asdict(obj)
-
-    # Handle enums
-    if isinstance(obj, Enum):
-        return _simple_serialize_defaults(obj.value)
-
-    if isinstance(obj, (set, tuple)):
-        if hasattr(obj, "_asdict") and callable(obj._asdict):  # pyright: ignore[reportAttributeAccessIssue]
-            return cast(dict[str, Any], obj._asdict())  # pyright: ignore[reportAttributeAccessIssue]
-        return list(obj)
-
-    if isinstance(obj, datetime):
-        return obj.isoformat()
-
-    if isinstance(obj, (timezone, ZoneInfo)):
-        return obj.tzname(None)
-
-    # Allow JSON-serializable primitives to pass through unchanged
-    if obj is None or isinstance(obj, (bool, int, float, str)):
-        return obj
-
-    return str(obj)
-
-
 def format_args_for_trace_json(
     signature: inspect.Signature, *args: Any, **kwargs: Any
 ) -> str:
     """Return a JSON string of inputs from the function signature."""
     result = format_args_for_trace(signature, *args, **kwargs)
-    return json.dumps(result, default=_simple_serialize_defaults)
+    return json.dumps(result, default=simple_serialize_defaults)
 
 
 def format_object_for_trace_json(
     input_object: Any,
 ) -> str:
     """Return a JSON string of inputs from the function signature."""
-    return json.dumps(input_object, default=_simple_serialize_defaults)
+    return json.dumps(input_object, default=simple_serialize_defaults)
 
 
 def format_args_for_trace(

src/uipath/core/utils/__init__.py

@@ -0,0 +1,5 @@
+"""Utility functions for uipath.core."""
+
+from .serialization import simple_serialize_defaults
+
+__all__ = ["simple_serialize_defaults"]

src/uipath/core/utils/serialization.py

@@ -0,0 +1,114 @@
+"""Serialization utilities for converting Python objects to JSON-serializable formats."""
+
+from dataclasses import asdict, is_dataclass
+from datetime import datetime, timezone
+from enum import Enum
+from typing import Any, cast
+from zoneinfo import ZoneInfo
+
+from pydantic import BaseModel
+
+
+def simple_serialize_defaults(
+    obj: Any,
+) -> dict[str, Any] | list[Any] | str | int | float | bool | None:
+    """Convert Python objects to JSON-serializable formats.
+
+    Handles common Python types that are not natively JSON-serializable:
+    - Pydantic models (v1 and v2)
+    - Dataclasses
+    - Enums
+    - Datetime objects
+    - Timezone objects
+    - Named tuples
+    - Sets and tuples
+
+    This function is designed to be used as the `default` parameter in json.dumps():
+    ```python
+    import json
+    result = json.dumps(obj, default=simple_serialize_defaults)
+    ```
+
+    Args:
+        obj: The object to serialize
+
+    Returns:
+        A JSON-serializable representation of the object:
+        - Pydantic models: dict from model_dump()
+        - Dataclasses: dict from asdict()
+        - Enums: the enum value (recursively serialized)
+        - datetime: ISO format string
+        - timezone/ZoneInfo: timezone name
+        - sets/tuples: converted to lists
+        - named tuples: converted to dict
+        - Primitives (None, bool, int, float, str): returned unchanged
+        - Other types: converted to string with str()
+
+    Examples:
+        >>> from datetime import datetime
+        >>> from pydantic import BaseModel
+        >>>
+        >>> class User(BaseModel):
+        ...     name: str
+        ...     created_at: datetime
+        >>>
+        >>> user = User(name="Alice", created_at=datetime.now())
+        >>> import json
+        >>> json.dumps(user, default=simple_serialize_defaults)
+        '{"name": "Alice", "created_at": "2024-01-01T12:00:00"}'
+    """
+    # Handle Pydantic BaseModel instances
+    if hasattr(obj, "model_dump") and not isinstance(obj, type):
+        return obj.model_dump(exclude_none=True, mode="json")
+
+    # Handle Pydantic model classes - convert to schema representation
+    if isinstance(obj, type) and issubclass(obj, BaseModel):
+        return {
+            "__class__": obj.__name__,
+            "__module__": obj.__module__,
+            "schema": obj.model_json_schema(),
+        }
+
+    # Handle Pydantic v1 models
+    if hasattr(obj, "dict") and not isinstance(obj, type):
+        return obj.dict()
+
+    # Handle objects with to_dict method
+    if hasattr(obj, "to_dict") and not isinstance(obj, type):
+        return obj.to_dict()
+
+    # Handle dataclasses
+    if is_dataclass(obj) and not isinstance(obj, type):
+        return asdict(obj)
+
+    # Handle enums - recursively serialize the value
+    if isinstance(obj, Enum):
+        return simple_serialize_defaults(obj.value)
+
+    # Handle sets and tuples
+    if isinstance(obj, (set, tuple)):
+        # Check if it's a named tuple (has _asdict method)
+        if hasattr(obj, "_asdict") and callable(
+            obj._asdict  # pyright: ignore[reportAttributeAccessIssue]
+        ):
+            return cast(
+                dict[str, Any],
+                obj._asdict(),  # pyright: ignore[reportAttributeAccessIssue]
+            )
+        # Convert to list
+        return list(obj)
+
+    # Handle datetime objects
+    if isinstance(obj, datetime):
+        return obj.isoformat()
+
+    # Handle timezone objects
+    if isinstance(obj, (timezone, ZoneInfo)):
+        return obj.tzname(None)
+
+    # Allow JSON-serializable primitives to pass through unchanged
+    if obj is None or isinstance(obj, (bool, int, float, str, list, dict)):
+        return obj
+
+    # Fallback: convert to string
+    return str(obj)

tests/utils/__init__.py

@@ -0,0 +1 @@
+"""Tests for uipath.core.utils."""