add attributes

Author: yaozheng-fang

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

@@ -62,8 +62,10 @@ def llm_openinference_instrumentation_veadk(**kwargs) -> str:
     "gen_ai.user.id": common_gen_ai_user_id,  # APMPlus required
     "gen_ai.session.id": common_gen_ai_session_id,  # APMPlus required
     "agent_name": common_gen_ai_agent_name,  # CozeLoop required
+    "agent.name": common_gen_ai_agent_name,  # TLS 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
+    "app.name": common_gen_ai_app_name,  # TLS required
+    "user.id": common_gen_ai_user_id,  # CozeLoop / TLS required
+    "session.id": common_gen_ai_session_id,  # CozeLoop / TLS required
     "cozeloop.report.source": common_cozeloop_report_source,  # CozeLoop required
 }

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

@@ -360,25 +360,33 @@ def llm_gen_ai_choice(params: LLMAttributesParams) -> ExtractorResponse:
 
 
 LLM_ATTRIBUTES = {
+    # ===== request attributes =====
     "gen_ai.request.model": llm_gen_ai_request_model,
     "gen_ai.request.type": llm_gen_ai_request_type,
-    "gen_ai.response.model": llm_gen_ai_response_model,
     "gen_ai.request.max_tokens": llm_gen_ai_request_max_tokens,
     "gen_ai.request.temperature": llm_gen_ai_request_temperature,
     "gen_ai.request.top_p": llm_gen_ai_request_top_p,
-    "gen_ai.prompt": llm_gen_ai_prompt,
-    "gen_ai.completion": llm_gen_ai_completion,
+    # ===== response attributes =====
+    "gen_ai.response.model": llm_gen_ai_response_model,
     "gen_ai.response.stop_reason": llm_gen_ai_response_stop_reason,
     "gen_ai.response.finish_reason": llm_gen_ai_response_finish_reason,
-    "gen_ai.usage.input_tokens": llm_gen_ai_usage_input_tokens,
-    "gen_ai.usage.output_tokens": llm_gen_ai_usage_output_tokens,
-    "gen_ai.usage.total_tokens": llm_gen_ai_usage_total_tokens,
-    "gen_ai.usage.cache_creation_input_tokens": llm_gen_ai_usage_cache_creation_input_tokens,
-    "gen_ai.usage.cache_read_input_tokens": llm_gen_ai_usage_cache_read_input_tokens,
+    # ===== streaming =====
     "gen_ai.is_streaming": llm_gen_ai_is_streaming,
+    # ===== span type =====
     "gen_ai.operation.name": llm_gen_ai_operation_name,
+    # ===== inputs and outputs =====
+    # events
     "gen_ai.system.message": llm_gen_ai_system_message,
     "gen_ai.user.message": llm_gen_ai_user_message,
     "gen_ai.assistant.message": llm_gen_ai_assistant_message,
     "gen_ai.choice": llm_gen_ai_choice,
+    # attributes
+    "gen_ai.prompt": llm_gen_ai_prompt,
+    "gen_ai.completion": llm_gen_ai_completion,
+    # ===== usage =====
+    "gen_ai.usage.input_tokens": llm_gen_ai_usage_input_tokens,
+    "gen_ai.usage.output_tokens": llm_gen_ai_usage_output_tokens,
+    "gen_ai.usage.total_tokens": llm_gen_ai_usage_total_tokens,
+    "gen_ai.usage.cache_creation_input_tokens": llm_gen_ai_usage_cache_creation_input_tokens,
+    "gen_ai.usage.cache_read_input_tokens": llm_gen_ai_usage_cache_read_input_tokens,
 }

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

@@ -48,6 +48,10 @@ def tool_cozeloop_input(params: ToolAttributesParams) -> ExtractorResponse:
     return ExtractorResponse(content=json.dumps(tool_input) or "<unknown_tool_input>")
 
 
+def tool_gen_ai_tool_name(params: ToolAttributesParams) -> ExtractorResponse:
+    return ExtractorResponse(content=params.tool.name or "<unknown_tool_name>")
+
+
 def tool_cozeloop_output(params: ToolAttributesParams) -> ExtractorResponse:
     function_response = params.function_response_event.get_function_responses()[0]
     tool_output = {
@@ -60,6 +64,7 @@ def tool_cozeloop_output(params: ToolAttributesParams) -> ExtractorResponse:
 
 TOOL_ATTRIBUTES = {
     "gen_ai.operation.name": tool_gen_ai_operation_name,
+    "gen_ai.tool.name": tool_gen_ai_tool_name,  # TLS required
     "cozeloop.input": tool_cozeloop_input,  # CozeLoop required
     "cozeloop.output": tool_cozeloop_output,  # CozeLoop required
 }