phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(fix) raise openai.NotFoundError

Browse source
This commit is contained in:
ishaan-jaff 2023-12-15 14:03:50 +05:30
parent 0b4f1dee42
commit 33afbd6ff5
1 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/utils.py
View file

@ -2862,7 +2862,7 @@ def get_llm_provider(model: str, custom_llm_provider: Optional[str] = None, api_
response=httpx.Response( response=httpx.Response(
status_code=404, status_code=404,
content=error_str, content=error_str,
request=httpx.request(method="completion", url="https://litellm.ai") # type: ignore request=httpx.request(method="completion", url="https://github.com/BerriAI/litellm") # type: ignore
), ),
llm_provider="" llm_provider=""
) )
@ -4207,6 +4207,14 @@ def exception_type(
model=model, model=model,
response=original_exception.response response=original_exception.response
) )
elif original_exception.status_code == 404:
exception_mapping_worked = True
raise NotFoundError(
message=f"OpenAIException - {original_exception.message}",
model=model,
llm_provider="openai",
response=original_exception.response
)
elif original_exception.status_code == 408: elif original_exception.status_code == 408:
exception_mapping_worked = True exception_mapping_worked = True
raise Timeout( raise Timeout(
@ -4214,7 +4222,7 @@ def exception_type(
model=model, model=model,
llm_provider="openai", llm_provider="openai",
) )
if original_exception.status_code == 422: elif original_exception.status_code == 422:
exception_mapping_worked = True exception_mapping_worked = True
raise BadRequestError( raise BadRequestError(
message=f"OpenAIException - {original_exception.message}", message=f"OpenAIException - {original_exception.message}",