mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-11 09:08:31 +00:00
fix: tests
This commit is contained in:
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
@@ -689,12 +689,12 @@ def test_agent_usage_metrics_are_captured_for_hierarchical_process():
|
||||
)
|
||||
|
||||
result = crew.kickoff()
|
||||
assert result == '"Howdy!"'
|
||||
assert result == "Howdy!"
|
||||
|
||||
assert crew.usage_metrics == {
|
||||
"total_tokens": 1671,
|
||||
"prompt_tokens": 1388,
|
||||
"completion_tokens": 283,
|
||||
"total_tokens": 1476,
|
||||
"prompt_tokens": 1322,
|
||||
"completion_tokens": 154,
|
||||
"successful_requests": 3,
|
||||
}
|
||||
|
||||
|
||||
@@ -413,7 +413,7 @@ def test_increment_delegations_for_hierarchical_process():
|
||||
agents=[scorer],
|
||||
tasks=[task],
|
||||
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:
|
||||
|
||||
Reference in New Issue
Block a user