Skip to content

Commit 68db19e

Browse files
fix: add kwargs for patched tracing functions (#498)
* fix: add kwargs for patched tracing functions * fix: args --------- Co-authored-by: hanzhi.421 <hanzhi.421@bytedance.com>
1 parent 42edc02 commit 68db19e

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

veadk/tracing/telemetry/telemetry.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -305,6 +305,7 @@ def trace_tool_call(
305305
tool: BaseTool,
306306
args: dict[str, Any],
307307
function_response_event: Event,
308+
**kwargs,
308309
) -> None:
309310
"""Trace a tool function call with comprehensive telemetry data.
310311
@@ -344,6 +345,8 @@ def trace_call_llm(
344345
event_id: str,
345346
llm_request: LlmRequest,
346347
llm_response: LlmResponse,
348+
*args,
349+
**kwargs,
347350
) -> None:
348351
"""Trace a language model call with comprehensive telemetry data.
349352
@@ -413,4 +416,4 @@ def trace_call_llm(
413416

414417

415418
# Do not modify this function
416-
def trace_send_data(): ...
419+
def trace_send_data(**kwargs): ...

0 commit comments

Comments
 (0)