mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-15 02:58:30 +00:00
28 lines
911 B
Python
28 lines
911 B
Python
from typing import Any, Dict
|
|
|
|
|
|
class TokenProcess:
|
|
total_tokens: int = 0
|
|
prompt_tokens: int = 0
|
|
completion_tokens: int = 0
|
|
successful_requests: int = 0
|
|
|
|
def sum_prompt_tokens(self, tokens: int):
|
|
self.prompt_tokens = self.prompt_tokens + tokens
|
|
self.total_tokens = self.total_tokens + tokens
|
|
|
|
def sum_completion_tokens(self, tokens: int):
|
|
self.completion_tokens = self.completion_tokens + tokens
|
|
self.total_tokens = self.total_tokens + tokens
|
|
|
|
def sum_successful_requests(self, requests: int):
|
|
self.successful_requests = self.successful_requests + requests
|
|
|
|
def get_summary(self) -> Dict[str, Any]:
|
|
return {
|
|
"total_tokens": self.total_tokens,
|
|
"prompt_tokens": self.prompt_tokens,
|
|
"completion_tokens": self.completion_tokens,
|
|
"successful_requests": self.successful_requests,
|
|
}
|