fix v0 tests, non-azure openai won't accept api_version (#33511)

Author: kristapratico

sdk/openai/azure-openai/tests/v0_tests/test_dall_e_v0.py

@@ -14,7 +14,8 @@ class TestDallE(AzureRecordedTestCase):
     @pytest.mark.parametrize("api_type", ALL)
     @configure_v0
     def test_image_create(self, set_vars, azure_openai_creds, api_type):
-        openai.api_version = "2023-10-01-preview"  # not supported by later versions
+        if api_type != OPENAI:
+            openai.api_version = "2023-10-01-preview"  # not supported by later versions
         image = openai.Image.create(
             prompt="a cute baby seal"
         )
@@ -25,7 +26,8 @@ def test_image_create(self, set_vars, azure_openai_creds, api_type):
     @pytest.mark.parametrize("api_type", [AZURE, OPENAI])
     @configure_v0
     def test_image_create_n(self, set_vars, azure_openai_creds, api_type):
-        openai.api_version = "2023-10-01-preview"  # not supported by later versions
+        if api_type != OPENAI:
+            openai.api_version = "2023-10-01-preview"  # not supported by later versions
         image = openai.Image.create(
             prompt="a cute baby seal",
             n=2
@@ -38,7 +40,8 @@ def test_image_create_n(self, set_vars, azure_openai_creds, api_type):
     @pytest.mark.parametrize("api_type", [AZURE, OPENAI])
     @configure_v0
     def test_image_create_size(self, set_vars, azure_openai_creds, api_type):
-        openai.api_version = "2023-10-01-preview"  # not supported by later versions
+        if api_type != OPENAI:
+            openai.api_version = "2023-10-01-preview"  # not supported by later versions
         image = openai.Image.create(
             prompt="a cute baby seal",
             size="256x256"
@@ -61,7 +64,8 @@ def test_image_create_response_format(self, set_vars, azure_openai_creds, api_ty
     @pytest.mark.parametrize("api_type", [AZURE, OPENAI])
     @configure_v0
     def test_image_create_user(self, set_vars, azure_openai_creds, api_type):
-        openai.api_version = "2023-10-01-preview"  # not supported by later versions
+        if api_type != OPENAI:
+            openai.api_version = "2023-10-01-preview"  # not supported by later versions
         image = openai.Image.create(
             prompt="a cute baby seal",
             user="krista"

sdk/openai/azure-openai/tests/v0_tests/test_dall_e_v0_async.py

@@ -15,7 +15,8 @@ class TestDallEAsync(AzureRecordedTestCase):
     @pytest.mark.parametrize("api_type", ALL)
     @configure_v0_async
     async def test_image_create(self, set_vars, azure_openai_creds, api_type):
-        openai.api_version = "2023-10-01-preview"  # not supported by later versions
+        if api_type != OPENAI:
+            openai.api_version = "2023-10-01-preview"  # not supported by later versions
         image = await openai.Image.acreate(
             prompt="a cute baby seal"
         )
@@ -27,7 +28,8 @@ async def test_image_create(self, set_vars, azure_openai_creds, api_type):
     @pytest.mark.parametrize("api_type", [AZURE, OPENAI])
     @configure_v0_async
     async def test_image_create_n(self, set_vars, azure_openai_creds, api_type):
-        openai.api_version = "2023-10-01-preview"  # not supported by later versions
+        if api_type != OPENAI:
+            openai.api_version = "2023-10-01-preview"  # not supported by later versions
         image = await openai.Image.acreate(
             prompt="a cute baby seal",
             n=2
@@ -41,7 +43,8 @@ async def test_image_create_n(self, set_vars, azure_openai_creds, api_type):
     @pytest.mark.parametrize("api_type", [AZURE, OPENAI])
     @configure_v0_async
     async def test_image_create_size(self, set_vars, azure_openai_creds, api_type):
-        openai.api_version = "2023-10-01-preview"  # not supported by later versions
+        if api_type != OPENAI:
+            openai.api_version = "2023-10-01-preview"  # not supported by later versions
         image = await openai.Image.acreate(
             prompt="a cute baby seal",
             size="256x256"
@@ -66,7 +69,8 @@ async def test_image_create_response_format(self, set_vars, azure_openai_creds,
     @pytest.mark.parametrize("api_type", [AZURE, OPENAI])
     @configure_v0_async
     async def test_image_create_user(self, set_vars, azure_openai_creds, api_type):
-        openai.api_version = "2023-10-01-preview"  # not supported by later versions
+        if api_type != OPENAI:
+            openai.api_version = "2023-10-01-preview"  # not supported by later versions
         image = await openai.Image.acreate(
             prompt="a cute baby seal",
             user="krista"