From 964cfce9d59df48af19960e6d86bb695fd8cb335 Mon Sep 17 00:00:00 2001 From: Darren Cohen <39422044+dargilco@users.noreply.github.com> Date: Sat, 3 May 2025 01:12:50 -0700 Subject: [PATCH] Re-emit from TypeSpec --- .../ai/projects/aio/operations/_patch_inference_async.py | 6 ++++-- .../ai/projects/aio/operations/_patch_telemetry_async.py | 6 ++++-- .../test_ai_project_evaluations_operations.py | 4 ++-- .../test_ai_project_evaluations_operations_async.py | 4 ++-- .../generated_tests/test_ai_project_red_teams_operations.py | 4 ++-- .../test_ai_project_red_teams_operations_async.py | 4 ++-- sdk/ai/azure-ai-projects/tests/samples/test_samples.py | 1 + sdk/ai/azure-ai-projects/tsp-location.yaml | 2 +- 8 files changed, 18 insertions(+), 13 deletions(-) diff --git a/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_patch_inference_async.py b/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_patch_inference_async.py index 35aa7ee8c9f7..b1e296e2c5c4 100644 --- a/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_patch_inference_async.py +++ b/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_patch_inference_async.py @@ -250,8 +250,10 @@ async def get_azure_openai_client( # If the connection uses API key authentication, we need to make another service call to get # the connection with API key populated. if connection.credentials.type == CredentialType.API_KEY: - connection = await self._outer_instance.connections._get_with_credentials( # pylint: disable=protected-access - name=connection_name, **kwargs + connection = ( + await self._outer_instance.connections._get_with_credentials( # pylint: disable=protected-access + name=connection_name, **kwargs + ) ) logger.debug("[InferenceOperations.get_azure_openai_client] connection = %s", str(connection)) diff --git a/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_patch_telemetry_async.py b/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_patch_telemetry_async.py index f09fc8de9efe..70637c9eb4de 100644 --- a/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_patch_telemetry_async.py +++ b/sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_patch_telemetry_async.py @@ -58,8 +58,10 @@ async def get_connection_string(self) -> str: if not connection_name: raise ResourceNotFoundError("No Application Insights connection found.") - connection = await self._outer_instance.connections._get_with_credentials( # pylint: disable=protected-access - name=connection_name + connection = ( + await self._outer_instance.connections._get_with_credentials( # pylint: disable=protected-access + name=connection_name + ) ) if isinstance(connection.credentials, ApiKeyCredentials): diff --git a/sdk/ai/azure-ai-projects/generated_tests/test_ai_project_evaluations_operations.py b/sdk/ai/azure-ai-projects/generated_tests/test_ai_project_evaluations_operations.py index d6f6c0869b48..e07aa0e02b47 100644 --- a/sdk/ai/azure-ai-projects/generated_tests/test_ai_project_evaluations_operations.py +++ b/sdk/ai/azure-ai-projects/generated_tests/test_ai_project_evaluations_operations.py @@ -34,9 +34,9 @@ def test_evaluations_list(self, aiproject_endpoint): @AIProjectPreparer() @recorded_by_proxy - def test_evaluations_create_run(self, aiproject_endpoint): + def test_evaluations_create(self, aiproject_endpoint): client = self.create_client(endpoint=aiproject_endpoint) - response = client.evaluations.create_run( + response = client.evaluations.create( evaluation={ "data": "input_data", "evaluators": {"str": {"id": "str", "dataMapping": {"str": "str"}, "initParams": {"str": {}}}}, diff --git a/sdk/ai/azure-ai-projects/generated_tests/test_ai_project_evaluations_operations_async.py b/sdk/ai/azure-ai-projects/generated_tests/test_ai_project_evaluations_operations_async.py index 3cf97659ac44..07f22bd9e58a 100644 --- a/sdk/ai/azure-ai-projects/generated_tests/test_ai_project_evaluations_operations_async.py +++ b/sdk/ai/azure-ai-projects/generated_tests/test_ai_project_evaluations_operations_async.py @@ -35,9 +35,9 @@ async def test_evaluations_list(self, aiproject_endpoint): @AIProjectPreparer() @recorded_by_proxy_async - async def test_evaluations_create_run(self, aiproject_endpoint): + async def test_evaluations_create(self, aiproject_endpoint): client = self.create_async_client(endpoint=aiproject_endpoint) - response = await client.evaluations.create_run( + response = await client.evaluations.create( evaluation={ "data": "input_data", "evaluators": {"str": {"id": "str", "dataMapping": {"str": "str"}, "initParams": {"str": {}}}}, diff --git a/sdk/ai/azure-ai-projects/generated_tests/test_ai_project_red_teams_operations.py b/sdk/ai/azure-ai-projects/generated_tests/test_ai_project_red_teams_operations.py index 4611d9ba846f..03d6211ace8e 100644 --- a/sdk/ai/azure-ai-projects/generated_tests/test_ai_project_red_teams_operations.py +++ b/sdk/ai/azure-ai-projects/generated_tests/test_ai_project_red_teams_operations.py @@ -34,9 +34,9 @@ def test_red_teams_list(self, aiproject_endpoint): @AIProjectPreparer() @recorded_by_proxy - def test_red_teams_create_run(self, aiproject_endpoint): + def test_red_teams_create(self, aiproject_endpoint): client = self.create_client(endpoint=aiproject_endpoint) - response = client.red_teams.create_run( + response = client.red_teams.create( red_team={ "attackStrategies": ["str"], "id": "str", diff --git a/sdk/ai/azure-ai-projects/generated_tests/test_ai_project_red_teams_operations_async.py b/sdk/ai/azure-ai-projects/generated_tests/test_ai_project_red_teams_operations_async.py index c65c7dc2a63a..6f17becd4537 100644 --- a/sdk/ai/azure-ai-projects/generated_tests/test_ai_project_red_teams_operations_async.py +++ b/sdk/ai/azure-ai-projects/generated_tests/test_ai_project_red_teams_operations_async.py @@ -35,9 +35,9 @@ async def test_red_teams_list(self, aiproject_endpoint): @AIProjectPreparer() @recorded_by_proxy_async - async def test_red_teams_create_run(self, aiproject_endpoint): + async def test_red_teams_create(self, aiproject_endpoint): client = self.create_async_client(endpoint=aiproject_endpoint) - response = await client.red_teams.create_run( + response = await client.red_teams.create( red_team={ "attackStrategies": ["str"], "id": "str", diff --git a/sdk/ai/azure-ai-projects/tests/samples/test_samples.py b/sdk/ai/azure-ai-projects/tests/samples/test_samples.py index d186464d31ce..3cab2b0861cb 100644 --- a/sdk/ai/azure-ai-projects/tests/samples/test_samples.py +++ b/sdk/ai/azure-ai-projects/tests/samples/test_samples.py @@ -1,3 +1,4 @@ +# pylint: disable=line-too-long,useless-suppression # ------------------------------------ # Copyright (c) Microsoft Corporation. # Licensed under the MIT License. diff --git a/sdk/ai/azure-ai-projects/tsp-location.yaml b/sdk/ai/azure-ai-projects/tsp-location.yaml index 29abf931aaf0..46ad7536ab04 100644 --- a/sdk/ai/azure-ai-projects/tsp-location.yaml +++ b/sdk/ai/azure-ai-projects/tsp-location.yaml @@ -1,4 +1,4 @@ directory: specification/ai/Azure.AI.Projects -commit: 3e0ddc7036c4d7bcb823dfb3398ad2e51b8e59dd +commit: 8de48d87b4a46f26917673d0457ef3662441d1a1 repo: Azure/azure-rest-api-specs additionalDirectories: