Skip to content

Token counter

TokenCounter

Source code in griptape/griptape/utils/token_counter.py 
 4
 5
 6
 7
 8
 9
10
11
@define
class TokenCounter:
    tokens: int = field(default=0, kw_only=True)

    def add_tokens(self, new_tokens: int) -> int:
        self.tokens += 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/griptape/utils/token_counter.py 
 8
 9
10
11
def add_tokens(self, new_tokens: int) -> int:
    self.tokens += new_tokens

    return self.tokens