Function build_node

src/google/adk/workflow/utils/_workflow_graph_utils.py:40–136

Converts a NodeLike to a BaseNode, wrapping async funcs in FunctionNode. Args: node_like: The item to convert to a BaseNode. name: If provided, overrides the name of the wrapped node. rerun_on_resume: If provided, overrides the rerun_on_resume property of the wrapped node. r

(
    node_like: NodeLike,
    *,
    name: str | None = None,
    rerun_on_resume: bool | None = None,
    retry_config: RetryConfig | None = None,
    timeout: float | None = None,
    auth_config: Any = None,
)

def build_node(
    node_like: NodeLike,
    *,
    name: str | None = None,
    rerun_on_resume: bool | None = None,
    retry_config: RetryConfig | None = None,
    timeout: float | None = None,
    auth_config: Any = None,
) -> BaseNode:
  """Converts a NodeLike to a BaseNode, wrapping async funcs in FunctionNode.

  Args:
    node_like: The item to convert to a BaseNode.
    name: If provided, overrides the name of the wrapped node.
    rerun_on_resume: If provided, overrides the rerun_on_resume property of the
      wrapped node.
    retry_config: If provided, overrides the retry_config property of the
      wrapped node.
    timeout: If provided, overrides the timeout property of the wrapped node.
    auth_config: If provided, passed to FunctionNode for authentication.

  Returns:
    A BaseNode instance.

  Raises:
    ValueError: If node_like is not a valid type (BaseNode, BaseAgent,
      BaseTool, callable, or 'START').
  """

  if node_like == 'START':
    return START

  # Lazy import to avoid circular dependency:
  # workflow_graph_utils -> agents.llm_agent -> ... -> workflow_graph_utils
  from ...agents.llm_agent import LlmAgent

  if isinstance(node_like, BaseNode):
    kwargs: dict[str, Any] = {}
    if name is not None:
      kwargs['name'] = name
    if rerun_on_resume is not None:
      kwargs['rerun_on_resume'] = rerun_on_resume
    if retry_config is not None:
      kwargs['retry_config'] = retry_config
    if timeout is not None:
      kwargs['timeout'] = timeout

    if isinstance(node_like, LlmAgent):
      if rerun_on_resume is None:
        kwargs['rerun_on_resume'] = True
      agent = node_like.clone(update=kwargs)
      # Preserve parent agent reference that was lost during clone
      agent.parent_agent = node_like.parent_agent

      if agent.mode is None:
        # Sub-agents dynamically attached to a parent agent default to 'chat'
        # mode to enable agent transfer.
        # Standalone agents in a workflow graph default to 'single_turn'.