Initial commit: AASRT v1.0.0 - AI Agent Security Reconnaissance Tool

src/utils/exceptions.py

@@ -0,0 +1,51 @@
+"""Custom exceptions for AASRT."""
+
+
+class AASRTException(Exception):
+    """Base exception for AASRT."""
+    pass
+
+
+class APIException(AASRTException):
+    """Raised when API call fails."""
+
+    def __init__(self, message: str, engine: str = None, status_code: int = None):
+        self.engine = engine
+        self.status_code = status_code
+        super().__init__(message)
+
+
+class RateLimitException(AASRTException):
+    """Raised when rate limit is exceeded."""
+
+    def __init__(self, message: str, engine: str = None, retry_after: int = None):
+        self.engine = engine
+        self.retry_after = retry_after
+        super().__init__(message)
+
+
+class ConfigurationException(AASRTException):
+    """Raised when configuration is invalid."""
+    pass
+
+
+class ValidationException(AASRTException):
+    """Raised when input validation fails."""
+    pass
+
+
+class AuthenticationException(AASRTException):
+    """Raised when authentication fails."""
+
+    def __init__(self, message: str, engine: str = None):
+        self.engine = engine
+        super().__init__(message)
+
+
+class TimeoutException(AASRTException):
+    """Raised when a request times out."""
+
+    def __init__(self, message: str, engine: str = None, timeout: int = None):
+        self.engine = engine
+        self.timeout = timeout
+        super().__init__(message)