Merge pull request #11 from UiPath/fix/configure_log_handler

fix: configure log handler

Author: cristipufu

pyproject.toml

@@ -1,14 +1,14 @@
 [project]
 name = "uipath-runtime"
-version = "0.0.4"
-description = "UiPath Runtime abstractions"
+version = "0.0.5"
+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",
+  "uipath-core>=0.0.3, <0.1.0",
 ]
 classifiers = [
   "Intended Audience :: Developers",

src/uipath/runtime/base.py

@@ -142,15 +142,16 @@ def __init__(
         delegate: T,
         trace_manager: UiPathTraceManager,
         root_span: str = "root",
+        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.execution_id = execution_id
-        self.log_handler: Optional[UiPathRuntimeExecutionLogHandler] = None
-        if execution_id is not None:
+        self.log_handler = log_handler
+        if execution_id is not None and log_handler is None:
             self.log_handler = UiPathRuntimeExecutionLogHandler(execution_id)
 
     async def execute(

tests/test_executor.py

@@ -128,21 +128,21 @@ async def test_multiple_factories_same_executor():
     # Execute runtime A
     runtime_a = factory_a.new_runtime(entrypoint="")
     execution_runtime_a = UiPathExecutionRuntime(
-        runtime_a, trace_manager, "runtime-a-span", "exec-a"
+        runtime_a, trace_manager, "runtime-a-span", execution_id="exec-a"
     )
     result_a = await execution_runtime_a.execute({"input": "a"})
 
     # Execute runtime B
     runtime_b = factory_b.new_runtime(entrypoint="")
     execution_runtime_b = UiPathExecutionRuntime(
-        runtime_b, trace_manager, "runtime-b-span", "exec-b"
+        runtime_b, trace_manager, "runtime-b-span", execution_id="exec-b"
     )
     result_b = await execution_runtime_b.execute({"input": "b"})
 
     # Execute runtime C with custom spans
     runtime_c = factory_c.new_runtime(entrypoint="")
     execution_runtime_c = UiPathExecutionRuntime(
-        runtime_c, trace_manager, "runtime-c-span", "exec-c"
+        runtime_c, trace_manager, "runtime-c-span", execution_id="exec-c"
     )
     result_c = await execution_runtime_c.execute({"input": "c"})