Spaces:
Runtime error
Runtime error
To count tokens you can use Swarms events and the `TokenCounter` util: | |
```python | |
from swarms import utils | |
from swarms.events import ( | |
StartPromptEvent, FinishPromptEvent, | |
) | |
from swarms.structures import Agent | |
token_counter = utils.TokenCounter() | |
agent = Agent( | |
event_listeners={ | |
StartPromptEvent: [ | |
lambda e: token_counter.add_tokens(e.token_count) | |
], | |
FinishPromptEvent: [ | |
lambda e: token_counter.add_tokens(e.token_count) | |
], | |
} | |
) | |
agent.run("tell me about large language models") | |
agent.run("tell me about GPT") | |
print(f"total tokens: {token_counter.tokens}") | |
``` |