Merge pull request #40 from UiPath/feature/add_traced_integration

feat(trace): fix tracing integration with sdk

Author: ionmincu

pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "uipath-langchain"
-version = "0.0.91"
+version = "0.0.92"
 description = "UiPath Langchain"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"
 dependencies = [
-    "uipath>=2.0.13, <2.1.0",
+    "uipath>=2.0.14, <2.1.0",
     "langgraph>=0.2.70",
     "langchain-core>=0.3.34",
     "langgraph-checkpoint-sqlite>=2.0.3",

src/uipath_langchain/tracers/__init__.py

@@ -2,4 +2,8 @@
 from .AsyncUiPathTracer import AsyncUiPathTracer
 from .UiPathTracer import UiPathTracer
 
-__all__ = ["AsyncUiPathTracer", "UiPathTracer", "_instrument_traceable_attributes"]
+__all__ = [
+    "AsyncUiPathTracer",
+    "UiPathTracer",
+    "_instrument_traceable_attributes",
+]

src/uipath_langchain/tracers/_instrument_traceable.py

@@ -7,7 +7,7 @@
 from typing import Any, Callable, Dict, List, Literal, Optional
 
 from langchain_core.callbacks import dispatch_custom_event
-from uipath.tracing import TracedDecoratorRegistry
+from uipath.tracing import TracingManager
 
 from ._events import CustomTraceEvents, FunctionCallEventData
 
@@ -373,7 +373,8 @@ def uipath_trace_decorator(func):
 # Register the UIPath traced decorator
 def register_uipath_tracing():
     """Register the UIPath tracing decorator with TracedDecoratorRegistry."""
-    TracedDecoratorRegistry.register_decorator("uipath", _uipath_traced)
+    # Reapply to all previously decorated functions
+    TracingManager.reapply_traced_decorator(_uipath_traced)
 
 
 # Apply the patch