Skip to content

Commit c7aada3

Browse files
committed
fix: gen_ai input ouput
1 parent c02ba62 commit c7aada3

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

veadk/tracing/telemetry/telemetry.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@
3030
ToolAttributesParams,
3131
)
3232
from veadk.utils.logger import get_logger
33+
from veadk.utils.misc import safe_json_serialize
3334

3435
logger = get_logger(__name__)
3536

@@ -92,7 +93,7 @@ def _set_agent_input_attribute(
9293
user_content = invocation_context.user_content
9394
if user_content and user_content.parts:
9495
# set gen_ai.input attribute required by APMPlus
95-
span.set_attribute("gen_ai.input", user_content.model_dump(exclude_none=True))
96+
span.set_attribute("gen_ai.input", safe_json_serialize(user_content.model_dump(exclude_none=True)))
9697

9798
span.add_event(
9899
"gen_ai.user.message",
@@ -132,7 +133,7 @@ def _set_agent_output_attribute(span: Span, llm_response: LlmResponse) -> None:
132133
content = llm_response.content
133134
if content and content.parts:
134135
# set gen_ai.output attribute required by APMPlus
135-
span.set_attribute("gen_ai.output", content.model_dump(exclude_none=True))
136+
span.set_attribute("gen_ai.output", safe_json_serialize(content.model_dump(exclude_none=True)))
136137

137138
for idx, part in enumerate(content.parts):
138139
if part.text:

0 commit comments

Comments
 (0)