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

package/src/operations/computeOperations.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 {
   ComputeResource,
   ComputeListNextOptionalParams,
@@ -39,7 +43,7 @@ import {
   ComputeStopOptionalParams,
   ComputeRestartOptionalParams,
   ComputeListNextResponse,
-  ComputeListNodesNextResponse
+  ComputeListNodesNextResponse,
 } from "../models";
 
 /// <reference lib="esnext.asynciterable" />
@@ -64,7 +68,7 @@ export class ComputeOperationsImpl implements ComputeOperations {
   public list(
     resourceGroupName: string,
     workspaceName: string,
-    options?: ComputeListOptionalParams
+    options?: ComputeListOptionalParams,
   ): PagedAsyncIterableIterator<ComputeResource> {
     const iter = this.listPagingAll(resourceGroupName, workspaceName, options);
     return {
@@ -82,9 +86,9 @@ export class ComputeOperationsImpl implements ComputeOperations {
           resourceGroupName,
           workspaceName,
           options,
-          settings
+          settings,
         );
-      }
+      },
     };
   }
 
@@ -92,7 +96,7 @@ export class ComputeOperationsImpl implements ComputeOperations {
     resourceGroupName: string,
     workspaceName: string,
     options?: ComputeListOptionalParams,
-    settings?: PageSettings
+    settings?: PageSettings,
   ): AsyncIterableIterator<ComputeResource[]> {
     let result: ComputeListResponse;
     let continuationToken = settings?.continuationToken;
@@ -108,7 +112,7 @@ export class ComputeOperationsImpl implements ComputeOperations {
         resourceGroupName,
         workspaceName,
         continuationToken,
-        options
+        options,
       );
       continuationToken = result.nextLink;
       let page = result.value || [];
@@ -120,12 +124,12 @@ export class ComputeOperationsImpl implements ComputeOperations {
   private async *listPagingAll(
     resourceGroupName: string,
     workspaceName: string,
-    options?: ComputeListOptionalParams
+    options?: ComputeListOptionalParams,
   ): AsyncIterableIterator<ComputeResource> {
     for await (const page of this.listPagingPage(
       resourceGroupName,
       workspaceName,
-      options
+      options,
     )) {
       yield* page;
     }
@@ -142,13 +146,13 @@ export class ComputeOperationsImpl implements ComputeOperations {
     resourceGroupName: string,
     workspaceName: string,
     computeName: string,
-    options?: ComputeListNodesOptionalParams
+    options?: ComputeListNodesOptionalParams,
   ): PagedAsyncIterableIterator<AmlComputeNodeInformation> {
     const iter = this.listNodesPagingAll(
       resourceGroupName,
       workspaceName,
       computeName,
-      options
+      options,
     );
     return {
       next() {
@@ -166,9 +170,9 @@ export class ComputeOperationsImpl implements ComputeOperations {
           workspaceName,
           computeName,
           options,
-          settings
+          settings,
         );
-      }
+      },
     };
   }
 
@@ -177,7 +181,7 @@ export class ComputeOperationsImpl implements ComputeOperations {
     workspaceName: string,
     computeName: string,
     options?: ComputeListNodesOptionalParams,
-    settings?: PageSettings
+    settings?: PageSettings,
   ): AsyncIterableIterator<AmlComputeNodeInformation[]> {
     let result: ComputeListNodesResponse;
     let continuationToken = settings?.continuationToken;
@@ -186,7 +190,7 @@ export class ComputeOperationsImpl implements ComputeOperations {
         resourceGroupName,
         workspaceName,
         computeName,
-        options
+        options,
       );
       let page = result.nodes || [];
       continuationToken = result.nextLink;
@@ -199,7 +203,7 @@ export class ComputeOperationsImpl implements ComputeOperations {
         workspaceName,
         computeName,
         continuationToken,
-        options
+        options,
       );
       continuationToken = result.nextLink;
       let page = result.nodes || [];
@@ -212,13 +216,13 @@ export class ComputeOperationsImpl implements ComputeOperations {
     resourceGroupName: string,
     workspaceName: string,
     computeName: string,
-    options?: ComputeListNodesOptionalParams
+    options?: ComputeListNodesOptionalParams,
   ): AsyncIterableIterator<AmlComputeNodeInformation> {
     for await (const page of this.listNodesPagingPage(
       resourceGroupName,
       workspaceName,
       computeName,
-      options
+      options,
     )) {
       yield* page;
     }
@@ -233,11 +237,11 @@ export class ComputeOperationsImpl implements ComputeOperations {
   private _list(
     resourceGroupName: string,
     workspaceName: string,
-    options?: ComputeListOptionalParams
+    options?: ComputeListOptionalParams,
   ): Promise<ComputeListResponse> {
     return this.client.sendOperationRequest(
       { resourceGroupName, workspaceName, options },
-      listOperationSpec
+      listOperationSpec,
     );
   }
 
@@ -253,11 +257,11 @@ export class ComputeOperationsImpl implements ComputeOperations {
     resourceGroupName: string,
     workspaceName: string,
     computeName: string,
-    options?: ComputeGetOptionalParams
+    options?: ComputeGetOptionalParams,
   ): Promise<ComputeGetResponse> {
     return this.client.sendOperationRequest(
       { resourceGroupName, workspaceName, computeName, options },
-      getOperationSpec
+      getOperationSpec,
     );
   }
 
@@ -276,30 +280,29 @@ export class ComputeOperationsImpl implements ComputeOperations {
     workspaceName: string,
     computeName: string,
     parameters: ComputeResource,
-    options?: ComputeCreateOrUpdateOptionalParams
+    options?: ComputeCreateOrUpdateOptionalParams,
   ): Promise<
-    PollerLike<
-      PollOperationState<ComputeCreateOrUpdateResponse>,
+    SimplePollerLike<
+      OperationState<ComputeCreateOrUpdateResponse>,
       ComputeCreateOrUpdateResponse
     >
   > {
     const directSendOperation = async (
       args: coreClient.OperationArguments,
-      spec: coreClient.OperationSpec
+      spec: coreClient.OperationSpec,
     ): Promise<ComputeCreateOrUpdateResponse> => {
       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);
@@ -308,8 +311,8 @@ export class ComputeOperationsImpl implements ComputeOperations {
         ...args,
         options: {
           ...args.options,
-          onResponse: callback
-        }
+          onResponse: callback,
+        },
       };
       const flatResponse = await directSendOperation(updatedArgs, spec);
       return {
@@ -317,19 +320,28 @@ export class ComputeOperationsImpl implements ComputeOperations {
         rawResponse: {
           statusCode: currentRawResponse!.status,
           body: currentRawResponse!.parsedBody,
-          headers: currentRawResponse!.headers.toJSON()
-        }
+          headers: currentRawResponse!.headers.toJSON(),
+        },
       };
     };
 
-    const lro = new LroImpl(
-      sendOperation,
-      { resourceGroupName, workspaceName, computeName, parameters, options },
-      createOrUpdateOperationSpec
-    );
-    const poller = new LroEngine(lro, {
-      resumeFrom: options?.resumeFrom,
-      intervalInMs: options?.updateIntervalInMs
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: {
+        resourceGroupName,
+        workspaceName,
+        computeName,
+        parameters,
+        options,
+      },
+      spec: createOrUpdateOperationSpec,
+    });
+    const poller = await createHttpPoller<
+      ComputeCreateOrUpdateResponse,
+      OperationState<ComputeCreateOrUpdateResponse>
+    >(lro, {
+      restoreFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
     });
     await poller.poll();
     return poller;
@@ -350,14 +362,14 @@ export class ComputeOperationsImpl implements ComputeOperations {
     workspaceName: string,
     computeName: string,
     parameters: ComputeResource,
-    options?: ComputeCreateOrUpdateOptionalParams
+    options?: ComputeCreateOrUpdateOptionalParams,
   ): Promise<ComputeCreateOrUpdateResponse> {
     const poller = await this.beginCreateOrUpdate(
       resourceGroupName,
       workspaceName,
       computeName,
       parameters,
-      options
+      options,
     );
     return poller.pollUntilDone();
   }
@@ -376,27 +388,29 @@ export class ComputeOperationsImpl implements ComputeOperations {
     workspaceName: string,
     computeName: string,
     parameters: ClusterUpdateParameters,
-    options?: ComputeUpdateOptionalParams
+    options?: ComputeUpdateOptionalParams,
   ): Promise<
-    PollerLike<PollOperationState<ComputeUpdateResponse>, ComputeUpdateResponse>
+    SimplePollerLike<
+      OperationState<ComputeUpdateResponse>,
+      ComputeUpdateResponse
+    >
   > {
     const directSendOperation = async (
       args: coreClient.OperationArguments,
-      spec: coreClient.OperationSpec
+      spec: coreClient.OperationSpec,
     ): Promise<ComputeUpdateResponse> => {
       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);
@@ -405,8 +419,8 @@ export class ComputeOperationsImpl implements ComputeOperations {
         ...args,
         options: {
           ...args.options,
-          onResponse: callback
-        }
+          onResponse: callback,
+        },
       };
       const flatResponse = await directSendOperation(updatedArgs, spec);
       return {
@@ -414,19 +428,28 @@ export class ComputeOperationsImpl implements ComputeOperations {
         rawResponse: {
           statusCode: currentRawResponse!.status,
           body: currentRawResponse!.parsedBody,
-          headers: currentRawResponse!.headers.toJSON()
-        }
+          headers: currentRawResponse!.headers.toJSON(),
+        },
       };
     };
 
-    const lro = new LroImpl(
-      sendOperation,
-      { resourceGroupName, workspaceName, computeName, parameters, options },
-      updateOperationSpec
-    );
-    const poller = new LroEngine(lro, {
-      resumeFrom: options?.resumeFrom,
-      intervalInMs: options?.updateIntervalInMs
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: {
+        resourceGroupName,
+        workspaceName,
+        computeName,
+        parameters,
+        options,
+      },
+      spec: updateOperationSpec,
+    });
+    const poller = await createHttpPoller<
+      ComputeUpdateResponse,
+      OperationState<ComputeUpdateResponse>
+    >(lro, {
+      restoreFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
     });
     await poller.poll();
     return poller;
@@ -446,14 +469,14 @@ export class ComputeOperationsImpl implements ComputeOperations {
     workspaceName: string,
     computeName: string,
     parameters: ClusterUpdateParameters,
-    options?: ComputeUpdateOptionalParams
+    options?: ComputeUpdateOptionalParams,
   ): Promise<ComputeUpdateResponse> {
     const poller = await this.beginUpdate(
       resourceGroupName,
       workspaceName,
       computeName,
       parameters,
-      options
+      options,
     );
     return poller.pollUntilDone();
   }
@@ -472,25 +495,24 @@ export class ComputeOperationsImpl implements ComputeOperations {
     workspaceName: string,
     computeName: string,
     underlyingResourceAction: UnderlyingResourceAction,
-    options?: ComputeDeleteOptionalParams
-  ): Promise<PollerLike<PollOperationState<void>, void>> {
+    options?: ComputeDeleteOptionalParams,
+  ): 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);
@@ -499,8 +521,8 @@ export class ComputeOperationsImpl implements ComputeOperations {
         ...args,
         options: {
           ...args.options,
-          onResponse: callback
-        }
+          onResponse: callback,
+        },
       };
       const flatResponse = await directSendOperation(updatedArgs, spec);
       return {
@@ -508,25 +530,25 @@ export class ComputeOperationsImpl implements ComputeOperations {
         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,
         computeName,
         underlyingResourceAction,
-        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;
@@ -546,14 +568,14 @@ export class ComputeOperationsImpl implements ComputeOperations {
     workspaceName: string,
     computeName: string,
     underlyingResourceAction: UnderlyingResourceAction,
-    options?: ComputeDeleteOptionalParams
+    options?: ComputeDeleteOptionalParams,
   ): Promise<void> {
     const poller = await this.beginDelete(
       resourceGroupName,
       workspaceName,
       computeName,
       underlyingResourceAction,
-      options
+      options,
     );
     return poller.pollUntilDone();
   }
@@ -569,11 +591,11 @@ export class ComputeOperationsImpl implements ComputeOperations {
     resourceGroupName: string,
     workspaceName: string,
     computeName: string,
-    options?: ComputeListNodesOptionalParams
+    options?: ComputeListNodesOptionalParams,
   ): Promise<ComputeListNodesResponse> {
     return this.client.sendOperationRequest(
       { resourceGroupName, workspaceName, computeName, options },
-      listNodesOperationSpec
+      listNodesOperationSpec,
     );
   }
 
@@ -588,11 +610,11 @@ export class ComputeOperationsImpl implements ComputeOperations {
     resourceGroupName: string,
     workspaceName: string,
     computeName: string,
-    options?: ComputeListKeysOptionalParams
+    options?: ComputeListKeysOptionalParams,
   ): Promise<ComputeListKeysResponse> {
     return this.client.sendOperationRequest(
       { resourceGroupName, workspaceName, computeName, options },
-      listKeysOperationSpec
+      listKeysOperationSpec,
     );
   }
 
@@ -607,25 +629,24 @@ export class ComputeOperationsImpl implements ComputeOperations {
     resourceGroupName: string,
     workspaceName: string,
     computeName: string,
-    options?: ComputeStartOptionalParams
-  ): Promise<PollerLike<PollOperationState<void>, void>> {
+    options?: ComputeStartOptionalParams,
+  ): 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);
@@ -634,8 +655,8 @@ export class ComputeOperationsImpl implements ComputeOperations {
         ...args,
         options: {
           ...args.options,
-          onResponse: callback
-        }
+          onResponse: callback,
+        },
       };
       const flatResponse = await directSendOperation(updatedArgs, spec);
       return {
@@ -643,19 +664,19 @@ export class ComputeOperationsImpl implements ComputeOperations {
         rawResponse: {
           statusCode: currentRawResponse!.status,
           body: currentRawResponse!.parsedBody,
-          headers: currentRawResponse!.headers.toJSON()
-        }
+          headers: currentRawResponse!.headers.toJSON(),
+        },
       };
     };
 
