andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-08 15:48:29 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix: tests

Browse Source
This commit is contained in:
Eduardo Chiarotti
2024-06-27 09:52:28 -07:00
parent ca3de82b09
commit 4b3f9aedb3
2 changed files with 433 additions and 659 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1084
tests/cassettes/test_agent_usage_metrics_are_captured_for_hierarchical_process.yaml
View File

File diff suppressed because it is too large Load Diff

8
tests/crew_test.py
View File

@@ -685,16 +685,16 @@ def test_agent_usage_metrics_are_captured_for_hierarchical_process():
agents=[agent],
tasks=[task],
process=Process.hierarchical,
manager_llm=ChatOpenAI(temperature=0, model="gpt-4"),
manager_llm=ChatOpenAI(temperature=0, model="gpt-4o"),
)
result = crew.kickoff()
assert result == "Howdy!"
assert crew.usage_metrics == {
"total_tokens": 1476,
"prompt_tokens": 1322,
"completion_tokens": 154,
"total_tokens": 1383,
"prompt_tokens": 1278,
"completion_tokens": 105,
"successful_requests": 3,
}