17 lines
430 B
Python
17 lines
430 B
Python
|
|
class InvokeError(Exception):
|
||
|
|
"""Base class for all LLM exceptions."""
|
||
|
|
|
||
|
|
description: str | None = None
|
||
|
|
|
||
|
|
def __init__(self, description: str | None = 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"
|