After Model • AI Agent SDK

</div>

This function is defined in the middleware module.

Decorator to mark a function as an after_model hook.

The function receives a ModelResponse and should return a (possibly modified) ModelResponse.

Signature

def after_model(func: AfterModelFn) -> AfterModelFn

Parameters

Returns

Usage

@after_model
    def log_response(response):
        print(f"Model returned: {response.content[:50]}")
        return response

Source