litellm-mirror/litellm/llms/predibase/common_utils.py

23 lines
598 B
Python

from typing import Optional, Union
import httpx
from litellm.llms.base_llm.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,
)