diff --git a/tests/crew_test.py b/tests/crew_test.py index 1549765d4..12911815d 100644 --- a/tests/crew_test.py +++ b/tests/crew_test.py @@ -692,8 +692,8 @@ def test_agent_usage_metrics_are_captured_for_hierarchical_process(): assert result == '"Howdy!"' assert crew.usage_metrics == { - "total_tokens": 507, - "prompt_tokens": 224, + "total_tokens": 1671, + "prompt_tokens": 1388, "completion_tokens": 283, "successful_requests": 3, } diff --git a/tests/task_test.py b/tests/task_test.py index 90cc36e82..ca3d51483 100644 --- a/tests/task_test.py +++ b/tests/task_test.py @@ -419,7 +419,7 @@ def test_increment_delegations_for_hierarchical_process(): with patch.object(Task, "increment_delegations") as increment_delegations: increment_delegations.return_value = None crew.kickoff() - increment_delegations.assert_called_once + increment_delegations.assert_called_once() @pytest.mark.vcr(filter_headers=["authorization"])