Merge pull request #3 from UiPath/fix/cleanup_traces

fix: cleanup unused methods

Author: ionmincu

src/uipath/core/tracing/_utils.py

@@ -2,8 +2,6 @@
 
 import inspect
 import json
-import random
-import uuid
 from collections.abc import Callable
 from dataclasses import asdict, is_dataclass
 from datetime import datetime, timezone
@@ -75,50 +73,6 @@ def _simple_serialize_defaults(obj):
     return str(obj)
 
 
-def span_id_to_uuid4(span_id: int) -> uuid.UUID:
-    """Convert a 64-bit span ID to a valid UUID4 format.
-
-    Creates a UUID where:
-    - The 64 least significant bits contain the span ID
-    - The UUID version (bits 48-51) is set to 4
-    - The UUID variant (bits 64-65) is set to binary 10
-    """
-    # Generate deterministic high bits using the span_id as seed
-    temp_random = random.Random(span_id)
-    high_bits = temp_random.getrandbits(64)
-
-    # Combine high bits and span ID into a 128-bit integer
-    combined = (high_bits << 64) | span_id
-
-    # Set version to 4 (UUID4)
-    combined = (combined & ~(0xF << 76)) | (0x4 << 76)
-
-    # Set variant to binary 10
-    combined = (combined & ~(0x3 << 62)) | (2 << 62)
-
-    # Convert to hex string in UUID format
-    hex_str = format(combined, "032x")
-    return uuid.UUID(hex_str)
-
-
-def trace_id_to_uuid4(trace_id: int) -> uuid.UUID:
-    """Convert a 128-bit trace ID to a valid UUID4 format.
-
-    Modifies the trace ID to conform to UUID4 requirements:
-    - The UUID version (bits 48-51) is set to 4
-    - The UUID variant (bits 64-65) is set to binary 10
-    """
-    # Set version to 4 (UUID4)
-    uuid_int = (trace_id & ~(0xF << 76)) | (0x4 << 76)
-
-    # Set variant to binary 10
-    uuid_int = (uuid_int & ~(0x3 << 62)) | (2 << 62)
-
-    # Convert to hex string in UUID format
-    hex_str = format(uuid_int, "032x")
-    return uuid.UUID(hex_str)
-
-
 def format_args_for_trace_json(
     signature: inspect.Signature, *args: Any, **kwargs: Any
 ) -> str:

src/uipath/core/tracing/decorators.py

@@ -3,11 +3,12 @@
 import inspect
 import json
 import logging
+import random
 from functools import wraps
 from typing import Any, Callable, Optional
 
 from opentelemetry import trace
-from opentelemetry.trace import NonRecordingSpan
+from opentelemetry.trace import NonRecordingSpan, SpanContext, TraceFlags
 from opentelemetry.trace.status import StatusCode
 
 from uipath.core.tracing._utils import (
@@ -36,10 +37,6 @@ def decorator(func):
         def get_span():
             if not recording:
                 # Create a valid but non-sampled trace context
-                import random
-
-                from opentelemetry.trace import SpanContext, TraceFlags
-
                 # Generate a valid trace ID (not INVALID)
                 trace_id = random.getrandbits(128)
                 span_id = random.getrandbits(64)
@@ -277,8 +274,6 @@ def decorator(func):
         # Decorate the function with only supported parameters
         decorated_func = tracer_impl(**supported_params)(func)
 
-        # Register both original and decorated function with parameters
-        UiPathTracingManager.register_traced_function(func, decorated_func, params)
         return decorated_func
 
     return decorator

src/uipath/core/tracing/manager.py

@@ -1,7 +1,7 @@
 """Tracing manager for handling tracer implementations and function registry."""
 
 import logging
-from typing import Any, Callable, List, Optional, Tuple
+from typing import Any, Callable, Optional
 
 from opentelemetry import context, trace
 from opentelemetry.trace import set_span_in_context
@@ -12,10 +12,6 @@
 class UiPathTracingManager:
     """Static utility class to manage tracing implementations and decorated functions."""
 
-    # Registry to track original functions, decorated functions, and their parameters
-    # Each entry is (original_func, decorated_func, params)
-    _traced_registry: List[Tuple[Callable[..., Any], Callable[..., Any], Any]] = []
-
     _current_span_provider: Optional[Callable[[], Any]] = None
 
     @classmethod
@@ -41,6 +37,7 @@ def get_parent_context():
         """
         # Always use the currently active OTel span if valid (recursion / children)
         current_span = trace.get_current_span()
+
         if current_span is not None and current_span.get_span_context().is_valid:
             return set_span_in_context(current_span)
 
@@ -54,18 +51,9 @@ def get_parent_context():
                 logger.warning(f"Error getting current span from provider: {e}")
 
         # Last fallback
-        return context.get_current()
-
-    @classmethod
-    def register_traced_function(cls, original_func, decorated_func, params):
-        """Register a function decorated with @traced and its parameters.
+        ctx = context.get_current()
 
-        Args:
-            original_func: The original function before decoration
-            decorated_func: The function after decoration
-            params: The parameters used for tracing
-        """
-        cls._traced_registry.append((original_func, decorated_func, params))
+        return ctx
 
 
 __all__ = ["UiPathTracingManager"]