Cleanup .pyc and .DS_Store, add new modules, remove legacy services
This commit is contained in:
@@ -186,3 +186,65 @@ class EveAINoManagementPartnerForTenant(EveAIException):
|
||||
super().__init__(message, status_code, payload)
|
||||
|
||||
|
||||
class EveAIQuotaExceeded(EveAIException):
|
||||
"""Base exception for quota-related errors"""
|
||||
|
||||
def __init__(self, message, quota_type, current_usage, limit, additional=0, status_code=400, payload=None):
|
||||
super().__init__(message, status_code, payload)
|
||||
self.quota_type = quota_type
|
||||
self.current_usage = current_usage
|
||||
self.limit = limit
|
||||
self.additional = additional
|
||||
|
||||
|
||||
class EveAIStorageQuotaExceeded(EveAIQuotaExceeded):
|
||||
"""Raised when storage quota is exceeded"""
|
||||
|
||||
def __init__(self, current_usage, limit, additional, status_code=400, payload=None):
|
||||
message = (f"Storage quota exceeded. Current: {current_usage:.1f}MB, "
|
||||
f"Additional: {additional:.1f}MB, Limit: {limit}MB")
|
||||
super().__init__(message, "storage", current_usage, limit, additional, status_code, payload)
|
||||
|
||||
|
||||
class EveAIEmbeddingQuotaExceeded(EveAIQuotaExceeded):
|
||||
"""Raised when embedding quota is exceeded"""
|
||||
|
||||
def __init__(self, current_usage, limit, additional, status_code=400, payload=None):
|
||||
message = (f"Embedding quota exceeded. Current: {current_usage:.1f}MB, "
|
||||
f"Additional: {additional:.1f}MB, Limit: {limit}MB")
|
||||
super().__init__(message, "embedding", current_usage, limit, additional, status_code, payload)
|
||||
|
||||
|
||||
class EveAIInteractionQuotaExceeded(EveAIQuotaExceeded):
|
||||
"""Raised when the interaction token quota is exceeded"""
|
||||
|
||||
def __init__(self, current_usage, limit, status_code=400, payload=None):
|
||||
message = (f"Interaction token quota exceeded. Current: {current_usage:.2f}M tokens, "
|
||||
f"Limit: {limit:.2f}M tokens")
|
||||
super().__init__(message, "interaction", current_usage, limit, 0, status_code, payload)
|
||||
|
||||
|
||||
class EveAIQuotaWarning(EveAIException):
|
||||
"""Warning when approaching quota limits (not blocking)"""
|
||||
|
||||
def __init__(self, message, quota_type, usage_percentage, status_code=200, payload=None):
|
||||
super().__init__(message, status_code, payload)
|
||||
self.quota_type = quota_type
|
||||
self.usage_percentage = usage_percentage
|
||||
|
||||
|
||||
class EveAILicensePeriodsExceeded(EveAIException):
|
||||
"""Raised when no more license periods can be created for a given license"""
|
||||
|
||||
def __init__(self, license_id, status_code=400, payload=None):
|
||||
message = f"No more license periods can be created for license with ID {license_id}. "
|
||||
super().__init__(message, status_code, payload)
|
||||
|
||||
|
||||
class EveAIPendingLicensePeriod(EveAIException):
|
||||
"""Raised when a license period is pending"""
|
||||
|
||||
def __init__(self, status_code=400, payload=None):
|
||||
message = f"Basic Fee Payment has not been received yet. Please ensure payment has been made, and please wait for payment to be processed."
|
||||
super().__init__(message, status_code, payload)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user