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

package/dist-esm/src/operations/computeOperations.js

@@ -10,8 +10,8 @@ import { setContinuationToken } from "../pagingHelper";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
 import * as Parameters from "../models/parameters";
-import { LroEngine } from "@azure/core-lro";
-import { LroImpl } from "../lroImpl";
+import { createHttpPoller, } from "@azure/core-lro";
+import { createLroSpec } from "../lroImpl";
 /// <reference lib="esnext.asynciterable" />
 /** Class containing ComputeOperations operations. */
 export class ComputeOperationsImpl {
@@ -42,7 +42,7 @@ export class ComputeOperationsImpl {
                     throw new Error("maxPageSize is not supported by this operation.");
                 }
                 return this.listPagingPage(resourceGroupName, workspaceName, options, settings);
-            }
+            },
         };
     }
     listPagingPage(resourceGroupName, workspaceName, options, settings) {
@@ -106,7 +106,7 @@ export class ComputeOperationsImpl {
                     throw new Error("maxPageSize is not supported by this operation.");
                 }
                 return this.listNodesPagingPage(resourceGroupName, workspaceName, computeName, options, settings);
-            }
+            },
         };
     }
     listNodesPagingPage(resourceGroupName, workspaceName, computeName, options, settings) {
@@ -184,7 +184,7 @@ export class ComputeOperationsImpl {
             const directSendOperation = (args, spec) => __awaiter(this, void 0, void 0, function* () {
                 return this.client.sendOperationRequest(args, spec);
             });
-            const sendOperation = (args, spec) => __awaiter(this, void 0, void 0, function* () {
+            const sendOperationFn = (args, spec) => __awaiter(this, void 0, void 0, function* () {
                 var _a;
                 let currentRawResponse = undefined;
                 const providedCallback = (_a = args.options) === null || _a === void 0 ? void 0 : _a.onResponse;
@@ -199,14 +199,24 @@ export class ComputeOperationsImpl {
                     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 === null || options === void 0 ? void 0 : options.resumeFrom,
-                intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs
+            const lro = createLroSpec({
+                sendOperationFn,
+                args: {
+                    resourceGroupName,
+                    workspaceName,
+                    computeName,
+                    parameters,
+                    options,
+                },
+                spec: createOrUpdateOperationSpec,
+            });
+            const poller = yield createHttpPoller(lro, {
+                restoreFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
+                intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs,
             });
             yield poller.poll();
             return poller;
@@ -242,7 +252,7 @@ export class ComputeOperationsImpl {
             const directSendOperation = (args, spec) => __awaiter(this, void 0, void 0, function* () {
                 return this.client.sendOperationRequest(args, spec);
             });
-            const sendOperation = (args, spec) => __awaiter(this, void 0, void 0, function* () {
+            const sendOperationFn = (args, spec) => __awaiter(this, void 0, void 0, function* () {
                 var _a;
                 let currentRawResponse = undefined;
                 const providedCallback = (_a = args.options) === null || _a === void 0 ? void 0 : _a.onResponse;
@@ -257,14 +267,24 @@ export class ComputeOperationsImpl {
                     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 === null || options === void 0 ? void 0 : options.resumeFrom,
-                intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs
+            const lro = createLroSpec({
+                sendOperationFn,
+                args: {
+                    resourceGroupName,
+                    workspaceName,
+                    computeName,
+                    parameters,
+                    options,
+                },
+                spec: updateOperationSpec,
+            });
+            const poller = yield createHttpPoller(lro, {
+                restoreFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
+                intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs,
             });
             yield poller.poll();
             return poller;
@@ -299,7 +319,7 @@ export class ComputeOperationsImpl {
             const directSendOperation = (args, spec) => __awaiter(this, void 0, void 0, function* () {
                 return this.client.sendOperationRequest(args, spec);
             });
-            const sendOperation = (args, spec) => __awaiter(this, void 0, void 0, function* () {
+            const sendOperationFn = (args, spec) => __awaiter(this, void 0, void 0, function* () {
                 var _a;
                 let currentRawResponse = undefined;
                 const providedCallback = (_a = args.options) === null || _a === void 0 ? void 0 : _a.onResponse;
@@ -314,20 +334,24 @@ export class ComputeOperationsImpl {
                     rawResponse: {
                         statusCode: currentRawResponse.status,
                         body: currentRawResponse.parsedBody,
-                        headers: currentRawResponse.headers.toJSON()
-                    }
+                        headers: currentRawResponse.headers.toJSON(),
+                    },
                 };
             });
-            const lro = new LroImpl(sendOperation, {
-                resourceGroupName,
-                workspaceName,
-                computeName,
-                underlyingResourceAction,
-                options
-            }, deleteOperationSpec);
-            const poller = new LroEngine(lro, {
-                resumeFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
-                intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs
+            const lro = createLroSpec({
+                sendOperationFn,
+                args: {
+                    resourceGroupName,
+                    workspaceName,
+                    computeName,
+                    underlyingResourceAction,
+                    options,
+                },
+                spec: deleteOperationSpec,
+            });
+            const poller = yield createHttpPoller(lro, {
+                restoreFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
+                intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs,
             });
             yield poller.poll();
             return poller;
@@ -380,7 +404,7 @@ export class ComputeOperationsImpl {
             const directSendOperation = (args, spec) => __awaiter(this, void 0, void 0, function* () {
                 return this.client.sendOperationRequest(args, spec);
             });
-            const sendOperation = (args, spec) => __awaiter(this, void 0, void 0, function* () {
+            const sendOperationFn = (args, spec) => __awaiter(this, void 0, void 0, function* () {
                 var _a;
                 let currentRawResponse = undefined;
                 const providedCallback = (_a = args.options) === null || _a === void 0 ? void 0 : _a.onResponse;
@@ -395,14 +419,18 @@ export class ComputeOperationsImpl {
                     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 === null || options === void 0 ? void 0 : options.resumeFrom,
-                intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs
+            const lro = createLroSpec({
+                sendOperationFn,
+                args: { resourceGroupName, workspaceName, computeName, options },
+                spec: startOperationSpec,
+            });
+            const poller = yield createHttpPoller(lro, {
+                restoreFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
+                intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs,
             });
             yield poller.poll();
             return poller;
@@ -433,7 +461,7 @@ export class ComputeOperationsImpl {
             const directSendOperation = (args, spec) => __awaiter(this, void 0, void 0, function* () {
                 return this.client.sendOperationRequest(args, spec);
             });
-            const sendOperation = (args, spec) => __awaiter(this, void 0, void 0, function* () {
+            const sendOperationFn = (args, spec) => __awaiter(this, void 0, void 0, function* () {
                 var _a;
                 let currentRawResponse = undefined;
                 const providedCallback = (_a = args.options) === null || _a === void 0 ? void 0 : _a.onResponse;
@@ -448,14 +476,18 @@ export class ComputeOperationsImpl {
                     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 === null || options === void 0 ? void 0 : options.resumeFrom,
-                intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs
+            const lro = createLroSpec({
+                sendOperationFn,
+                args: { resourceGroupName, workspaceName, computeName, options },
+                spec: stopOperationSpec,
+            });
+            const poller = yield createHttpPoller(lro, {
+                restoreFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
+                intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs,
             });
             yield poller.poll();
             return poller;
@@ -486,7 +518,7 @@ export class ComputeOperationsImpl {
             const directSendOperation = (args, spec) => __awaiter(this, void 0, void 0, function* () {
                 return this.client.sendOperationRequest(args, spec);
             });
-            const sendOperation = (args, spec) => __awaiter(this, void 0, void 0, function* () {
+            const sendOperationFn = (args, spec) => __awaiter(this, void 0, void 0, function* () {
                 var _a;
                 let currentRawResponse = undefined;
                 const providedCallback = (_a = args.options) === null || _a === void 0 ? void 0 : _a.onResponse;
@@ -501,14 +533,18 @@ export class ComputeOperationsImpl {
                     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 === null || options === void 0 ? void 0 : options.resumeFrom,
-                intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs
+            const lro = createLroSpec({
+                sendOperationFn,
+                args: { resourceGroupName, workspaceName, computeName, options },
+                spec: restartOperationSpec,
+            });
+            const poller = yield createHttpPoller(lro, {
+                restoreFrom: options === null || options === void 0 ? void 0 : options.resumeFrom,
+                intervalInMs: options === null || options === void 0 ? void 0 : options.updateIntervalInMs,
             });
             yield poller.poll();
             return poller;
@@ -556,32 +592,32 @@ const listOperationSpec = {
     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 = {
     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: [
@@ -589,30 +625,30 @@ const getOperationSpec = {
         Parameters.subscriptionId,
         Parameters.resourceGroupName,
         Parameters.workspaceName,
-        Parameters.computeName
+        Parameters.computeName,
     ],
     headerParameters: [Parameters.accept],
-    serializer
+    serializer,
 };
 const createOrUpdateOperationSpec = {
     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],
@@ -621,31 +657,31 @@ const createOrUpdateOperationSpec = {
         Parameters.subscriptionId,
         Parameters.resourceGroupName,
         Parameters.workspaceName,
-        Parameters.computeName
+        Parameters.computeName,
     ],
     headerParameters: [Parameters.accept, Parameters.contentType],
     mediaType: "json",
-    serializer
+    serializer,
 };
 const updateOperationSpec = {
     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],
@@ -654,11 +690,11 @@ const updateOperationSpec = {
         Parameters.subscriptionId,
         Parameters.resourceGroupName,
         Parameters.workspaceName,
-        Parameters.computeName
+        Parameters.computeName,
     ],
     headerParameters: [Parameters.accept, Parameters.contentType],
     mediaType: "json",
-    serializer
+    serializer,
 };
 const deleteOperationSpec = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}",
@@ -669,8 +705,8 @@ const deleteOperationSpec = {
         202: {},
         204: {},
         default: {
-            bodyMapper: Mappers.ErrorResponse
-        }
+            bodyMapper: Mappers.ErrorResponse,
+        },
     },
     queryParameters: [Parameters.apiVersion, Parameters.underlyingResourceAction],
     urlParameters: [
@@ -678,21 +714,21 @@ const deleteOperationSpec = {
         Parameters.subscriptionId,
         Parameters.resourceGroupName,
         Parameters.workspaceName,
-        Parameters.computeName
+        Parameters.computeName,
     ],
     headerParameters: [Parameters.accept],
-    serializer
+    serializer,
 };
 const listNodesOperationSpec = {
     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: [
@@ -700,21 +736,21 @@ const listNodesOperationSpec = {
         Parameters.subscriptionId,
         Parameters.resourceGroupName,
         Parameters.workspaceName,
-        Parameters.computeName
+        Parameters.computeName,
     ],
     headerParameters: [Parameters.accept],
-    serializer
+    serializer,
 };
 const listKeysOperationSpec = {
     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: [
@@ -722,10 +758,10 @@ const listKeysOperationSpec = {
         Parameters.subscriptionId,
         Parameters.resourceGroupName,
         Parameters.workspaceName,
-        Parameters.computeName
+        Parameters.computeName,
     ],
     headerParameters: [Parameters.accept],
-    serializer
+    serializer,
 };
 const startOperationSpec = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}/start",
@@ -736,8 +772,8 @@ const startOperationSpec = {
         202: {},
         204: {},
         default: {
-            bodyMapper: Mappers.ErrorResponse
-        }
+            bodyMapper: Mappers.ErrorResponse,
+        },
     },
     queryParameters: [Parameters.apiVersion],
     urlParameters: [
@@ -745,10 +781,10 @@ const startOperationSpec = {
         Parameters.subscriptionId,
         Parameters.resourceGroupName,
         Parameters.workspaceName,
-        Parameters.computeName
+        Parameters.computeName,
     ],
     headerParameters: [Parameters.accept],
-    serializer
+    serializer,
 };
 const stopOperationSpec = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}/stop",
@@ -759,8 +795,8 @@ const stopOperationSpec = {
         202: {},
         204: {},
         default: {
-            bodyMapper: Mappers.ErrorResponse
-        }
+            bodyMapper: Mappers.ErrorResponse,
+        },
     },
     queryParameters: [Parameters.apiVersion],
     urlParameters: [
@@ -768,10 +804,10 @@ const stopOperationSpec = {
         Parameters.subscriptionId,
         Parameters.resourceGroupName,
         Parameters.workspaceName,
-        Parameters.computeName
+        Parameters.computeName,
     ],
     headerParameters: [Parameters.accept],
-    serializer
+    serializer,
 };
 const restartOperationSpec = {
     path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}/restart",
@@ -782,8 +818,8 @@ const restartOperationSpec = {
         202: {},
         204: {},
         default: {
-            bodyMapper: Mappers.ErrorResponse
-        }
+            bodyMapper: Mappers.ErrorResponse,
+        },
     },
     queryParameters: [Parameters.apiVersion],
     urlParameters: [
@@ -791,54 +827,52 @@ const restartOperationSpec = {
         Parameters.subscriptionId,
         Parameters.resourceGroupName,
         Parameters.workspaceName,
-        Parameters.computeName
+        Parameters.computeName,
     ],
     headerParameters: [Parameters.accept],
-    serializer
+    serializer,
 };
 const listNextOperationSpec = {
     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 = {
     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,
 };
 //# sourceMappingURL=computeOperations.js.map