diff --git a/tests/agent_test.py b/tests/agent_test.py index 6879a4519..da8cf1d0e 100644 --- a/tests/agent_test.py +++ b/tests/agent_test.py @@ -466,6 +466,8 @@ def test_agent_custom_max_iterations(): tools=[get_final_answer], ) assert private_mock.call_count == 2 + # Verify that have_forced_answer was set to True after max iterations + assert agent.agent_executor.have_forced_answer == True, "have_forced_answer should be True after max iterations" @pytest.mark.vcr(filter_headers=["authorization"]) @@ -566,6 +568,8 @@ def test_agent_moved_on_after_max_iterations(): tools=[get_final_answer], ) assert output == "The final answer is 42." + # Verify that have_forced_answer was set to True after max iterations + assert agent.agent_executor.have_forced_answer == True, "have_forced_answer should be True after max iterations" @pytest.mark.vcr(filter_headers=["authorization"]) diff --git a/tests/crew_test.py b/tests/crew_test.py index 2003ddada..7582ded33 100644 --- a/tests/crew_test.py +++ b/tests/crew_test.py @@ -3125,4 +3125,4 @@ def test_multimodal_agent_live_image_analysis(): # Verify we got a meaningful response assert isinstance(result.raw, str) assert len(result.raw) > 100 # Expecting a detailed analysis - assert "error" not in result.raw.lower() # No error messages in response \ No newline at end of file + assert "error" not in result.raw.lower() # No error messages in response