Function run_hooks

lazyllm/hook.py:191–204

(hook_objs, phase: str, *phase_args)

def run_hooks(hook_objs, phase: str, *phase_args):
    if phase not in ('post_hook', 'on_error', 'finalize'):
        raise ValueError(f'Invalid hook phase: {phase}')
    strict_errors = []
    ordered_hooks = hook_objs[::-1]
    for hook_obj in ordered_hooks:
        try:
            getattr(hook_obj, phase)(*phase_args)
        except Exception as e:
            if _hook_error_mode(hook_obj) == 'raise':
                strict_errors.append((hook_obj, e))
            else:
                LOG.warning(f'Hook `{type(hook_obj).__name__}` {phase} failed and will be skipped: {e}')
    _raise_hook_phase_errors(phase, strict_errors)


@contextmanager