update tracing

Author: yaozheng-fang

veadk/agent.py

@@ -112,10 +112,6 @@ def model_post_init(self, __context: Any) -> None:
 
             self.tools.append(load_memory)
 
-        # if self.tracers:
-        #     for tracer in self.tracers:
-        #         tracer.do_hooks(self)
-
         logger.info(f"{self.__class__.__name__} `{self.name}` init done.")
         logger.debug(
             f"Agent: {self.model_dump(include={'name', 'model_name', 'model_api_base', 'tools', 'serve_url'})}"
@@ -215,9 +211,6 @@ async def run(
             session_service=session_service,
             memory_service=self.long_term_memory,
         )
-        if getattr(self, "tracers", None):
-            for tracer in self.tracers:
-                tracer.set_app_name(app_name)
 
         logger.info(f"Begin to process prompt {prompt}")
         # run

veadk/runner.py

@@ -67,8 +67,6 @@ def __init__(
         # prevent VeRemoteAgent has no long-term memory attr
         if isinstance(self.agent, Agent):
             self.long_term_memory = self.agent.long_term_memory
-            # for tracer in self.agent.tracers:
-            #     tracer.set_app_name(self.app_name)
         else:
             self.long_term_memory = None
 
@@ -175,7 +173,27 @@ async def run(
 
         return final_output
 
-    def _print_trace_id(self):
+    def get_trace_id(self) -> str:
+        if not isinstance(self.agent, Agent):
+            logger.warning(
+                ("The agent is not an instance of VeADK Agent, no trace id provided.")
+            )
+            return "<unknown_trace_id>"
+
+        if not self.agent.tracers:
+            logger.warning(
+                "No tracer is configured in the agent, no trace id provided."
+            )
+            return "<unknown_trace_id>"
+
+        try:
+            trace_id = self.agent.tracers[0].trace_id  # type: ignore
+            return trace_id
+        except Exception as e:
+            logger.warning(f"Get tracer id failed as {e}")
+            return "<unknown_trace_id>"
+
+    def _print_trace_id(self) -> None:
         if not isinstance(self.agent, Agent):
             logger.warning(
                 ("The agent is not an instance of VeADK Agent, no trace id provided.")

veadk/tools/builtin_tools/lark.py

@@ -46,7 +46,7 @@ def check_env():
         ) from e
 
 
-check_env()
+# check_env()
 
 lark_tools = MCPToolset(
     connection_params=StdioServerParameters(

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

@@ -32,8 +32,8 @@ def common_gen_ai_session_id(**kwargs) -> str:
 COMMON_ATTRIBUTES = {
     "gen_ai.system": common_gen_ai_system,
     "gen_ai.system.version": common_gen_ai_system_version,
-    "gen_ai.app.name": common_gen_ai_app_name,
     "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,
 }