From 5cb5879d7ca4848a0454a0786cbe376764e18a4c Mon Sep 17 00:00:00 2001 From: Eduardo Chiarotti Date: Thu, 4 Jul 2024 22:08:22 -0300 Subject: [PATCH] fix: test changing prompt tokens to get error on CI --- .../agents/agent_builder/utilities/base_token_process.py | 4 ++-- 1 file changed, 2 insertions(+), 2 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 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, }