-    const lro = new LroImpl(
-      sendOperation,
-      { resourceGroupName, workspaceName, computeName, options },
-      startOperationSpec
-    );
-    const poller = new LroEngine(lro, {
-      resumeFrom: options?.resumeFrom,
-      intervalInMs: options?.updateIntervalInMs
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: { resourceGroupName, workspaceName, computeName, options },
+      spec: startOperationSpec,
+    });
+    const poller = await createHttpPoller<void, OperationState<void>>(lro, {
+      restoreFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
     });
     await poller.poll();
     return poller;
@@ -672,13 +693,13 @@ export class ComputeOperationsImpl implements ComputeOperations {
     resourceGroupName: string,
     workspaceName: string,
     computeName: string,
-    options?: ComputeStartOptionalParams
+    options?: ComputeStartOptionalParams,
   ): Promise<void> {
     const poller = await this.beginStart(
       resourceGroupName,
       workspaceName,
       computeName,
-      options
+      options,
     );
     return poller.pollUntilDone();
   }
@@ -694,25 +715,24 @@ export class ComputeOperationsImpl implements ComputeOperations {
     resourceGroupName: string,
     workspaceName: string,
     computeName: string,
-    options?: ComputeStopOptionalParams
-  ): Promise<PollerLike<PollOperationState<void>, void>> {
+    options?: ComputeStopOptionalParams,
+  ): 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);
@@ -721,8 +741,8 @@ export class ComputeOperationsImpl implements ComputeOperations {
         ...args,
         options: {
           ...args.options,
-          onResponse: callback
-        }
+          onResponse: callback,
+        },
       };
       const flatResponse = await directSendOperation(updatedArgs, spec);
       return {
@@ -730,19 +750,19 @@ export class ComputeOperationsImpl implements ComputeOperations {
         rawResponse: {
           statusCode: currentRawResponse!.status,
           body: currentRawResponse!.parsedBody,
-          headers: currentRawResponse!.headers.toJSON()
-        }
+          headers: currentRawResponse!.headers.toJSON(),
+        },
       };
     };
 
