Source code for agent_inspect.exception.exception

from agent_inspect.exception.error_codes import (
    EvaluationComponent,
    UserProxyComponent,
    ToolComponent,
)




class EvaluationError(Exception):
    """Base exception class for Evaluation errors."""

    def __init__(self, internal_code: str, message: str):
        self.internal_code = EvaluationComponent.EVALUATION_ERROR_CODE.value + internal_code
        self.message = f"Internal Code: {self.internal_code}, Error Message: {message}"
        super().__init__(self.message)





class InvalidInputValueError(ValueError):
    def __init__(
        self,
        internal_code: str,
        message: str,
        component_code=EvaluationComponent.EVALUATION_ERROR_CODE.value,
    ):
        self.internal_code = component_code + internal_code
        self.message = f"Internal Code: {self.internal_code}, Error Message: {message}"
        super().__init__(self.message)





class UserProxyError(Exception):
    def __init__(self, internal_code: str, message: str):
        self.internal_code = UserProxyComponent.USER_PROXY_ERROR_CODE.value + internal_code
        self.message = f"Internal Code: {self.internal_code}, Error Message: {message}"
        super().__init__(self.message)





class ToolError(Exception):
    """Base exception class for Tool errors."""

    def __init__(self, internal_code: str, message: str):
        self.internal_code = ToolComponent.TOOL_ERROR_CODE.value + internal_code
        self.message = f"Internal Code: {self.internal_code}, Error Message: {message}"
        super().__init__(self.message)