From fedb4113294b09a1da4399dbb1b902b24979bc9b Mon Sep 17 00:00:00 2001 From: Shashank Ram Date: Tue, 31 Mar 2026 09:32:27 -0700 Subject: [PATCH] agentsts: add debug logs to MCP tool handlers Adds debug logs to MCP tool handlers to help debug issues. Signed-off-by: Shashank Ram --- python/packages/agentsts-adk/src/agentsts/adk/_base.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/python/packages/agentsts-adk/src/agentsts/adk/_base.py b/python/packages/agentsts-adk/src/agentsts/adk/_base.py index 9c11310ee..11a516535 100644 --- a/python/packages/agentsts-adk/src/agentsts/adk/_base.py +++ b/python/packages/agentsts-adk/src/agentsts/adk/_base.py @@ -122,17 +122,22 @@ def add_to_agent(self, agent: BaseAgent): Add the plugin to an ADK LLM agent by updating its MCP toolset Call this once when setting up the agent; do not call it at runtime. """ + agent_name = getattr(agent, "name", "unknown") + logger.debug(f"add_to_agent called for agent {agent_name}") + if not isinstance(agent, LlmAgent): + logger.debug(f"add_to_agent: agent {agent_name} is not LlmAgent, skipping") return if not agent.tools: + logger.debug(f"add_to_agent: agent {agent_name} has no tools, skipping") return for tool in agent.tools: if isinstance(tool, McpToolset): mcp_toolset = tool mcp_toolset._header_provider = self.header_provider - logger.debug("Updated tool connection params to include access token from STS server") + logger.debug(f"add_to_agent: updated MCP tool's header provider for agent {agent_name}") def header_provider(self, readonly_context: Optional[ReadonlyContext]) -> Dict[str, str]: # access saved token @@ -140,6 +145,7 @@ def header_provider(self, readonly_context: Optional[ReadonlyContext]) -> Dict[s if not cache_entry: return {} + logger.debug("Using cached access token for tool invocation") return { "Authorization": f"Bearer {cache_entry.token}", }