-    const lro = new LroImpl(
-      sendOperation,
-      { resourceGroupName, workspaceName, computeName, options },
-      stopOperationSpec
-    );
-    const poller = new LroEngine(lro, {
-      resumeFrom: options?.resumeFrom,
-      intervalInMs: options?.updateIntervalInMs
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: { resourceGroupName, workspaceName, computeName, options },
+      spec: stopOperationSpec,
+    });
+    const poller = await createHttpPoller<void, OperationState<void>>(lro, {
+      restoreFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
     });
     await poller.poll();
     return poller;
@@ -759,13 +779,13 @@ export class ComputeOperationsImpl implements ComputeOperations {
     resourceGroupName: string,
     workspaceName: string,
     computeName: string,
-    options?: ComputeStopOptionalParams
+    options?: ComputeStopOptionalParams,
   ): Promise<void> {
     const poller = await this.beginStop(
       resourceGroupName,
       workspaceName,
       computeName,
-      options
+      options,
     );
     return poller.pollUntilDone();
   }
@@ -781,25 +801,24 @@ export class ComputeOperationsImpl implements ComputeOperations {
     resourceGroupName: string,
     workspaceName: string,
     computeName: string,
-    options?: ComputeRestartOptionalParams
-  ): Promise<PollerLike<PollOperationState<void>, void>> {
+    options?: ComputeRestartOptionalParams,
+  ): 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);
@@ -808,8 +827,8 @@ export class ComputeOperationsImpl implements ComputeOperations {
         ...args,
         options: {
           ...args.options,
-          onResponse: callback
-        }
+          onResponse: callback,
+        },
       };
       const flatResponse = await directSendOperation(updatedArgs, spec);
       return {
@@ -817,19 +836,19 @@ export class ComputeOperationsImpl implements ComputeOperations {
         rawResponse: {
           statusCode: currentRawResponse!.status,
           body: currentRawResponse!.parsedBody,
-          headers: currentRawResponse!.headers.toJSON()
-        }
+          headers: currentRawResponse!.headers.toJSON(),
+        },
       };
     };
 
