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 9daf70d83..42ac25ae0 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 = 0 + self.prompt_tokens = 10 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": 25, + "prompt_tokens": self.prompt_tokens, "completion_tokens": self.completion_tokens, "successful_requests": self.successful_requests, }