fix: tests

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

View File

@@ -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,
}

View File

@@ -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: