scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-08 22:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

Merge pull request #3403 from msabramo/disambiguate-invalid-model-name-errors

Browse source 
Disambiguate invalid model name errors
This commit is contained in:
Ishaan Jaff 2024-05-02 15:09:59 -07:00 • committed by GitHub
commit f893f8bc55
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 8 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
litellm/proxy/proxy_server.py
View file

@ -3644,7 +3644,7 @@ async def chat_completion(
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail={
"error": "Invalid model name passed in model="
"error": "chat_completion: Invalid model name passed in model="
+ data.get("model", "")
},
)
@ -3833,7 +3833,7 @@ async def completion(
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail={
"error": "Invalid model name passed in model="
"error": "completion: Invalid model name passed in model="
+ data.get("model", "")
},
)
@ -4041,7 +4041,7 @@ async def embeddings(
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail={
"error": "Invalid model name passed in model="
"error": "embeddings: Invalid model name passed in model="
+ data.get("model", "")
},
)
@ -4197,7 +4197,7 @@ async def image_generation(
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail={
"error": "Invalid model name passed in model="
"error": "image_generation: Invalid model name passed in model="
+ data.get("model", "")
},
)
@ -4372,7 +4372,7 @@ async def audio_transcriptions(
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail={
"error": "Invalid model name passed in model="
"error": "audio_transcriptions: Invalid model name passed in model="
+ data.get("model", "")
},
)
@ -4538,7 +4538,7 @@ async def moderations(
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail={
"error": "Invalid model name passed in model="
"error": "moderations: Invalid model name passed in model="
+ data.get("model", "")
},
)

4
litellm/tests/test_proxy_exception_mapping.py
View file

@ -169,7 +169,7 @@ def test_chat_completion_exception_any_model(client):
)
assert isinstance(openai_exception, openai.BadRequestError)
_error_message = openai_exception.message
assert "Invalid model name passed in model=Lite-GPT-12" in str(_error_message)
assert "chat_completion: Invalid model name passed in model=Lite-GPT-12" in str(_error_message)
except Exception as e:
pytest.fail(f"LiteLLM Proxy test failed. Exception {str(e)}")
@ -197,7 +197,7 @@ def test_embedding_exception_any_model(client):
print("Exception raised=", openai_exception)
assert isinstance(openai_exception, openai.BadRequestError)
_error_message = openai_exception.message
assert "Invalid model name passed in model=Lite-GPT-12" in str(_error_message)
assert "embeddings: Invalid model name passed in model=Lite-GPT-12" in str(_error_message)
except Exception as e:
pytest.fail(f"LiteLLM Proxy test failed. Exception {str(e)}")