> ## Documentation Index
> Fetch the complete documentation index at: https://praison.ai/docs/llms.txt
> Use this file to discover all available pages before exploring further.

# Get Token Budget • AI Agent SDK

> get_token_budget: Get TokenBudget instance for this config.

# get\_token\_budget

<div className="flex items-center gap-2">
  <Badge color="purple">Method</Badge>
</div>

> This is a method of the [**RetrievalConfig**](../classes/RetrievalConfig) class in the [**retrieval\_config**](../modules/retrieval_config) module.

Get TokenBudget instance for this config.

## Signature

```python theme={"theme":{"light":"vitesse-light","dark":"vitesse-dark"}}
def get_token_budget(model_name: Optional[str]) -> Any
```

## Parameters

<ParamField query="model_name" type="Optional" required={false}>
  Optional model name for context window detection
</ParamField>

### Returns

<ResponseField name="Returns" type="Any">
  TokenBudget configured for this retrieval config
</ResponseField>

## Uses

* `get_model_context_window`
* `TokenBudget`

## Source

<Card title="View on GitHub" icon="github" href="https://github.com/MervinPraison/PraisonAI/blob/main/src/praisonai-agents/praisonaiagents/rag/retrieval_config.py#L116">
  `praisonaiagents/rag/retrieval_config.py` at line 116
</Card>