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, }