tracing: add openllmetry

Author: ionmincu

pyproject.toml

@@ -1,11 +1,12 @@
 [project]
 name = "uipath-llamaindex"
-version = "0.0.6"
+version = "0.0.7"
 description = "UiPath LlamaIndex SDK"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"
 dependencies = [
     "llama-index>=0.12.36",
+    "opentelemetry-instrumentation-llamaindex>=0.40.7",
     "uipath>=2.0.56",
 ]
 classifiers = [

src/uipath_llamaindex/_cli/cli_run.py

@@ -1,11 +1,14 @@
 import asyncio
 import logging
+from contextlib import suppress
 from os import environ as env
 from typing import Optional
 
 from dotenv import load_dotenv
+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 ._runtime._context import UiPathLlamaIndexRuntimeContext
 from ._runtime._exception import UiPathLlamaIndexRuntimeError
@@ -30,6 +33,10 @@ def llamaindex_run_middleware(
     try:
 
         async def execute():
+
+            with suppress(Exception): 
+                LlamaIndexInstrumentor().instrument(tracer_provider=get_trace_provider())
+
             context = UiPathLlamaIndexRuntimeContext.from_config(
                 env.get("UIPATH_CONFIG_PATH", "uipath.json")
             )