24 lines
603 B
Python
24 lines
603 B
Python
|
from typing import Optional, Union
|
||
|
|
|
||
|
|
import httpx
|
||
|
|
|
||
|
|
from litellm.llms.base_llm.chat.transformation import BaseLLMException
|
||
|
|
|
||
|
|
|
||
|
|
class PredibaseError(BaseLLMException):
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
status_code: int,
|
||
|
|
message: str,
|
||
|
|
request: Optional[httpx.Request] = None,
|
||
|
|
response: Optional[httpx.Response] = None,
|
||
|
|
headers: Optional[Union[httpx.Headers, dict]] = None,
|
||
|
|
):
|
||
|
|
super().__init__(
|
||
|
|
status_code=status_code,
|
||
|
|
message=message,
|
||
|
|
request=request,
|
||
|
|
response=response,
|
||
|
|
headers=headers,
|
||
|
|
)
|