andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 16:48:30 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix: correct default model (gpt-4o), correct token counts, and correct TaskOutput attributes (added agent) (#749)

Browse Source 
* fix: 'from datetime import datetime for logging' to print the timestamp

* fix: correct default model (gpt-4o), correct token counts, and correct TaskOutput attributes (added agent)

* test: verify Task callback data is an instance of TaskOutput
This commit is contained in:
Matt Thompson
2024-06-11 20:29:22 +02:00
committed by GitHub
parent e202592715
commit 4f093e2626
4 changed files with 59 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/crew_test.py
View File

@@ -445,10 +445,14 @@ def test_async_task_execution():
start.return_value = thread
with patch.object(threading.Thread, "join", wraps=thread.join()) as join:
list_ideas.output = TaskOutput(
description="A 4 paragraph article about AI.", raw_output="ok"
description="A 4 paragraph article about AI.",
raw_output="ok",
agent="writer",
)
list_important_history.output = TaskOutput(
description="A 4 paragraph article about AI.", raw_output="ok"
description="A 4 paragraph article about AI.",
raw_output="ok",
agent="writer",
)
crew.kickoff()
start.assert_called()
@@ -677,9 +681,10 @@ def test_agent_usage_metrics_are_captured_for_hierarchical_process():
result = crew.kickoff()
assert result == '"Howdy!"'
print(crew.usage_metrics)
assert crew.usage_metrics == {
"total_tokens": 1666,
"prompt_tokens": 1383,
"total_tokens": 1664,
"prompt_tokens": 1381,
"completion_tokens": 283,
"successful_requests": 3,
}