-    const lro = new LroImpl(
-      sendOperation,
-      { resourceGroupName, workspaceName, computeName, options },
-      restartOperationSpec
-    );
-    const poller = new LroEngine(lro, {
-      resumeFrom: options?.resumeFrom,
-      intervalInMs: options?.updateIntervalInMs
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: { resourceGroupName, workspaceName, computeName, options },
+      spec: restartOperationSpec,
+    });
+    const poller = await createHttpPoller<void, OperationState<void>>(lro, {
+      restoreFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
     });
     await poller.poll();
     return poller;
@@ -846,13 +865,13 @@ export class ComputeOperationsImpl implements ComputeOperations {
     resourceGroupName: string,
     workspaceName: string,
     computeName: string,
-    options?: ComputeRestartOptionalParams
+    options?: ComputeRestartOptionalParams,
   ): Promise<void> {
     const poller = await this.beginRestart(
       resourceGroupName,
       workspaceName,
       computeName,
-      options
+      options,
     );
     return poller.pollUntilDone();
   }
@@ -868,11 +887,11 @@ export class ComputeOperationsImpl implements ComputeOperations {
     resourceGroupName: string,
     workspaceName: string,
     nextLink: string,
-    options?: ComputeListNextOptionalParams
+    options?: ComputeListNextOptionalParams,
   ): Promise<ComputeListNextResponse> {
     return this.client.sendOperationRequest(
       { resourceGroupName, workspaceName, nextLink, options },
-      listNextOperationSpec
+      listNextOperationSpec,
     );
   }
 
@@ -889,11 +908,11 @@ export class ComputeOperationsImpl implements ComputeOperations {
     workspaceName: string,
     computeName: string,
     nextLink: string,
-    options?: ComputeListNodesNextOptionalParams
+    options?: ComputeListNodesNextOptionalParams,
   ): Promise<ComputeListNodesNextResponse> {
     return this.client.sendOperationRequest(
       { resourceGroupName, workspaceName, computeName, nextLink, options },
-      listNodesNextOperationSpec
+      listNodesNextOperationSpec,
     );
   }
 }
@@ -901,38 +920,36 @@ export class ComputeOperationsImpl implements ComputeOperations {
 const serializer = coreClient.createSerializer(Mappers, /* isXml */ false);
 
 const listOperationSpec: coreClient.OperationSpec = {
-  path:
-    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes",
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes",
   httpMethod: "GET",
   responses: {
     200: {
-      bodyMapper: Mappers.PaginatedComputeResourcesList
+      bodyMapper: Mappers.PaginatedComputeResourcesList,
     },
     default: {
-      bodyMapper: Mappers.ErrorResponse
-    }
+      bodyMapper: Mappers.ErrorResponse,
+    },
   },
   queryParameters: [Parameters.apiVersion, Parameters.skip],
   urlParameters: [
     Parameters.$host,
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
-    Parameters.workspaceName
+    Parameters.workspaceName,
   ],
   headerParameters: [Parameters.accept],
-  serializer
+  serializer,
 };
 const getOperationSpec: coreClient.OperationSpec = {
-  path:
-    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}",
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}",
   httpMethod: "GET",
   responses: {
     200: {
-      bodyMapper: Mappers.ComputeResource
+      bodyMapper: Mappers.ComputeResource,
     },
     default: {
-      bodyMapper: Mappers.ErrorResponse
-    }
+      bodyMapper: Mappers.ErrorResponse,
+    },
   },
   queryParameters: [Parameters.apiVersion],
   urlParameters: [
@@ -940,31 +957,30 @@ const getOperationSpec: coreClient.OperationSpec = {
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.workspaceName,
-    Parameters.computeName
+    Parameters.computeName,
   ],
   headerParameters: [Parameters.accept],
-  serializer
+  serializer,
 };
 const createOrUpdateOperationSpec: coreClient.OperationSpec = {
-  path:
-    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}",
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}",
   httpMethod: "PUT",
   responses: {
     200: {
-      bodyMapper: Mappers.ComputeResource
+      bodyMapper: Mappers.ComputeResource,
     },
     201: {
-      bodyMapper: Mappers.ComputeResource
+      bodyMapper: Mappers.ComputeResource,
     },
     202: {
-      bodyMapper: Mappers.ComputeResource
+      bodyMapper: Mappers.ComputeResource,
     },
     204: {
-      bodyMapper: Mappers.ComputeResource
+      bodyMapper: Mappers.ComputeResource,
     },
     default: {
-      bodyMapper: Mappers.ErrorResponse
-    }
+      bodyMapper: Mappers.ErrorResponse,
+    },
   },
   requestBody: Parameters.parameters4,
   queryParameters: [Parameters.apiVersion],
@@ -973,32 +989,31 @@ const createOrUpdateOperationSpec: coreClient.OperationSpec = {
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.workspaceName,
-    Parameters.computeName
+    Parameters.computeName,
   ],
   headerParameters: [Parameters.accept, Parameters.contentType],
   mediaType: "json",
-  serializer
+  serializer,
 };
 const updateOperationSpec: coreClient.OperationSpec = {
-  path:
-    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}",
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}",
   httpMethod: "PATCH",
   responses: {
     200: {
-      bodyMapper: Mappers.ComputeResource
+      bodyMapper: Mappers.ComputeResource,
     },
     201: {
-      bodyMapper: Mappers.ComputeResource
+      bodyMapper: Mappers.ComputeResource,
     },
     202: {
-      bodyMapper: Mappers.ComputeResource
+      bodyMapper: Mappers.ComputeResource,
     },
     204: {
-      bodyMapper: Mappers.ComputeResource
+      bodyMapper: Mappers.ComputeResource,
     },
     default: {
-      bodyMapper: Mappers.ErrorResponse
-    }
+      bodyMapper: Mappers.ErrorResponse,
+    },
   },
   requestBody: Parameters.parameters5,
   queryParameters: [Parameters.apiVersion],
@@ -1007,15 +1022,14 @@ const updateOperationSpec: coreClient.OperationSpec = {
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.workspaceName,
-    Parameters.computeName
+    Parameters.computeName,
   ],
   headerParameters: [Parameters.accept, Parameters.contentType],
   mediaType: "json",
