From e4363f9ed8325334c31f23ac3b905b2b53d84f72 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=A3o=20Moura?= <joaomdmoura@gmail.com>
Date: Fri, 5 Apr 2024 08:33:31 -0300
Subject: [PATCH] updating tests

---
 tests/crew_test.py | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/tests/crew_test.py b/tests/crew_test.py
index 60644be51d..d69d49c28e 100644
--- a/tests/crew_test.py
+++ b/tests/crew_test.py
@@ -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": 17,
-        "prompt_tokens": 161,
-        "successful_requests": 1,
-        "total_tokens": 178,
+        "completion_tokens": 51,
+        "prompt_tokens": 483,
+        "successful_requests": 3,
+        "total_tokens": 534,
     }
 
 
@@ -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": 1641,
-        "prompt_tokens": 1358,
-        "completion_tokens": 283,
-        "successful_requests": 3,
+        "total_tokens": 2592,
+        "prompt_tokens": 2048,
+        "completion_tokens": 544,
+        "successful_requests": 6,
     }