> ## 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.
# Wrap Tool Call • AI Agent SDK
> wrap_tool_call: Decorator to mark a function as a wrap_tool_call middleware.
# wrap\_tool\_call
Function
> This function is defined in the [**middleware**](../modules/middleware) module.
Decorator to mark a function as a wrap\_tool\_call middleware.
The function receives (request, call\_next) and should call call\_next(request)
to continue the chain, or return early to short-circuit.
```mermaid theme={"theme":{"light":"vitesse-light","dark":"vitesse-dark"}}
%%{init: {'theme': 'base', 'themeVariables': { 'primaryColor': '#8B0000', 'primaryTextColor': '#fff', 'primaryBorderColor': '#710101', 'lineColor': '#189AB4', 'secondaryColor': '#189AB4', 'tertiaryColor': '#fff' }}}%%
graph LR
agent["Agent"] -- "uses" --> tool["Tool: wrap_tool_call"]
tool -- "returns" --> result["Result"]
style tool fill:#189AB4,color:#fff
style agent fill:#8B0000,color:#fff
style result fill:#8B0000,color:#fff
```
## Signature
```python theme={"theme":{"light":"vitesse-light","dark":"vitesse-dark"}}
def wrap_tool_call(func: WrapToolCallFn) -> WrapToolCallFn
```
## Parameters
No description available.
### Returns
The result of the operation.
## Usage
```python theme={"theme":{"light":"vitesse-light","dark":"vitesse-dark"}}
@wrap_tool_call
def retry_flaky_tool(request, call_next):
last_error = None
for _ in range(3):
try:
return call_next(request)
except Exception as e:
last_error = e
raise last_error
```
## Source
`praisonaiagents/hooks/middleware.py` at line 272
***
## Related Documentation