diff --git a/tests/image_gen_tests/test_image_generation.py b/tests/image_gen_tests/test_image_generation.py index 118c818150..0f980854c0 100644 --- a/tests/image_gen_tests/test_image_generation.py +++ b/tests/image_gen_tests/test_image_generation.py @@ -187,46 +187,6 @@ class TestAzureOpenAIDalle3(BaseImageGenTest): -@pytest.mark.flaky(retries=3, delay=1) -def test_image_generation_azure_dall_e_3(): - try: - litellm.set_verbose = True - response = litellm.image_generation( - prompt="A cute baby sea otter", - model="azure/dall-e-3-test", - api_version="2023-12-01-preview", - api_base=os.getenv("AZURE_SWEDEN_API_BASE"), - api_key=os.getenv("AZURE_SWEDEN_API_KEY"), - metadata={ - "model_info": { - "base_model": "azure/dall-e-3", - } - }, - ) - print(f"response: {response}") - - print("response", response._hidden_params) - assert len(response.data) > 0 - except litellm.InternalServerError as e: - pass - except litellm.ContentPolicyViolationError: - pass # OpenAI randomly raises these errors - skip when they occur - except litellm.InternalServerError: - pass - except litellm.RateLimitError as e: - pass - except Exception as e: - if "Your task failed as a result of our safety system." in str(e): - pass - if "Connection error" in str(e): - pass - else: - pytest.fail(f"An exception occurred - {str(e)}") - - -# asyncio.run(test_async_image_generation_openai()) - - @pytest.mark.skip(reason="model EOL") @pytest.mark.asyncio async def test_aimage_generation_bedrock_with_optional_params():