tracing: wait for tracers

Author: ionmincu

pyproject.toml

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

src/uipath_llamaindex/_cli/cli_run.py

@@ -8,7 +8,7 @@
 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
+from uipath.tracing import get_trace_provider, wait_for_tracers
 
 from ._runtime._context import UiPathLlamaIndexRuntimeContext
 from ._runtime._exception import UiPathLlamaIndexRuntimeError
@@ -66,7 +66,10 @@ async def execute():
             async with UiPathLlamaIndexRuntime.from_context(context) as runtime:
                 await runtime.execute()
 
-        asyncio.run(execute())
+        try:
+            asyncio.run(execute())
+        finally:
+            wait_for_tracers()
 
         return MiddlewareResult(should_continue=False, error_message=None)