Skip to content

token_counter

TokenCounter

Source code in griptape/utils/token_counter.py
@define
class TokenCounter:
    tokens: int = field(default=0, kw_only=True)

    def add_tokens(self, new_tokens: int | float) -> int:
        self.tokens += int(new_tokens)

        return self.tokens

tokens: int = field(default=0, kw_only=True) class-attribute instance-attribute

add_tokens(new_tokens)

Source code in griptape/utils/token_counter.py
def add_tokens(self, new_tokens: int | float) -> int:
    self.tokens += int(new_tokens)

    return self.tokens