Function rebuild_tests

src/google/adk/cli/agent_test_runner.py:727–954

Discovers test files and rebuilds them by running the agent live.

(path: str)

def rebuild_tests(path: str):
  """Discovers test files and rebuilds them by running the agent live."""
  import json
  import sys

  from google.adk.apps.app import App
  from google.adk.events.event import Event as AdkEvent

  path_obj = Path(path)
  if path_obj.is_dir():
    folder = path
    expected_name = None
  else:
    folder = str(path_obj.parent.parent)
    expected_name = path_obj.name

  test_files = get_test_files(folder)
  if not test_files:
    print(f"No test files found in {folder}")
    return

  for item in test_files:
    agent_dir, test_file = item.values
    if expected_name and test_file.name != expected_name:
      continue
    print(f"Rebuilding {test_file}...")

    # Add agent_dir.parent to sys.path so relative imports work
    sys_path_saved = list(sys.path)
    sys.path.insert(0, str(agent_dir.parent))

    try:
      import random

      loader = AgentLoader(str(agent_dir.parent))
      loader.remove_agent_from_cache(agent_dir.name)
      agent_or_app = loader.load_agent(agent_dir.name)

      root_agent = (
          agent_or_app.root_agent
          if isinstance(agent_or_app, App)
          else agent_or_app
      )
      _make_nodes_sequential(root_agent)

      with open(test_file, "r") as f:
        session_data = json.load(f)

      events_data = session_data.get("events", [])
      if not events_data:
        print(f"No events in {test_file}, skipping.")
        continue

      # Extract user messages
      user_messages = []
      for event in events_data:
        content = _extract_user_content(event)
        if content: