exceptions.py

"""
exceptions.py — Пользовательские исключения для агента.
"""


class AgentException(Exception):
    """Базовое исключение агента."""
    def __init__(self, message: str, details: dict = None):
        super().__init__(message)
        self.message = message
        self.details = details or {}
    
    def __str__(self):
        if self.details:
            return f"{self.message} | Детали: {self.details}"
        return self.message


class ModelError(AgentException):
    """Ошибка загрузки или работы модели."""
    pass


class ModelNotLoadedError(ModelError):
    """Модель не загружена."""
    pass


class ModelInferenceError(ModelError):
    """Ошибка инференса."""
    pass


class SafetyError(AgentException):
    """Ошибка безопасности."""
    pass


class DangerousCommandError(SafetyError):
    """Обнаружена опасная команда."""
    def __init__(self, command: str, pattern: str):
        super().__init__(
            f"Обнаружена опасная команда: {command}",
            {"command": command, "pattern": pattern}
        )


class ConfirmationRequiredError(SafetyError):
    """Требуется подтверждение пользователя."""
    def __init__(self, command: str, reason: str):
        super().__init__(
            f"Требуется подтверждение: {reason}",
            {"command": command, "reason": reason}
        )


class ToolError(AgentException):
    """Ошибка инструмента."""
    pass


class ToolNotFoundError(ToolError):
    """Инструмент не найден."""
    pass


class ToolExecutionError(ToolError):
    """Ошибка выполнения инструмента."""
    def __init__(self, tool_name: str, error: str, command: str = None):
        details = {"tool": tool_name, "error": error}
        if command:
            details["command"] = command
        super().__init__(f"Ошибка инструмента {tool_name}: {error}", details)


class ToolTimeoutError(ToolError):
    """Таймаут инструмента."""
    def __init__(self, tool_name: str, timeout: int):
        super().__init__(
            f"Инструмент {tool_name} превысил таймаут {timeout}с",
            {"tool": tool_name, "timeout": timeout}
        )


class MemoryError(AgentException):
    """Ошибка памяти."""
    pass


class MemoryStorageError(MemoryError):
    """Ошибка сохранения в память."""
    pass


class MemoryRetrievalError(MemoryError):
    """Ошибка извлечения из памяти."""
    pass


class WebError(AgentException):
    """Ошибка веб-инструментов."""
    pass


class WebRequestError(WebError):
    """Ошибка HTTP-запроса."""
    def __init__(self, url: str, status_code: int = None, error: str = None):
        super().__init__(
            f"Ошибка запроса к {url}",
            {"url": url, "status_code": status_code, "error": error}
        )


class WebParseError(WebError):
    """Ошибка парсинга веб-страницы."""
    pass


class ConfigurationError(AgentException):
    """Ошибка конфигурации."""
    pass


class StepLimitExceededError(AgentException):
    """Превышен лимит шагов."""
    def __init__(self, max_steps: int):
        super().__init__(
            f"Превышен лимит шагов ({max_steps})",
            {"max_steps": max_steps}
        )


class RetryLimitExceededError(AgentException):
    """Превышен лимит повторных попыток."""
    def __init__(self, max_retries: int):
        super().__init__(
            f"Превышен лимит повторных попыток ({max_retries})",
            {"max_retries": max_retries}
        )