mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-05-01 15:22:37 +00:00
refactor: improve token counter implementation
- Fix import sorting in tests - Add docstrings and type validation - Add comprehensive test cases - Add validation for negative token counts Addresses review feedback on #2198 Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
@@ -18,7 +18,18 @@ class TokenProcess:
|
||||
self.total_tokens += tokens
|
||||
|
||||
def sum_cached_prompt_tokens(self, tokens: int | None) -> None:
|
||||
"""
|
||||
Adds the given token count to cached prompt tokens.
|
||||
|
||||
Args:
|
||||
tokens (int | None): Number of tokens to add. None values are ignored.
|
||||
|
||||
Raises:
|
||||
ValueError: If tokens is negative.
|
||||
"""
|
||||
if tokens is not None:
|
||||
if tokens < 0:
|
||||
raise ValueError("Token count cannot be negative")
|
||||
self.cached_prompt_tokens += tokens
|
||||
|
||||
def sum_successful_requests(self, requests: int) -> None:
|
||||
|
||||
Reference in New Issue
Block a user