diff --git a/libs/prebuilt/tests/test_react_agent.py b/libs/prebuilt/tests/test_react_agent.py index 4f91d4610..8fb0a042e 100644 --- a/libs/prebuilt/tests/test_react_agent.py +++ b/libs/prebuilt/tests/test_react_agent.py @@ -2091,7 +2091,7 @@ def test_post_model_hook_with_structured_output() -> None: }, {"post_model_hook": {"flag": True}}, { - "generate_structured_response": { + "respond": { "structured_response": WeatherResponse(temperature=75.0) } }, @@ -2159,3 +2159,4 @@ def test_create_react_agent_inject_vars_with_post_model_hook( assert result["foo"] == 2 +