Skip to content

base

_base

Shared error-handling helpers for axm_ast tool implementations.

log_and_fallback(logger, exc, fallback)

Log an exception at WARNING with exc_info and return fallback.

Used inside helpers that return a dict (or other structured value) instead of a ToolResult — they can't use :func:safe_execute but still need centralized logging on the failure boundary.

Source code in packages/axm-ast/src/axm_ast/tools/_base.py
Python
43
44
45
46
47
48
49
50
51
52
53
54
55
def log_and_fallback[T](
    logger: logging.Logger,
    exc: BaseException,
    fallback: T,
) -> T:
    """Log an exception at ``WARNING`` with ``exc_info`` and return *fallback*.

    Used inside helpers that return a ``dict`` (or other structured value)
    instead of a ``ToolResult`` — they can't use :func:`safe_execute` but
    still need centralized logging on the failure boundary.
    """
    logger.warning("Tool helper failed: %s", exc, exc_info=True)
    return fallback

safe_execute(method)

Wrap a tool execute method to log + return structured failures.

The wrapper logs any uncaught exception at WARNING on the calling module's logger with exc_info=True, then returns a ToolResult(success=False, error=str(exc)) so callers never see a raised exception.

Source code in packages/axm-ast/src/axm_ast/tools/_base.py
Python
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
def safe_execute(
    method: Callable[..., ToolResult],
) -> Callable[..., ToolResult]:
    """Wrap a tool ``execute`` method to log + return structured failures.

    The wrapper logs any uncaught exception at ``WARNING`` on the calling
    module's logger with ``exc_info=True``, then returns a
    ``ToolResult(success=False, error=str(exc))`` so callers never see a
    raised exception.
    """
    logger = logging.getLogger(method.__module__)

    @functools.wraps(method)
    def wrapper(self: Any, *args: Any, **kwargs: Any) -> ToolResult:
        try:
            return method(self, *args, **kwargs)
        except Exception as exc:  # noqa: BLE001 — final boundary
            logger.warning(
                "Tool %s failed: %s",
                type(self).__name__,
                exc,
                exc_info=True,
            )
            return ToolResult(success=False, error=str(exc))

    return wrapper