fix attr

Author: yaozheng-fang

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

@@ -49,10 +49,15 @@ def common_cozeloop_report_source(**kwargs) -> str:
     return "veadk"
 
 
+def llm_openinference_instrumentation_veadk(**kwargs) -> str:
+    return VERSION
+
+
 COMMON_ATTRIBUTES = {
     "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,
+    "openinference.instrumentation.veadk": llm_openinference_instrumentation_veadk,
     "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

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

@@ -13,7 +13,6 @@
 # limitations under the License.
 
 import json
-from xml.etree.ElementTree import VERSION
 
 from veadk.tracing.telemetry.attributes.extractors.types import (
     ExtractorResponse,
@@ -360,12 +359,6 @@ def llm_gen_ai_choice(params: LLMAttributesParams) -> ExtractorResponse:
     return ExtractorResponse(type="event", content=message)
 
 
-def llm_openinference_instrumentation_veadk(
-    params: LLMAttributesParams,
-) -> ExtractorResponse:
-    return ExtractorResponse(content=VERSION)
-
-
 LLM_ATTRIBUTES = {
     "gen_ai.request.model": llm_gen_ai_request_model,
     "gen_ai.request.type": llm_gen_ai_request_type,
@@ -388,5 +381,4 @@ def llm_openinference_instrumentation_veadk(
     "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,
-    "openinference.instrumentation.veadk": llm_openinference_instrumentation_veadk,
 }