trace: add openinference

Author: ionmincu

pyproject.toml

@@ -6,7 +6,7 @@ readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"
 dependencies = [
     "llama-index>=0.12.36",
-    "opentelemetry-instrumentation-llamaindex>=0.40.7",
+    "openinference-instrumentation-llama-index>=4.2.1",
     "uipath>=2.0.56",
 ]
 classifiers = [
@@ -84,4 +84,4 @@ addopts = "-ra -q"
 name = "testpypi"
 url = "https://test.pypi.org/simple/"
 publish-url = "https://test.pypi.org/legacy/"
-explicit = true
+explicit = true

samples/quickstart-agent/pyproject.toml

@@ -9,3 +9,12 @@ dependencies = [
     "uipath-llamaindex>=0.0.11",
     "llama-index-llms-openai>=0.2.2"
 ]
+
+[[tool.uv.index]]
+name = "testpypi"
+url = "https://test.pypi.org/simple/"
+publish-url = "https://test.pypi.org/legacy/"
+explicit = true
+
+[tool.uv.sources]
+uipath = { index = "testpypi" }

src/uipath_llamaindex/_cli/_runtime/_runtime.py

@@ -1,7 +1,10 @@
 import json
 import logging
+from contextlib import suppress
 from typing import Optional
 
+from openinference.instrumentation.llama_index import LlamaIndexInstrumentor
+from opentelemetry import trace
 from uipath import UiPath
 from uipath._cli._runtime._contracts import (
     UiPathBaseRuntime,
@@ -38,6 +41,11 @@ async def execute(self) -> Optional[UiPathRuntimeResult]:
         """
         await self.validate()
 
+        with suppress(Exception):
+            LlamaIndexInstrumentor().instrument(
+                tracer_provider=trace.get_tracer_provider()
+            )
+
         try:
             start_event_class = self.context.workflow._start_event_class
 

src/uipath_llamaindex/_cli/cli_run.py

@@ -1,15 +1,11 @@
 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 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 wait_for_tracers
 
 from ._runtime._context import UiPathLlamaIndexRuntimeContext
 from ._runtime._exception import UiPathLlamaIndexRuntimeError
@@ -34,10 +30,6 @@ def llamaindex_run_middleware(
     try:
 
         async def execute():
-
-            with suppress(Exception): 
-                LlamaIndexInstrumentor().instrument(tracer_provider=trace.get_tracer_provider())
-
             context = UiPathLlamaIndexRuntimeContext.from_config(
                 env.get("UIPATH_CONFIG_PATH", "uipath.json")
             )
@@ -67,10 +59,7 @@ async def execute():
             async with UiPathLlamaIndexRuntime.from_context(context) as runtime:
                 await runtime.execute()
 
-        try:
-            asyncio.run(execute())
-        finally:
-            wait_for_tracers()
+        asyncio.run(execute())
 
         return MiddlewareResult(should_continue=False, error_message=None)