using latest version for auto rest (Azure#40787)

* using latest version for autorest
* fetching the component with correct name

---------

Co-authored-by: Copilot <[email protected]>

Author: achauhan-scc

sdk/ml/azure-ai-ml/azure/ai/ml/operations/_component_operations.py

@@ -2,50 +2,42 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # ---------------------------------------------------------
 
+import collections
+import hashlib
+
 # pylint: disable=protected-access,too-many-lines
 import time
-import collections
 import types
 from functools import partial
 from inspect import Parameter, signature
 from os import PathLike
 from pathlib import Path
 from typing import Any, Callable, Dict, Iterable, List, Optional, Tuple, Union, cast
-import hashlib
 
 from azure.ai.ml._restclient.v2021_10_01_dataplanepreview import (
     AzureMachineLearningWorkspaces as ServiceClient102021Dataplane,
 )
-from azure.ai.ml._restclient.v2024_01_01_preview import (
-    AzureMachineLearningWorkspaces as ServiceClient012024,
-)
-from azure.ai.ml._restclient.v2024_01_01_preview.models import (
-    ComponentVersion,
-    ListViewType,
-)
+from azure.ai.ml._restclient.v2024_01_01_preview import AzureMachineLearningWorkspaces as ServiceClient012024
+from azure.ai.ml._restclient.v2024_01_01_preview.models import ComponentVersion, ListViewType
 from azure.ai.ml._scope_dependent_operations import (
     OperationConfig,
     OperationsContainer,
     OperationScope,
     _ScopeDependentOperations,
 )
-from azure.ai.ml._telemetry import (
-    ActivityType,
-    monitor_with_activity,
-    monitor_with_telemetry_mixin,
-)
+from azure.ai.ml._telemetry import ActivityType, monitor_with_activity, monitor_with_telemetry_mixin
 from azure.ai.ml._utils._asset_utils import (
+    IgnoreFile,
     _archive_or_restore,
     _create_or_update_autoincrement,
     _get_file_hash,
     _get_latest,
     _get_next_version_from_container,
     _resolve_label_to_asset,
+    create_catalog_files,
+    delete_two_catalog_files,
     get_ignore_file,
     get_upload_files_from_folder,
-    IgnoreFile,
-    delete_two_catalog_files,
-    create_catalog_files,
 )
 from azure.ai.ml._utils._azureml_polling import AzureMLPolling
 from azure.ai.ml._utils._endpoint_utils import polling_wait
@@ -59,12 +51,7 @@
     LROConfigurations,
 )
 from azure.ai.ml.entities import Component, ValidationResult
-from azure.ai.ml.exceptions import (
-    ComponentException,
-    ErrorCategory,
-    ErrorTarget,
-    ValidationException,
-)
+from azure.ai.ml.exceptions import ComponentException, ErrorCategory, ErrorTarget, ValidationException
 from azure.core.exceptions import HttpResponseError, ResourceNotFoundError
 
 from .._utils._cache_utils import CachedNodeResolver
@@ -675,7 +662,7 @@ def create_or_update(
         )
 
         if not result:
-            component = self.get(name=component.name, version=component.version)
+            component = self.get(name=name, version=version)
         else:
             component = Component._from_rest_object(result)
 