-  serializer
+  serializer,
 };
 const deleteOperationSpec: coreClient.OperationSpec = {
-  path:
-    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}",
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}",
   httpMethod: "DELETE",
   responses: {
     200: {},
@@ -1023,8 +1037,8 @@ const deleteOperationSpec: coreClient.OperationSpec = {
     202: {},
     204: {},
     default: {
-      bodyMapper: Mappers.ErrorResponse
-    }
+      bodyMapper: Mappers.ErrorResponse,
+    },
   },
   queryParameters: [Parameters.apiVersion, Parameters.underlyingResourceAction],
   urlParameters: [
@@ -1032,22 +1046,21 @@ const deleteOperationSpec: coreClient.OperationSpec = {
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.workspaceName,
-    Parameters.computeName
+    Parameters.computeName,
   ],
   headerParameters: [Parameters.accept],
-  serializer
+  serializer,
 };
 const listNodesOperationSpec: coreClient.OperationSpec = {
-  path:
-    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}/listNodes",
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}/listNodes",
   httpMethod: "POST",
   responses: {
     200: {
-      bodyMapper: Mappers.AmlComputeNodesInformation
+      bodyMapper: Mappers.AmlComputeNodesInformation,
     },
     default: {
-      bodyMapper: Mappers.ErrorResponse
-    }
+      bodyMapper: Mappers.ErrorResponse,
+    },
   },
   queryParameters: [Parameters.apiVersion],
   urlParameters: [
@@ -1055,22 +1068,21 @@ const listNodesOperationSpec: coreClient.OperationSpec = {
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.workspaceName,
-    Parameters.computeName
+    Parameters.computeName,
   ],
   headerParameters: [Parameters.accept],
-  serializer
+  serializer,
 };
 const listKeysOperationSpec: coreClient.OperationSpec = {
-  path:
-    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}/listKeys",
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}/listKeys",
   httpMethod: "POST",
   responses: {
     200: {
-      bodyMapper: Mappers.ComputeSecrets
+      bodyMapper: Mappers.ComputeSecrets,
     },
     default: {
-      bodyMapper: Mappers.ErrorResponse
-    }
+      bodyMapper: Mappers.ErrorResponse,
+    },
   },
   queryParameters: [Parameters.apiVersion],
   urlParameters: [
@@ -1078,14 +1090,13 @@ const listKeysOperationSpec: coreClient.OperationSpec = {
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.workspaceName,
-    Parameters.computeName
+    Parameters.computeName,
   ],
   headerParameters: [Parameters.accept],
-  serializer
+  serializer,
 };
 const startOperationSpec: coreClient.OperationSpec = {
-  path:
-    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}/start",
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}/start",
   httpMethod: "POST",
   responses: {
     200: {},
@@ -1093,8 +1104,8 @@ const startOperationSpec: coreClient.OperationSpec = {
     202: {},
     204: {},
     default: {
-      bodyMapper: Mappers.ErrorResponse
-    }
+      bodyMapper: Mappers.ErrorResponse,
+    },
   },
   queryParameters: [Parameters.apiVersion],
   urlParameters: [
@@ -1102,14 +1113,13 @@ const startOperationSpec: coreClient.OperationSpec = {
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.workspaceName,
-    Parameters.computeName
+    Parameters.computeName,
   ],
   headerParameters: [Parameters.accept],
-  serializer
+  serializer,
 };
 const stopOperationSpec: coreClient.OperationSpec = {
-  path:
-    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}/stop",
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}/stop",
   httpMethod: "POST",
   responses: {
     200: {},
@@ -1117,8 +1127,8 @@ const stopOperationSpec: coreClient.OperationSpec = {
     202: {},
     204: {},
     default: {
-      bodyMapper: Mappers.ErrorResponse
-    }
+      bodyMapper: Mappers.ErrorResponse,
+    },
   },
   queryParameters: [Parameters.apiVersion],
   urlParameters: [
@@ -1126,14 +1136,13 @@ const stopOperationSpec: coreClient.OperationSpec = {
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.workspaceName,
-    Parameters.computeName
+    Parameters.computeName,
   ],
   headerParameters: [Parameters.accept],
-  serializer
+  serializer,
 };
 const restartOperationSpec: coreClient.OperationSpec = {
-  path:
-    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}/restart",
+  path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}/restart",
   httpMethod: "POST",
   responses: {
     200: {},
@@ -1141,8 +1150,8 @@ const restartOperationSpec: coreClient.OperationSpec = {
     202: {},
     204: {},
     default: {
-      bodyMapper: Mappers.ErrorResponse
-    }
+      bodyMapper: Mappers.ErrorResponse,
+    },
   },
   queryParameters: [Parameters.apiVersion],
   urlParameters: [
@@ -1150,53 +1159,51 @@ const restartOperationSpec: coreClient.OperationSpec = {
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.workspaceName,
-    Parameters.computeName
+    Parameters.computeName,
   ],
   headerParameters: [Parameters.accept],
-  serializer
+  serializer,
 };
 const listNextOperationSpec: coreClient.OperationSpec = {
   path: "{nextLink}",
   httpMethod: "GET",
   responses: {
     200: {
-      bodyMapper: Mappers.PaginatedComputeResourcesList
+      bodyMapper: Mappers.PaginatedComputeResourcesList,
     },
     default: {
-      bodyMapper: Mappers.ErrorResponse
-    }
+      bodyMapper: Mappers.ErrorResponse,
+    },
   },
-  queryParameters: [Parameters.apiVersion, Parameters.skip],
   urlParameters: [
     Parameters.$host,
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.workspaceName,
-    Parameters.nextLink
+    Parameters.nextLink,
   ],
   headerParameters: [Parameters.accept],
-  serializer
+  serializer,
 };
 const listNodesNextOperationSpec: coreClient.OperationSpec = {
   path: "{nextLink}",
   httpMethod: "GET",
   responses: {
     200: {
-      bodyMapper: Mappers.AmlComputeNodesInformation
+      bodyMapper: Mappers.AmlComputeNodesInformation,
     },
     default: {
-      bodyMapper: Mappers.ErrorResponse
-    }
+      bodyMapper: Mappers.ErrorResponse,
+    },
   },
-  queryParameters: [Parameters.apiVersion],
   urlParameters: [
     Parameters.$host,
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.workspaceName,
     Parameters.nextLink,
-    Parameters.computeName
+    Parameters.computeName,
   ],
   headerParameters: [Parameters.accept],
-  serializer
+  serializer,
 };