Track token usage for an AI API call
AI agents call track_usage to retrieve information from Llm Token Tracker without modifying anything — typically the context-gathering step in research, monitoring, and reporting workflows, before the agent takes action elsewhere.
Even though track_usage only reads data, uncontrolled read access leaks sensitive information and racks up API costs — an agent caught in a retry loop can make thousands of calls a minute without anyone noticing.
Attacks that exploit this kind of access
Track token usage for an AI API call. It is categorised as a Read tool in the Llm Token Tracker MCP Server, which means it retrieves data without modifying state.
Register the Llm Token Tracker MCP server in PolicyLayer and add a rule for track_usage: allow, deny, rate-limit, or require approval. Point your MCP client at the PolicyLayer proxy URL and the rule is enforced on every call, before it reaches Llm Token Tracker. Nothing to install.
track_usage is a Read tool with low risk. Read-only tools are generally safe to allow by default.
Yes. Add a rate_limit block to the track_usage rule in your PolicyLayer policy. For example, setting max: 10 and window: 60 limits the tool to 10 calls per minute. Rate limits are tracked per agent session and reset automatically.
Set action: deny in the PolicyLayer policy for track_usage. The AI agent will receive a policy violation error and cannot call the tool. You can also include a reason field to explain why the tool is blocked.
track_usage is provided by the Llm Token Tracker MCP server (wn01011/llm-token-tracker). PolicyLayer sits as a proxy in front of this server to enforce policies before tool calls reach the server.