Merge pull request #12 from UiPath/feature/add_tracing_3

tracer: fix tracerprovider

Author: ionmincu

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-llamaindex"
-version = "0.0.8"
+version = "0.0.9"
 description = "UiPath LlamaIndex SDK"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"

src/uipath_llamaindex/_cli/cli_run.py

@@ -5,10 +5,11 @@
 from typing import Optional
 
 from dotenv import load_dotenv
+from opentelemetry import trace
 from opentelemetry.instrumentation.llamaindex import LlamaIndexInstrumentor
 from uipath._cli._runtime._contracts import UiPathTraceContext
 from uipath._cli.middlewares import MiddlewareResult
-from uipath.tracing import get_trace_provider, wait_for_tracers
+from uipath.tracing import wait_for_tracers
 
 from ._runtime._context import UiPathLlamaIndexRuntimeContext
 from ._runtime._exception import UiPathLlamaIndexRuntimeError
@@ -35,7 +36,7 @@ def llamaindex_run_middleware(
         async def execute():
 
             with suppress(Exception): 
-                LlamaIndexInstrumentor().instrument(tracer_provider=get_trace_provider())
+                LlamaIndexInstrumentor().instrument(tracer_provider=trace.get_tracer_provider())
 
             context = UiPathLlamaIndexRuntimeContext.from_config(
                 env.get("UIPATH_CONFIG_PATH", "uipath.json")