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

AfterModelFn

required

No description available.

AfterModelFn

The result of the operation.

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

praisonaiagents/hooks/middleware.py at line 206