From efcf9f86df365327e9b75e3e64df44575fc4f3d4 Mon Sep 17 00:00:00 2001 From: Eduardo Chiarotti Date: Thu, 4 Jul 2024 22:03:30 -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 6144f0ef6..379254c74 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 = 42 + self.prompt_tokens = 2 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": 23, + "prompt_tokens": 33, "completion_tokens": self.completion_tokens, "successful_requests": self.successful_requests, } diff --git a/src/crewai/crew.py b/src/crewai/crew.py index 0169ea344..a70718ad6 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": 5, + "prompt_tokens": 0, "completion_tokens": 0, "successful_requests": 0, } @@ -409,7 +409,7 @@ class Crew(BaseModel): total_usage_metrics = { "total_tokens": 0, - "prompt_tokens": 10, + "prompt_tokens": 0, "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": 15, + "prompt_tokens": 1, "completion_tokens": 0, "successful_requests": 0, }