uipath-langchain 0.0.77__tar.gz → 0.0.78__tar.gz

{uipath_langchain-0.0.77 → uipath_langchain-0.0.78}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath-langchain
-Version: 0.0.77
+Version: 0.0.78
 Summary: UiPath Langchain
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-python

{uipath_langchain-0.0.77 → uipath_langchain-0.0.78}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-langchain"
-version = "0.0.77"
+version = "0.0.78"
 description = "UiPath Langchain"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.9"

{uipath_langchain-0.0.77 → uipath_langchain-0.0.78}/uipath_langchain/tracers/AsyncUiPathTracer.py

@@ -13,13 +13,15 @@ from langchain_core.tracers.base import AsyncBaseTracer
 from langchain_core.tracers.schemas import Run
 from pydantic import PydanticDeprecationWarning
 
+from ._utils import _simple_serialize_defaults
+
 logger = logging.getLogger(__name__)
 
 
 class Status:
     SUCCESS = 1
     ERROR = 2
-    INTERRUPTED = 1  # intentional equal to SUCCESS
+    INTERRUPTED = 3
 
 
 class AsyncUiPathTracer(AsyncBaseTracer):
@@ -235,7 +237,7 @@ class AsyncUiPathTracer(AsyncBaseTracer):
 
     def _safe_json_dump(self, obj) -> str:
         try:
-            json_str = json.dumps(obj, default=str)
+            json_str = json.dumps(obj, default=_simple_serialize_defaults)
             return json_str
         except Exception as e:
             logger.warning(e)

{uipath_langchain-0.0.77 → uipath_langchain-0.0.78}/uipath_langchain/tracers/UiPathTracer.py

@@ -11,6 +11,8 @@ from langchain_core.tracers.base import BaseTracer
 from langchain_core.tracers.schemas import Run
 from pydantic import PydanticDeprecationWarning
 
+from ._utils import _simple_serialize_defaults
+
 logger = logging.getLogger(__name__)
 
 
@@ -142,7 +144,7 @@ class UiPathTracer(BaseTracer):
 
     def _safe_json_dump(self, obj) -> str:
         try:
-            json_str = json.dumps(obj, default=str)
+            json_str = json.dumps(obj, default=_simple_serialize_defaults)
             return json_str
         except Exception as e:
             logger.warning(e)

uipath_langchain-0.0.78/uipath_langchain/tracers/_utils.py

@@ -0,0 +1,24 @@
+import datetime
+from zoneinfo import ZoneInfo
+
+
+def _simple_serialize_defaults(obj):
+    if hasattr(obj, "model_dump"):
+        return obj.model_dump(exclude_none=True, mode="json")
+    if hasattr(obj, "dict"):
+        return obj.dict()
+    if hasattr(obj, "to_dict"):
+        return obj.to_dict()
+
+    if isinstance(obj, (set, tuple)):
+        if hasattr(obj, "_asdict") and callable(obj._asdict):
+            return obj._asdict()
+        return list(obj)
+
+    if isinstance(obj, datetime.datetime):
+        return obj.isoformat()
+
+    if isinstance(obj, (datetime.timezone, ZoneInfo)):
+        return obj.tzname(None)
+
+    return str(obj)