Merge pull request #599 from UiPath/fix/remove-exporters-from-runtimes

Fix/remove exporters from runtimes

Author: radu-mocanu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath"
-version = "2.1.55"
+version = "2.1.56"
 description = "Python SDK and CLI for UiPath Platform, enabling programmatic interaction with automation services, process management, and deployment tools."
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"

src/uipath/_cli/_runtime/_contracts.py

@@ -24,7 +24,7 @@
 from pydantic import BaseModel, Field
 
 from uipath.agent.conversation import UiPathConversationEvent, UiPathConversationMessage
-from uipath.tracing import LlmOpsHttpExporter, TracingManager
+from uipath.tracing import TracingManager
 
 from ._logging import LogsInterceptor
 
@@ -698,8 +698,6 @@ def __init__(
         self.tracer_provider: TracerProvider = TracerProvider()
         self.tracer_span_processors: List[SpanProcessor] = []
         trace.set_tracer_provider(self.tracer_provider)
-        if os.getenv("UIPATH_JOB_KEY"):
-            self.add_span_exporter(LlmOpsHttpExporter())
 
     def add_span_exporter(
         self,

src/uipath/_cli/cli_eval.py

@@ -14,6 +14,7 @@
 from uipath._cli._runtime._runtime import UiPathScriptRuntime
 from uipath._cli.middlewares import Middlewares
 from uipath.eval._helpers import auto_discover_entrypoint
+from uipath.tracing import LlmOpsHttpExporter
 
 from .._utils.constants import ENV_JOB_ID
 from ..telemetry import track
@@ -108,6 +109,8 @@ def generate_runtime_context(**context_kwargs) -> UiPathRuntimeContext:
                 UiPathRuntimeContext,
                 context_generator=generate_runtime_context,
             )
+            if not no_report or eval_context.job_id:
+                runtime_factory.add_span_exporter(LlmOpsHttpExporter())
 
             async def execute():
                 async with UiPathEvalRuntime.from_eval_context(

src/uipath/_cli/cli_run.py

@@ -7,6 +7,7 @@
 import click
 
 from uipath._cli._utils._debug import setup_debugging
+from uipath.tracing import LlmOpsHttpExporter
 
 from .._utils.constants import (
     ENV_JOB_ID,
@@ -102,6 +103,7 @@ def run(
             runtime_factory = UiPathRuntimeFactory(
                 UiPathScriptRuntime, UiPathRuntimeContext
             )
+
             context = UiPathRuntimeContext.with_defaults(
                 entrypoint=entrypoint,
                 input=input,
@@ -110,6 +112,8 @@ def run(
                 execution_output_file=output_file,
                 debug=debug,
             )
+            if context.job_id:
+                runtime_factory.add_span_exporter(LlmOpsHttpExporter())
 
             asyncio.run(runtime_factory.execute(context))