Function test_nl_planner

tests/integration/test_system_instruction.py:146–157

(agent_runner: TestRunner)

    indirect=True,
)
def test_nl_planner(agent_runner: TestRunner):
  session = Session(context={"customerId": "1234567890"})
  si = UnitFlow()._build_system_instruction(
      InvocationContext(
          invocation_id="1234567890",
          agent=agent_runner.agent,
          session=session,
      )
  )

  for line in nl_planner_si.splitlines():
    assert line in si


@pytest.mark.parametrize(