diff --git a/litellm/llms/bedrock_httpx.py b/litellm/llms/bedrock_httpx.py index ffc096f762..c433c32b7d 100644 --- a/litellm/llms/bedrock_httpx.py +++ b/litellm/llms/bedrock_httpx.py @@ -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 ), ) diff --git a/litellm/tests/test_bedrock_completion.py b/litellm/tests/test_bedrock_completion.py index 4da18144d0..c331021213 100644 --- a/litellm/tests/test_bedrock_completion.py +++ b/litellm/tests/test_bedrock_completion.py @@ -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=[