uipath 2.1.57__py3-none-any.whl → 2.1.58__py3-none-any.whl

uipath/tracing/_traced.py

@@ -215,11 +215,15 @@ def _opentelemetry_traced(
                 if input_processor:
                     processed_inputs = input_processor(json.loads(inputs))
                     inputs = json.dumps(processed_inputs, default=str)
-                span.set_attribute("inputs", inputs)
+                span.set_attribute("input.mime_type", "application/json")
+                span.set_attribute("input.value", inputs)
 
                 result = func(*args, **kwargs)
                 output = output_processor(result) if output_processor else result
-                span.set_attribute("output", json.dumps(output, default=str))
+                span.set_attribute(
+                    "output.value", _SpanUtils.format_object_for_trace_json(output)
+                )
+                span.set_attribute("output.mime_type", "application/json")
                 return result
             except Exception as e:
                 span.record_exception(e)
@@ -249,11 +253,15 @@ def _opentelemetry_traced(
                 if input_processor:
                     processed_inputs = input_processor(json.loads(inputs))
                     inputs = json.dumps(processed_inputs, default=str)
-                span.set_attribute("inputs", inputs)
+                span.set_attribute("input.mime_type", "application/json")
+                span.set_attribute("input.value", inputs)
 
                 result = await func(*args, **kwargs)
                 output = output_processor(result) if output_processor else result
-                span.set_attribute("output", json.dumps(output, default=str))
+                span.set_attribute(
+                    "output.value", _SpanUtils.format_object_for_trace_json(output)
+                )
+                span.set_attribute("output.mime_type", "application/json")
                 return result
             except Exception as e:
                 span.record_exception(e)
@@ -285,7 +293,8 @@ def _opentelemetry_traced(
                 if input_processor:
                     processed_inputs = input_processor(json.loads(inputs))
                     inputs = json.dumps(processed_inputs, default=str)
-                span.set_attribute("inputs", inputs)
+                span.set_attribute("input.mime_type", "application/json")
+                span.set_attribute("input.value", inputs)
 
                 outputs = []
                 for item in func(*args, **kwargs):
@@ -293,7 +302,10 @@ def _opentelemetry_traced(
                     span.add_event(f"Yielded: {item}")
                     yield item
                 output = output_processor(outputs) if output_processor else outputs
-                span.set_attribute("output", json.dumps(output, default=str))
+                span.set_attribute(
+                    "output.value", _SpanUtils.format_object_for_trace_json(output)
+                )
+                span.set_attribute("output.mime_type", "application/json")
             except Exception as e:
                 span.record_exception(e)
                 span.set_status(
@@ -324,7 +336,8 @@ def _opentelemetry_traced(
                 if input_processor:
                     processed_inputs = input_processor(json.loads(inputs))
                     inputs = json.dumps(processed_inputs, default=str)
-                span.set_attribute("inputs", inputs)
+                span.set_attribute("input.mime_type", "application/json")
+                span.set_attribute("input.value", inputs)
 
                 outputs = []
                 async for item in func(*args, **kwargs):
@@ -332,7 +345,10 @@ def _opentelemetry_traced(
                     span.add_event(f"Yielded: {item}")
                     yield item
                 output = output_processor(outputs) if output_processor else outputs
-                span.set_attribute("output", json.dumps(output, default=str))
+                span.set_attribute(
+                    "output.value", _SpanUtils.format_object_for_trace_json(output)
+                )
+                span.set_attribute("output.mime_type", "application/json")
             except Exception as e:
                 span.record_exception(e)
                 span.set_status(

uipath/tracing/_utils.py

@@ -4,10 +4,12 @@ import logging
 import os
 import random
 import uuid
-from dataclasses import dataclass, field
-from datetime import datetime
+from dataclasses import asdict, dataclass, field, is_dataclass
+from datetime import datetime, timezone
+from enum import Enum
 from os import environ as env
 from typing import Any, Dict, Optional
+from zoneinfo import ZoneInfo
 
 from opentelemetry.sdk.trace import ReadableSpan
 from opentelemetry.trace import StatusCode
@@ -15,6 +17,36 @@ from opentelemetry.trace import StatusCode
 logger = logging.getLogger(__name__)
 
 
+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()
+
+    # Handle dataclasses
+    if is_dataclass(obj) and not isinstance(obj, type):
+        return asdict(obj)
+
+    # Handle enums
+    if isinstance(obj, Enum):
+        return obj.value
+
+    if isinstance(obj, (set, tuple)):
+        if hasattr(obj, "_asdict") and callable(obj._asdict):
+            return obj._asdict()
+        return list(obj)
+
+    if isinstance(obj, datetime):
+        return obj.isoformat()
+
+    if isinstance(obj, (timezone, ZoneInfo)):
+        return obj.tzname(None)
+
+    return str(obj)
+
+
 @dataclass
 class UiPathSpan:
     """Represents a span in the UiPath tracing system."""
@@ -150,30 +182,32 @@ class _SpanUtils:
             status = 2  # Error
             attributes_dict["error"] = otel_span.status.description
 
-        original_inputs = attributes_dict.get("inputs", None)
-        original_outputs = attributes_dict.get("outputs", None)
+        original_inputs = attributes_dict.get("input", None)
+        original_outputs = attributes_dict.get("output", None)
 
         if original_inputs:
             try:
                 if isinstance(original_inputs, str):
                     json_inputs = json.loads(original_inputs)
-                    attributes_dict["inputs"] = json_inputs
+                    attributes_dict["input.value"] = json_inputs
+                    attributes_dict["input.mime_type"] = "application/json"
                 else:
-                    attributes_dict["inputs"] = original_inputs
+                    attributes_dict["input.value"] = original_inputs
             except Exception as e:
-                print(f"Error parsing inputs: {e}")
-                attributes_dict["inputs"] = str(original_inputs)
+                logger.warning(f"Error parsing inputs: {e}")
+                attributes_dict["input.value"] = str(original_inputs)
 
         if original_outputs:
             try:
                 if isinstance(original_outputs, str):
                     json_outputs = json.loads(original_outputs)
-                    attributes_dict["outputs"] = json_outputs
+                    attributes_dict["output.value"] = json_outputs
+                    attributes_dict["output.mime_type"] = "application/json"
                 else:
-                    attributes_dict["outputs"] = original_outputs
+                    attributes_dict["output.value"] = original_outputs
             except Exception as e:
-                print(f"Error parsing outputs: {e}")
-                attributes_dict["outputs"] = str(original_outputs)
+                logger.warning(f"Error parsing output: {e}")
+                attributes_dict["output.value"] = str(original_outputs)
 
         # Add events as additional attributes if they exist
         if otel_span.events:
@@ -233,7 +267,14 @@ class _SpanUtils:
     ) -> str:
         """Return a JSON string of inputs from the function signature."""
         result = _SpanUtils.format_args_for_trace(signature, *args, **kwargs)
-        return json.dumps(result, default=str)
+        return json.dumps(result, default=_simple_serialize_defaults)
+
+    @staticmethod
+    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)
 
     @staticmethod
     def format_args_for_trace(

{uipath-2.1.57.dist-info → uipath-2.1.58.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath
-Version: 2.1.57
+Version: 2.1.58
 Summary: Python SDK and CLI for UiPath Platform, enabling programmatic interaction with automation services, process management, and deployment tools.
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-python

{uipath-2.1.57.dist-info → uipath-2.1.58.dist-info}/RECORD

@@ -146,12 +146,12 @@ uipath/telemetry/_constants.py,sha256=uRDuEZayBYtBA0tMx-2AS_D-oiVA7oKgp9zid9jNat
 uipath/telemetry/_track.py,sha256=G_Pyq8n8iMvoCWhUpWedlptXUSuUSbQBBzGxsh4DW9c,4654
 uipath/tracing/__init__.py,sha256=GKRINyWdHVrDsI-8mrZDLdf0oey6GHGlNZTOADK-kgc,224
 uipath/tracing/_otel_exporters.py,sha256=X7cnuGqvxGbACZuFD2XYTWXwIse8pokOEAjeTPE6DCQ,3158
-uipath/tracing/_traced.py,sha256=CsuNMy67R5HCunuu84LnRNwcadNLhtHLMEawGaNsTYs,18435
-uipath/tracing/_utils.py,sha256=wJRELaPu69iY0AhV432Dk5QYf_N_ViRU4kAUG1BI1ew,10384
+uipath/tracing/_traced.py,sha256=CJIp6wqIoHVTmh9-Kuk7diD4OJDTAeA6x3zI9w0i4lI,19307
+uipath/tracing/_utils.py,sha256=ny3VgfUPT_5fQ0BAFl7hitR2jH6nNuuZEpR7DKVd8QE,11739
 uipath/utils/__init__.py,sha256=VD-KXFpF_oWexFg6zyiWMkxl2HM4hYJMIUDZ1UEtGx0,105
 uipath/utils/_endpoints_manager.py,sha256=iRTl5Q0XAm_YgcnMcJOXtj-8052sr6jpWuPNz6CgT0Q,8408
-uipath-2.1.57.dist-info/METADATA,sha256=LZV3q6VHuyZ0YY919szKSEItcd9Ir0p4wocixl1r7T0,6482
-uipath-2.1.57.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-uipath-2.1.57.dist-info/entry_points.txt,sha256=9C2_29U6Oq1ExFu7usihR-dnfIVNSKc-0EFbh0rskB4,43
-uipath-2.1.57.dist-info/licenses/LICENSE,sha256=-KBavWXepyDjimmzH5fVAsi-6jNVpIKFc2kZs0Ri4ng,1058
-uipath-2.1.57.dist-info/RECORD,,
+uipath-2.1.58.dist-info/METADATA,sha256=V3aUe96vDwV0iGCY6Uyr752tATUAiDRVYDonJ5Zckvs,6482
+uipath-2.1.58.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+uipath-2.1.58.dist-info/entry_points.txt,sha256=9C2_29U6Oq1ExFu7usihR-dnfIVNSKc-0EFbh0rskB4,43
+uipath-2.1.58.dist-info/licenses/LICENSE,sha256=-KBavWXepyDjimmzH5fVAsi-6jNVpIKFc2kZs0Ri4ng,1058
+uipath-2.1.58.dist-info/RECORD,,