diff --git a/tests/lib/test_bedrock.py b/tests/lib/test_bedrock.py index bb29f106..280cb3ae 100644 --- a/tests/lib/test_bedrock.py +++ b/tests/lib/test_bedrock.py @@ -92,6 +92,7 @@ async def test_messages_retries_async(respx_mock: MockRouter) -> None: == "https://bedrock-runtime.us-east-1.amazonaws.com/model/anthropic.claude-3-sonnet-20240229-v1:0/invoke" ) + @pytest.mark.respx() def test_application_inference_profile(respx_mock: MockRouter) -> None: respx_mock.post(re.compile(r"https://bedrock-runtime\.us-east-1\.amazonaws\.com/model/.*/invoke")).mock( @@ -113,7 +114,6 @@ def test_application_inference_profile(respx_mock: MockRouter) -> None: ) calls = cast("list[MockRequestCall]", respx_mock.calls) - assert len(calls) == 2 assert ( @@ -123,4 +123,4 @@ def test_application_inference_profile(respx_mock: MockRouter) -> None: assert ( calls[1].request.url == "https://bedrock-runtime.us-east-1.amazonaws.com/model/arn:aws:bedrock:us-east-1:123456789012:application-inference-profile%2Fjf2sje1c0jnb/invoke" - ) \ No newline at end of file + )