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

fix(utils.py): fix exception raised

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-08 07:42:17 +05:30
parent 3469b5b911
commit c04fa54d19
1 changed files with 13 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

14
litellm/utils.py
View file

@ -4123,7 +4123,19 @@ def get_llm_provider(
)
return model, custom_llm_provider, dynamic_api_key, api_base
except Exception as e:
raise Exception(f"GetLLMProvider Exception - {str(e)}\n\noriginal model: {model}")
if isinstance(e, litellm.exceptions.BadRequestError):
raise e
else:
raise litellm.exceptions.BadRequestError( # type: ignore
message=f"GetLLMProvider Exception - {str(e)}\n\noriginal model: {model}",
model=model,
response=httpx.Response(
status_code=400,
content=error_str,
request=httpx.request(method="completion", url="https://github.com/BerriAI/litellm"), # type: ignore
),
llm_provider="",
)
def get_api_key(llm_provider: str, dynamic_api_key: Optional[str]):