Function build_test_agent

tests/unittests/telemetry/functional_test_helpers.py:363–387

Builds the canonical 1-tool, 2-LLM-turn agent.

(*, failing: bool = False)

def build_test_agent(*, failing: bool = False) -> Agent:
  """Builds the canonical 1-tool, 2-LLM-turn agent."""
  mock_model = MockModel.create(
      responses=[
          _make_llm_response(
              Part.from_function_call(name=TOOL_NAME, args=TOOL_ARGS)
          ),
          _make_llm_response(Part.from_text(text=FINAL_TEXT)),
      ]
  )

  def some_tool(arg1: str) -> str:
    """A sample tool."""
    if failing:
      raise ValueError("This tool always fails")

    return f"{TOOL_RESULT_PREFIX}{arg1}"

  return Agent(
      name=AGENT_NAME,
      description=AGENT_DESCRIPTION,
      instruction=BASE_INSTRUCTION,
      model=mock_model,
      tools=[FunctionTool(some_tool)],
  )


def build_test_runner(*, failing: bool = False) -> TestInMemoryRunner: