Merge pull request #225 from UiPath/feature/traces_test

Testcases: add asserts for traces

Author: ionmincu

pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "uipath-langchain"
-version = "0.0.142"
+version = "0.0.143"
 description = "UiPath Langchain"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"
 dependencies = [
-    "uipath>=2.1.101, <2.2.0",
+    "uipath>=2.1.103, <2.2.0",
     "langgraph>=0.5.0, <0.7.0",
     "langchain-core>=0.3.34",
     "langgraph-checkpoint-sqlite>=2.0.3",

src/uipath_langchain/_cli/cli_debug.py

@@ -12,8 +12,9 @@
     UiPathRuntimeFactory,
 )
 from uipath._cli.middlewares import MiddlewareResult
+from uipath.tracing import LlmOpsHttpExporter
 
-from .._tracing import LangChainExporter, _instrument_traceable_attributes
+from .._tracing import _instrument_traceable_attributes
 from ._runtime._exception import LangGraphRuntimeError
 from ._runtime._runtime import (  # type: ignore[attr-defined]
     LangGraphRuntimeContext,
@@ -61,7 +62,9 @@ def generate_runtime(
             )
 
             if context.job_id:
-                runtime_factory.add_span_exporter(LangChainExporter())
+                runtime_factory.add_span_exporter(
+                    LlmOpsHttpExporter(extra_process_spans=True)
+                )
 
             runtime_factory.add_instrumentor(LangChainInstrumentor, get_current_span)
 

src/uipath_langchain/_cli/cli_eval.py

@@ -15,12 +15,12 @@
 from uipath._cli.middlewares import MiddlewareResult
 from uipath._events._event_bus import EventBus
 from uipath.eval._helpers import auto_discover_entrypoint
+from uipath.tracing import LlmOpsHttpExporter
 
 from uipath_langchain._cli._runtime._context import LangGraphRuntimeContext
 from uipath_langchain._cli._runtime._runtime import LangGraphScriptRuntime
 from uipath_langchain._cli._utils._graph import LangGraphConfig
 from uipath_langchain._tracing import (
-    LangChainExporter,
     _instrument_traceable_attributes,
 )
 
@@ -41,7 +41,7 @@ def langgraph_eval_middleware(
 
         if kwargs.get("register_progress_reporter", False):
             progress_reporter = StudioWebProgressReporter(
-                spans_exporter=LangChainExporter()
+                spans_exporter=LlmOpsHttpExporter(extra_process_spans=True)
             )
             asyncio.run(progress_reporter.subscribe_to_eval_runtime_events(event_bus))
         console_reporter = ConsoleProgressReporter()
@@ -76,7 +76,9 @@ def generate_runtime(ctx: LangGraphRuntimeContext) -> LangGraphScriptRuntime:
         )
 
         if eval_context.job_id:
-            runtime_factory.add_span_exporter(LangChainExporter())
+            runtime_factory.add_span_exporter(
+                LlmOpsHttpExporter(extra_process_spans=True)
+            )
 
         runtime_factory.add_instrumentor(LangChainInstrumentor, get_current_span)
 

src/uipath_langchain/_cli/cli_run.py

@@ -13,8 +13,11 @@
 )
 from uipath._cli.middlewares import MiddlewareResult
 from uipath._events._events import UiPathAgentStateEvent
+from uipath.tracing import JsonLinesFileExporter, LlmOpsHttpExporter
 
-from .._tracing import LangChainExporter, _instrument_traceable_attributes
+from .._tracing import (
+    _instrument_traceable_attributes,
+)
 from ._runtime._exception import LangGraphRuntimeError
 from ._runtime._runtime import (  # type: ignore[attr-defined]
     LangGraphRuntimeContext,
@@ -24,7 +27,11 @@
 
 
 def langgraph_run_middleware(
-    entrypoint: Optional[str], input: Optional[str], resume: bool, **kwargs
+    entrypoint: Optional[str],
+    input: Optional[str],
+    resume: bool,
+    trace_file: Optional[str] = None,
+    **kwargs,
 ) -> MiddlewareResult:
     """Middleware to handle LangGraph execution"""
     config = LangGraphConfig()
@@ -61,8 +68,13 @@ def generate_runtime(
 
             runtime_factory.add_instrumentor(LangChainInstrumentor, get_current_span)
 
+            if trace_file:
+                runtime_factory.add_span_exporter(JsonLinesFileExporter(trace_file))
+
             if context.job_id:
-                runtime_factory.add_span_exporter(LangChainExporter())
+                runtime_factory.add_span_exporter(
+                    LlmOpsHttpExporter(extra_process_spans=True)
+                )
                 await runtime_factory.execute(context)
             else:
                 debug_bridge: UiPathDebugBridge = ConsoleDebugBridge()

src/uipath_langchain/_tracing/__init__.py

@@ -1,4 +1,5 @@
 from ._instrument_traceable import _instrument_traceable_attributes
-from ._oteladapter import LangChainExporter
 
-__all__ = ["LangChainExporter", "_instrument_traceable_attributes"]
+__all__ = [
+    "_instrument_traceable_attributes",
+]