# TokenStateMiddleware

> **Class** in `deepagents_cli`

📖 [View in docs](https://reference.langchain.com/python/deepagents-cli/token_state/TokenStateMiddleware)

Schema-only middleware that registers `_context_tokens` in the state schema.

## Signature

```python
TokenStateMiddleware()
```

## Extends

- `AgentMiddleware`

## Properties

- `state_schema`

---

[View source on GitHub](https://github.com/langchain-ai/deepagents/blob/a9e6e4f7ad7fe161dd9affc3d74bb19784aca70b/libs/cli/deepagents_cli/token_state.py#L28)