added usage_metrics to full output (#756)

* added extra parameter for kickoff to return token usage count after result

* added output_token_usage to class and in full_output

* logger duplicated

* added more types

* added usage_metrics to full output instead

* added more to the description on full_output

* possible mispacing
This commit is contained in:
Lorenze Jay
2024-06-12 10:18:52 -07:00
committed by GitHub
parent 9e2eb2700a
commit 14e7046980
4 changed files with 62 additions and 46 deletions

View File

@@ -384,6 +384,12 @@ def test_crew_full_ouput():
assert result == {
"final_output": "Hello! It is a delight to receive your message. I trust this response finds you in good spirits. It's indeed a pleasure to connect with you too.",
"tasks_outputs": [task1.output, task2.output],
"usage_metrics": {
"completion_tokens": 109,
"prompt_tokens": 330,
"successful_requests": 2,
"total_tokens": 439,
},
}