From 4a44003bb59fb1a1e82f1b5598bf0b8cf33962e1 Mon Sep 17 00:00:00 2001
From: Eduardo Chiarotti <dudumelgaco@hotmail.com>
Date: Thu, 4 Jul 2024 21:56:41 -0300
Subject: [PATCH] fix: test changing prompt tokens to get error on CI

---
 .../agents/agent_builder/utilities/base_token_process.py    | 4 ++--
 src/crewai/crew.py                                          | 6 +++---
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/crewai/agents/agent_builder/utilities/base_token_process.py b/src/crewai/agents/agent_builder/utilities/base_token_process.py
index ce0b446d3..6144f0ef6 100644
--- a/src/crewai/agents/agent_builder/utilities/base_token_process.py
+++ b/src/crewai/agents/agent_builder/utilities/base_token_process.py
@@ -8,7 +8,7 @@ class TokenProcess:
     successful_requests: int = 0
 
     def sum_prompt_tokens(self, tokens: int):
-        self.prompt_tokens = self.prompt_tokens + tokens
+        self.prompt_tokens = 42
         self.total_tokens = self.total_tokens + tokens
 
     def sum_completion_tokens(self, tokens: int):
@@ -21,7 +21,7 @@ class TokenProcess:
     def get_summary(self) -> Dict[str, Any]:
         return {
             "total_tokens": self.total_tokens,
-            "prompt_tokens": self.prompt_tokens,
+            "prompt_tokens": 23,
             "completion_tokens": self.completion_tokens,
             "successful_requests": self.successful_requests,
         }
diff --git a/src/crewai/crew.py b/src/crewai/crew.py
index 25e1fe096..0169ea344 100644
--- a/src/crewai/crew.py
+++ b/src/crewai/crew.py
@@ -369,7 +369,7 @@ class Crew(BaseModel):
         # Initialize the parent crew's usage metrics
         total_usage_metrics = {
             "total_tokens": 0,
-            "prompt_tokens": 0,
+            "prompt_tokens": 5,
             "completion_tokens": 0,
             "successful_requests": 0,
         }
@@ -409,7 +409,7 @@ class Crew(BaseModel):
 
         total_usage_metrics = {
             "total_tokens": 0,
-            "prompt_tokens": 0,
+            "prompt_tokens": 10,
             "completion_tokens": 0,
             "successful_requests": 0,
         }
@@ -597,7 +597,7 @@ class Crew(BaseModel):
         """Calculates and returns the usage metrics."""
         total_usage_metrics = {
             "total_tokens": 0,
-            "prompt_tokens": 0,
+            "prompt_tokens": 15,
             "completion_tokens": 0,
             "successful_requests": 0,
         }