a78339a040
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: -LAN- <laipz8200@outlook.com>
20 lines
466 B
Python
20 lines
466 B
Python
from typing import Optional
|
|
|
|
|
|
class InvokeError(Exception):
|
|
"""Base class for all LLM exceptions."""
|
|
|
|
description: Optional[str] = None
|
|
|
|
def __init__(self, description: Optional[str] = None):
|
|
self.description = description
|
|
|
|
def __str__(self):
|
|
return self.description or self.__class__.__name__
|
|
|
|
|
|
class InvokeRateLimitError(InvokeError):
|
|
"""Raised when the Invoke returns rate limit error."""
|
|
|
|
description = "Rate Limit Error"
|