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

updating tests

Browse Source
This commit is contained in:
João Moura
2024-04-05 08:33:31 -03:00
parent 9b37aab61a
commit 4c274dbca7
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/crew_test.py
View File

@@ -648,10 +648,10 @@ def test_agent_usage_metrics_are_captured_for_sequential_process():
result = crew.kickoff() result = crew.kickoff()
assert result == "Howdy!" assert result == "Howdy!"
assert crew.usage_metrics == { assert crew.usage_metrics == {
"completion_tokens": 17, "completion_tokens": 51,
"prompt_tokens": 161, "prompt_tokens": 483,
"successful_requests": 1, "successful_requests": 3,
"total_tokens": 178, "total_tokens": 534,
} }
@@ -678,10 +678,10 @@ def test_agent_usage_metrics_are_captured_for_hierarchical_process():
result = crew.kickoff() result = crew.kickoff()
assert result == '"Howdy!"' assert result == '"Howdy!"'
assert crew.usage_metrics == { assert crew.usage_metrics == {
"total_tokens": 1641, "total_tokens": 2592,
"prompt_tokens": 1358, "prompt_tokens": 2048,
"completion_tokens": 283, "completion_tokens": 544,
"successful_requests": 3, "successful_requests": 6,
} }