fix: remove do hooks of tracer

Author: zakahan

veadk/agents/loop_agent.py

@@ -19,7 +19,6 @@
 from pydantic import ConfigDict, Field
 from typing_extensions import Any
 
-from veadk.agent import Agent
 from veadk.prompts.agent_default_prompt import DEFAULT_DESCRIPTION, DEFAULT_INSTRUCTION
 from veadk.tracing.base_tracer import BaseTracer
 from veadk.utils.logger import get_logger
@@ -50,21 +49,7 @@ class LoopAgent(GoogleADKLoopAgent):
     tracers: list[BaseTracer] = []
     """The tracers provided to agent."""
 
-    def set_sub_agents_tracer(self, tracer) -> None:
-        from veadk.agents.parallel_agent import ParallelAgent
-        from veadk.agents.sequential_agent import SequentialAgent
-
-        for sub_agent in self.sub_agents:
-            if isinstance(sub_agent, Agent):
-                tracer.do_hooks(sub_agent)
-            elif isinstance(sub_agent, (SequentialAgent, LoopAgent, ParallelAgent)):
-                sub_agent.set_sub_agents_tracer(tracer)
-
     def model_post_init(self, __context: Any) -> None:
         super().model_post_init(None)  # for sub_agents init
 
-        if self.tracers:
-            for tracer in self.tracers:
-                self.set_sub_agents_tracer(tracer)
-
         logger.info(f"{self.__class__.__name__} `{self.name}` init done.")

veadk/agents/parallel_agent.py

@@ -19,7 +19,6 @@
 from pydantic import ConfigDict, Field
 from typing_extensions import Any
 
-from veadk.agent import Agent
 from veadk.prompts.agent_default_prompt import DEFAULT_DESCRIPTION, DEFAULT_INSTRUCTION
 from veadk.tracing.base_tracer import BaseTracer
 from veadk.utils.logger import get_logger
@@ -50,24 +49,12 @@ class ParallelAgent(GoogleADKParallelAgent):
     tracers: list[BaseTracer] = []
     """The tracers provided to agent."""
 
-    def set_sub_agents_tracer(self, tracer) -> None:
-        from veadk.agents.loop_agent import LoopAgent
-        from veadk.agents.sequential_agent import SequentialAgent
-
-        for sub_agent in self.sub_agents:
-            if isinstance(sub_agent, Agent):
-                tracer.do_hooks(sub_agent)
-            elif isinstance(sub_agent, (SequentialAgent, LoopAgent, ParallelAgent)):
-                sub_agent.set_sub_agents_tracer(tracer)
-
     def model_post_init(self, __context: Any) -> None:
         super().model_post_init(None)  # for sub_agents init
 
         if self.tracers:
             logger.warning(
                 "Enable tracing in ParallelAgent may cause OpenTelemetry context error. Issue see https://github.com/google/adk-python/issues/1670"
             )
-            for tracer in self.tracers:
-                self.set_sub_agents_tracer(tracer)
 
         logger.info(f"{self.__class__.__name__} `{self.name}` init done.")

veadk/agents/sequential_agent.py

@@ -19,7 +19,6 @@
 from pydantic import ConfigDict, Field
 from typing_extensions import Any
 
-from veadk.agent import Agent
 from veadk.prompts.agent_default_prompt import DEFAULT_DESCRIPTION, DEFAULT_INSTRUCTION
 from veadk.tracing.base_tracer import BaseTracer
 from veadk.utils.logger import get_logger
@@ -50,21 +49,7 @@ class SequentialAgent(GoogleADKSequentialAgent):
     tracers: list[BaseTracer] = []
     """The tracers provided to agent."""
 
-    def set_sub_agents_tracer(self, tracer) -> None:
-        from veadk.agents.loop_agent import LoopAgent
-        from veadk.agents.parallel_agent import ParallelAgent
-
-        for sub_agent in self.sub_agents:
-            if isinstance(sub_agent, Agent):
-                tracer.do_hooks(sub_agent)
-            elif isinstance(sub_agent, (SequentialAgent, LoopAgent, ParallelAgent)):
-                sub_agent.set_sub_agents_tracer(tracer)
-
     def model_post_init(self, __context: Any) -> None:
         super().model_post_init(None)  # for sub_agents init
 
-        if self.tracers:
-            for tracer in self.tracers:
-                self.set_sub_agents_tracer(tracer)
-
         logger.info(f"{self.__class__.__name__} `{self.name}` init done.")

veadk/integrations/ve_faas/template/{{cookiecutter.local_dir_name}}/src/app.py

@@ -74,7 +74,6 @@ def load_tracer() -> None:
         name="veadk_tracer", app_name=agent_run_config.app_name, exporters=exporters
     )
     agent_run_config.agent.tracers.extend([tracer])
-    tracer.do_hooks(agent=agent_run_config.agent)
 
 
 def build_mcp_run_agent_func() -> Callable: