Krrish Dholakia 2024-12-11 01:00:33 -08:00
parent 78d132c1fb
commit b9b34a7b99
8 changed files with 255 additions and 176 deletions

View file

@ -0,0 +1,23 @@
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,
)