a78339a040
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: -LAN- <laipz8200@outlook.com>
58 lines
1.3 KiB
Python
58 lines
1.3 KiB
Python
from typing import Optional
|
|
|
|
|
|
class LLMError(ValueError):
|
|
"""Base class for all LLM exceptions."""
|
|
|
|
description: Optional[str] = None
|
|
|
|
def __init__(self, description: Optional[str] = None):
|
|
self.description = description
|
|
|
|
|
|
class LLMBadRequestError(LLMError):
|
|
"""Raised when the LLM returns bad request."""
|
|
|
|
description = "Bad Request"
|
|
|
|
|
|
class ProviderTokenNotInitError(ValueError):
|
|
"""
|
|
Custom exception raised when the provider token is not initialized.
|
|
"""
|
|
|
|
description = "Provider Token Not Init"
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
self.description = args[0] if args else self.description
|
|
|
|
|
|
class QuotaExceededError(ValueError):
|
|
"""
|
|
Custom exception raised when the quota for a provider has been exceeded.
|
|
"""
|
|
|
|
description = "Quota Exceeded"
|
|
|
|
|
|
class AppInvokeQuotaExceededError(ValueError):
|
|
"""
|
|
Custom exception raised when the quota for an app has been exceeded.
|
|
"""
|
|
|
|
description = "App Invoke Quota Exceeded"
|
|
|
|
|
|
class ModelCurrentlyNotSupportError(ValueError):
|
|
"""
|
|
Custom exception raised when the model not support
|
|
"""
|
|
|
|
description = "Model Currently Not Support"
|
|
|
|
|
|
class InvokeRateLimitError(ValueError):
|
|
"""Raised when the Invoke returns rate limit error."""
|
|
|
|
description = "Rate Limit Error"
|