update attributes

Author: yaozheng-fang

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

@@ -2,7 +2,9 @@
 
 
 def common_gen_ai_system(**kwargs) -> str:
-    return "veadk"
+    """This field will be parsed as `model_provider` in Volcengine CozeLoop platform."""
+    model_provider = kwargs.get("model_provider")
+    return model_provider or "<unknown_model_provider>"
 
 
 def common_gen_ai_system_version(**kwargs) -> str:
@@ -33,7 +35,10 @@ def common_gen_ai_session_id(**kwargs) -> str:
     "gen_ai.system": common_gen_ai_system,
     "gen_ai.system.version": common_gen_ai_system_version,
     "gen_ai.agent.name": common_gen_ai_agent_name,
-    "gen_ai.app.name": common_gen_ai_app_name,
-    "gen_ai.user.id": common_gen_ai_user_id,
-    "gen_ai.session.id": common_gen_ai_session_id,
+    "gen_ai.app.name": common_gen_ai_app_name,  # APMPlus required
+    "gen_ai.user.id": common_gen_ai_user_id,  # APMPlus required
+    "gen_ai.session.id": common_gen_ai_session_id,  # APMPlus required
+    "app.name": common_gen_ai_app_name,  # CozeLoop required
+    "user.id": common_gen_ai_user_id,  # CozeLoop required
+    "session.id": common_gen_ai_session_id,  # CozeLoop required
 }

veadk/tracing/telemetry/telemetry.py

@@ -37,6 +37,10 @@ def set_common_attributes(current_span: _Span, **kwargs) -> None:
 
     common_attributes = ATTRIBUTES.get("common", {})
     for span in spans_in_current_trace:
+        if span.name.startswith("invocation"):
+            span.set_attribute("gen_ai.operation.name", "chain")
+        elif span.name.startswith("agent_run"):
+            span.set_attribute("gen_ai.operation.name", "agent")
         for attr_name, attr_extractor in common_attributes.items():
             value = attr_extractor(**kwargs)
             span.set_attribute(attr_name, value)
@@ -65,12 +69,20 @@ def trace_call_llm(
 ) -> None:
     span = trace.get_current_span()
 
+    from veadk.agent import Agent
+
     set_common_attributes(
         current_span=span,  # type: ignore
         agent_name=invocation_context.agent.name,
-        app_name=invocation_context.app_name,
         user_id=invocation_context.user_id,
+        app_name=invocation_context.app_name,
         session_id=invocation_context.session.id,
+        model_provider=invocation_context.agent.model_provider
+        if isinstance(invocation_context.agent, Agent)
+        else "",
+        model_name=invocation_context.agent.model_name
+        if isinstance(invocation_context.agent, Agent)
+        else "",
     )
 
     llm_attributes_mapping = ATTRIBUTES.get("llm", {})