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: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], agents=[agent],
tasks=[task], tasks=[task],
process=Process.hierarchical, process=Process.hierarchical,
manager_llm=ChatOpenAI(temperature=0, model="gpt-4"), manager_llm=ChatOpenAI(temperature=0, model="gpt-4o"),
) )
result = crew.kickoff() result = crew.kickoff()
assert result == "Howdy!" assert result == "Howdy!"
assert crew.usage_metrics == { assert crew.usage_metrics == {
"total_tokens": 1476, "total_tokens": 1383,
"prompt_tokens": 1322, "prompt_tokens": 1278,
"completion_tokens": 154, "completion_tokens": 105,
"successful_requests": 3, "successful_requests": 3,
} }