diff --git a/tests/test_openai_endpoints.py b/tests/test_openai_endpoints.py index 0d15c598b..51ec70371 100644 --- a/tests/test_openai_endpoints.py +++ b/tests/test_openai_endpoints.py @@ -198,6 +198,10 @@ async def image_generation(session, key): print() if status != 200: + if ( + "Connection error" in response_text + ): # OpenAI endpoint returns a connection error + pass raise Exception(f"Request did not return a 200 status code: {status}")