Praison AI

Get Token Budget • AI Agent SDK

get_token_budget: Get TokenBudget instance for this config.

get_token_budget

Method

</div>

This is a method of the RetrievalConfig class in the retrieval_config module.

Get TokenBudget instance for this config.

Signature

python
def get_token_budget(model_name: Optional[str]) -> Any

Parameters

ParamField Optional[str]

Optional model name for context window detection

Returns

ResponseField Any

TokenBudget configured for this retrieval config

Uses

  • get_model_context_window
  • TokenBudget

Source

View on GitHub

praisonaiagents/rag/retrieval_config.py at line 128