Token usage management

View how to attribute AI spend to people and projects, forecast it against budget and measure the value it delivers.

How token usage management works in Drivetrain

Start by connecting your AI providers. Then track your AI spend, budget and forecast it, and show the value it delivers.

1
Connect AI providers and set up spend attribution
- Connect providers such as Anthropic and OpenAI once, and Drivetrain pulls in usage data from then on.
- Bring in the API key, owner, model, token counts, timestamps and spend for every request.
- Give each user and project its own API key, so every dollar of spend has an owner.
2
Break down AI spend by provider, model, project and owner
- See total spend, active projects and API keys in one view.
- Track spend over time and break it down by provider, model, project or user.
- Drill down to a single API key to trace where a charge came from.
3
Track token usage and blended cost by model
- Follow total token volume over time, split by provider.
- Compare the blended cost per thousand tokens across providers and models.
- Model what happens to cost if work moves from a higher-cost model to a lower-cost one.
4
Compare AI spend against budget
- See actual spend against the AI budget and how much of the budget is used.
- See where actuals end and the forecast begins, so you know whether spend is ahead of or behind budget.
- Break budget and spend down by project and owner.
5
Forecast AI spend from usage and headcount drivers
- Set budgets for each project and track actual spend against them.
- Layer in drivers such as usage growth, headcount growth, model mix and spend limits, and see them flow through the forecast.
- Assign a per-employee AI budget by team or department.
6
Measure the output and value behind AI spend
- Connect engineering AI usage to developer activity, from sessions and commits to pull requests.
- See what each session or commit costs.
- Translate productivity gains into an estimated business value.

How Drivetrain simplifies token usage management

Forecast AI spend from usage, headcount and model mix, so the bill matches what you expected.
See which people, projects and models drive AI spend, not just the total on the invoice.
Weigh what AI costs against the output and value it produces.

FAQs

Which AI providers can Drivetrain connect to?

Drivetrain connects to AI providers such as Anthropic and OpenAI. Each connection is set up once, and usage data flows in from then on.

What usage data does Drivetrain pull from AI providers?

API key, owner, model, token counts, timestamps and spend. You can add business context on top, such as the project or team each API key belongs to.

Can Drivetrain show AI spend by person or project?

Yes. Drivetrain breaks AI spend down by provider, model, project and user, down to a single API key. Giving each user and project its own API key makes this more accurate.

Can Drivetrain forecast AI spend?

Yes. Drivetrain shows a spend forecast alongside actuals and budget. You can add drivers such as usage growth, headcount growth, model mix and spend limits, and set a per-employee AI budget by team or department.

Can Drivetrain show what happens if we move work to a cheaper model?

Yes. Drivetrain compares blended cost per thousand tokens across models, so you can model the cost of shifting work from one model to another and check the effect on gross margin.