Tokens Module¶
Token counting and cost estimation.
Classes¶
| Class | Description |
|---|---|
| TokenCounter | Count tokens in text |
| UsageTracker | Track usage across requests |
| CostEstimator | Estimate costs |
Quick Example¶
from mamba_agents.tokens import TokenCounter, UsageTracker, CostEstimator
# Count tokens
counter = TokenCounter()
count = counter.count("Hello, world!")
# Track usage
tracker = UsageTracker()
tracker.record_usage(prompt_tokens=100, completion_tokens=50)
usage = tracker.get_total_usage()
# Estimate cost
estimator = CostEstimator()
cost = estimator.estimate(usage, model="gpt-4o")