Skip to content

Commit 059a7c0

Browse files
committed
feat(tracer): add support for custom names on traced
1 parent ec90a1a commit 059a7c0

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

pyproject.toml

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[project]
22
name = "uipath-langchain"
3-
version = "0.0.97"
3+
version = "0.0.98"
44
description = "UiPath Langchain"
55
readme = { file = "README.md", content-type = "text/markdown" }
66
requires-python = ">=3.10"
@@ -59,6 +59,9 @@ langchain = [
5959
"uipath-langchain>=0.0.2"
6060
]
6161

62+
[tool.hatch.build.targets.wheel]
63+
packages = ["src/uipath_langchain"]
64+
6265
[tool.ruff]
6366
line-length = 88
6467
indent-width = 4

src/uipath_langchain/tracers/_instrument_traceable.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -334,16 +334,20 @@ def _create_appropriate_wrapper(
334334

335335

336336
def _uipath_traced(
337+
name: Optional[str] = None,
337338
run_type: Optional[str] = None,
338339
span_type: Optional[str] = None,
339340
input_processor: Optional[Callable[..., Any]] = None,
340341
output_processor: Optional[Callable[..., Any]] = None,
342+
*args: Any,
343+
**kwargs: Any,
341344
):
342345
"""Decorator factory that creates traced functions using dispatch_trace_event."""
343346

344347
def decorator(func):
345348
return _create_traced_wrapper(
346349
func=func,
350+
func_name=name,
347351
run_type=run_type,
348352
span_type=span_type,
349353
input_processor=input_processor,

0 commit comments

Comments
 (0)