fix: tests

This commit is contained in:
Eduardo Chiarotti
2024-06-27 09:14:27 -07:00
parent da9cc5f097
commit a5a235de62
2 changed files with 3 additions and 3 deletions

View File

@@ -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,
}

View File

@@ -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"])