Function handle_tool_error

core/tools.py:109–149

Decorator for graceful error handling of tool operations. Args: tool_name: Name of the tool fallback_value: Value to return if tool fails log_level: Log level for errors Example: @handle_tool_error("nmap", fallback_value=[]) async def sca

(
    tool_name: str,
    fallback_value=None,
    log_level: str = "warning",
)

def handle_tool_error(
    tool_name: str,
    fallback_value=None,
    log_level: str = "warning",
):
    """Decorator for graceful error handling of tool operations.

    Args:
        tool_name: Name of the tool
        fallback_value: Value to return if tool fails
        log_level: Log level for errors

    Example:
        @handle_tool_error("nmap", fallback_value=[])
        async def scan_ports(target):
            # nmap code
            pass
    """
    def decorator(func):
        async def wrapper(*args, **kwargs):
            try:
                return await func(*args, **kwargs)
            except ToolNotFoundError as e:
                logger.log(
                    level=getattr(logger, log_level.upper()),
                    msg=f"Skipping {tool_name}: {e}"
                )
                return fallback_value
            except ToolExecutionError as e:
                logger.log(
                    level=getattr(logger, log_level.upper()),
                    msg=f"{tool_name} execution failed: {e.error}"
                )
                return fallback_value
            except Exception as e:
                logger.error(f"Unexpected error in {tool_name}: {e}")
                return fallback_value

        return wrapper

    return decorator