Skip to content

Commit f005e0d

Browse files
authored
fix(tracing): fix coze tracer (#171)
1 parent 311a4b0 commit f005e0d

File tree

1 file changed

+18
-2
lines changed

1 file changed

+18
-2
lines changed

veadk/tracing/telemetry/attributes/extractors/llm_attributes_extractors.py

Lines changed: 18 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -244,8 +244,24 @@ def llm_gen_ai_messages(params: LLMAttributesParams) -> ExtractorResponse:
244244
)
245245
if part.inline_data:
246246
if len(content.parts) == 1:
247-
# TODO(qingliang)
248-
pass
247+
part = content.parts[0]
248+
user_event["gen_ai.user.message"].update(
249+
{
250+
"parts.0.type": "image_url",
251+
"parts.0.image_url.name": (
252+
part.inline_data.display_name.split(
253+
"/"
254+
)[-1]
255+
if part.inline_data.display_name
256+
else "<unknown_image_name>"
257+
),
258+
"parts.0.image_url.url": (
259+
part.inline_data.display_name
260+
if part.inline_data.display_name
261+
else "<unknown_image_url>"
262+
),
263+
}
264+
)
249265
else:
250266
user_event["gen_ai.user.message"].update(
251267
{

0 commit comments

Comments
 (0)