phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(bedrock_httpx.py): fix error code for not found provider/model combo to be 404

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-13 20:35:18 -07:00
parent d6125ca257
commit c1279ed809
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/bedrock_httpx.py
View file

@ -1055,8 +1055,8 @@ class BedrockLLM(BaseLLM):
},
)
raise BedrockError(
status_code=400,
message="Bedrock HTTPX: Unsupported provider={}, model={}".format(
status_code=404,
message="Bedrock HTTPX: Unknown provider={}, model={}".format(
provider, model
),
)

4
litellm/tests/test_bedrock_completion.py
View file

@ -1159,8 +1159,8 @@ def test_bedrock_tools_pt_invalid_names():
assert result[1]["toolSpec"]["name"] == "another_invalid_name"
def test_bad_request_error():
with pytest.raises(litellm.BadRequestError):
def test_not_found_error():
with pytest.raises(litellm.NotFoundError):
completion(
model="bedrock/bad_model",
messages=[