Update agent tests to reflect changes in expected call counts and improve response formatting in YAML cassette. Adjusted mock call count from 2 to 3 and refined interaction formats for clarity and consistency.

2026-01-20 21:38:14 +00:00 · 2025-03-31 15:34:13 -07:00
parent 99614f83e0
commit 8f5d6f720d
2 changed files with 261 additions and 92 deletions
--- a/tests/agent_test.py
+++ b/tests/agent_test.py
@@ -443,7 +443,7 @@ def test_agent_custom_max_iterations():
            task=task,
            tools=[get_final_answer],
        )
-        assert private_mock.call_count == 2
+        assert private_mock.call_count == 3


@pytest.mark.vcr(filter_headers=["authorization"])