feat: agent run span

Author: feng-95

veadk/tracing/telemetry/exporters/inmemory_exporter.py

@@ -77,7 +77,7 @@ def __init__(self, exporter: _InMemoryExporter) -> None:
         super().__init__(exporter)
 
     def on_start(self, span, parent_context) -> None:
-        if span.name.startswith("invocation") or span.name.startswith("invoke") :
+        if span.name.startswith("invocation"):
             span.set_attribute("gen_ai.operation.name", "chain")
             span.set_attribute("gen_ai.span.kind", "workflow")
             span.set_attribute("gen_ai.usage.total_tokens", 0)
@@ -88,7 +88,7 @@ def on_start(self, span, parent_context) -> None:
             token = attach(ctx)  # mount context on `invocation` root span in Google ADK
             setattr(span, "_invocation_token", token)  # for later detach
 
-        if span.name.startswith("agent_run"):
+        if span.name.startswith("agent_run") or span.name.startswith("invoke_agent"):
             span.set_attribute("gen_ai.operation.name", "agent")
             span.set_attribute("gen_ai.span.kind", "agent")
 

veadk/tracing/telemetry/telemetry.py

@@ -93,8 +93,7 @@ def _set_agent_input_attribute(
     user_content = invocation_context.user_content
     if user_content and user_content.parts:
         # set gen_ai.input attribute required by APMPlus
-        # TODO: 优化 gen_ai.input，目前无法序列化
-        span.set_attribute("gen_ai.input", safe_json_serialize(user_content.parts))
+        span.set_attribute("gen_ai.input", user_content.model_dump(exclude_none=True))
 
         span.add_event(
             "gen_ai.user.message",
@@ -132,15 +131,9 @@ def _set_agent_input_attribute(
 
 def _set_agent_output_attribute(span: Span, llm_response: LlmResponse) -> None:
     content = llm_response.content
-    print(f"### response:  {llm_response}")
-    print(f"### output_transcription:  {llm_response.output_transcription}")
-    print(f"### input_transcription:  {llm_response.input_transcription}")
-
-
     if content and content.parts:
         # set gen_ai.output attribute required by APMPlus
-        span.set_attribute("gen_ai.output", safe_json_serialize(content.parts))
-        print("### parts: ", content.parts)
+        span.set_attribute("gen_ai.output", content.model_dump(exclude_none=True))
         for idx, part in enumerate(content.parts):
             if part.text:
                 span.add_event(
@@ -163,7 +156,7 @@ def set_common_attributes_on_model_span(
         invocation_span: Span = get_value("invocation_span_instance")  # type: ignore
         agent_run_span: Span = get_value("agent_run_span_instance")  # type: ignore
 
-        if invocation_span and (invocation_span.name.startswith("invocation") or invocation_span.name.startswith("invoke")):
+        if invocation_span and invocation_span.name.startswith("invocation"):
             _set_agent_input_attribute(invocation_span, invocation_context)
             _set_agent_output_attribute(invocation_span, llm_response)
             for attr_name, attr_extractor in common_attributes.items():
@@ -190,7 +183,7 @@ def set_common_attributes_on_model_span(
                 "gen_ai.usage.total_tokens", accumulated_total_token_usage
             )
 
-        if agent_run_span and agent_run_span.name.startswith("agent_run"):
+        if agent_run_span and ( agent_run_span.name.startswith("agent_run") or agent_run_span.name.startswith("invoke_agent") ):
             _set_agent_input_attribute(agent_run_span, invocation_context)
             _set_agent_output_attribute(agent_run_span, llm_response)
             for attr_name, attr_extractor in common_attributes.items():