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

fix: tests

Browse Source
This commit is contained in:
Eduardo Chiarotti
2024-06-27 09:46:19 -07:00
parent a5a235de62
commit ca3de82b09
4 changed files with 1593 additions and 205550 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1598
tests/cassettes/test_agent_usage_metrics_are_captured_for_hierarchical_process.yaml
View File

File diff suppressed because it is too large Load Diff

205535
tests/cassettes/test_increment_delegations_for_hierarchical_process.yaml
View File

File diff suppressed because it is too large Load Diff

8
tests/crew_test.py
View File

@@ -689,12 +689,12 @@ 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": 1671, "total_tokens": 1476,
"prompt_tokens": 1388, "prompt_tokens": 1322,
"completion_tokens": 283, "completion_tokens": 154,
"successful_requests": 3, "successful_requests": 3,
} }

2
tests/task_test.py
View File

@@ -413,7 +413,7 @@ def test_increment_delegations_for_hierarchical_process():
agents=[scorer], agents=[scorer],
tasks=[task], tasks=[task],
process=Process.hierarchical, process=Process.hierarchical,
manager_llm=ChatOpenAI(model="gpt-4-0125-preview"), manager_llm=ChatOpenAI(model="gpt-4o"),
) )
with patch.object(Task, "increment_delegations") as increment_delegations: with patch.object(Task, "increment_delegations") as increment_delegations: