andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-03 00:02:36 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix: handle None values in token counter

Browse Source 
- Update sum_cached_prompt_tokens to handle None values gracefully
- Add unit tests for token counting with None values
- Fixes #2197

Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-22 01:29:47 +00:00
parent b50772a38b
commit 3f02d10626
2 changed files with 20 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/crewai/agents/agent_builder/utilities/base_token_process.py
View File

@@ -17,8 +17,9 @@ class TokenProcess:
self.completion_tokens += tokens
self.total_tokens += tokens
def sum_cached_prompt_tokens(self, tokens: int) -> None:
self.cached_prompt_tokens += tokens
def sum_cached_prompt_tokens(self, tokens: int | None) -> None:
if tokens is not None:
self.cached_prompt_tokens += tokens
def sum_successful_requests(self, requests: int) -> None:
self.successful_requests += requests