Skip to content

Commit d12bdea

Browse files
committed
fix: format
1 parent c7aada3 commit d12bdea

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

veadk/tracing/telemetry/telemetry.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -93,8 +93,10 @@ def _set_agent_input_attribute(
9393
user_content = invocation_context.user_content
9494
if user_content and user_content.parts:
9595
# set gen_ai.input attribute required by APMPlus
96-
span.set_attribute("gen_ai.input", safe_json_serialize(user_content.model_dump(exclude_none=True)))
97-
96+
span.set_attribute(
97+
"gen_ai.input",
98+
safe_json_serialize(user_content.model_dump(exclude_none=True)),
99+
)
98100
span.add_event(
99101
"gen_ai.user.message",
100102
{
@@ -133,7 +135,9 @@ def _set_agent_output_attribute(span: Span, llm_response: LlmResponse) -> None:
133135
content = llm_response.content
134136
if content and content.parts:
135137
# set gen_ai.output attribute required by APMPlus
136-
span.set_attribute("gen_ai.output", safe_json_serialize(content.model_dump(exclude_none=True)))
138+
span.set_attribute(
139+
"gen_ai.output", safe_json_serialize(content.model_dump(exclude_none=True))
140+
)
137141

138142
for idx, part in enumerate(content.parts):
139143
if part.text:

0 commit comments

Comments
 (0)