@@ -915,9 +902,7 @@ def _resolve_binding_on_supported_fields_for_node(cls, node: BaseNode) -> None:
         :param node: The node
         :type node: BaseNode
         """
-        from azure.ai.ml.entities._job.pipeline._attr_dict import (
-            try_get_non_arbitrary_attr,
-        )
+        from azure.ai.ml.entities._job.pipeline._attr_dict import try_get_non_arbitrary_attr
         from azure.ai.ml.entities._job.pipeline._io import PipelineInput
 
         # compute binding to pipeline input is supported on node.

sdk/ml/azure-ai-ml/scripts/regenerate_restclient.py

@@ -1,3 +1,11 @@
+# coding: utf-8
+
+# -------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+# --------------------------------------------------------------------------
+
 import logging
 import os
 import subprocess
@@ -44,27 +52,27 @@ def _run_command(
 
     t0 = time.perf_counter()
     try:
-        logger.debug("Executing {0} in {1}".format(commands, cwd))
+        logger.debug("Executing %s in %s", commands, cwd)
         out = ""
-        p = subprocess.Popen(commands, stdout=subprocess.PIPE, stderr=stderr, cwd=cwd, shell=shell, env=env)
-        for line in p.stdout:
-            line = line.decode("utf-8").rstrip()
-            if line and line.strip():
-                logger.debug(line)
-                if stream_stdout:
-                    sys.stdout.write(line)
-                    sys.stdout.write("\n")
-                out += line
-                out += "\n"
-        p.communicate()
-        retcode = p.poll()
-        if throw_on_retcode:
-            if retcode:
-                raise subprocess.CalledProcessError(retcode, p.args, output=out, stderr=p.stderr)
+        with subprocess.Popen(commands, stdout=subprocess.PIPE, stderr=stderr, cwd=cwd, shell=shell, env=env) as p:
+            for line in p.stdout:
+                line = line.decode("utf-8").rstrip()
+                if line and line.strip():
+                    logger.debug(line)
+                    if stream_stdout:
+                        sys.stdout.write(line)
+                        sys.stdout.write("\n")
+                    out += line
+                    out += "\n"
+            p.communicate()
+            retcode = p.poll()
+            if throw_on_retcode:
+                if retcode:
+                    raise subprocess.CalledProcessError(retcode, p.args, output=out, stderr=p.stderr)
         return retcode, out
     finally:
         t1 = time.perf_counter()
-        logger.debug("Execution took {0}s for {1} in {2}".format(t1 - t0, commands, cwd))
+        logger.debug("Execution took %ss for %s in %s", t1 - t0, commands, cwd)
 
 
 def run_command(
@@ -90,13 +98,13 @@ def download_file(from_url: str, to_path: Path, with_file_name: str) -> None:
     print_blue(f"- Downloading {with_file_name} from {from_url} to {to_path}")
 
     try:
-        response = urlopen(from_url)
-    except Exception:
+        with urlopen(from_url) as response:
+            with open(f"{to_path}/{with_file_name}", "w", encoding="utf-8") as f:
+                f.write(response.read().decode("utf-8"))
+    except (OSError, URLError, HTTPError) as e:
         sys.exit(
-            f"Connection error while trying to download file from {from_url}. Please try running the script again."
+            f"Connection error while trying to download file from {from_url}: {e}. Please try running the script again."
         )
-    with open(f"{to_path}/{with_file_name}", "w") as f:
-        f.write(response.read().decode("utf-8"))
 
 
 def regenerate_restclient(api_tag, verbose):
@@ -115,7 +123,7 @@ def regenerate_restclient(api_tag, verbose):
         "--python",
         "--track2",
         "--version=3.6.2",
-        "--use=@autorest/python@5.12.6",
+        "--use=@autorest/python@latest",
         f"--python-sdks-folder={restclient_path.absolute()}",
         "--package-version=0.1.0",
         tag_arg,
@@ -138,8 +146,11 @@ def regenerate_restclient(api_tag, verbose):
         "-a",
         "--api-tag",
         required=False,
-        help="""Specifies which API to generate using autorest. If not supplied, all APIs are targeted.
-            Must match the name of a tag in the sdk/ml/azure-ai-ml/swagger/machinelearningservices/resource-manager/readme.md file.""",
+        help=(
+            "Specifies which API to generate using autorest. If not supplied, all APIs are targeted.\n"
+            "Must match the name of a tag in the sdk/ml/azure-ai-ml/swagger/machinelearningservices/"
+            "resource-manager/readme.md file."
+        ),
     )
     parser.add_argument("-v", "--verbose", action="store_true", required=False, help="turn on verbose output")
 

sdk/ml/azure-ai-ml/swagger/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/preview/2025-01-01-preview/mfe.json

@@ -4324,6 +4324,10 @@
                 "format": "int32",
                 "maximum": 600,
                 "minimum": 10
+              },
+              "Azure-AsyncOperation": {
+                "description": "URI to poll for asynchronous operation status.",
+                "type": "string"
               }
             }
           },
@@ -4344,7 +4348,7 @@
         },
         "x-ms-long-running-operation": true,
         "x-ms-long-running-operation-options": {
-          "final-state-via": "location"
+          "final-state-via": "azure-async-operation"
         }
       },
       "get": {