smal fixes and better guardrail for parsing small models tools usage

This commit is contained in:
João Moura
2024-05-02 02:21:42 -03:00
parent 84775373a6
commit 2ae6fc4bd8
3 changed files with 6180 additions and 5907 deletions

File diff suppressed because it is too large Load Diff

View File

@@ -648,10 +648,10 @@ def test_agent_usage_metrics_are_captured_for_sequential_process():
result = crew.kickoff()
assert result == "Howdy!"
assert crew.usage_metrics == {
"completion_tokens": 51,
"prompt_tokens": 483,
"successful_requests": 3,
"total_tokens": 534,
"completion_tokens": 17,
"prompt_tokens": 160,
"successful_requests": 1,
"total_tokens": 177,
}
@@ -678,10 +678,10 @@ def test_agent_usage_metrics_are_captured_for_hierarchical_process():
result = crew.kickoff()
assert result == '"Howdy!"'
assert crew.usage_metrics == {
"total_tokens": 2592,
"prompt_tokens": 2048,
"completion_tokens": 544,
"successful_requests": 6,
"total_tokens": 1650,
"prompt_tokens": 1367,
"completion_tokens": 283,
"successful_requests": 3,
}