andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 00:28:31 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix: tests

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/crew_test.py
View File

@@ -692,8 +692,8 @@ def test_agent_usage_metrics_are_captured_for_hierarchical_process():
assert result == '"Howdy!"' assert result == '"Howdy!"'
assert crew.usage_metrics == { assert crew.usage_metrics == {
"total_tokens": 507, "total_tokens": 1671,
"prompt_tokens": 224, "prompt_tokens": 1388,
"completion_tokens": 283, "completion_tokens": 283,
"successful_requests": 3, "successful_requests": 3,
} }

2
tests/task_test.py
View File

@@ -419,7 +419,7 @@ def test_increment_delegations_for_hierarchical_process():
with patch.object(Task, "increment_delegations") as increment_delegations: with patch.object(Task, "increment_delegations") as increment_delegations:
increment_delegations.return_value = None increment_delegations.return_value = None
crew.kickoff() crew.kickoff()
increment_delegations.assert_called_once increment_delegations.assert_called_once()
@pytest.mark.vcr(filter_headers=["authorization"]) @pytest.mark.vcr(filter_headers=["authorization"])