After Model • AI Agent SDK

after_model: Decorator to mark a function as an after_model hook.

after_model

Function

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

func · AfterModelFn · required

No description available.

Returns

Returns · AfterModelFn

The result of the operation.

Usage

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

Source

View on GitHub

praisonaiagents/hooks/middleware.py at line 206