Merge pull request #15 from UiPath/fix/add_attributes

fix: add attributes to execution runtime

Author: cristipufu

pyproject.toml

@@ -1,22 +1,19 @@
 [project]
 name = "uipath-runtime"
-version = "0.0.7"
+version = "0.0.8"
 description = "Runtime abstractions and interfaces for building agents and automation scripts in the UiPath ecosystem"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
 dependencies = [
-  "opentelemetry-sdk>=1.38.0",
-  "opentelemetry-instrumentation>=0.59b0",
   "pydantic>=2.12.3",
-  "uipath-core>=0.0.3, <0.1.0",
+  "uipath-core>=0.0.4, <0.1.0",
 ]
 classifiers = [
   "Intended Audience :: Developers",
   "Topic :: Software Development :: Build Tools",
   "Programming Language :: Python :: 3.11",
   "Programming Language :: Python :: 3.12",
   "Programming Language :: Python :: 3.13",
-
 ]
 maintainers = [
   { name = "Marius Cosareanu", email = "marius.cosareanu@uipath.com" },

src/uipath/runtime/base.py

@@ -147,13 +147,15 @@ def __init__(
         delegate: UiPathRuntimeProtocol,
         trace_manager: UiPathTraceManager,
         root_span: str = "root",
+        span_attributes: Optional[dict[str, str]] = None,
         log_handler: Optional[UiPathRuntimeExecutionLogHandler] = None,
         execution_id: Optional[str] = None,
     ):
         """Initialize the executor."""
         self.delegate = delegate
         self.trace_manager = trace_manager
         self.root_span = root_span
+        self.span_attributes = span_attributes
         self.execution_id = execution_id
         self.log_handler = log_handler
         if execution_id is not None and log_handler is None:
@@ -174,7 +176,9 @@ async def execute(
         try:
             if self.execution_id:
                 with self.trace_manager.start_execution_span(
-                    self.root_span, execution_id=self.execution_id
+                    self.root_span,
+                    execution_id=self.execution_id,
+                    attributes=self.span_attributes,
                 ):
                     return await self.delegate.execute(input, options=options)
             else:
@@ -209,7 +213,9 @@ async def stream(
         try:
             if self.execution_id:
                 with self.trace_manager.start_execution_span(
-                    self.root_span, execution_id=self.execution_id
+                    self.root_span,
+                    execution_id=self.execution_id,
+                    attributes=self.span_attributes,
                 ):
                     async for event in self.delegate.stream(input, options=options):
                         yield event

src/uipath/runtime/context.py

@@ -9,10 +9,8 @@
     Optional,
     TypeVar,
 )
-from uuid import uuid4
 
 from pydantic import BaseModel
-from uipath.core.tracing.context import UiPathTraceContext
 
 from uipath.runtime.errors import (
     UiPathErrorCategory,
@@ -34,7 +32,6 @@ class UiPathRuntimeContext(BaseModel):
     entrypoint: Optional[str] = None
     input: Optional[dict[str, Any]] = None
     job_id: Optional[str] = None
-    trace_context: Optional[UiPathTraceContext] = None
     config_path: str = "uipath.json"
     runtime_dir: Optional[str] = "__uipath"
     result_file: str = "output.json"
@@ -208,19 +205,6 @@ def with_defaults(cls: type[C], config_path: Optional[str] = None, **kwargs) ->
         base.job_id = os.environ.get("UIPATH_JOB_KEY")
         base.logs_min_level = os.environ.get("LOG_LEVEL", "INFO")
 
-        base.trace_context = UiPathTraceContext(
-            trace_id=os.environ.get("UIPATH_TRACE_ID"),
-            parent_span_id=os.environ.get("UIPATH_PARENT_SPAN_ID"),
-            root_span_id=os.environ.get("UIPATH_ROOT_SPAN_ID"),
-            enabled=tracing_enabled,
-            job_id=os.environ.get("UIPATH_JOB_KEY"),
-            org_id=os.environ.get("UIPATH_ORGANIZATION_ID"),
-            tenant_id=os.environ.get("UIPATH_TENANT_ID"),
-            process_key=os.environ.get("UIPATH_PROCESS_UUID"),
-            folder_key=os.environ.get("UIPATH_FOLDER_KEY"),
-            reference_id=os.environ.get("UIPATH_JOB_KEY") or str(uuid4()),
-        )
-
         # Override with kwargs
         for k, v in kwargs.items():
             setattr(base, k, v)