@azure/arm-machinelearning 2.1.2-alpha.20240522.1 → 2.2.0-alpha.20240523.2

package/src/operations/batchDeployments.ts

@@ -13,8 +13,12 @@ import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
 import * as Parameters from "../models/parameters";
 import { AzureMachineLearningWorkspaces } from "../azureMachineLearningWorkspaces";
-import { PollerLike, PollOperationState, LroEngine } from "@azure/core-lro";
-import { LroImpl } from "../lroImpl";
+import {
+  SimplePollerLike,
+  OperationState,
+  createHttpPoller,
+} from "@azure/core-lro";
+import { createLroSpec } from "../lroImpl";
 import {
   BatchDeployment,
   BatchDeploymentsListNextOptionalParams,
@@ -28,7 +32,7 @@ import {
   BatchDeploymentsUpdateResponse,
   BatchDeploymentsCreateOrUpdateOptionalParams,
   BatchDeploymentsCreateOrUpdateResponse,
-  BatchDeploymentsListNextResponse
+  BatchDeploymentsListNextResponse,
 } from "../models";
 
 /// <reference lib="esnext.asynciterable" />
@@ -55,13 +59,13 @@ export class BatchDeploymentsImpl implements BatchDeployments {
     resourceGroupName: string,
     workspaceName: string,
     endpointName: string,
-    options?: BatchDeploymentsListOptionalParams
+    options?: BatchDeploymentsListOptionalParams,
   ): PagedAsyncIterableIterator<BatchDeployment> {
     const iter = this.listPagingAll(
       resourceGroupName,
       workspaceName,
       endpointName,
-      options
+      options,
     );
     return {
       next() {
@@ -79,9 +83,9 @@ export class BatchDeploymentsImpl implements BatchDeployments {
           workspaceName,
           endpointName,
           options,
-          settings
+          settings,
         );
-      }
+      },
     };
   }
 
