increase live test timeout + fixes for GA version (#34861)

Author: kristapratico

sdk/openai/azure-openai/tests/test_chat_completions.py

@@ -713,8 +713,11 @@ def test_chat_completion_byod(self, client, api_type, api_version, **kwargs):
                         "type": "azure_search",
                         "parameters": {
                             "endpoint": os.environ[ENV_AZURE_OPENAI_SEARCH_ENDPOINT],
-                            "key": os.environ[ENV_AZURE_OPENAI_SEARCH_KEY],
-                            "index_name": os.environ[ENV_AZURE_OPENAI_SEARCH_INDEX]
+                            "index_name": os.environ[ENV_AZURE_OPENAI_SEARCH_INDEX],
+                            "authentication": {
+                                "type": "api_key",
+                                "key": os.environ[ENV_AZURE_OPENAI_SEARCH_KEY],
+                            }
                         }
                     }
                 ],
@@ -749,8 +752,11 @@ def test_streamed_chat_completions_byod(self, client, api_type, api_version, **k
                         "type": "azure_search",
                         "parameters": {
                             "endpoint": os.environ[ENV_AZURE_OPENAI_SEARCH_ENDPOINT],
-                            "key": os.environ[ENV_AZURE_OPENAI_SEARCH_KEY],
-                            "index_name": os.environ[ENV_AZURE_OPENAI_SEARCH_INDEX]
+                            "index_name": os.environ[ENV_AZURE_OPENAI_SEARCH_INDEX],
+                            "authentication": {
+                                "type": "api_key",
+                                "key": os.environ[ENV_AZURE_OPENAI_SEARCH_KEY],
+                            }
                         }
                     }
                 ],

sdk/openai/azure-openai/tests/test_chat_completions_async.py

@@ -729,8 +729,11 @@ async def test_chat_completion_byod(self, client_async, api_type, api_version, *
                         "type": "azure_search",
                         "parameters": {
                             "endpoint": os.environ[ENV_AZURE_OPENAI_SEARCH_ENDPOINT],
-                            "key": os.environ[ENV_AZURE_OPENAI_SEARCH_KEY],
-                            "index_name": os.environ[ENV_AZURE_OPENAI_SEARCH_INDEX]
+                            "index_name": os.environ[ENV_AZURE_OPENAI_SEARCH_INDEX],
+                            "authentication": {
+                                "type": "api_key",
+                                "key": os.environ[ENV_AZURE_OPENAI_SEARCH_KEY],
+                            }
                         }
                     }
                 ],
@@ -766,8 +769,11 @@ async def test_streamed_chat_completions_byod(self, client_async, api_type, api_
                         "type": "azure_search",
                         "parameters": {
                             "endpoint": os.environ[ENV_AZURE_OPENAI_SEARCH_ENDPOINT],
-                            "key": os.environ[ENV_AZURE_OPENAI_SEARCH_KEY],
-                            "index_name": os.environ[ENV_AZURE_OPENAI_SEARCH_INDEX]
+                            "index_name": os.environ[ENV_AZURE_OPENAI_SEARCH_INDEX],
+                            "authentication": {
+                                "type": "api_key",
+                                "key": os.environ[ENV_AZURE_OPENAI_SEARCH_KEY],
+                            }
                         }
                     }
                 ],

sdk/openai/tests.yml

@@ -4,6 +4,7 @@ extends:
     template: /eng/pipelines/templates/stages/archetype-sdk-tests.yml
     parameters:
       ServiceDirectory: openai
+      TestTimeoutInMinutes: 200
       ToxTestEnv: optional
       MatrixFilters:
         - PythonVersion=3.11.*