Function _get_output_schema

src/google/adk/tools/agent_tool.py:82–104

Extracts the output_schema from an agent. For LlmAgent, returns its output_schema directly. For agents with sub_agents, recursively searches the last sub-agent for an output_schema. Args: agent: The agent to extract output_schema from. Returns: The output_schema if found, None o

(agent: BaseAgent)

def _get_output_schema(agent: BaseAgent) -> Optional[SchemaType]:
  """Extracts the output_schema from an agent.

  For LlmAgent, returns its output_schema directly.
  For agents with sub_agents, recursively searches the last sub-agent for an
  output_schema.

  Args:
    agent: The agent to extract output_schema from.

  Returns:
    The output_schema if found, None otherwise.
  """
  from ..agents.llm_agent import LlmAgent

  if isinstance(agent, LlmAgent):
    return agent.output_schema

  # For composite agents, check the last sub-agent
  if agent.sub_agents:
    return _get_output_schema(agent.sub_agents[-1])

  return None


class AgentTool(BaseTool):