@@ -90,7 +94,7 @@ export class BatchDeploymentsImpl implements BatchDeployments {
     workspaceName: string,
     endpointName: string,
     options?: BatchDeploymentsListOptionalParams,
-    settings?: PageSettings
+    settings?: PageSettings,
   ): AsyncIterableIterator<BatchDeployment[]> {
     let result: BatchDeploymentsListResponse;
     let continuationToken = settings?.continuationToken;
@@ -99,7 +103,7 @@ export class BatchDeploymentsImpl implements BatchDeployments {
         resourceGroupName,
         workspaceName,
         endpointName,
-        options
+        options,
       );
       let page = result.value || [];
       continuationToken = result.nextLink;
@@ -112,7 +116,7 @@ export class BatchDeploymentsImpl implements BatchDeployments {
         workspaceName,
         endpointName,
         continuationToken,
-        options
+        options,
       );
       continuationToken = result.nextLink;
       let page = result.value || [];
@@ -125,13 +129,13 @@ export class BatchDeploymentsImpl implements BatchDeployments {
     resourceGroupName: string,
     workspaceName: string,
     endpointName: string,
-    options?: BatchDeploymentsListOptionalParams
+    options?: BatchDeploymentsListOptionalParams,
   ): AsyncIterableIterator<BatchDeployment> {
     for await (const page of this.listPagingPage(
       resourceGroupName,
       workspaceName,
       endpointName,
-      options
+      options,
     )) {
       yield* page;
     }
@@ -148,11 +152,11 @@ export class BatchDeploymentsImpl implements BatchDeployments {
     resourceGroupName: string,
     workspaceName: string,
     endpointName: string,
-    options?: BatchDeploymentsListOptionalParams
+    options?: BatchDeploymentsListOptionalParams,
   ): Promise<BatchDeploymentsListResponse> {
     return this.client.sendOperationRequest(
       { resourceGroupName, workspaceName, endpointName, options },
-      listOperationSpec
+      listOperationSpec,
     );
   }
 
@@ -169,25 +173,24 @@ export class BatchDeploymentsImpl implements BatchDeployments {
     workspaceName: string,
     endpointName: string,
     deploymentName: string,
-    options?: BatchDeploymentsDeleteOptionalParams
-  ): Promise<PollerLike<PollOperationState<void>, void>> {
+    options?: BatchDeploymentsDeleteOptionalParams,
+  ): Promise<SimplePollerLike<OperationState<void>, void>> {
     const directSendOperation = async (
       args: coreClient.OperationArguments,
-      spec: coreClient.OperationSpec
+      spec: coreClient.OperationSpec,
     ): Promise<void> => {
       return this.client.sendOperationRequest(args, spec);
     };
-    const sendOperation = async (
+    const sendOperationFn = async (
       args: coreClient.OperationArguments,
-      spec: coreClient.OperationSpec
+      spec: coreClient.OperationSpec,
     ) => {
-      let currentRawResponse:
-        | coreClient.FullOperationResponse
-        | undefined = undefined;
+      let currentRawResponse: coreClient.FullOperationResponse | undefined =
+        undefined;
       const providedCallback = args.options?.onResponse;
       const callback: coreClient.RawResponseCallback = (
         rawResponse: coreClient.FullOperationResponse,
-        flatResponse: unknown
+        flatResponse: unknown,
       ) => {
         currentRawResponse = rawResponse;
         providedCallback?.(rawResponse, flatResponse);
@@ -196,8 +199,8 @@ export class BatchDeploymentsImpl implements BatchDeployments {
         ...args,
         options: {
           ...args.options,
-          onResponse: callback
-        }
+          onResponse: callback,
+        },
       };
       const flatResponse = await directSendOperation(updatedArgs, spec);
       return {
@@ -205,25 +208,25 @@ export class BatchDeploymentsImpl implements BatchDeployments {
         rawResponse: {
           statusCode: currentRawResponse!.status,
           body: currentRawResponse!.parsedBody,
-          headers: currentRawResponse!.headers.toJSON()
-        }
+          headers: currentRawResponse!.headers.toJSON(),
+        },
       };
     };
 
-    const lro = new LroImpl(
-      sendOperation,
-      {
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: {
         resourceGroupName,
         workspaceName,
         endpointName,
         deploymentName,
-        options
+        options,
       },
-      deleteOperationSpec
-    );
-    const poller = new LroEngine(lro, {
-      resumeFrom: options?.resumeFrom,
-      intervalInMs: options?.updateIntervalInMs
+      spec: deleteOperationSpec,
+    });
+    const poller = await createHttpPoller<void, OperationState<void>>(lro, {
+      restoreFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
     });
     await poller.poll();
     return poller;
@@ -242,14 +245,14 @@ export class BatchDeploymentsImpl implements BatchDeployments {
     workspaceName: string,
     endpointName: string,
     deploymentName: string,
-    options?: BatchDeploymentsDeleteOptionalParams
+    options?: BatchDeploymentsDeleteOptionalParams,
   ): Promise<void> {
     const poller = await this.beginDelete(
       resourceGroupName,
       workspaceName,
       endpointName,
       deploymentName,
-      options
+      options,
     );
     return poller.pollUntilDone();
   }
@@ -267,7 +270,7 @@ export class BatchDeploymentsImpl implements BatchDeployments {
     workspaceName: string,
     endpointName: string,
     deploymentName: string,
-    options?: BatchDeploymentsGetOptionalParams
+    options?: BatchDeploymentsGetOptionalParams,
   ): Promise<BatchDeploymentsGetResponse> {
     return this.client.sendOperationRequest(
       {
@@ -275,9 +278,9 @@ export class BatchDeploymentsImpl implements BatchDeployments {
         workspaceName,
         endpointName,
         deploymentName,
-        options
+        options,
       },
-      getOperationSpec
+      getOperationSpec,
     );
   }
 
@@ -296,30 +299,29 @@ export class BatchDeploymentsImpl implements BatchDeployments {
     endpointName: string,
     deploymentName: string,
     body: PartialBatchDeploymentPartialMinimalTrackedResourceWithProperties,
-    options?: BatchDeploymentsUpdateOptionalParams
+    options?: BatchDeploymentsUpdateOptionalParams,
   ): Promise<
-    PollerLike<
-      PollOperationState<BatchDeploymentsUpdateResponse>,
+    SimplePollerLike<
+      OperationState<BatchDeploymentsUpdateResponse>,
       BatchDeploymentsUpdateResponse
     >
   > {
     const directSendOperation = async (
       args: coreClient.OperationArguments,
-      spec: coreClient.OperationSpec
+      spec: coreClient.OperationSpec,
     ): Promise<BatchDeploymentsUpdateResponse> => {
       return this.client.sendOperationRequest(args, spec);
     };
-    const sendOperation = async (
+    const sendOperationFn = async (
       args: coreClient.OperationArguments,
-      spec: coreClient.OperationSpec
+      spec: coreClient.OperationSpec,
     ) => {
-      let currentRawResponse:
-        | coreClient.FullOperationResponse
-        | undefined = undefined;
+      let currentRawResponse: coreClient.FullOperationResponse | undefined =
+        undefined;
       const providedCallback = args.options?.onResponse;
       const callback: coreClient.RawResponseCallback = (
         rawResponse: coreClient.FullOperationResponse,
-        flatResponse: unknown
+        flatResponse: unknown,
       ) => {
         currentRawResponse = rawResponse;
         providedCallback?.(rawResponse, flatResponse);
@@ -328,8 +330,8 @@ export class BatchDeploymentsImpl implements BatchDeployments {
         ...args,
         options: {
           ...args.options,
-          onResponse: callback
-        }
+          onResponse: callback,
+        },
       };
       const flatResponse = await directSendOperation(updatedArgs, spec);
       return {
@@ -337,26 +339,29 @@ export class BatchDeploymentsImpl implements BatchDeployments {
         rawResponse: {
           statusCode: currentRawResponse!.status,
           body: currentRawResponse!.parsedBody,
-          headers: currentRawResponse!.headers.toJSON()
-        }
+          headers: currentRawResponse!.headers.toJSON(),
+        },
       };
     };
 
-    const lro = new LroImpl(
-      sendOperation,
-      {
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: {
         resourceGroupName,
         workspaceName,
         endpointName,
         deploymentName,
         body,
-        options
+        options,
       },
-      updateOperationSpec
-    );
-    const poller = new LroEngine(lro, {
-      resumeFrom: options?.resumeFrom,
-      intervalInMs: options?.updateIntervalInMs
+      spec: updateOperationSpec,
+    });
+    const poller = await createHttpPoller<
+      BatchDeploymentsUpdateResponse,
+      OperationState<BatchDeploymentsUpdateResponse>
+    >(lro, {
+      restoreFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
     });
     await poller.poll();
     return poller;
@@ -377,7 +382,7 @@ export class BatchDeploymentsImpl implements BatchDeployments {
     endpointName: string,
     deploymentName: string,
     body: PartialBatchDeploymentPartialMinimalTrackedResourceWithProperties,
-    options?: BatchDeploymentsUpdateOptionalParams
+    options?: BatchDeploymentsUpdateOptionalParams,
   ): Promise<BatchDeploymentsUpdateResponse> {
     const poller = await this.beginUpdate(
       resourceGroupName,
@@ -385,7 +390,7 @@ export class BatchDeploymentsImpl implements BatchDeployments {
       endpointName,
       deploymentName,
       body,
-      options
+      options,
     );
     return poller.pollUntilDone();
   }
@@ -405,30 +410,29 @@ export class BatchDeploymentsImpl implements BatchDeployments {
     endpointName: string,
     deploymentName: string,
     body: BatchDeployment,
-    options?: BatchDeploymentsCreateOrUpdateOptionalParams
+    options?: BatchDeploymentsCreateOrUpdateOptionalParams,
   ): Promise<
-    PollerLike<
-      PollOperationState<BatchDeploymentsCreateOrUpdateResponse>,
+    SimplePollerLike<
+      OperationState<BatchDeploymentsCreateOrUpdateResponse>,
       BatchDeploymentsCreateOrUpdateResponse
     >
   > {
     const directSendOperation = async (
       args: coreClient.OperationArguments,
-      spec: coreClient.OperationSpec
+      spec: coreClient.OperationSpec,
     ): Promise<BatchDeploymentsCreateOrUpdateResponse> => {
       return this.client.sendOperationRequest(args, spec);
     };
-    const sendOperation = async (
+    const sendOperationFn = async (
       args: coreClient.OperationArguments,
-      spec: coreClient.OperationSpec
+      spec: coreClient.OperationSpec,
     ) => {
-      let currentRawResponse:
-        | coreClient.FullOperationResponse
-        | undefined = undefined;
+      let currentRawResponse: coreClient.FullOperationResponse | undefined =
+        undefined;
       const providedCallback = args.options?.onResponse;
       const callback: coreClient.RawResponseCallback = (
         rawResponse: coreClient.FullOperationResponse,
-        flatResponse: unknown
+        flatResponse: unknown,
       ) => {
         currentRawResponse = rawResponse;
         providedCallback?.(rawResponse, flatResponse);
@@ -437,8 +441,8 @@ export class BatchDeploymentsImpl implements BatchDeployments {
         ...args,
         options: {
           ...args.options,
-          onResponse: callback
-        }
+          onResponse: callback,
+        },
       };
       const flatResponse = await directSendOperation(updatedArgs, spec);
       return {
@@ -446,26 +450,29 @@ export class BatchDeploymentsImpl implements BatchDeployments {
         rawResponse: {
           statusCode: currentRawResponse!.status,
           body: currentRawResponse!.parsedBody,
-          headers: currentRawResponse!.headers.toJSON()
-        }
+          headers: currentRawResponse!.headers.toJSON(),
+        },
       };
     };
 
-    const lro = new LroImpl(
-      sendOperation,
-      {
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: {
         resourceGroupName,
         workspaceName,
         endpointName,
         deploymentName,
         body,
-        options
+        options,
       },
-      createOrUpdateOperationSpec
-    );
-    const poller = new LroEngine(lro, {
-      resumeFrom: options?.resumeFrom,
-      intervalInMs: options?.updateIntervalInMs
+      spec: createOrUpdateOperationSpec,
+    });
+    const poller = await createHttpPoller<
+      BatchDeploymentsCreateOrUpdateResponse,
+      OperationState<BatchDeploymentsCreateOrUpdateResponse>
+    >(lro, {
+      restoreFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
     });
     await poller.poll();
     return poller;
@@ -486,7 +493,7 @@ export class BatchDeploymentsImpl implements BatchDeployments {
     endpointName: string,
     deploymentName: string,
     body: BatchDeployment,
-    options?: BatchDeploymentsCreateOrUpdateOptionalParams
+    options?: BatchDeploymentsCreateOrUpdateOptionalParams,
   ): Promise<BatchDeploymentsCreateOrUpdateResponse> {
     const poller = await this.beginCreateOrUpdate(
       resourceGroupName,
@@ -494,7 +501,7 @@ export class BatchDeploymentsImpl implements BatchDeployments {
       endpointName,
       deploymentName,
       body,
-      options
+      options,
     );
     return poller.pollUntilDone();
   }
@@ -512,11 +519,11 @@ export class BatchDeploymentsImpl implements BatchDeployments {
     workspaceName: string,
     endpointName: string,
     nextLink: string,
-    options?: BatchDeploymentsListNextOptionalParams
+    options?: BatchDeploymentsListNextOptionalParams,
   ): Promise<BatchDeploymentsListNextResponse> {
     return this.client.sendOperationRequest(
       { resourceGroupName, workspaceName, endpointName, nextLink, options },
-      listNextOperationSpec
+      listNextOperationSpec,
     );
   }
 }
@@ -524,36 +531,34 @@ export class BatchDeploymentsImpl implements BatchDeployments {
 const serializer = coreClient.createSerializer(Mappers, /* isXml */ false);
 
 const listOperationSpec: coreClient.OperationSpec = {
-  path:
-    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/batchEndpoints/{endpointName}/deployments",
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/batchEndpoints/{endpointName}/deployments",
   httpMethod: "GET",
   responses: {
     200: {
-      bodyMapper: Mappers.BatchDeploymentTrackedResourceArmPaginatedResult
+      bodyMapper: Mappers.BatchDeploymentTrackedResourceArmPaginatedResult,
     },
     default: {
-      bodyMapper: Mappers.ErrorResponse
-    }
+      bodyMapper: Mappers.ErrorResponse,
+    },
   },
   queryParameters: [
     Parameters.apiVersion,
     Parameters.skip,
     Parameters.orderBy,
-    Parameters.top
+    Parameters.top,
   ],
   urlParameters: [
     Parameters.$host,
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.workspaceName,
-    Parameters.endpointName
+    Parameters.endpointName,
   ],
   headerParameters: [Parameters.accept],
-  serializer
+  serializer,
 };
 const deleteOperationSpec: coreClient.OperationSpec = {
-  path:
-    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/batchEndpoints/{endpointName}/deployments/{deploymentName}",
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/batchEndpoints/{endpointName}/deployments/{deploymentName}",
   httpMethod: "DELETE",
   responses: {
     200: {},
@@ -561,8 +566,8 @@ const deleteOperationSpec: coreClient.OperationSpec = {
     202: {},
     204: {},
     default: {
-      bodyMapper: Mappers.ErrorResponse
-    }
+      bodyMapper: Mappers.ErrorResponse,
+    },
   },
   queryParameters: [Parameters.apiVersion],
   urlParameters: [
@@ -571,22 +576,21 @@ const deleteOperationSpec: coreClient.OperationSpec = {
     Parameters.resourceGroupName,
     Parameters.workspaceName,
     Parameters.endpointName,
-    Parameters.deploymentName
+    Parameters.deploymentName,
   ],
   headerParameters: [Parameters.accept],
-  serializer
+  serializer,
 };
 const getOperationSpec: coreClient.OperationSpec = {
-  path:
-    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/batchEndpoints/{endpointName}/deployments/{deploymentName}",
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/batchEndpoints/{endpointName}/deployments/{deploymentName}",
   httpMethod: "GET",
   responses: {
     200: {
-      bodyMapper: Mappers.BatchDeployment
+      bodyMapper: Mappers.BatchDeployment,
     },
     default: {
-      bodyMapper: Mappers.ErrorResponse
-    }
+      bodyMapper: Mappers.ErrorResponse,
+    },
   },
   queryParameters: [Parameters.apiVersion],
   urlParameters: [
@@ -595,31 +599,30 @@ const getOperationSpec: coreClient.OperationSpec = {
     Parameters.resourceGroupName,
     Parameters.workspaceName,
     Parameters.endpointName,
-    Parameters.deploymentName
+    Parameters.deploymentName,
   ],
   headerParameters: [Parameters.accept],
-  serializer
+  serializer,
 };
 const updateOperationSpec: coreClient.OperationSpec = {
-  path:
-    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/batchEndpoints/{endpointName}/deployments/{deploymentName}",
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/batchEndpoints/{endpointName}/deployments/{deploymentName}",
   httpMethod: "PATCH",
   responses: {
     200: {
-      bodyMapper: Mappers.BatchDeployment
+      bodyMapper: Mappers.BatchDeployment,
     },
     201: {
-      bodyMapper: Mappers.BatchDeployment
+      bodyMapper: Mappers.BatchDeployment,
     },
     202: {
-      bodyMapper: Mappers.BatchDeployment
+      bodyMapper: Mappers.BatchDeployment,
     },
     204: {
-      bodyMapper: Mappers.BatchDeployment
+      bodyMapper: Mappers.BatchDeployment,
     },
     default: {
-      bodyMapper: Mappers.ErrorResponse
-    }
+      bodyMapper: Mappers.ErrorResponse,
+    },
   },
   requestBody: Parameters.body2,
   queryParameters: [Parameters.apiVersion],
@@ -629,32 +632,31 @@ const updateOperationSpec: coreClient.OperationSpec = {
     Parameters.resourceGroupName,
     Parameters.workspaceName,
     Parameters.endpointName1,
-    Parameters.deploymentName1
+    Parameters.deploymentName1,
   ],
   headerParameters: [Parameters.accept, Parameters.contentType],
   mediaType: "json",
-  serializer
+  serializer,
 };
 const createOrUpdateOperationSpec: coreClient.OperationSpec = {
-  path:
-    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/batchEndpoints/{endpointName}/deployments/{deploymentName}",
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/batchEndpoints/{endpointName}/deployments/{deploymentName}",
   httpMethod: "PUT",
   responses: {
     200: {
-      bodyMapper: Mappers.BatchDeployment
+      bodyMapper: Mappers.BatchDeployment,
     },
     201: {
-      bodyMapper: Mappers.BatchDeployment
+      bodyMapper: Mappers.BatchDeployment,
     },
     202: {
-      bodyMapper: Mappers.BatchDeployment
+      bodyMapper: Mappers.BatchDeployment,
     },
     204: {
-      bodyMapper: Mappers.BatchDeployment
+      bodyMapper: Mappers.BatchDeployment,
     },
     default: {
-      bodyMapper: Mappers.ErrorResponse
-    }
+      bodyMapper: Mappers.ErrorResponse,
+    },
   },
   requestBody: Parameters.body3,
   queryParameters: [Parameters.apiVersion],
@@ -664,37 +666,31 @@ const createOrUpdateOperationSpec: coreClient.OperationSpec = {
     Parameters.resourceGroupName,
     Parameters.workspaceName,
     Parameters.endpointName1,
-    Parameters.deploymentName1
+    Parameters.deploymentName1,
   ],
   headerParameters: [Parameters.accept, Parameters.contentType],
   mediaType: "json",
-  serializer
+  serializer,
 };
 const listNextOperationSpec: coreClient.OperationSpec = {
   path: "{nextLink}",
   httpMethod: "GET",
   responses: {
     200: {
-      bodyMapper: Mappers.BatchDeploymentTrackedResourceArmPaginatedResult
+      bodyMapper: Mappers.BatchDeploymentTrackedResourceArmPaginatedResult,
     },
     default: {
-      bodyMapper: Mappers.ErrorResponse
-    }
+      bodyMapper: Mappers.ErrorResponse,
+    },
   },
-  queryParameters: [
-    Parameters.apiVersion,
-    Parameters.skip,
-    Parameters.orderBy,
-    Parameters.top
-  ],
   urlParameters: [
     Parameters.$host,
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.workspaceName,
     Parameters.nextLink,
-    Parameters.endpointName
+    Parameters.endpointName,
   ],
   headerParameters: [Parameters.accept],
-  serializer
+  serializer,
 };