src/writing_studio/core/exceptions.py

"""Custom exceptions for Writing Studio."""


class WritingStudioException(Exception):
    """Base exception for Writing Studio."""

    def __init__(self, message: str, details: dict = None):
        """
        Initialize exception.

        Args:
            message: Error message
            details: Additional error details
        """
        self.message = message
        self.details = details or {}
        super().__init__(self.message)


class ModelLoadError(WritingStudioException):
    """Exception raised when model loading fails."""

    pass


class TextGenerationError(WritingStudioException):
    """Exception raised during text generation."""

    pass


class ValidationError(WritingStudioException):
    """Exception raised for input validation failures."""

    pass


class RateLimitExceeded(WritingStudioException):
    """Exception raised when rate limit is exceeded."""

    pass


class ConfigurationError(WritingStudioException):
    """Exception raised for configuration issues."""

    pass


class CacheError(WritingStudioException):
    """Exception raised for caching issues."""

    pass