Search⌘ K
AI Features
Home
Courses
Agentic System Design

SetupAgent and Environment Grounding

Explore the SetupAgent's role in initializing shared state, loading configurations, and creating environment contexts to support downstream agents. Understand how to prepare system artifacts and generate baseline rollouts that enable reward learning pipelines to function cohesively and reliably.

We'll cover the following...

In this EUREKA-like pipeline, the SetupAgent is the first step in the execution flow. It initializes shared state that downstream agents depend on, including loaded configuration, an artifacts directory, and the initial environment context. It also generates a baseline rollout that serves as a visual reference before reward evolution begins. In this lesson, we’ll walk through agents/setup_agent.py.

Code walk-through

We begin by importing everything the SetupAgent needs to run.

import yaml
from google.adk.agents import BaseAgent
from google.adk.agents.invocation_context import InvocationContext
from google.adk.events import Event
from typing import AsyncGenerator
from loguru import logger
Imports: Declaring what this agent is allowed to do

Let’s unpack this step by step:

  • We import BaseAgent because every executable unit in an ADK workflow must subclass it. This is what allows ADK to recognize SetupAgent as a runnable agent.

  • InvocationContext is how ADK passes execution state into the agent. This object gives us access to the shared session state that persists across agents.

  • Event signals completion back to ADK. Even when an agent doesn’t emit meaningful output, it still needs to yield an event to satisfy the execution contract.

  • AsyncGenerator appears because ADK agents are asynchronous and stream events as they execute.

  • yaml is used to load configuration files, and logger gives us structured logging during setup.

At this point, we’ve declared:

“This file defines an ADK agent that reads configuration and mutates shared execution state.”

Next, we import the tools that the SetupAgent will rely on.

from ..tools.brax_env import get_env_and_source
from ..tools.artifacts import init_run_dir, init_leaderboard, save_json

These imports are especially important because they show what this agent does not do itself.

  • get_env_and_source(...) encapsulates environment creation and source extraction. The SetupAgent will call this function, but it does not manage Brax internals directly. ...