From 0324a41819210653ae562420c69b48e3a0fd1497 Mon Sep 17 00:00:00 2001 From: Eduardo Chiarotti Date: Thu, 4 Jul 2024 22:05:42 -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 | 2 +- 2 files changed, 3 insertions(+), 3 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 379254c74..9daf70d83 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 = 2 + self.prompt_tokens = 0 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": 33, + "prompt_tokens": 25, "completion_tokens": self.completion_tokens, "successful_requests": self.successful_requests, } diff --git a/src/crewai/crew.py b/src/crewai/crew.py index a70718ad6..a297bdff2 100644 --- a/src/crewai/crew.py +++ b/src/crewai/crew.py @@ -597,7 +597,7 @@ class Crew(BaseModel): """Calculates and returns the usage metrics.""" total_usage_metrics = { "total_tokens": 0, - "prompt_tokens": 1, + "prompt_tokens": 5, "completion_tokens": 0, "successful_requests": 0, }