(feat) add openai.NotFoundError

This commit is contained in:
ishaan-jaff 2023-12-15 10:18:02 +05:30
parent 4fd59c5847
commit 353433e5ce
3 changed files with 28 additions and 3 deletions

View file

@ -12,6 +12,7 @@
from openai import (
AuthenticationError,
BadRequestError,
NotFoundError,
RateLimitError,
APIStatusError,
OpenAIError,
@ -36,6 +37,20 @@ class AuthenticationError(AuthenticationError): # type: ignore
body=None
) # Call the base class constructor with the parameters it needs
# raise when invalid models passed, example gpt-8
class NotFoundError(NotFoundError): # type: ignore
def __init__(self, message, model, llm_provider, response: httpx.Response):
self.status_code = 404
self.message = message
self.model = model
self.llm_provider = llm_provider
super().__init__(
self.message,
response=response,
body=None
) # Call the base class constructor with the parameters it needs
class BadRequestError(BadRequestError): # type: ignore
def __init__(self, message, model, llm_provider, response: httpx.Response):
self.status_code = 400