From a5a235de62e2bd0352368ba97f8a8b258138e5b6 Mon Sep 17 00:00:00 2001 From: Eduardo Chiarotti Date: Thu, 27 Jun 2024 09:14:27 -0700 Subject: [PATCH] fix: tests --- tests/crew_test.py | 4 ++-- tests/task_test.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) 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"])