errors.py

"""Custom exceptions for the A2A client."""

from a2a.types import JSONRPCErrorResponse


class A2AClientError(Exception):
    """Base exception for A2A Client errors."""


class A2AClientHTTPError(A2AClientError):
    """Client exception for HTTP errors received from the server."""

    def __init__(self, status_code: int, message: str):
        """Initializes the A2AClientHTTPError.

        Args:
            status_code: The HTTP status code of the response.
            message: A descriptive error message.
        """
        self.status_code = status_code
        self.message = message
        super().__init__(f'HTTP Error {status_code}: {message}')

    def __repr__(self) -> str:
        """Returns an unambiguous representation showing structured attributes."""
        return (
            f'{self.__class__.__name__}('
            f'status_code={self.status_code!r}, '
            f'message={self.message!r})'
        )


class A2AClientJSONError(A2AClientError):
    """Client exception for JSON errors during response parsing or validation."""

    def __init__(self, message: str):
        """Initializes the A2AClientJSONError.

        Args:
            message: A descriptive error message.
        """
        self.message = message
        super().__init__(f'JSON Error: {message}')

    def __repr__(self) -> str:
        """Returns an unambiguous representation showing structured attributes."""
        return f'{self.__class__.__name__}(message={self.message!r})'


class A2AClientTimeoutError(A2AClientError):
    """Client exception for timeout errors during a request."""

    def __init__(self, message: str):
        """Initializes the A2AClientTimeoutError.

        Args:
            message: A descriptive error message.
        """
        self.message = message
        super().__init__(f'Timeout Error: {message}')

    def __repr__(self) -> str:
        """Returns an unambiguous representation showing structured attributes."""
        return f'{self.__class__.__name__}(message={self.message!r})'


class A2AClientInvalidArgsError(A2AClientError):
    """Client exception for invalid arguments passed to a method."""

    def __init__(self, message: str):
        """Initializes the A2AClientInvalidArgsError.

        Args:
            message: A descriptive error message.
        """
        self.message = message
        super().__init__(f'Invalid arguments error: {message}')

    def __repr__(self) -> str:
        """Returns an unambiguous representation showing structured attributes."""
        return f'{self.__class__.__name__}(message={self.message!r})'


class A2AClientInvalidStateError(A2AClientError):
    """Client exception for an invalid client state."""

    def __init__(self, message: str):
        """Initializes the A2AClientInvalidStateError.

        Args:
            message: A descriptive error message.
        """
        self.message = message
        super().__init__(f'Invalid state error: {message}')

    def __repr__(self) -> str:
        """Returns an unambiguous representation showing structured attributes."""
        return f'{self.__class__.__name__}(message={self.message!r})'


class A2AClientJSONRPCError(A2AClientError):
    """Client exception for JSON-RPC errors returned by the server."""

    def __init__(self, error: JSONRPCErrorResponse):
        """Initializes the A2AClientJsonRPCError.

        Args:
            error: The JSON-RPC error object.
        """
        self.error = error.error
        super().__init__(f'JSON-RPC Error {error.error}')

    def __repr__(self) -> str:
        """Returns an unambiguous representation showing the JSON-RPC error object."""
        return f'{self.__class__.__name__}({self.error!r})'