@azure/ai-text-analytics 5.1.1-alpha.20210730.2 → 5.1.1-alpha.20220725.1

package/dist-esm/src/lro/analyze/operation.js

@@ -1,11 +1,9 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT license.
 import { __asyncGenerator, __await } from "tslib";
-import { createAnalyzeActionsResult } from "../../analyzeActionsResult";
-import { getOperationId, handleInvalidDocumentBatch, nextLinkToTopAndSkip } from "../../util";
+import { createAnalyzeActionsResult, } from "../../analyzeActionsResult";
+import { getOperationId, nextLinkToTopAndSkip, throwError } from "../../util";
 import { AnalysisPollOperation } from "../poller";
-import { SpanStatusCode } from "@azure/core-tracing";
-import { createSpan } from "../../tracing";
 import { logger } from "../../logger";
 /**
  * @internal
@@ -19,7 +17,7 @@ function getMetaInfoFromResponse(response) {
         actionsSucceededCount: response.tasks.completed,
         actionsFailedCount: response.tasks.failed,
         actionsInProgressCount: response.tasks.inProgress,
-        displayName: response.displayName
+        displayName: response.displayName,
     };
 }
 /**
@@ -30,10 +28,11 @@ function getMetaInfoFromResponse(response) {
 export class BeginAnalyzeActionsPollerOperation extends AnalysisPollOperation {
     constructor(state, 
     // eslint-disable-next-line @azure/azure-sdk/ts-use-interface-parameters
-    client, documents, actions, options = {}) {
+    client, tracing, documents, actions, options = {}) {
         super(state);
         this.state = state;
         this.client = client;
+        this.tracing = tracing;
         this.documents = documents;
         this.actions = actions;
         this.options = options;
@@ -55,7 +54,7 @@ export class BeginAnalyzeActionsPollerOperation extends AnalysisPollOperation {
             byPage: (settings) => {
                 const pageOptions = Object.assign(Object.assign({}, options), { top: settings === null || settings === void 0 ? void 0 : settings.maxPageSize });
                 return this._listAnalyzeActionsResultsPaged(operationId, pageOptions);
-            }
+            },
         };
     }
     /**
@@ -76,77 +75,36 @@ export class BeginAnalyzeActionsPollerOperation extends AnalysisPollOperation {
      * returns an iterator to arrays of the sorted results of an analyze actions operation.
      */
     async _listAnalyzeActionsResultsSinglePage(operationId, options) {
-        const { span, updatedOptions: finalOptions } = createSpan("TextAnalyticsClient-_listAnalyzeActionsResultsSinglePage", options || {});
-        try {
-            const response = await this.client.analyzeStatus(operationId, finalOptions);
-            const result = createAnalyzeActionsResult(response, this.documents);
-            return response.nextLink
-                ? Object.assign({ result }, nextLinkToTopAndSkip(response.nextLink)) : { result };
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        const response = await this.tracing.withSpan("TextAnalyticsClient-_listAnalyzeActionsResultsSinglePage", options || {}, (finalOptions) => this.client.analyzeStatus(operationId, finalOptions));
+        const result = createAnalyzeActionsResult(response, this.documents);
+        return response.nextLink ? Object.assign({ result }, nextLinkToTopAndSkip(response.nextLink)) : { result };
     }
     /**
      * returns whether the analyze actions operation is done and if so returns also
      * statistics.
      */
     async getAnalyzeActionsOperationStatus(operationId, options) {
-        const { span, updatedOptions: finalOptions } = createSpan("TextAnalyticsClient-getAnalyzeActionsOperationStatus", options || {});
-        try {
-            const response = await this.client.analyzeStatus(operationId, finalOptions);
-            switch (response.status) {
-                case "notStarted":
-                case "running":
-                    break;
-                default: {
-                    return {
-                        done: true,
-                        statistics: response.statistics,
-                        operationMetdata: getMetaInfoFromResponse(response)
-                    };
-                }
+        const response = await this.tracing.withSpan("TextAnalyticsClient-getAnalyzeActionsOperationStatus", options || {}, (finalOptions) => this.client.analyzeStatus(operationId, finalOptions));
+        switch (response.status) {
+            case "notStarted":
+            case "running":
+                break;
+            default: {
+                return {
+                    done: true,
+                    statistics: response.statistics,
+                    operationMetdata: getMetaInfoFromResponse(response),
+                };
             }
-            return { done: false, operationMetdata: getMetaInfoFromResponse(response) };
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message
-            });
-            throw e;
-        }
-        finally {
-            span.end();
         }
+        return { done: false, operationMetdata: getMetaInfoFromResponse(response) };
     }
     async beginAnalyzeActions(documents, actions, options) {
-        const { span, updatedOptions: finalOptions } = createSpan("TextAnalyticsClient-beginAnalyze", options || {});
-        try {
-            return await this.client.analyze(Object.assign({ body: {
-                    analysisInput: { documents: documents },
-                    tasks: actions,
-                    displayName: options === null || options === void 0 ? void 0 : options.displayName
-                } }, finalOptions));
-        }
-        catch (e) {
-            const exception = handleInvalidDocumentBatch(e);
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: exception.message
-            });
-            throw exception;
-        }
-        finally {
-            span.end();
-        }
+        return this.tracing.withSpan("TextAnalyticsClient-beginAnalyze", options || {}, (finalOptions) => throwError(this.client.analyze(Object.assign({ body: {
+                analysisInput: { documents: documents },
+                tasks: actions,
+                displayName: options === null || options === void 0 ? void 0 : options.displayName,
+            } }, finalOptions))));
     }
     async update(options = {}) {
         var _a;
@@ -158,7 +116,7 @@ export class BeginAnalyzeActionsPollerOperation extends AnalysisPollOperation {
                 displayName: this.options.displayName,
                 tracingOptions: this.options.tracingOptions,
                 requestOptions: this.options.requestOptions,
-                abortSignal: updatedAbortSignal ? updatedAbortSignal : this.options.abortSignal
+                abortSignal: updatedAbortSignal ? updatedAbortSignal : this.options.abortSignal,
             });
             if (!response.operationLocation) {
                 throw new Error("Expects a valid 'operationLocation' to retrieve analyze results but did not find any");
@@ -168,7 +126,7 @@ export class BeginAnalyzeActionsPollerOperation extends AnalysisPollOperation {
         const operationStatus = await this.getAnalyzeActionsOperationStatus(state.operationId, {
             abortSignal: updatedAbortSignal ? updatedAbortSignal : options.abortSignal,
             includeStatistics: this.options.includeStatistics,
-            tracingOptions: this.options.tracingOptions
+            tracingOptions: this.options.tracingOptions,
         });
         state.createdOn = operationStatus.operationMetdata.createdOn;
         state.expiresOn = operationStatus.operationMetdata.expiresOn;
@@ -184,7 +142,7 @@ export class BeginAnalyzeActionsPollerOperation extends AnalysisPollOperation {
                 tracingOptions: this.options.tracingOptions,
                 includeStatistics: this.options.includeStatistics,
                 onResponse: this.options.onResponse,
-                serializerOptions: this.options.serializerOptions
+                serializerOptions: this.options.serializerOptions,
             });
             // Attach stats if the service starts to return them
             // https://github.com/Azure/azure-sdk-for-js/issues/14139

package/dist-esm/src/lro/analyze/operation.js.map

@@ -1 +1 @@
-{"version":3,"file":"operation.js","sourceRoot":"","sources":["../../../../src/lro/analyze/operation.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;;AAelC,OAAO,EAIL,0BAA0B,EAC3B,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EAAE,cAAc,EAAE,0BAA0B,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAC;AAC9F,OAAO,EAAE,qBAAqB,EAAiD,MAAM,WAAW,CAAC;AAEjG,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAC3C,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAoFtC;;GAEG;AACH,SAAS,uBAAuB,CAC9B,QAAyB;IAEzB,OAAO;QACL,SAAS,EAAE,QAAQ,CAAC,eAAe;QACnC,cAAc,EAAE,QAAQ,CAAC,kBAAkB;QAC3C,SAAS,EAAE,QAAQ,CAAC,kBAAkB;QACtC,MAAM,EAAE,QAAQ,CAAC,MAAM;QACvB,qBAAqB,EAAE,QAAQ,CAAC,KAAK,CAAC,SAAS;QAC/C,kBAAkB,EAAE,QAAQ,CAAC,KAAK,CAAC,MAAM;QACzC,sBAAsB,EAAE,QAAQ,CAAC,KAAK,CAAC,UAAU;QACjD,WAAW,EAAE,QAAQ,CAAC,WAAW;KAClC,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAM,OAAO,kCAAmC,SAAQ,qBAGvD;IACC,YACS,KAAmC;IAC1C,wEAAwE;IAChE,MAAc,EACd,SAA8B,EAC9B,OAAyB,EACzB,UAAsC,EAAE;QAEhD,KAAK,CAAC,KAAK,CAAC,CAAC;QAPN,UAAK,GAAL,KAAK,CAA8B;QAElC,WAAM,GAAN,MAAM,CAAQ;QACd,cAAS,GAAT,SAAS,CAAqB;QAC9B,YAAO,GAAP,OAAO,CAAkB;QACzB,YAAO,GAAP,OAAO,CAAiC;IAGlD,CAAC;IAED;;;;OAIG;IACK,yBAAyB,CAC/B,WAAmB,EACnB,UAAgD,EAAE;QAElD,MAAM,IAAI,GAAG,IAAI,CAAC,+BAA+B,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QACxE,OAAO;YACL,IAAI;gBACF,OAAO,IAAI,CAAC,IAAI,EAAE,CAAC;YACrB,CAAC;YACD,CAAC,MAAM,CAAC,aAAa,CAAC;gBACpB,OAAO,IAAI,CAAC;YACd,CAAC;YACD,MAAM,EAAE,CAAC,QAAuB,EAAE,EAAE;gBAClC,MAAM,WAAW,mCAAQ,OAAO,KAAE,GAAG,EAAE,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,CAAE,WAAW,GAAE,CAAC;gBAC/D,OAAO,IAAI,CAAC,+BAA+B,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC;YACxE,CAAC;SACF,CAAC;IACJ,CAAC;IAED;;OAEG;IACY,+BAA+B,CAC5C,WAAmB,EACnB,OAA8C;;YAE9C,IAAI,QAAQ,GAAG,cAAM,IAAI,CAAC,oCAAoC,CAAC,WAAW,EAAE,OAAO,CAAC,CAAA,CAAC;YACrF,oBAAM,QAAQ,CAAC,MAAM,CAAA,CAAC;YACtB,OAAO,QAAQ,CAAC,IAAI,EAAE;gBACpB,MAAM,uBAAuB,mCACxB,OAAO,KACV,GAAG,EAAE,QAAQ,CAAC,GAAG,EACjB,IAAI,EAAE,QAAQ,CAAC,IAAI,GACpB,CAAC;gBACF,QAAQ,GAAG,cAAM,IAAI,CAAC,oCAAoC,CACxD,WAAW,EACX,uBAAuB,CACxB,CAAA,CAAC;gBACF,oBAAM,QAAQ,CAAC,MAAM,CAAA,CAAC;aACvB;QACH,CAAC;KAAA;IAED;;OAEG;IACK,KAAK,CAAC,oCAAoC,CAChD,WAAmB,EACnB,OAA8C;QAE9C,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,YAAY,EAAE,GAAG,UAAU,CACvD,0DAA0D,EAC1D,OAAO,IAAI,EAAE,CACd,CAAC;QACF,IAAI;YACF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;YAC5E,MAAM,MAAM,GAAG,0BAA0B,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;YACpE,OAAO,QAAQ,CAAC,QAAQ;gBACtB,CAAC,iBAAG,MAAM,IAAK,oBAAoB,CAAC,QAAQ,CAAC,QAAQ,CAAC,EACtD,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC;SAChB;QAAC,OAAO,CAAC,EAAE;YACV,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,cAAc,CAAC,KAAK;gBAC1B,OAAO,EAAE,CAAC,CAAC,OAAO;aACnB,CAAC,CAAC;YACH,MAAM,CAAC,CAAC;SACT;gBAAS;YACR,IAAI,CAAC,GAAG,EAAE,CAAC;SACZ;IACH,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,gCAAgC,CAC5C,WAAmB,EACnB,OAA8C;QAE9C,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,YAAY,EAAE,GAAG,UAAU,CACvD,sDAAsD,EACtD,OAAO,IAAI,EAAE,CACd,CAAC;QACF,IAAI;YACF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;YAC5E,QAAQ,QAAQ,CAAC,MAAM,EAAE;gBACvB,KAAK,YAAY,CAAC;gBAClB,KAAK,SAAS;oBACZ,MAAM;gBACR,OAAO,CAAC,CAAC;oBACP,OAAO;wBACL,IAAI,EAAE,IAAI;wBACV,UAAU,EAAE,QAAQ,CAAC,UAAU;wBAC/B,gBAAgB,EAAE,uBAAuB,CAAC,QAAQ,CAAC;qBACpD,CAAC;iBACH;aACF;YACD,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,gBAAgB,EAAE,uBAAuB,CAAC,QAAQ,CAAC,EAAE,CAAC;SAC7E;QAAC,OAAO,CAAC,EAAE;YACV,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,cAAc,CAAC,KAAK;gBAC1B,OAAO,EAAE,CAAC,CAAC,OAAO;aACnB,CAAC,CAAC;YACH,MAAM,CAAC,CAAC;SACT;gBAAS;YACR,IAAI,CAAC,GAAG,EAAE,CAAC;SACZ;IACH,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAC/B,SAA8B,EAC9B,OAAyB,EACzB,OAAqC;QAErC,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,YAAY,EAAE,GAAG,UAAU,CACvD,kCAAkC,EAClC,OAAO,IAAI,EAAE,CACd,CAAC;QAEF,IAAI;YACF,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,iBAC9B,IAAI,EAAE;oBACJ,aAAa,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE;oBACvC,KAAK,EAAE,OAAO;oBACd,WAAW,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW;iBAClC,IACE,YAAY,EACf,CAAC;SACJ;QAAC,OAAO,CAAC,EAAE;YACV,MAAM,SAAS,GAAG,0BAA0B,CAAC,CAAC,CAAC,CAAC;YAChD,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,cAAc,CAAC,KAAK;gBAC1B,OAAO,EAAE,SAAS,CAAC,OAAO;aAC3B,CAAC,CAAC;YACH,MAAM,SAAS,CAAC;SACjB;gBAAS;YACR,IAAI,CAAC,GAAG,EAAE,CAAC;SACZ;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CACV,UAGI,EAAE;;QAEN,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACzB,MAAM,kBAAkB,GAAG,OAAO,CAAC,WAAW,CAAC;QAC/C,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE;YACpB,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC;YACvB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,EAAE;gBAC5E,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW;gBACrC,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;gBAC3C,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;gBAC3C,WAAW,EAAE,kBAAkB,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW;aAChF,CAAC,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,iBAAiB,EAAE;gBAC/B,MAAM,IAAI,KAAK,CACb,sFAAsF,CACvF,CAAC;aACH;YACD,KAAK,CAAC,WAAW,GAAG,cAAc,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC;SAChE;QAED,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,gCAAgC,CAAC,KAAK,CAAC,WAAY,EAAE;YACtF,WAAW,EAAE,kBAAkB,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW;YAC1E,iBAAiB,EAAE,IAAI,CAAC,OAAO,CAAC,iBAAiB;YACjD,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;SAC5C,CAAC,CAAC;QAEH,KAAK,CAAC,SAAS,GAAG,eAAe,CAAC,gBAAgB,CAAC,SAAS,CAAC;QAC7D,KAAK,CAAC,SAAS,GAAG,eAAe,CAAC,gBAAgB,CAAC,SAAS,CAAC;QAC7D,KAAK,CAAC,cAAc,GAAG,eAAe,CAAC,gBAAgB,CAAC,cAAc,CAAC;QACvE,KAAK,CAAC,MAAM,GAAG,eAAe,CAAC,gBAAgB,CAAC,MAAM,CAAC;QACvD,KAAK,CAAC,qBAAqB,GAAG,eAAe,CAAC,gBAAgB,CAAC,qBAAqB,CAAC;QACrF,KAAK,CAAC,kBAAkB,GAAG,eAAe,CAAC,gBAAgB,CAAC,kBAAkB,CAAC;QAC/E,KAAK,CAAC,sBAAsB,GAAG,eAAe,CAAC,gBAAgB,CAAC,sBAAsB,CAAC;QACvF,KAAK,CAAC,WAAW,GAAG,MAAA,eAAe,CAAC,gBAAgB,0CAAE,WAAW,CAAC;QAElE,IAAI,CAAC,KAAK,CAAC,WAAW,IAAI,eAAe,CAAC,IAAI,EAAE;YAC9C,MAAM,aAAa,GAAG,IAAI,CAAC,yBAAyB,CAAC,KAAK,CAAC,WAAY,EAAE;gBACvE,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW;gBACrC,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;gBAC3C,iBAAiB,EAAE,IAAI,CAAC,OAAO,CAAC,iBAAiB;gBACjD,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU;gBACnC,iBAAiB,EAAE,IAAI,CAAC,OAAO,CAAC,iBAAiB;aAClD,CAAC,CAAC;YACH,oDAAoD;YACpD,yDAAyD;YACzD,gDAAgD;YAChD,2CAA2C;YAC3C,MAAM;YACN,KAAK,CAAC,MAAM,GAAG,aAAa,CAAC;YAC7B,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC;SAC1B;QAED,IAAI,OAAO,OAAO,CAAC,YAAY,KAAK,UAAU,EAAE;YAC9C,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;SAC7B;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,MAAM;QACV,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACzB,MAAM,CAAC,OAAO,CAAC,iEAAiE,CAAC,CAAC;QAClF,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC;QACzB,OAAO,IAAI,CAAC;IACd,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\nimport { OperationOptions } from \"@azure/core-client\";\n\nimport { AbortSignalLike } from \"@azure/abort-controller\";\n\nimport {\n  AnalyzeJobState,\n  GeneratedClientAnalyzeResponse as BeginAnalyzeResponse,\n  GeneratedClientAnalyzeStatusOptionalParams as AnalyzeActionsOperationStatusOptions,\n  JobManifestTasks as GeneratedActions,\n  State,\n  TextDocumentBatchStatistics,\n  TextDocumentInput\n} from \"../../generated/models\";\nimport {\n  AnalyzeActionsResult,\n  PagedAsyncIterableAnalyzeActionsResult,\n  PagedAnalyzeActionsResult,\n  createAnalyzeActionsResult\n} from \"../../analyzeActionsResult\";\nimport { PageSettings } from \"@azure/core-paging\";\nimport { getOperationId, handleInvalidDocumentBatch, nextLinkToTopAndSkip } from \"../../util\";\nimport { AnalysisPollOperation, AnalysisPollOperationState, OperationMetadata } from \"../poller\";\nimport { GeneratedClient as Client } from \"../../generated\";\nimport { SpanStatusCode } from \"@azure/core-tracing\";\nimport { createSpan } from \"../../tracing\";\nimport { logger } from \"../../logger\";\nexport { State };\n\n/**\n * @internal\n */\ninterface AnalyzeResultsWithPagination {\n  result: AnalyzeActionsResult;\n  top?: number;\n  skip?: number;\n}\n\n/**\n * The metadata for beginAnalyzeActions operations.\n */\nexport interface AnalyzeActionsOperationMetadata extends OperationMetadata {\n  /**\n   * Number of successfully completed actions.\n   */\n  actionsSucceededCount: number;\n  /**\n   * Number of failed actions.\n   */\n  actionsFailedCount: number;\n  /**\n   * Number of actions still in progress.\n   */\n  actionsInProgressCount: number;\n  /**\n   * The operation's display name.\n   */\n  displayName?: string;\n}\n\n/**\n * @internal\n */\ninterface AnalyzeActionsOperationStatus {\n  done: boolean;\n  /**\n   * Statistics about the input document batch and how it was processed\n   * by the service. This property will have a value when includeStatistics is set to true\n   * in the client call.\n   */\n  statistics?: TextDocumentBatchStatistics;\n  operationMetdata: Omit<AnalyzeActionsOperationMetadata, \"operationId\">;\n}\n\n/**\n * @internal\n */\ninterface BeginAnalyzeInternalOptions extends OperationOptions {\n  displayName?: string;\n}\n\n/**\n * Options for the begin analyze actions operation.\n */\nexport interface BeginAnalyzeActionsOptions extends OperationOptions {\n  /**\n   * Delay to wait until next poll, in milliseconds.\n   */\n  updateIntervalInMs?: number;\n  /**\n   * A serialized poller which can be used to resume an existing paused Long-Running-Operation.\n   */\n  resumeFrom?: string;\n  /**\n   * If set to true, response will contain input and document level statistics.\n   */\n  includeStatistics?: boolean;\n  /**\n   * The operation's display name.\n   */\n  displayName?: string;\n}\n\n/**\n * The state of the begin analyze polling operation.\n */\nexport interface AnalyzeActionsOperationState\n  extends AnalysisPollOperationState<PagedAnalyzeActionsResult>,\n    AnalyzeActionsOperationMetadata {}\n\n/**\n * @internal\n */\nfunction getMetaInfoFromResponse(\n  response: AnalyzeJobState\n): Omit<AnalyzeActionsOperationMetadata, \"operationId\"> {\n  return {\n    createdOn: response.createdDateTime,\n    lastModifiedOn: response.lastUpdateDateTime,\n    expiresOn: response.expirationDateTime,\n    status: response.status,\n    actionsSucceededCount: response.tasks.completed,\n    actionsFailedCount: response.tasks.failed,\n    actionsInProgressCount: response.tasks.inProgress,\n    displayName: response.displayName\n  };\n}\n\n/**\n * Class that represents a poller that waits for results of the analyze\n * operation.\n * @internal\n */\nexport class BeginAnalyzeActionsPollerOperation extends AnalysisPollOperation<\n  AnalyzeActionsOperationState,\n  PagedAnalyzeActionsResult\n> {\n  constructor(\n    public state: AnalyzeActionsOperationState,\n    // eslint-disable-next-line @azure/azure-sdk/ts-use-interface-parameters\n    private client: Client,\n    private documents: TextDocumentInput[],\n    private actions: GeneratedActions,\n    private options: BeginAnalyzeActionsOptions = {}\n  ) {\n    super(state);\n  }\n\n  /**\n   * should be called only after all the status of the analyze actions operations became\n   * \"succeeded\" and it returns an iterator for the results and provides a\n   * byPage method to return the results paged.\n   */\n  private listAnalyzeActionsResults(\n    operationId: string,\n    options: AnalyzeActionsOperationStatusOptions = {}\n  ): PagedAsyncIterableAnalyzeActionsResult {\n    const iter = this._listAnalyzeActionsResultsPaged(operationId, options);\n    return {\n      next() {\n        return iter.next();\n      },\n      [Symbol.asyncIterator]() {\n        return this;\n      },\n      byPage: (settings?: PageSettings) => {\n        const pageOptions = { ...options, top: settings?.maxPageSize };\n        return this._listAnalyzeActionsResultsPaged(operationId, pageOptions);\n      }\n    };\n  }\n\n  /**\n   * returns an iterator to arrays of the results of an analyze actions operation.\n   */\n  private async *_listAnalyzeActionsResultsPaged(\n    operationId: string,\n    options?: AnalyzeActionsOperationStatusOptions\n  ): AsyncIterableIterator<AnalyzeActionsResult> {\n    let response = await this._listAnalyzeActionsResultsSinglePage(operationId, options);\n    yield response.result;\n    while (response.skip) {\n      const optionsWithContinuation: AnalyzeActionsOperationStatusOptions = {\n        ...options,\n        top: response.top,\n        skip: response.skip\n      };\n      response = await this._listAnalyzeActionsResultsSinglePage(\n        operationId,\n        optionsWithContinuation\n      );\n      yield response.result;\n    }\n  }\n\n  /**\n   * returns an iterator to arrays of the sorted results of an analyze actions operation.\n   */\n  private async _listAnalyzeActionsResultsSinglePage(\n    operationId: string,\n    options?: AnalyzeActionsOperationStatusOptions\n  ): Promise<AnalyzeResultsWithPagination> {\n    const { span, updatedOptions: finalOptions } = createSpan(\n      \"TextAnalyticsClient-_listAnalyzeActionsResultsSinglePage\",\n      options || {}\n    );\n    try {\n      const response = await this.client.analyzeStatus(operationId, finalOptions);\n      const result = createAnalyzeActionsResult(response, this.documents);\n      return response.nextLink\n        ? { result, ...nextLinkToTopAndSkip(response.nextLink) }\n        : { result };\n    } catch (e) {\n      span.setStatus({\n        code: SpanStatusCode.ERROR,\n        message: e.message\n      });\n      throw e;\n    } finally {\n      span.end();\n    }\n  }\n\n  /**\n   * returns whether the analyze actions operation is done and if so returns also\n   * statistics.\n   */\n  private async getAnalyzeActionsOperationStatus(\n    operationId: string,\n    options?: AnalyzeActionsOperationStatusOptions\n  ): Promise<AnalyzeActionsOperationStatus> {\n    const { span, updatedOptions: finalOptions } = createSpan(\n      \"TextAnalyticsClient-getAnalyzeActionsOperationStatus\",\n      options || {}\n    );\n    try {\n      const response = await this.client.analyzeStatus(operationId, finalOptions);\n      switch (response.status) {\n        case \"notStarted\":\n        case \"running\":\n          break;\n        default: {\n          return {\n            done: true,\n            statistics: response.statistics,\n            operationMetdata: getMetaInfoFromResponse(response)\n          };\n        }\n      }\n      return { done: false, operationMetdata: getMetaInfoFromResponse(response) };\n    } catch (e) {\n      span.setStatus({\n        code: SpanStatusCode.ERROR,\n        message: e.message\n      });\n      throw e;\n    } finally {\n      span.end();\n    }\n  }\n\n  private async beginAnalyzeActions(\n    documents: TextDocumentInput[],\n    actions: GeneratedActions,\n    options?: BeginAnalyzeInternalOptions\n  ): Promise<BeginAnalyzeResponse> {\n    const { span, updatedOptions: finalOptions } = createSpan(\n      \"TextAnalyticsClient-beginAnalyze\",\n      options || {}\n    );\n\n    try {\n      return await this.client.analyze({\n        body: {\n          analysisInput: { documents: documents },\n          tasks: actions,\n          displayName: options?.displayName\n        },\n        ...finalOptions\n      });\n    } catch (e) {\n      const exception = handleInvalidDocumentBatch(e);\n      span.setStatus({\n        code: SpanStatusCode.ERROR,\n        message: exception.message\n      });\n      throw exception;\n    } finally {\n      span.end();\n    }\n  }\n\n  async update(\n    options: {\n      abortSignal?: AbortSignalLike;\n      fireProgress?: (state: AnalyzeActionsOperationState) => void;\n    } = {}\n  ): Promise<BeginAnalyzeActionsPollerOperation> {\n    const state = this.state;\n    const updatedAbortSignal = options.abortSignal;\n    if (!state.isStarted) {\n      state.isStarted = true;\n      const response = await this.beginAnalyzeActions(this.documents, this.actions, {\n        displayName: this.options.displayName,\n        tracingOptions: this.options.tracingOptions,\n        requestOptions: this.options.requestOptions,\n        abortSignal: updatedAbortSignal ? updatedAbortSignal : this.options.abortSignal\n      });\n      if (!response.operationLocation) {\n        throw new Error(\n          \"Expects a valid 'operationLocation' to retrieve analyze results but did not find any\"\n        );\n      }\n      state.operationId = getOperationId(response.operationLocation);\n    }\n\n    const operationStatus = await this.getAnalyzeActionsOperationStatus(state.operationId!, {\n      abortSignal: updatedAbortSignal ? updatedAbortSignal : options.abortSignal,\n      includeStatistics: this.options.includeStatistics,\n      tracingOptions: this.options.tracingOptions\n    });\n\n    state.createdOn = operationStatus.operationMetdata.createdOn;\n    state.expiresOn = operationStatus.operationMetdata.expiresOn;\n    state.lastModifiedOn = operationStatus.operationMetdata.lastModifiedOn;\n    state.status = operationStatus.operationMetdata.status;\n    state.actionsSucceededCount = operationStatus.operationMetdata.actionsSucceededCount;\n    state.actionsFailedCount = operationStatus.operationMetdata.actionsFailedCount;\n    state.actionsInProgressCount = operationStatus.operationMetdata.actionsInProgressCount;\n    state.displayName = operationStatus.operationMetdata?.displayName;\n\n    if (!state.isCompleted && operationStatus.done) {\n      const pagedIterator = this.listAnalyzeActionsResults(state.operationId!, {\n        abortSignal: this.options.abortSignal,\n        tracingOptions: this.options.tracingOptions,\n        includeStatistics: this.options.includeStatistics,\n        onResponse: this.options.onResponse,\n        serializerOptions: this.options.serializerOptions\n      });\n      // Attach stats if the service starts to return them\n      // https://github.com/Azure/azure-sdk-for-js/issues/14139\n      // state.result = Object.assign(pagedIterator, {\n      //   statistics: operationStatus.statistics\n      // });\n      state.result = pagedIterator;\n      state.isCompleted = true;\n    }\n\n    if (typeof options.fireProgress === \"function\") {\n      options.fireProgress(state);\n    }\n    return this;\n  }\n\n  async cancel(): Promise<BeginAnalyzeActionsPollerOperation> {\n    const state = this.state;\n    logger.warning(`The service does not yet support cancellation for beginAnalyze.`);\n    state.isCancelled = true;\n    return this;\n  }\n}\n"]}
+{"version":3,"file":"operation.js","sourceRoot":"","sources":["../../../../src/lro/analyze/operation.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;;AAclC,OAAO,EAIL,0BAA0B,GAC3B,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EAAE,cAAc,EAAE,oBAAoB,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAC9E,OAAO,EAAE,qBAAqB,EAAiD,MAAM,WAAW,CAAC;AAEjG,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAoFtC;;GAEG;AACH,SAAS,uBAAuB,CAC9B,QAAyB;IAEzB,OAAO;QACL,SAAS,EAAE,QAAQ,CAAC,eAAe;QACnC,cAAc,EAAE,QAAQ,CAAC,kBAAkB;QAC3C,SAAS,EAAE,QAAQ,CAAC,kBAAkB;QACtC,MAAM,EAAE,QAAQ,CAAC,MAAM;QACvB,qBAAqB,EAAE,QAAQ,CAAC,KAAK,CAAC,SAAS;QAC/C,kBAAkB,EAAE,QAAQ,CAAC,KAAK,CAAC,MAAM;QACzC,sBAAsB,EAAE,QAAQ,CAAC,KAAK,CAAC,UAAU;QACjD,WAAW,EAAE,QAAQ,CAAC,WAAW;KAClC,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAM,OAAO,kCAAmC,SAAQ,qBAGvD;IACC,YACS,KAAmC;IAC1C,wEAAwE;IAChE,MAAc,EACd,OAAsB,EACtB,SAA8B,EAC9B,OAAyB,EACzB,UAAsC,EAAE;QAEhD,KAAK,CAAC,KAAK,CAAC,CAAC;QARN,UAAK,GAAL,KAAK,CAA8B;QAElC,WAAM,GAAN,MAAM,CAAQ;QACd,YAAO,GAAP,OAAO,CAAe;QACtB,cAAS,GAAT,SAAS,CAAqB;QAC9B,YAAO,GAAP,OAAO,CAAkB;QACzB,YAAO,GAAP,OAAO,CAAiC;IAGlD,CAAC;IAED;;;;OAIG;IACK,yBAAyB,CAC/B,WAAmB,EACnB,UAAgD,EAAE;QAElD,MAAM,IAAI,GAAG,IAAI,CAAC,+BAA+B,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QACxE,OAAO;YACL,IAAI;gBACF,OAAO,IAAI,CAAC,IAAI,EAAE,CAAC;YACrB,CAAC;YACD,CAAC,MAAM,CAAC,aAAa,CAAC;gBACpB,OAAO,IAAI,CAAC;YACd,CAAC;YACD,MAAM,EAAE,CAAC,QAAuB,EAAE,EAAE;gBAClC,MAAM,WAAW,mCAAQ,OAAO,KAAE,GAAG,EAAE,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,CAAE,WAAW,GAAE,CAAC;gBAC/D,OAAO,IAAI,CAAC,+BAA+B,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC;YACxE,CAAC;SACF,CAAC;IACJ,CAAC;IAED;;OAEG;IACY,+BAA+B,CAC5C,WAAmB,EACnB,OAA8C;;YAE9C,IAAI,QAAQ,GAAG,cAAM,IAAI,CAAC,oCAAoC,CAAC,WAAW,EAAE,OAAO,CAAC,CAAA,CAAC;YACrF,oBAAM,QAAQ,CAAC,MAAM,CAAA,CAAC;YACtB,OAAO,QAAQ,CAAC,IAAI,EAAE;gBACpB,MAAM,uBAAuB,mCACxB,OAAO,KACV,GAAG,EAAE,QAAQ,CAAC,GAAG,EACjB,IAAI,EAAE,QAAQ,CAAC,IAAI,GACpB,CAAC;gBACF,QAAQ,GAAG,cAAM,IAAI,CAAC,oCAAoC,CACxD,WAAW,EACX,uBAAuB,CACxB,CAAA,CAAC;gBACF,oBAAM,QAAQ,CAAC,MAAM,CAAA,CAAC;aACvB;QACH,CAAC;KAAA;IAED;;OAEG;IACK,KAAK,CAAC,oCAAoC,CAChD,WAAmB,EACnB,OAA8C;QAE9C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAC1C,0DAA0D,EAC1D,OAAO,IAAI,EAAE,EACb,CAAC,YAAY,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,WAAW,EAAE,YAAY,CAAC,CACvE,CAAC;QACF,MAAM,MAAM,GAAG,0BAA0B,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QACpE,OAAO,QAAQ,CAAC,QAAQ,CAAC,CAAC,iBAAG,MAAM,IAAK,oBAAoB,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAG,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC;IACjG,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,gCAAgC,CAC5C,WAAmB,EACnB,OAA8C;QAE9C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAC1C,sDAAsD,EACtD,OAAO,IAAI,EAAE,EACb,CAAC,YAAY,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,WAAW,EAAE,YAAY,CAAC,CACvE,CAAC;QACF,QAAQ,QAAQ,CAAC,MAAM,EAAE;YACvB,KAAK,YAAY,CAAC;YAClB,KAAK,SAAS;gBACZ,MAAM;YACR,OAAO,CAAC,CAAC;gBACP,OAAO;oBACL,IAAI,EAAE,IAAI;oBACV,UAAU,EAAE,QAAQ,CAAC,UAAU;oBAC/B,gBAAgB,EAAE,uBAAuB,CAAC,QAAQ,CAAC;iBACpD,CAAC;aACH;SACF;QACD,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,gBAAgB,EAAE,uBAAuB,CAAC,QAAQ,CAAC,EAAE,CAAC;IAC9E,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAC/B,SAA8B,EAC9B,OAAyB,EACzB,OAAqC;QAErC,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAC1B,kCAAkC,EAClC,OAAO,IAAI,EAAE,EACb,CAAC,YAAY,EAAE,EAAE,CACf,UAAU,CACR,IAAI,CAAC,MAAM,CAAC,OAAO,iBACjB,IAAI,EAAE;gBACJ,aAAa,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE;gBACvC,KAAK,EAAE,OAAO;gBACd,WAAW,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW;aAClC,IACE,YAAY,EACf,CACH,CACJ,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CACV,UAGI,EAAE;;QAEN,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACzB,MAAM,kBAAkB,GAAG,OAAO,CAAC,WAAW,CAAC;QAC/C,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE;YACpB,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC;YACvB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,EAAE;gBAC5E,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW;gBACrC,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;gBAC3C,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;gBAC3C,WAAW,EAAE,kBAAkB,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW;aAChF,CAAC,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,iBAAiB,EAAE;gBAC/B,MAAM,IAAI,KAAK,CACb,sFAAsF,CACvF,CAAC;aACH;YACD,KAAK,CAAC,WAAW,GAAG,cAAc,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC;SAChE;QAED,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,gCAAgC,CAAC,KAAK,CAAC,WAAY,EAAE;YACtF,WAAW,EAAE,kBAAkB,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW;YAC1E,iBAAiB,EAAE,IAAI,CAAC,OAAO,CAAC,iBAAiB;YACjD,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;SAC5C,CAAC,CAAC;QAEH,KAAK,CAAC,SAAS,GAAG,eAAe,CAAC,gBAAgB,CAAC,SAAS,CAAC;QAC7D,KAAK,CAAC,SAAS,GAAG,eAAe,CAAC,gBAAgB,CAAC,SAAS,CAAC;QAC7D,KAAK,CAAC,cAAc,GAAG,eAAe,CAAC,gBAAgB,CAAC,cAAc,CAAC;QACvE,KAAK,CAAC,MAAM,GAAG,eAAe,CAAC,gBAAgB,CAAC,MAAM,CAAC;QACvD,KAAK,CAAC,qBAAqB,GAAG,eAAe,CAAC,gBAAgB,CAAC,qBAAqB,CAAC;QACrF,KAAK,CAAC,kBAAkB,GAAG,eAAe,CAAC,gBAAgB,CAAC,kBAAkB,CAAC;QAC/E,KAAK,CAAC,sBAAsB,GAAG,eAAe,CAAC,gBAAgB,CAAC,sBAAsB,CAAC;QACvF,KAAK,CAAC,WAAW,GAAG,MAAA,eAAe,CAAC,gBAAgB,0CAAE,WAAW,CAAC;QAElE,IAAI,CAAC,KAAK,CAAC,WAAW,IAAI,eAAe,CAAC,IAAI,EAAE;YAC9C,MAAM,aAAa,GAAG,IAAI,CAAC,yBAAyB,CAAC,KAAK,CAAC,WAAY,EAAE;gBACvE,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW;gBACrC,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;gBAC3C,iBAAiB,EAAE,IAAI,CAAC,OAAO,CAAC,iBAAiB;gBACjD,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU;gBACnC,iBAAiB,EAAE,IAAI,CAAC,OAAO,CAAC,iBAAiB;aAClD,CAAC,CAAC;YACH,oDAAoD;YACpD,yDAAyD;YACzD,gDAAgD;YAChD,2CAA2C;YAC3C,MAAM;YACN,KAAK,CAAC,MAAM,GAAG,aAAa,CAAC;YAC7B,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC;SAC1B;QAED,IAAI,OAAO,OAAO,CAAC,YAAY,KAAK,UAAU,EAAE;YAC9C,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;SAC7B;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,MAAM;QACV,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACzB,MAAM,CAAC,OAAO,CAAC,iEAAiE,CAAC,CAAC;QAClF,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC;QACzB,OAAO,IAAI,CAAC;IACd,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\nimport { OperationOptions } from \"@azure/core-client\";\n\nimport { AbortSignalLike } from \"@azure/abort-controller\";\n\nimport {\n  AnalyzeStatusOptionalParams as AnalyzeActionsOperationStatusOptions,\n  AnalyzeJobState,\n  AnalyzeResponse as BeginAnalyzeResponse,\n  JobManifestTasks as GeneratedActions,\n  TextDocumentBatchStatistics,\n  TextDocumentInput,\n} from \"../../generated/models\";\nimport {\n  AnalyzeActionsResult,\n  PagedAnalyzeActionsResult,\n  PagedAsyncIterableAnalyzeActionsResult,\n  createAnalyzeActionsResult,\n} from \"../../analyzeActionsResult\";\nimport { PageSettings } from \"@azure/core-paging\";\nimport { getOperationId, nextLinkToTopAndSkip, throwError } from \"../../util\";\nimport { AnalysisPollOperation, AnalysisPollOperationState, OperationMetadata } from \"../poller\";\nimport { GeneratedClient as Client } from \"../../generated\";\nimport { logger } from \"../../logger\";\nimport { TracingClient } from \"@azure/core-tracing\";\n\n/**\n * @internal\n */\ninterface AnalyzeResultsWithPagination {\n  result: AnalyzeActionsResult;\n  top?: number;\n  skip?: number;\n}\n\n/**\n * The metadata for beginAnalyzeActions operations.\n */\nexport interface AnalyzeActionsOperationMetadata extends OperationMetadata {\n  /**\n   * Number of successfully completed actions.\n   */\n  actionsSucceededCount: number;\n  /**\n   * Number of failed actions.\n   */\n  actionsFailedCount: number;\n  /**\n   * Number of actions still in progress.\n   */\n  actionsInProgressCount: number;\n  /**\n   * The operation's display name.\n   */\n  displayName?: string;\n}\n\n/**\n * @internal\n */\ninterface AnalyzeActionsOperationStatus {\n  done: boolean;\n  /**\n   * Statistics about the input document batch and how it was processed\n   * by the service. This property will have a value when includeStatistics is set to true\n   * in the client call.\n   */\n  statistics?: TextDocumentBatchStatistics;\n  operationMetdata: Omit<AnalyzeActionsOperationMetadata, \"operationId\">;\n}\n\n/**\n * @internal\n */\ninterface BeginAnalyzeInternalOptions extends OperationOptions {\n  displayName?: string;\n}\n\n/**\n * Options for the begin analyze actions operation.\n */\nexport interface BeginAnalyzeActionsOptions extends OperationOptions {\n  /**\n   * Delay to wait until next poll, in milliseconds.\n   */\n  updateIntervalInMs?: number;\n  /**\n   * A serialized poller which can be used to resume an existing paused Long-Running-Operation.\n   */\n  resumeFrom?: string;\n  /**\n   * If set to true, response will contain input and document level statistics.\n   */\n  includeStatistics?: boolean;\n  /**\n   * The operation's display name.\n   */\n  displayName?: string;\n}\n\n/**\n * The state of the begin analyze polling operation.\n */\nexport interface AnalyzeActionsOperationState\n  extends AnalysisPollOperationState<PagedAnalyzeActionsResult>,\n    AnalyzeActionsOperationMetadata {}\n\n/**\n * @internal\n */\nfunction getMetaInfoFromResponse(\n  response: AnalyzeJobState\n): Omit<AnalyzeActionsOperationMetadata, \"operationId\"> {\n  return {\n    createdOn: response.createdDateTime,\n    lastModifiedOn: response.lastUpdateDateTime,\n    expiresOn: response.expirationDateTime,\n    status: response.status,\n    actionsSucceededCount: response.tasks.completed,\n    actionsFailedCount: response.tasks.failed,\n    actionsInProgressCount: response.tasks.inProgress,\n    displayName: response.displayName,\n  };\n}\n\n/**\n * Class that represents a poller that waits for results of the analyze\n * operation.\n * @internal\n */\nexport class BeginAnalyzeActionsPollerOperation extends AnalysisPollOperation<\n  AnalyzeActionsOperationState,\n  PagedAnalyzeActionsResult\n> {\n  constructor(\n    public state: AnalyzeActionsOperationState,\n    // eslint-disable-next-line @azure/azure-sdk/ts-use-interface-parameters\n    private client: Client,\n    private tracing: TracingClient,\n    private documents: TextDocumentInput[],\n    private actions: GeneratedActions,\n    private options: BeginAnalyzeActionsOptions = {}\n  ) {\n    super(state);\n  }\n\n  /**\n   * should be called only after all the status of the analyze actions operations became\n   * \"succeeded\" and it returns an iterator for the results and provides a\n   * byPage method to return the results paged.\n   */\n  private listAnalyzeActionsResults(\n    operationId: string,\n    options: AnalyzeActionsOperationStatusOptions = {}\n  ): PagedAsyncIterableAnalyzeActionsResult {\n    const iter = this._listAnalyzeActionsResultsPaged(operationId, options);\n    return {\n      next() {\n        return iter.next();\n      },\n      [Symbol.asyncIterator]() {\n        return this;\n      },\n      byPage: (settings?: PageSettings) => {\n        const pageOptions = { ...options, top: settings?.maxPageSize };\n        return this._listAnalyzeActionsResultsPaged(operationId, pageOptions);\n      },\n    };\n  }\n\n  /**\n   * returns an iterator to arrays of the results of an analyze actions operation.\n   */\n  private async *_listAnalyzeActionsResultsPaged(\n    operationId: string,\n    options?: AnalyzeActionsOperationStatusOptions\n  ): AsyncIterableIterator<AnalyzeActionsResult> {\n    let response = await this._listAnalyzeActionsResultsSinglePage(operationId, options);\n    yield response.result;\n    while (response.skip) {\n      const optionsWithContinuation: AnalyzeActionsOperationStatusOptions = {\n        ...options,\n        top: response.top,\n        skip: response.skip,\n      };\n      response = await this._listAnalyzeActionsResultsSinglePage(\n        operationId,\n        optionsWithContinuation\n      );\n      yield response.result;\n    }\n  }\n\n  /**\n   * returns an iterator to arrays of the sorted results of an analyze actions operation.\n   */\n  private async _listAnalyzeActionsResultsSinglePage(\n    operationId: string,\n    options?: AnalyzeActionsOperationStatusOptions\n  ): Promise<AnalyzeResultsWithPagination> {\n    const response = await this.tracing.withSpan(\n      \"TextAnalyticsClient-_listAnalyzeActionsResultsSinglePage\",\n      options || {},\n      (finalOptions) => this.client.analyzeStatus(operationId, finalOptions)\n    );\n    const result = createAnalyzeActionsResult(response, this.documents);\n    return response.nextLink ? { result, ...nextLinkToTopAndSkip(response.nextLink) } : { result };\n  }\n\n  /**\n   * returns whether the analyze actions operation is done and if so returns also\n   * statistics.\n   */\n  private async getAnalyzeActionsOperationStatus(\n    operationId: string,\n    options?: AnalyzeActionsOperationStatusOptions\n  ): Promise<AnalyzeActionsOperationStatus> {\n    const response = await this.tracing.withSpan(\n      \"TextAnalyticsClient-getAnalyzeActionsOperationStatus\",\n      options || {},\n      (finalOptions) => this.client.analyzeStatus(operationId, finalOptions)\n    );\n    switch (response.status) {\n      case \"notStarted\":\n      case \"running\":\n        break;\n      default: {\n        return {\n          done: true,\n          statistics: response.statistics,\n          operationMetdata: getMetaInfoFromResponse(response),\n        };\n      }\n    }\n    return { done: false, operationMetdata: getMetaInfoFromResponse(response) };\n  }\n\n  private async beginAnalyzeActions(\n    documents: TextDocumentInput[],\n    actions: GeneratedActions,\n    options?: BeginAnalyzeInternalOptions\n  ): Promise<BeginAnalyzeResponse> {\n    return this.tracing.withSpan(\n      \"TextAnalyticsClient-beginAnalyze\",\n      options || {},\n      (finalOptions) =>\n        throwError(\n          this.client.analyze({\n            body: {\n              analysisInput: { documents: documents },\n              tasks: actions,\n              displayName: options?.displayName,\n            },\n            ...finalOptions,\n          })\n        )\n    );\n  }\n\n  async update(\n    options: {\n      abortSignal?: AbortSignalLike;\n      fireProgress?: (state: AnalyzeActionsOperationState) => void;\n    } = {}\n  ): Promise<BeginAnalyzeActionsPollerOperation> {\n    const state = this.state;\n    const updatedAbortSignal = options.abortSignal;\n    if (!state.isStarted) {\n      state.isStarted = true;\n      const response = await this.beginAnalyzeActions(this.documents, this.actions, {\n        displayName: this.options.displayName,\n        tracingOptions: this.options.tracingOptions,\n        requestOptions: this.options.requestOptions,\n        abortSignal: updatedAbortSignal ? updatedAbortSignal : this.options.abortSignal,\n      });\n      if (!response.operationLocation) {\n        throw new Error(\n          \"Expects a valid 'operationLocation' to retrieve analyze results but did not find any\"\n        );\n      }\n      state.operationId = getOperationId(response.operationLocation);\n    }\n\n    const operationStatus = await this.getAnalyzeActionsOperationStatus(state.operationId!, {\n      abortSignal: updatedAbortSignal ? updatedAbortSignal : options.abortSignal,\n      includeStatistics: this.options.includeStatistics,\n      tracingOptions: this.options.tracingOptions,\n    });\n\n    state.createdOn = operationStatus.operationMetdata.createdOn;\n    state.expiresOn = operationStatus.operationMetdata.expiresOn;\n    state.lastModifiedOn = operationStatus.operationMetdata.lastModifiedOn;\n    state.status = operationStatus.operationMetdata.status;\n    state.actionsSucceededCount = operationStatus.operationMetdata.actionsSucceededCount;\n    state.actionsFailedCount = operationStatus.operationMetdata.actionsFailedCount;\n    state.actionsInProgressCount = operationStatus.operationMetdata.actionsInProgressCount;\n    state.displayName = operationStatus.operationMetdata?.displayName;\n\n    if (!state.isCompleted && operationStatus.done) {\n      const pagedIterator = this.listAnalyzeActionsResults(state.operationId!, {\n        abortSignal: this.options.abortSignal,\n        tracingOptions: this.options.tracingOptions,\n        includeStatistics: this.options.includeStatistics,\n        onResponse: this.options.onResponse,\n        serializerOptions: this.options.serializerOptions,\n      });\n      // Attach stats if the service starts to return them\n      // https://github.com/Azure/azure-sdk-for-js/issues/14139\n      // state.result = Object.assign(pagedIterator, {\n      //   statistics: operationStatus.statistics\n      // });\n      state.result = pagedIterator;\n      state.isCompleted = true;\n    }\n\n    if (typeof options.fireProgress === \"function\") {\n      options.fireProgress(state);\n    }\n    return this;\n  }\n\n  async cancel(): Promise<BeginAnalyzeActionsPollerOperation> {\n    const state = this.state;\n    logger.warning(`The service does not yet support cancellation for beginAnalyze.`);\n    state.isCancelled = true;\n    return this;\n  }\n}\n"]}

package/dist-esm/src/lro/analyze/poller.js

@@ -2,7 +2,7 @@
 // Licensed under the MIT license.
 import { delay } from "../../util";
 import { AnalysisPoller } from "../poller";
-import { BeginAnalyzeActionsPollerOperation } from "./operation";
+import { BeginAnalyzeActionsPollerOperation, } from "./operation";
 /**
  * Class that represents a poller that waits for the analyze actions results.
  * @internal
@@ -10,12 +10,12 @@ import { BeginAnalyzeActionsPollerOperation } from "./operation";
 export class BeginAnalyzeActionsPoller extends AnalysisPoller {
     // eslint-disable-next-line @azure/azure-sdk/ts-use-interface-parameters
     constructor(pollerOptions) {
-        const { client, documents, actions, options, updateIntervalInMs = 5000, resumeFrom } = pollerOptions;
+        const { client, tracing, documents, actions, options, updateIntervalInMs = 5000, resumeFrom, } = pollerOptions;
         let state;
         if (resumeFrom) {
             state = JSON.parse(resumeFrom).state;
         }
-        const operation = new BeginAnalyzeActionsPollerOperation((state || {}), client, documents, actions, options);
+        const operation = new BeginAnalyzeActionsPollerOperation((state || {}), client, tracing, documents, actions, options);
         super(operation);
         this.updateIntervalInMs = updateIntervalInMs;
     }

package/dist-esm/src/lro/analyze/poller.js.map

@@ -1 +1 @@
-{"version":3,"file":"poller.js","sourceRoot":"","sources":["../../../../src/lro/analyze/poller.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAKlC,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAEnC,OAAO,EAAE,cAAc,EAAyB,MAAM,WAAW,CAAC;AAClE,OAAO,EACL,kCAAkC,EAGnC,MAAM,aAAa,CAAC;AAkBrB;;;GAGG;AACH,MAAM,OAAO,yBAA0B,SAAQ,cAG9C;IACC,wEAAwE;IACxE,YAAY,aAA0C;QACpD,MAAM,EACJ,MAAM,EACN,SAAS,EACT,OAAO,EACP,OAAO,EACP,kBAAkB,GAAG,IAAI,EACzB,UAAU,EACX,GAAG,aAAa,CAAC;QAElB,IAAI,KAA+C,CAAC;QAEpD,IAAI,UAAU,EAAE;YACd,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC;SACtC;QACD,MAAM,SAAS,GAAG,IAAI,kCAAkC,CACtD,CAAC,KAAK,IAAI,EAAE,CAAQ,EACpB,MAAM,EACN,SAAS,EACT,OAAO,EACP,OAAO,CACR,CAAC;QAEF,KAAK,CAAC,SAAS,CAAC,CAAC;QAEjB,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;IAC/C,CAAC;IAEM,KAAK;QACV,OAAO,KAAK,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;IACxC,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\nimport { PollerLike } from \"@azure/core-lro\";\nimport { PagedAnalyzeActionsResult } from \"../../analyzeActionsResult\";\nimport { JobManifestTasks as GeneratedActions } from \"../../generated/models\";\nimport { delay } from \"../../util\";\n\nimport { AnalysisPoller, AnalysisPollerOptions } from \"../poller\";\nimport {\n  BeginAnalyzeActionsPollerOperation,\n  AnalyzeActionsOperationState,\n  BeginAnalyzeActionsOptions\n} from \"./operation\";\n\n/**\n * @internal\n */\nexport interface AnalyzeActionsPollerOptions extends AnalysisPollerOptions {\n  actions: GeneratedActions;\n  readonly options: BeginAnalyzeActionsOptions;\n}\n\n/**\n * Result type of the Begin Analyze Actions Long-Running-Operation (LRO).\n */\nexport type AnalyzeActionsPollerLike = PollerLike<\n  AnalyzeActionsOperationState,\n  PagedAnalyzeActionsResult\n>;\n\n/**\n * Class that represents a poller that waits for the analyze actions results.\n * @internal\n */\nexport class BeginAnalyzeActionsPoller extends AnalysisPoller<\n  AnalyzeActionsOperationState,\n  PagedAnalyzeActionsResult\n> {\n  // eslint-disable-next-line @azure/azure-sdk/ts-use-interface-parameters\n  constructor(pollerOptions: AnalyzeActionsPollerOptions) {\n    const {\n      client,\n      documents,\n      actions,\n      options,\n      updateIntervalInMs = 5000,\n      resumeFrom\n    } = pollerOptions;\n\n    let state: AnalyzeActionsOperationState | undefined;\n\n    if (resumeFrom) {\n      state = JSON.parse(resumeFrom).state;\n    }\n    const operation = new BeginAnalyzeActionsPollerOperation(\n      (state || {}) as any,\n      client,\n      documents,\n      actions,\n      options\n    );\n\n    super(operation);\n\n    this.updateIntervalInMs = updateIntervalInMs;\n  }\n\n  public delay(): Promise<void> {\n    return delay(this.updateIntervalInMs);\n  }\n}\n"]}
+{"version":3,"file":"poller.js","sourceRoot":"","sources":["../../../../src/lro/analyze/poller.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAKlC,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAEnC,OAAO,EAAE,cAAc,EAAyB,MAAM,WAAW,CAAC;AAClE,OAAO,EAGL,kCAAkC,GACnC,MAAM,aAAa,CAAC;AAkBrB;;;GAGG;AACH,MAAM,OAAO,yBAA0B,SAAQ,cAG9C;IACC,wEAAwE;IACxE,YAAY,aAA0C;QACpD,MAAM,EACJ,MAAM,EACN,OAAO,EACP,SAAS,EACT,OAAO,EACP,OAAO,EACP,kBAAkB,GAAG,IAAI,EACzB,UAAU,GACX,GAAG,aAAa,CAAC;QAElB,IAAI,KAA+C,CAAC;QAEpD,IAAI,UAAU,EAAE;YACd,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC;SACtC;QACD,MAAM,SAAS,GAAG,IAAI,kCAAkC,CACtD,CAAC,KAAK,IAAI,EAAE,CAAQ,EACpB,MAAM,EACN,OAAO,EACP,SAAS,EACT,OAAO,EACP,OAAO,CACR,CAAC;QAEF,KAAK,CAAC,SAAS,CAAC,CAAC;QAEjB,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;IAC/C,CAAC;IAEM,KAAK;QACV,OAAO,KAAK,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;IACxC,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\nimport { PollerLike } from \"@azure/core-lro\";\nimport { PagedAnalyzeActionsResult } from \"../../analyzeActionsResult\";\nimport { JobManifestTasks as GeneratedActions } from \"../../generated/models\";\nimport { delay } from \"../../util\";\n\nimport { AnalysisPoller, AnalysisPollerOptions } from \"../poller\";\nimport {\n  AnalyzeActionsOperationState,\n  BeginAnalyzeActionsOptions,\n  BeginAnalyzeActionsPollerOperation,\n} from \"./operation\";\n\n/**\n * @internal\n */\ninterface AnalyzeActionsPollerOptions extends AnalysisPollerOptions {\n  actions: GeneratedActions;\n  readonly options: BeginAnalyzeActionsOptions;\n}\n\n/**\n * Result type of the Begin Analyze Actions Long-Running-Operation (LRO).\n */\nexport type AnalyzeActionsPollerLike = PollerLike<\n  AnalyzeActionsOperationState,\n  PagedAnalyzeActionsResult\n>;\n\n/**\n * Class that represents a poller that waits for the analyze actions results.\n * @internal\n */\nexport class BeginAnalyzeActionsPoller extends AnalysisPoller<\n  AnalyzeActionsOperationState,\n  PagedAnalyzeActionsResult\n> {\n  // eslint-disable-next-line @azure/azure-sdk/ts-use-interface-parameters\n  constructor(pollerOptions: AnalyzeActionsPollerOptions) {\n    const {\n      client,\n      tracing,\n      documents,\n      actions,\n      options,\n      updateIntervalInMs = 5000,\n      resumeFrom,\n    } = pollerOptions;\n\n    let state: AnalyzeActionsOperationState | undefined;\n\n    if (resumeFrom) {\n      state = JSON.parse(resumeFrom).state;\n    }\n    const operation = new BeginAnalyzeActionsPollerOperation(\n      (state || {}) as any,\n      client,\n      tracing,\n      documents,\n      actions,\n      options\n    );\n\n    super(operation);\n\n    this.updateIntervalInMs = updateIntervalInMs;\n  }\n\n  public delay(): Promise<void> {\n    return delay(this.updateIntervalInMs);\n  }\n}\n"]}

package/dist-esm/src/lro/health/operation.js

@@ -1,12 +1,10 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT license.
 import { __asyncDelegator, __asyncGenerator, __asyncValues, __await } from "tslib";
-import { makeHealthcareEntitiesResult, makeHealthcareEntitiesErrorResult } from "../../analyzeHealthcareEntitiesResult";
-import { addStrEncodingParam, getOperationId, handleInvalidDocumentBatch, nextLinkToTopAndSkip } from "../../util";
-import { AnalysisPollOperation } from "../poller";
+import { makeHealthcareEntitiesErrorResult, makeHealthcareEntitiesResult, } from "../../analyzeHealthcareEntitiesResult";
+import { addStrEncodingParam, getOperationId, nextLinkToTopAndSkip, throwError, } from "../../util";
+import { AnalysisPollOperation, } from "../poller";
 import { processAndCombineSuccessfulAndErroneousDocuments } from "../../textAnalyticsResult";
-import { SpanStatusCode } from "@azure/core-tracing";
-import { createSpan } from "../../tracing";
 /**
  * @internal
  */
@@ -15,7 +13,7 @@ function getMetaInfoFromResponse(response) {
         createdOn: response.createdDateTime,
         lastModifiedOn: response.lastUpdateDateTime,
         expiresOn: response.expirationDateTime,
-        status: response.status
+        status: response.status,
     };
 }
 /**
@@ -25,10 +23,11 @@ function getMetaInfoFromResponse(response) {
 export class BeginAnalyzeHealthcarePollerOperation extends AnalysisPollOperation {
     constructor(state, 
     // eslint-disable-next-line @azure/azure-sdk/ts-use-interface-parameters
-    client, documents, options = {}) {
+    client, tracing, documents, options = {}) {
         super(state);
         this.state = state;
         this.client = client;
+        this.tracing = tracing;
         this.documents = documents;
         this.options = options;
     }
@@ -49,7 +48,7 @@ export class BeginAnalyzeHealthcarePollerOperation extends AnalysisPollOperation
             byPage: (settings) => {
                 const pageOptions = Object.assign(Object.assign({}, options), { top: settings === null || settings === void 0 ? void 0 : settings.maxPageSize });
                 return this._listHealthcareEntitiesPaged(operationId, pageOptions);
-            }
+            },
         };
     }
     /**
@@ -91,27 +90,14 @@ export class BeginAnalyzeHealthcarePollerOperation extends AnalysisPollOperation
      * returns an iterator to arrays of the sorted results of a healthcare operation.
      */
     async _listHealthcareEntitiesSinglePage(operationId, options) {
-        const { span, updatedOptions: finalOptions } = createSpan("TextAnalyticsClient-_listHealthcareEntitiesSinglePage", options || {});
-        try {
-            const response = await this.client.healthStatus(operationId, finalOptions);
-            if (response.results) {
-                const result = processAndCombineSuccessfulAndErroneousDocuments(this.documents, response.results, makeHealthcareEntitiesResult, makeHealthcareEntitiesErrorResult);
-                return response.nextLink
-                    ? Object.assign({ result }, nextLinkToTopAndSkip(response.nextLink)) : { result };
-            }
-            else {
-                throw new Error("Healthcare action has succeeded but the there are no results!");
-            }
+        const response = await this.tracing.withSpan("TextAnalyticsClient-_listHealthcareEntitiesSinglePage", options || {}, (finalOptions) => this.client.healthStatus(operationId, finalOptions));
+        if (response.results) {
+            const result = processAndCombineSuccessfulAndErroneousDocuments(this.documents, response.results, makeHealthcareEntitiesResult, makeHealthcareEntitiesErrorResult);
+            return response.nextLink
+                ? Object.assign({ result }, nextLinkToTopAndSkip(response.nextLink)) : { result };
         }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message
-            });
-            throw e;
-        }
-        finally {
-            span.end();
+        else {
+            throw new Error("Healthcare action has succeeded but the there are no results!");
         }
     }
     /**
@@ -120,61 +106,34 @@ export class BeginAnalyzeHealthcarePollerOperation extends AnalysisPollOperation
      */
     async getHealthStatus(operationId, options) {
         var _a;
-        const { span, updatedOptions: finalOptions } = createSpan("TextAnalyticsClient-getHealthStatus", options || {});
-        try {
-            const response = await this.client.healthStatus(operationId, finalOptions);
-            switch (response.status) {
-                case "notStarted":
-                case "running":
-                    break;
-                case "failed": {
-                    const errors = (_a = response.errors) === null || _a === void 0 ? void 0 : _a.map((e) => `  code ${e.code}, message: '${e.message}'`).join("\n");
-                    const message = `Healthcare analysis failed. Error(s): ${errors || ""}`;
-                    throw new Error(message);
+        const response = await this.tracing.withSpan("TextAnalyticsClient-getHealthStatus", options || {}, (finalOptions) => this.client.healthStatus(operationId, finalOptions));
+        switch (response.status) {
+            case "notStarted":
+            case "running":
+                break;
+            case "failed": {
+                const errors = (_a = response.errors) === null || _a === void 0 ? void 0 : _a.map((e) => `  code ${e.code}, message: '${e.message}'`).join("\n");
+                const message = `Healthcare analysis failed. Error(s): ${errors || ""}`;
+                throw new Error(message);
+            }
+            default: {
+                if (response.results) {
+                    return {
+                        done: true,
+                        statistics: response.results.statistics,
+                        modelVersion: response.results.modelVersion,
+                        operationMetdata: getMetaInfoFromResponse(response),
+                    };
                 }
-                default: {
-                    if (response.results) {
-                        return {
-                            done: true,
-                            statistics: response.results.statistics,
-                            modelVersion: response.results.modelVersion,
-                            operationMetdata: getMetaInfoFromResponse(response)
-                        };
-                    }
-                    else {
-                        throw new Error("Healthcare action has finished but the there are no results!");
-                    }
+                else {
+                    throw new Error("Healthcare action has finished but the there are no results!");
                 }
             }
-            return { done: false, operationMetdata: getMetaInfoFromResponse(response) };
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message
-            });
-            throw e;
-        }
-        finally {
-            span.end();
         }
+        return { done: false, operationMetdata: getMetaInfoFromResponse(response) };
     }
     async beginAnalyzeHealthcare(documents, options) {
-        const { span, updatedOptions: finalOptions } = createSpan("TextAnalyticsClient-beginAnalyzeHealthcare", addStrEncodingParam(options || {}));
-        try {
-            return await this.client.health({ documents: documents }, finalOptions);
-        }
-        catch (e) {
-            const exception = handleInvalidDocumentBatch(e);
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: exception.message
-            });
-            throw exception;
-        }
-        finally {
-            span.end();
-        }
+        return this.tracing.withSpan("TextAnalyticsClient-beginAnalyzeHealthcare", addStrEncodingParam(options || {}), (finalOptions) => throwError(this.client.health({ documents: documents }, finalOptions)));
     }
     async update(options = {}) {
         const state = this.state;
@@ -187,7 +146,7 @@ export class BeginAnalyzeHealthcarePollerOperation extends AnalysisPollOperation
                 abortSignal: updatedAbortSignal ? updatedAbortSignal : options.abortSignal,
                 modelVersion: this.options.modelVersion,
                 stringIndexType: this.options.stringIndexType,
-                loggingOptOut: this.options.disableServiceLogs
+                loggingOptOut: this.options.disableServiceLogs,
             });
             if (!response.operationLocation) {
                 throw new Error("Expects a valid 'operationLocation' to retrieve health results but did not find any");
@@ -199,7 +158,7 @@ export class BeginAnalyzeHealthcarePollerOperation extends AnalysisPollOperation
             includeStatistics: this.options.includeStatistics,
             tracingOptions: this.options.tracingOptions,
             onResponse: this.options.onResponse,
-            serializerOptions: this.options.serializerOptions
+            serializerOptions: this.options.serializerOptions,
         });
         state.createdOn = operationStatus.operationMetdata.createdOn;
         state.expiresOn = operationStatus.operationMetdata.expiresOn;
@@ -208,11 +167,11 @@ export class BeginAnalyzeHealthcarePollerOperation extends AnalysisPollOperation
         if (!state.isCompleted && operationStatus.done) {
             const pagedIterator = this.listHealthcareEntitiesByPage(state.operationId, {
                 abortSignal: this.options.abortSignal,
-                tracingOptions: this.options.tracingOptions
+                tracingOptions: this.options.tracingOptions,
             });
             state.result = Object.assign(pagedIterator, {
                 statistics: operationStatus.statistics,
-                modelVersion: operationStatus.modelVersion
+                modelVersion: operationStatus.modelVersion,
             });
             state.isCompleted = true;
         }
@@ -226,7 +185,7 @@ export class BeginAnalyzeHealthcarePollerOperation extends AnalysisPollOperation
         if (state.operationId) {
             await this.client.cancelHealthJob(state.operationId, {
                 abortSignal: this.options.abortSignal,
-                tracingOptions: this.options.tracingOptions
+                tracingOptions: this.options.tracingOptions,
             });
         }
         state.isCancelled = true;

package/dist-esm/src/lro/health/operation.js.map

@@ -1 +1 @@
-{"version":3,"file":"operation.js","sourceRoot":"","sources":["../../../../src/lro/health/operation.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;;AAalC,OAAO,EAKL,4BAA4B,EAC5B,iCAAiC,EAClC,MAAM,uCAAuC,CAAC;AAE/C,OAAO,EACL,mBAAmB,EACnB,cAAc,EACd,0BAA0B,EAC1B,oBAAoB,EAErB,MAAM,YAAY,CAAC;AACpB,OAAO,EACL,qBAAqB,EAGtB,MAAM,WAAW,CAAC;AAEnB,OAAO,EAAE,gDAAgD,EAAE,MAAM,2BAA2B,CAAC;AAC7F,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAiF3C;;GAEG;AACH,SAAS,uBAAuB,CAC9B,QAA4B;IAE5B,OAAO;QACL,SAAS,EAAE,QAAQ,CAAC,eAAe;QACnC,cAAc,EAAE,QAAQ,CAAC,kBAAkB;QAC3C,SAAS,EAAE,QAAQ,CAAC,kBAAkB;QACtC,MAAM,EAAE,QAAQ,CAAC,MAAM;KACxB,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,OAAO,qCAAsC,SAAQ,qBAG1D;IACC,YACS,KAAsC;IAC7C,wEAAwE;IAChE,MAAc,EACd,SAA8B,EAC9B,UAAiD,EAAE;QAE3D,KAAK,CAAC,KAAK,CAAC,CAAC;QANN,UAAK,GAAL,KAAK,CAAiC;QAErC,WAAM,GAAN,MAAM,CAAQ;QACd,cAAS,GAAT,SAAS,CAAqB;QAC9B,YAAO,GAAP,OAAO,CAA4C;IAG7D,CAAC;IAED;;;;OAIG;IACK,4BAA4B,CAClC,WAAmB,EACnB,UAAsC,EAAE;QAExC,MAAM,IAAI,GAAG,IAAI,CAAC,uBAAuB,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QAChE,OAAO;YACL,IAAI;gBACF,OAAO,IAAI,CAAC,IAAI,EAAE,CAAC;YACrB,CAAC;YACD,CAAC,MAAM,CAAC,aAAa,CAAC;gBACpB,OAAO,IAAI,CAAC;YACd,CAAC;YACD,MAAM,EAAE,CAAC,QAAuB,EAAE,EAAE;gBAClC,MAAM,WAAW,mCAAQ,OAAO,KAAE,GAAG,EAAE,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,CAAE,WAAW,GAAE,CAAC;gBAC/D,OAAO,IAAI,CAAC,4BAA4B,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC;YACrE,CAAC;SACF,CAAC;IACJ,CAAC;IAED;;OAEG;IACY,uBAAuB,CACpC,WAAmB,EACnB,OAAoC;;;;gBAEpC,KAAyB,IAAA,KAAA,cAAA,IAAI,CAAC,4BAA4B,CAAC,WAAW,EAAE,OAAO,CAAC,CAAA,IAAA;oBAArE,MAAM,IAAI,WAAA,CAAA;oBACnB,cAAA,KAAK,CAAC,CAAC,iBAAA,cAAA,IAAI,CAAA,CAAA,CAAA,CAAC;iBACb;;;;;;;;;QACH,CAAC;KAAA;IAED;;OAEG;IACY,4BAA4B,CACzC,WAAmB,EACnB,OAAoC;;YAEpC,IAAI,QAAQ,GAAG,cAAM,IAAI,CAAC,iCAAiC,CAAC,WAAW,EAAE,OAAO,CAAC,CAAA,CAAC;YAClF,oBAAM,QAAQ,CAAC,MAAM,CAAA,CAAC;YACtB,OAAO,QAAQ,CAAC,IAAI,EAAE;gBACpB,MAAM,uBAAuB,mCACxB,OAAO,KACV,GAAG,EAAE,QAAQ,CAAC,GAAG,EACjB,IAAI,EAAE,QAAQ,CAAC,IAAI,GACpB,CAAC;gBACF,QAAQ,GAAG,cAAM,IAAI,CAAC,iCAAiC,CAAC,WAAW,EAAE,uBAAuB,CAAC,CAAA,CAAC;gBAC9F,oBAAM,QAAQ,CAAC,MAAM,CAAA,CAAC;aACvB;QACH,CAAC;KAAA;IAED;;OAEG;IACK,KAAK,CAAC,iCAAiC,CAC7C,WAAmB,EACnB,OAAoC;QAEpC,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,YAAY,EAAE,GAAG,UAAU,CACvD,uDAAuD,EACvD,OAAO,IAAI,EAAE,CACd,CAAC;QACF,IAAI;YACF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;YAC3E,IAAI,QAAQ,CAAC,OAAO,EAAE;gBACpB,MAAM,MAAM,GAAG,gDAAgD,CAC7D,IAAI,CAAC,SAAS,EACd,QAAQ,CAAC,OAAO,EAChB,4BAA4B,EAC5B,iCAAiC,CAClC,CAAC;gBACF,OAAO,QAAQ,CAAC,QAAQ;oBACtB,CAAC,iBAAG,MAAM,IAAK,oBAAoB,CAAC,QAAQ,CAAC,QAAQ,CAAC,EACtD,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC;aAChB;iBAAM;gBACL,MAAM,IAAI,KAAK,CAAC,+DAA+D,CAAC,CAAC;aAClF;SACF;QAAC,OAAO,CAAC,EAAE;YACV,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,cAAc,CAAC,KAAK;gBAC1B,OAAO,EAAE,CAAC,CAAC,OAAO;aACnB,CAAC,CAAC;YACH,MAAM,CAAC,CAAC;SACT;gBAAS;YACR,IAAI,CAAC,GAAG,EAAE,CAAC;SACZ;IACH,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,eAAe,CAC3B,WAAmB,EACnB,OAAoC;;QAEpC,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,YAAY,EAAE,GAAG,UAAU,CACvD,qCAAqC,EACrC,OAAO,IAAI,EAAE,CACd,CAAC;QACF,IAAI;YACF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;YAC3E,QAAQ,QAAQ,CAAC,MAAM,EAAE;gBACvB,KAAK,YAAY,CAAC;gBAClB,KAAK,SAAS;oBACZ,MAAM;gBACR,KAAK,QAAQ,CAAC,CAAC;oBACb,MAAM,MAAM,GAAG,MAAA,QAAQ,CAAC,MAAM,0CAC1B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,IAAI,eAAe,CAAC,CAAC,OAAO,GAAG,EACvD,IAAI,CAAC,IAAI,CAAC,CAAC;oBACd,MAAM,OAAO,GAAG,yCAAyC,MAAM,IAAI,EAAE,EAAE,CAAC;oBACxE,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC;iBAC1B;gBACD,OAAO,CAAC,CAAC;oBACP,IAAI,QAAQ,CAAC,OAAO,EAAE;wBACpB,OAAO;4BACL,IAAI,EAAE,IAAI;4BACV,UAAU,EAAE,QAAQ,CAAC,OAAO,CAAC,UAAU;4BACvC,YAAY,EAAE,QAAQ,CAAC,OAAO,CAAC,YAAY;4BAC3C,gBAAgB,EAAE,uBAAuB,CAAC,QAAQ,CAAC;yBACpD,CAAC;qBACH;yBAAM;wBACL,MAAM,IAAI,KAAK,CAAC,8DAA8D,CAAC,CAAC;qBACjF;iBACF;aACF;YACD,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,gBAAgB,EAAE,uBAAuB,CAAC,QAAQ,CAAC,EAAE,CAAC;SAC7E;QAAC,OAAO,CAAC,EAAE;YACV,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,cAAc,CAAC,KAAK;gBAC1B,OAAO,EAAE,CAAC,CAAC,OAAO;aACnB,CAAC,CAAC;YACH,MAAM,CAAC,CAAC;SACT;gBAAS;YACR,IAAI,CAAC,GAAG,EAAE,CAAC;SACZ;IACH,CAAC;IAEO,KAAK,CAAC,sBAAsB,CAClC,SAA8B,EAC9B,OAA+C;QAE/C,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,YAAY,EAAE,GAAG,UAAU,CACvD,4CAA4C,EAC5C,mBAAmB,CAAC,OAAO,IAAI,EAAE,CAAC,CACnC,CAAC;QAEF,IAAI;YACF,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,EAAE,YAAY,CAAC,CAAC;SACzE;QAAC,OAAO,CAAC,EAAE;YACV,MAAM,SAAS,GAAG,0BAA0B,CAAC,CAAC,CAAC,CAAC;YAChD,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,cAAc,CAAC,KAAK;gBAC1B,OAAO,EAAE,SAAS,CAAC,OAAO;aAC3B,CAAC,CAAC;YACH,MAAM,SAAS,CAAC;SACjB;gBAAS;YACR,IAAI,CAAC,GAAG,EAAE,CAAC;SACZ;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CACV,UAGI,EAAE;QAEN,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACzB,MAAM,kBAAkB,GAAG,OAAO,CAAC,WAAW,CAAC;QAC/C,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE;YACpB,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC;YACvB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,SAAS,EAAE;gBACjE,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;gBAC3C,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;gBAC3C,WAAW,EAAE,kBAAkB,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW;gBAC1E,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,YAAY;gBACvC,eAAe,EAAE,IAAI,CAAC,OAAO,CAAC,eAAe;gBAC7C,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,kBAAkB;aAC/C,CAAC,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,iBAAiB,EAAE;gBAC/B,MAAM,IAAI,KAAK,CACb,qFAAqF,CACtF,CAAC;aACH;YACD,KAAK,CAAC,WAAW,GAAG,cAAc,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC;SAChE;QACD,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,WAAY,EAAE;YACrE,WAAW,EAAE,kBAAkB,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW;YAC1E,iBAAiB,EAAE,IAAI,CAAC,OAAO,CAAC,iBAAiB;YACjD,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;YAC3C,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU;YACnC,iBAAiB,EAAE,IAAI,CAAC,OAAO,CAAC,iBAAiB;SAClD,CAAC,CAAC;QAEH,KAAK,CAAC,SAAS,GAAG,eAAe,CAAC,gBAAgB,CAAC,SAAS,CAAC;QAC7D,KAAK,CAAC,SAAS,GAAG,eAAe,CAAC,gBAAgB,CAAC,SAAS,CAAC;QAC7D,KAAK,CAAC,cAAc,GAAG,eAAe,CAAC,gBAAgB,CAAC,cAAc,CAAC;QACvE,KAAK,CAAC,MAAM,GAAG,eAAe,CAAC,gBAAgB,CAAC,MAAM,CAAC;QAEvD,IAAI,CAAC,KAAK,CAAC,WAAW,IAAI,eAAe,CAAC,IAAI,EAAE;YAC9C,MAAM,aAAa,GAAG,IAAI,CAAC,4BAA4B,CAAC,KAAK,CAAC,WAAY,EAAE;gBAC1E,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW;gBACrC,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;aAC5C,CAAC,CAAC;YACH,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,aAAa,EAAE;gBAC1C,UAAU,EAAE,eAAe,CAAC,UAAU;gBACtC,YAAY,EAAE,eAAe,CAAC,YAAa;aAC5C,CAAC,CAAC;YACH,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC;SAC1B;QACD,IAAI,OAAO,OAAO,CAAC,YAAY,KAAK,UAAU,EAAE;YAC9C,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;SAC7B;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,MAAM;QACV,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACzB,IAAI,KAAK,CAAC,WAAW,EAAE;YACrB,MAAM,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,KAAK,CAAC,WAAW,EAAE;gBACnD,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW;gBACrC,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;aAC5C,CAAC,CAAC;SACJ;QACD,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC;QACzB,OAAO,IAAI,CAAC;IACd,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\nimport { AbortSignalLike } from \"@azure/abort-controller\";\nimport { OperationOptions } from \"@azure/core-client\";\n\nimport {\n  GeneratedClientHealthResponse as BeginAnalyzeHealthcareResponse,\n  GeneratedClientHealthStatusOptionalParams as HealthcareJobStatusOptions,\n  HealthcareJobState,\n  State,\n  TextDocumentBatchStatistics,\n  TextDocumentInput\n} from \"../../generated/models\";\nimport {\n  AnalyzeHealthcareEntitiesResult,\n  AnalyzeHealthcareEntitiesResultArray,\n  PagedAsyncIterableAnalyzeHealthcareEntitiesResult,\n  PagedAnalyzeHealthcareEntitiesResult,\n  makeHealthcareEntitiesResult,\n  makeHealthcareEntitiesErrorResult\n} from \"../../analyzeHealthcareEntitiesResult\";\nimport { PageSettings } from \"@azure/core-paging\";\nimport {\n  addStrEncodingParam,\n  getOperationId,\n  handleInvalidDocumentBatch,\n  nextLinkToTopAndSkip,\n  StringIndexType\n} from \"../../util\";\nimport {\n  AnalysisPollOperation,\n  AnalysisPollOperationState,\n  OperationMetadata as AnalyzeHealthcareEntitiesOperationMetadata\n} from \"../poller\";\nimport { GeneratedClient as Client } from \"../../generated\";\nimport { processAndCombineSuccessfulAndErroneousDocuments } from \"../../textAnalyticsResult\";\nimport { SpanStatusCode } from \"@azure/core-tracing\";\nimport { createSpan } from \"../../tracing\";\nimport { TextAnalyticsOperationOptions } from \"../../textAnalyticsOperationOptions\";\nexport { State };\n\n/**\n * @internal\n */\ninterface AnalyzeHealthcareEntitiesResultWithPagination {\n  result: AnalyzeHealthcareEntitiesResultArray;\n  top?: number;\n  skip?: number;\n}\n\n/**\n * @internal\n */\ninterface HealthcareJobStatus {\n  done: boolean;\n  /**\n   * Statistics about the input document batch and how it was processed\n   * by the service. This property will have a value when includeStatistics is set to true\n   * in the client call.\n   */\n  statistics?: TextDocumentBatchStatistics;\n  /**\n   * The version of the text analytics model used by this operation on this\n   * batch of input documents.\n   */\n  modelVersion?: string;\n  operationMetdata: Omit<AnalyzeHealthcareEntitiesOperationMetadata, \"operationId\">;\n}\n\n/**\n * @internal\n */\ninterface BeginAnalyzeHealthcareInternalOptions extends OperationOptions {\n  /**\n   * This value indicates which model will be used for scoring. If a model-version is\n   * not specified, the API should default to the latest, non-preview version.\n   * For supported model versions, see operation-specific documentation, for example:\n   * https://docs.microsoft.com/azure/cognitive-services/text-analytics/how-tos/text-analytics-how-to-sentiment-analysis#model-versioning\n   */\n  modelVersion?: string;\n  /**\n   * Specifies the measurement unit used to calculate the offset and length properties.\n   * Possible units are \"TextElements_v8\", \"UnicodeCodePoint\", and \"Utf16CodeUnit\".\n   * The default is the JavaScript's default which is \"Utf16CodeUnit\".\n   */\n  stringIndexType?: StringIndexType;\n  /**\n   * If set to false, you opt-in to have your text input logged for troubleshooting. By default, Text Analytics\n   * will not log your input text for healthcare entities analysis. Setting this parameter to false,\n   * enables input logging.\n   */\n  loggingOptOut?: boolean;\n}\n\n/**\n * Options for the begin analyze healthcare entities operation.\n */\nexport interface BeginAnalyzeHealthcareEntitiesOptions extends TextAnalyticsOperationOptions {\n  /**\n   * Delay to wait until next poll, in milliseconds.\n   */\n  stringIndexType?: StringIndexType;\n  /**\n   * Delay to wait until next poll, in milliseconds.\n   */\n  updateIntervalInMs?: number;\n  /**\n   * A serialized poller which can be used to resume an existing paused Long-Running-Operation.\n   */\n  resumeFrom?: string;\n}\n\n/**\n * The state of the begin analyze healthcare polling operation.\n */\nexport interface AnalyzeHealthcareOperationState\n  extends AnalysisPollOperationState<PagedAnalyzeHealthcareEntitiesResult> {}\n\n/**\n * @internal\n */\nfunction getMetaInfoFromResponse(\n  response: HealthcareJobState\n): Omit<AnalyzeHealthcareEntitiesOperationMetadata, \"operationId\"> {\n  return {\n    createdOn: response.createdDateTime,\n    lastModifiedOn: response.lastUpdateDateTime,\n    expiresOn: response.expirationDateTime,\n    status: response.status\n  };\n}\n\n/**\n * Class that represents a poller that waits for the healthcare results.\n * @internal\n */\nexport class BeginAnalyzeHealthcarePollerOperation extends AnalysisPollOperation<\n  AnalyzeHealthcareOperationState,\n  PagedAnalyzeHealthcareEntitiesResult\n> {\n  constructor(\n    public state: AnalyzeHealthcareOperationState,\n    // eslint-disable-next-line @azure/azure-sdk/ts-use-interface-parameters\n    private client: Client,\n    private documents: TextDocumentInput[],\n    private options: BeginAnalyzeHealthcareEntitiesOptions = {}\n  ) {\n    super(state);\n  }\n\n  /**\n   * should be called only after all the status of the healthcare operations became\n   * \"succeeded\" and it returns an iterator for the results and provides a\n   * byPage method to return the results paged.\n   */\n  private listHealthcareEntitiesByPage(\n    operationId: string,\n    options: HealthcareJobStatusOptions = {}\n  ): PagedAsyncIterableAnalyzeHealthcareEntitiesResult {\n    const iter = this._listHealthcareEntities(operationId, options);\n    return {\n      next() {\n        return iter.next();\n      },\n      [Symbol.asyncIterator]() {\n        return this;\n      },\n      byPage: (settings?: PageSettings) => {\n        const pageOptions = { ...options, top: settings?.maxPageSize };\n        return this._listHealthcareEntitiesPaged(operationId, pageOptions);\n      }\n    };\n  }\n\n  /**\n   * returns an iterator to the results of a healthcare operation.\n   */\n  private async *_listHealthcareEntities(\n    operationId: string,\n    options?: HealthcareJobStatusOptions\n  ): AsyncIterableIterator<AnalyzeHealthcareEntitiesResult> {\n    for await (const page of this._listHealthcareEntitiesPaged(operationId, options)) {\n      yield* page;\n    }\n  }\n\n  /**\n   * returns an iterator to arrays of the results of a healthcare operation.\n   */\n  private async *_listHealthcareEntitiesPaged(\n    operationId: string,\n    options?: HealthcareJobStatusOptions\n  ): AsyncIterableIterator<AnalyzeHealthcareEntitiesResultArray> {\n    let response = await this._listHealthcareEntitiesSinglePage(operationId, options);\n    yield response.result;\n    while (response.skip) {\n      const optionsWithContinuation: HealthcareJobStatusOptions = {\n        ...options,\n        top: response.top,\n        skip: response.skip\n      };\n      response = await this._listHealthcareEntitiesSinglePage(operationId, optionsWithContinuation);\n      yield response.result;\n    }\n  }\n\n  /**\n   * returns an iterator to arrays of the sorted results of a healthcare operation.\n   */\n  private async _listHealthcareEntitiesSinglePage(\n    operationId: string,\n    options?: HealthcareJobStatusOptions\n  ): Promise<AnalyzeHealthcareEntitiesResultWithPagination> {\n    const { span, updatedOptions: finalOptions } = createSpan(\n      \"TextAnalyticsClient-_listHealthcareEntitiesSinglePage\",\n      options || {}\n    );\n    try {\n      const response = await this.client.healthStatus(operationId, finalOptions);\n      if (response.results) {\n        const result = processAndCombineSuccessfulAndErroneousDocuments(\n          this.documents,\n          response.results,\n          makeHealthcareEntitiesResult,\n          makeHealthcareEntitiesErrorResult\n        );\n        return response.nextLink\n          ? { result, ...nextLinkToTopAndSkip(response.nextLink) }\n          : { result };\n      } else {\n        throw new Error(\"Healthcare action has succeeded but the there are no results!\");\n      }\n    } catch (e) {\n      span.setStatus({\n        code: SpanStatusCode.ERROR,\n        message: e.message\n      });\n      throw e;\n    } finally {\n      span.end();\n    }\n  }\n\n  /**\n   * returns whether the healthcare operation is done and if so returns also\n   * statistics and the model version used.\n   */\n  private async getHealthStatus(\n    operationId: string,\n    options?: HealthcareJobStatusOptions\n  ): Promise<HealthcareJobStatus> {\n    const { span, updatedOptions: finalOptions } = createSpan(\n      \"TextAnalyticsClient-getHealthStatus\",\n      options || {}\n    );\n    try {\n      const response = await this.client.healthStatus(operationId, finalOptions);\n      switch (response.status) {\n        case \"notStarted\":\n        case \"running\":\n          break;\n        case \"failed\": {\n          const errors = response.errors\n            ?.map((e) => `  code ${e.code}, message: '${e.message}'`)\n            .join(\"\\n\");\n          const message = `Healthcare analysis failed. Error(s): ${errors || \"\"}`;\n          throw new Error(message);\n        }\n        default: {\n          if (response.results) {\n            return {\n              done: true,\n              statistics: response.results.statistics,\n              modelVersion: response.results.modelVersion,\n              operationMetdata: getMetaInfoFromResponse(response)\n            };\n          } else {\n            throw new Error(\"Healthcare action has finished but the there are no results!\");\n          }\n        }\n      }\n      return { done: false, operationMetdata: getMetaInfoFromResponse(response) };\n    } catch (e) {\n      span.setStatus({\n        code: SpanStatusCode.ERROR,\n        message: e.message\n      });\n      throw e;\n    } finally {\n      span.end();\n    }\n  }\n\n  private async beginAnalyzeHealthcare(\n    documents: TextDocumentInput[],\n    options?: BeginAnalyzeHealthcareInternalOptions\n  ): Promise<BeginAnalyzeHealthcareResponse> {\n    const { span, updatedOptions: finalOptions } = createSpan(\n      \"TextAnalyticsClient-beginAnalyzeHealthcare\",\n      addStrEncodingParam(options || {})\n    );\n\n    try {\n      return await this.client.health({ documents: documents }, finalOptions);\n    } catch (e) {\n      const exception = handleInvalidDocumentBatch(e);\n      span.setStatus({\n        code: SpanStatusCode.ERROR,\n        message: exception.message\n      });\n      throw exception;\n    } finally {\n      span.end();\n    }\n  }\n\n  async update(\n    options: {\n      abortSignal?: AbortSignalLike;\n      fireProgress?: (state: AnalyzeHealthcareOperationState) => void;\n    } = {}\n  ): Promise<BeginAnalyzeHealthcarePollerOperation> {\n    const state = this.state;\n    const updatedAbortSignal = options.abortSignal;\n    if (!state.isStarted) {\n      state.isStarted = true;\n      const response = await this.beginAnalyzeHealthcare(this.documents, {\n        requestOptions: this.options.requestOptions,\n        tracingOptions: this.options.tracingOptions,\n        abortSignal: updatedAbortSignal ? updatedAbortSignal : options.abortSignal,\n        modelVersion: this.options.modelVersion,\n        stringIndexType: this.options.stringIndexType,\n        loggingOptOut: this.options.disableServiceLogs\n      });\n      if (!response.operationLocation) {\n        throw new Error(\n          \"Expects a valid 'operationLocation' to retrieve health results but did not find any\"\n        );\n      }\n      state.operationId = getOperationId(response.operationLocation);\n    }\n    const operationStatus = await this.getHealthStatus(state.operationId!, {\n      abortSignal: updatedAbortSignal ? updatedAbortSignal : options.abortSignal,\n      includeStatistics: this.options.includeStatistics,\n      tracingOptions: this.options.tracingOptions,\n      onResponse: this.options.onResponse,\n      serializerOptions: this.options.serializerOptions\n    });\n\n    state.createdOn = operationStatus.operationMetdata.createdOn;\n    state.expiresOn = operationStatus.operationMetdata.expiresOn;\n    state.lastModifiedOn = operationStatus.operationMetdata.lastModifiedOn;\n    state.status = operationStatus.operationMetdata.status;\n\n    if (!state.isCompleted && operationStatus.done) {\n      const pagedIterator = this.listHealthcareEntitiesByPage(state.operationId!, {\n        abortSignal: this.options.abortSignal,\n        tracingOptions: this.options.tracingOptions\n      });\n      state.result = Object.assign(pagedIterator, {\n        statistics: operationStatus.statistics,\n        modelVersion: operationStatus.modelVersion!\n      });\n      state.isCompleted = true;\n    }\n    if (typeof options.fireProgress === \"function\") {\n      options.fireProgress(state);\n    }\n    return this;\n  }\n\n  async cancel(): Promise<BeginAnalyzeHealthcarePollerOperation> {\n    const state = this.state;\n    if (state.operationId) {\n      await this.client.cancelHealthJob(state.operationId, {\n        abortSignal: this.options.abortSignal,\n        tracingOptions: this.options.tracingOptions\n      });\n    }\n    state.isCancelled = true;\n    return this;\n  }\n}\n"]}
+{"version":3,"file":"operation.js","sourceRoot":"","sources":["../../../../src/lro/health/operation.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;;AAYlC,OAAO,EAKL,iCAAiC,EACjC,4BAA4B,GAC7B,MAAM,uCAAuC,CAAC;AAE/C,OAAO,EAEL,mBAAmB,EACnB,cAAc,EACd,oBAAoB,EACpB,UAAU,GACX,MAAM,YAAY,CAAC;AACpB,OAAO,EACL,qBAAqB,GAGtB,MAAM,WAAW,CAAC;AAEnB,OAAO,EAAE,gDAAgD,EAAE,MAAM,2BAA2B,CAAC;AAiF7F;;GAEG;AACH,SAAS,uBAAuB,CAC9B,QAA4B;IAE5B,OAAO;QACL,SAAS,EAAE,QAAQ,CAAC,eAAe;QACnC,cAAc,EAAE,QAAQ,CAAC,kBAAkB;QAC3C,SAAS,EAAE,QAAQ,CAAC,kBAAkB;QACtC,MAAM,EAAE,QAAQ,CAAC,MAAM;KACxB,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,OAAO,qCAAsC,SAAQ,qBAG1D;IACC,YACS,KAAsC;IAC7C,wEAAwE;IAChE,MAAc,EACd,OAAsB,EACtB,SAA8B,EAC9B,UAAiD,EAAE;QAE3D,KAAK,CAAC,KAAK,CAAC,CAAC;QAPN,UAAK,GAAL,KAAK,CAAiC;QAErC,WAAM,GAAN,MAAM,CAAQ;QACd,YAAO,GAAP,OAAO,CAAe;QACtB,cAAS,GAAT,SAAS,CAAqB;QAC9B,YAAO,GAAP,OAAO,CAA4C;IAG7D,CAAC;IAED;;;;OAIG;IACK,4BAA4B,CAClC,WAAmB,EACnB,UAAsC,EAAE;QAExC,MAAM,IAAI,GAAG,IAAI,CAAC,uBAAuB,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QAChE,OAAO;YACL,IAAI;gBACF,OAAO,IAAI,CAAC,IAAI,EAAE,CAAC;YACrB,CAAC;YACD,CAAC,MAAM,CAAC,aAAa,CAAC;gBACpB,OAAO,IAAI,CAAC;YACd,CAAC;YACD,MAAM,EAAE,CAAC,QAAuB,EAAE,EAAE;gBAClC,MAAM,WAAW,mCAAQ,OAAO,KAAE,GAAG,EAAE,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,CAAE,WAAW,GAAE,CAAC;gBAC/D,OAAO,IAAI,CAAC,4BAA4B,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC;YACrE,CAAC;SACF,CAAC;IACJ,CAAC;IAED;;OAEG;IACY,uBAAuB,CACpC,WAAmB,EACnB,OAAoC;;;;gBAEpC,KAAyB,IAAA,KAAA,cAAA,IAAI,CAAC,4BAA4B,CAAC,WAAW,EAAE,OAAO,CAAC,CAAA,IAAA;oBAArE,MAAM,IAAI,WAAA,CAAA;oBACnB,cAAA,KAAK,CAAC,CAAC,iBAAA,cAAA,IAAI,CAAA,CAAA,CAAA,CAAC;iBACb;;;;;;;;;QACH,CAAC;KAAA;IAED;;OAEG;IACY,4BAA4B,CACzC,WAAmB,EACnB,OAAoC;;YAEpC,IAAI,QAAQ,GAAG,cAAM,IAAI,CAAC,iCAAiC,CAAC,WAAW,EAAE,OAAO,CAAC,CAAA,CAAC;YAClF,oBAAM,QAAQ,CAAC,MAAM,CAAA,CAAC;YACtB,OAAO,QAAQ,CAAC,IAAI,EAAE;gBACpB,MAAM,uBAAuB,mCACxB,OAAO,KACV,GAAG,EAAE,QAAQ,CAAC,GAAG,EACjB,IAAI,EAAE,QAAQ,CAAC,IAAI,GACpB,CAAC;gBACF,QAAQ,GAAG,cAAM,IAAI,CAAC,iCAAiC,CAAC,WAAW,EAAE,uBAAuB,CAAC,CAAA,CAAC;gBAC9F,oBAAM,QAAQ,CAAC,MAAM,CAAA,CAAC;aACvB;QACH,CAAC;KAAA;IAED;;OAEG;IACK,KAAK,CAAC,iCAAiC,CAC7C,WAAmB,EACnB,OAAoC;QAEpC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAC1C,uDAAuD,EACvD,OAAO,IAAI,EAAE,EACb,CAAC,YAAY,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,WAAW,EAAE,YAAY,CAAC,CACtE,CAAC;QACF,IAAI,QAAQ,CAAC,OAAO,EAAE;YACpB,MAAM,MAAM,GAAG,gDAAgD,CAC7D,IAAI,CAAC,SAAS,EACd,QAAQ,CAAC,OAAO,EAChB,4BAA4B,EAC5B,iCAAiC,CAClC,CAAC;YACF,OAAO,QAAQ,CAAC,QAAQ;gBACtB,CAAC,iBAAG,MAAM,IAAK,oBAAoB,CAAC,QAAQ,CAAC,QAAQ,CAAC,EACtD,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC;SAChB;aAAM;YACL,MAAM,IAAI,KAAK,CAAC,+DAA+D,CAAC,CAAC;SAClF;IACH,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,eAAe,CAC3B,WAAmB,EACnB,OAAoC;;QAEpC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAC1C,qCAAqC,EACrC,OAAO,IAAI,EAAE,EACb,CAAC,YAAY,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,WAAW,EAAE,YAAY,CAAC,CACtE,CAAC;QACF,QAAQ,QAAQ,CAAC,MAAM,EAAE;YACvB,KAAK,YAAY,CAAC;YAClB,KAAK,SAAS;gBACZ,MAAM;YACR,KAAK,QAAQ,CAAC,CAAC;gBACb,MAAM,MAAM,GAAG,MAAA,QAAQ,CAAC,MAAM,0CAC1B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,IAAI,eAAe,CAAC,CAAC,OAAO,GAAG,EACvD,IAAI,CAAC,IAAI,CAAC,CAAC;gBACd,MAAM,OAAO,GAAG,yCAAyC,MAAM,IAAI,EAAE,EAAE,CAAC;gBACxE,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC;aAC1B;YACD,OAAO,CAAC,CAAC;gBACP,IAAI,QAAQ,CAAC,OAAO,EAAE;oBACpB,OAAO;wBACL,IAAI,EAAE,IAAI;wBACV,UAAU,EAAE,QAAQ,CAAC,OAAO,CAAC,UAAU;wBACvC,YAAY,EAAE,QAAQ,CAAC,OAAO,CAAC,YAAY;wBAC3C,gBAAgB,EAAE,uBAAuB,CAAC,QAAQ,CAAC;qBACpD,CAAC;iBACH;qBAAM;oBACL,MAAM,IAAI,KAAK,CAAC,8DAA8D,CAAC,CAAC;iBACjF;aACF;SACF;QACD,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,gBAAgB,EAAE,uBAAuB,CAAC,QAAQ,CAAC,EAAE,CAAC;IAC9E,CAAC;IAEO,KAAK,CAAC,sBAAsB,CAClC,SAA8B,EAC9B,OAA+C;QAE/C,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAC1B,4CAA4C,EAC5C,mBAAmB,CAAC,OAAO,IAAI,EAAE,CAAC,EAClC,CAAC,YAAY,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,EAAE,YAAY,CAAC,CAAC,CACzF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CACV,UAGI,EAAE;QAEN,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACzB,MAAM,kBAAkB,GAAG,OAAO,CAAC,WAAW,CAAC;QAC/C,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE;YACpB,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC;YACvB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,SAAS,EAAE;gBACjE,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;gBAC3C,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;gBAC3C,WAAW,EAAE,kBAAkB,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW;gBAC1E,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,YAAY;gBACvC,eAAe,EAAE,IAAI,CAAC,OAAO,CAAC,eAAe;gBAC7C,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,kBAAkB;aAC/C,CAAC,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,iBAAiB,EAAE;gBAC/B,MAAM,IAAI,KAAK,CACb,qFAAqF,CACtF,CAAC;aACH;YACD,KAAK,CAAC,WAAW,GAAG,cAAc,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC;SAChE;QACD,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,WAAY,EAAE;YACrE,WAAW,EAAE,kBAAkB,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW;YAC1E,iBAAiB,EAAE,IAAI,CAAC,OAAO,CAAC,iBAAiB;YACjD,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;YAC3C,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU;YACnC,iBAAiB,EAAE,IAAI,CAAC,OAAO,CAAC,iBAAiB;SAClD,CAAC,CAAC;QAEH,KAAK,CAAC,SAAS,GAAG,eAAe,CAAC,gBAAgB,CAAC,SAAS,CAAC;QAC7D,KAAK,CAAC,SAAS,GAAG,eAAe,CAAC,gBAAgB,CAAC,SAAS,CAAC;QAC7D,KAAK,CAAC,cAAc,GAAG,eAAe,CAAC,gBAAgB,CAAC,cAAc,CAAC;QACvE,KAAK,CAAC,MAAM,GAAG,eAAe,CAAC,gBAAgB,CAAC,MAAM,CAAC;QAEvD,IAAI,CAAC,KAAK,CAAC,WAAW,IAAI,eAAe,CAAC,IAAI,EAAE;YAC9C,MAAM,aAAa,GAAG,IAAI,CAAC,4BAA4B,CAAC,KAAK,CAAC,WAAY,EAAE;gBAC1E,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW;gBACrC,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;aAC5C,CAAC,CAAC;YACH,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,aAAa,EAAE;gBAC1C,UAAU,EAAE,eAAe,CAAC,UAAU;gBACtC,YAAY,EAAE,eAAe,CAAC,YAAa;aAC5C,CAAC,CAAC;YACH,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC;SAC1B;QACD,IAAI,OAAO,OAAO,CAAC,YAAY,KAAK,UAAU,EAAE;YAC9C,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;SAC7B;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,MAAM;QACV,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACzB,IAAI,KAAK,CAAC,WAAW,EAAE;YACrB,MAAM,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,KAAK,CAAC,WAAW,EAAE;gBACnD,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW;gBACrC,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;aAC5C,CAAC,CAAC;SACJ;QACD,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC;QACzB,OAAO,IAAI,CAAC;IACd,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\nimport { AbortSignalLike } from \"@azure/abort-controller\";\nimport { OperationOptions } from \"@azure/core-client\";\n\nimport {\n  HealthResponse as BeginAnalyzeHealthcareResponse,\n  HealthcareJobState,\n  HealthStatusOptionalParams as HealthcareJobStatusOptions,\n  TextDocumentBatchStatistics,\n  TextDocumentInput,\n} from \"../../generated/models\";\nimport {\n  AnalyzeHealthcareEntitiesResult,\n  AnalyzeHealthcareEntitiesResultArray,\n  PagedAnalyzeHealthcareEntitiesResult,\n  PagedAsyncIterableAnalyzeHealthcareEntitiesResult,\n  makeHealthcareEntitiesErrorResult,\n  makeHealthcareEntitiesResult,\n} from \"../../analyzeHealthcareEntitiesResult\";\nimport { PageSettings } from \"@azure/core-paging\";\nimport {\n  StringIndexType,\n  addStrEncodingParam,\n  getOperationId,\n  nextLinkToTopAndSkip,\n  throwError,\n} from \"../../util\";\nimport {\n  AnalysisPollOperation,\n  AnalysisPollOperationState,\n  OperationMetadata as AnalyzeHealthcareEntitiesOperationMetadata,\n} from \"../poller\";\nimport { GeneratedClient as Client } from \"../../generated\";\nimport { processAndCombineSuccessfulAndErroneousDocuments } from \"../../textAnalyticsResult\";\nimport { TextAnalyticsOperationOptions } from \"../../textAnalyticsOperationOptions\";\nimport { TracingClient } from \"@azure/core-tracing\";\n\n/**\n * @internal\n */\ninterface AnalyzeHealthcareEntitiesResultWithPagination {\n  result: AnalyzeHealthcareEntitiesResultArray;\n  top?: number;\n  skip?: number;\n}\n\n/**\n * @internal\n */\ninterface HealthcareJobStatus {\n  done: boolean;\n  /**\n   * Statistics about the input document batch and how it was processed\n   * by the service. This property will have a value when includeStatistics is set to true\n   * in the client call.\n   */\n  statistics?: TextDocumentBatchStatistics;\n  /**\n   * The version of the text analytics model used by this operation on this\n   * batch of input documents.\n   */\n  modelVersion?: string;\n  operationMetdata: Omit<AnalyzeHealthcareEntitiesOperationMetadata, \"operationId\">;\n}\n\n/**\n * @internal\n */\ninterface BeginAnalyzeHealthcareInternalOptions extends OperationOptions {\n  /**\n   * This value indicates which model will be used for scoring. If a model-version is\n   * not specified, the API should default to the latest, non-preview version.\n   * For supported model versions, see operation-specific documentation, for example:\n   * https://docs.microsoft.com/azure/cognitive-services/text-analytics/how-tos/text-analytics-how-to-sentiment-analysis#model-versioning\n   */\n  modelVersion?: string;\n  /**\n   * Specifies the measurement unit used to calculate the offset and length properties.\n   * Possible units are \"TextElements_v8\", \"UnicodeCodePoint\", and \"Utf16CodeUnit\".\n   * The default is the JavaScript's default which is \"Utf16CodeUnit\".\n   */\n  stringIndexType?: StringIndexType;\n  /**\n   * If set to false, you opt-in to have your text input logged for troubleshooting. By default, Text Analytics\n   * will not log your input text for healthcare entities analysis. Setting this parameter to false,\n   * enables input logging.\n   */\n  loggingOptOut?: boolean;\n}\n\n/**\n * Options for the begin analyze healthcare entities operation.\n */\nexport interface BeginAnalyzeHealthcareEntitiesOptions extends TextAnalyticsOperationOptions {\n  /**\n   * Delay to wait until next poll, in milliseconds.\n   */\n  stringIndexType?: StringIndexType;\n  /**\n   * Delay to wait until next poll, in milliseconds.\n   */\n  updateIntervalInMs?: number;\n  /**\n   * A serialized poller which can be used to resume an existing paused Long-Running-Operation.\n   */\n  resumeFrom?: string;\n}\n\n/**\n * The state of the begin analyze healthcare polling operation.\n */\nexport interface AnalyzeHealthcareOperationState\n  extends AnalysisPollOperationState<PagedAnalyzeHealthcareEntitiesResult> {}\n\n/**\n * @internal\n */\nfunction getMetaInfoFromResponse(\n  response: HealthcareJobState\n): Omit<AnalyzeHealthcareEntitiesOperationMetadata, \"operationId\"> {\n  return {\n    createdOn: response.createdDateTime,\n    lastModifiedOn: response.lastUpdateDateTime,\n    expiresOn: response.expirationDateTime,\n    status: response.status,\n  };\n}\n\n/**\n * Class that represents a poller that waits for the healthcare results.\n * @internal\n */\nexport class BeginAnalyzeHealthcarePollerOperation extends AnalysisPollOperation<\n  AnalyzeHealthcareOperationState,\n  PagedAnalyzeHealthcareEntitiesResult\n> {\n  constructor(\n    public state: AnalyzeHealthcareOperationState,\n    // eslint-disable-next-line @azure/azure-sdk/ts-use-interface-parameters\n    private client: Client,\n    private tracing: TracingClient,\n    private documents: TextDocumentInput[],\n    private options: BeginAnalyzeHealthcareEntitiesOptions = {}\n  ) {\n    super(state);\n  }\n\n  /**\n   * should be called only after all the status of the healthcare operations became\n   * \"succeeded\" and it returns an iterator for the results and provides a\n   * byPage method to return the results paged.\n   */\n  private listHealthcareEntitiesByPage(\n    operationId: string,\n    options: HealthcareJobStatusOptions = {}\n  ): PagedAsyncIterableAnalyzeHealthcareEntitiesResult {\n    const iter = this._listHealthcareEntities(operationId, options);\n    return {\n      next() {\n        return iter.next();\n      },\n      [Symbol.asyncIterator]() {\n        return this;\n      },\n      byPage: (settings?: PageSettings) => {\n        const pageOptions = { ...options, top: settings?.maxPageSize };\n        return this._listHealthcareEntitiesPaged(operationId, pageOptions);\n      },\n    };\n  }\n\n  /**\n   * returns an iterator to the results of a healthcare operation.\n   */\n  private async *_listHealthcareEntities(\n    operationId: string,\n    options?: HealthcareJobStatusOptions\n  ): AsyncIterableIterator<AnalyzeHealthcareEntitiesResult> {\n    for await (const page of this._listHealthcareEntitiesPaged(operationId, options)) {\n      yield* page;\n    }\n  }\n\n  /**\n   * returns an iterator to arrays of the results of a healthcare operation.\n   */\n  private async *_listHealthcareEntitiesPaged(\n    operationId: string,\n    options?: HealthcareJobStatusOptions\n  ): AsyncIterableIterator<AnalyzeHealthcareEntitiesResultArray> {\n    let response = await this._listHealthcareEntitiesSinglePage(operationId, options);\n    yield response.result;\n    while (response.skip) {\n      const optionsWithContinuation: HealthcareJobStatusOptions = {\n        ...options,\n        top: response.top,\n        skip: response.skip,\n      };\n      response = await this._listHealthcareEntitiesSinglePage(operationId, optionsWithContinuation);\n      yield response.result;\n    }\n  }\n\n  /**\n   * returns an iterator to arrays of the sorted results of a healthcare operation.\n   */\n  private async _listHealthcareEntitiesSinglePage(\n    operationId: string,\n    options?: HealthcareJobStatusOptions\n  ): Promise<AnalyzeHealthcareEntitiesResultWithPagination> {\n    const response = await this.tracing.withSpan(\n      \"TextAnalyticsClient-_listHealthcareEntitiesSinglePage\",\n      options || {},\n      (finalOptions) => this.client.healthStatus(operationId, finalOptions)\n    );\n    if (response.results) {\n      const result = processAndCombineSuccessfulAndErroneousDocuments(\n        this.documents,\n        response.results,\n        makeHealthcareEntitiesResult,\n        makeHealthcareEntitiesErrorResult\n      );\n      return response.nextLink\n        ? { result, ...nextLinkToTopAndSkip(response.nextLink) }\n        : { result };\n    } else {\n      throw new Error(\"Healthcare action has succeeded but the there are no results!\");\n    }\n  }\n\n  /**\n   * returns whether the healthcare operation is done and if so returns also\n   * statistics and the model version used.\n   */\n  private async getHealthStatus(\n    operationId: string,\n    options?: HealthcareJobStatusOptions\n  ): Promise<HealthcareJobStatus> {\n    const response = await this.tracing.withSpan(\n      \"TextAnalyticsClient-getHealthStatus\",\n      options || {},\n      (finalOptions) => this.client.healthStatus(operationId, finalOptions)\n    );\n    switch (response.status) {\n      case \"notStarted\":\n      case \"running\":\n        break;\n      case \"failed\": {\n        const errors = response.errors\n          ?.map((e) => `  code ${e.code}, message: '${e.message}'`)\n          .join(\"\\n\");\n        const message = `Healthcare analysis failed. Error(s): ${errors || \"\"}`;\n        throw new Error(message);\n      }\n      default: {\n        if (response.results) {\n          return {\n            done: true,\n            statistics: response.results.statistics,\n            modelVersion: response.results.modelVersion,\n            operationMetdata: getMetaInfoFromResponse(response),\n          };\n        } else {\n          throw new Error(\"Healthcare action has finished but the there are no results!\");\n        }\n      }\n    }\n    return { done: false, operationMetdata: getMetaInfoFromResponse(response) };\n  }\n\n  private async beginAnalyzeHealthcare(\n    documents: TextDocumentInput[],\n    options?: BeginAnalyzeHealthcareInternalOptions\n  ): Promise<BeginAnalyzeHealthcareResponse> {\n    return this.tracing.withSpan(\n      \"TextAnalyticsClient-beginAnalyzeHealthcare\",\n      addStrEncodingParam(options || {}),\n      (finalOptions) => throwError(this.client.health({ documents: documents }, finalOptions))\n    );\n  }\n\n  async update(\n    options: {\n      abortSignal?: AbortSignalLike;\n      fireProgress?: (state: AnalyzeHealthcareOperationState) => void;\n    } = {}\n  ): Promise<BeginAnalyzeHealthcarePollerOperation> {\n    const state = this.state;\n    const updatedAbortSignal = options.abortSignal;\n    if (!state.isStarted) {\n      state.isStarted = true;\n      const response = await this.beginAnalyzeHealthcare(this.documents, {\n        requestOptions: this.options.requestOptions,\n        tracingOptions: this.options.tracingOptions,\n        abortSignal: updatedAbortSignal ? updatedAbortSignal : options.abortSignal,\n        modelVersion: this.options.modelVersion,\n        stringIndexType: this.options.stringIndexType,\n        loggingOptOut: this.options.disableServiceLogs,\n      });\n      if (!response.operationLocation) {\n        throw new Error(\n          \"Expects a valid 'operationLocation' to retrieve health results but did not find any\"\n        );\n      }\n      state.operationId = getOperationId(response.operationLocation);\n    }\n    const operationStatus = await this.getHealthStatus(state.operationId!, {\n      abortSignal: updatedAbortSignal ? updatedAbortSignal : options.abortSignal,\n      includeStatistics: this.options.includeStatistics,\n      tracingOptions: this.options.tracingOptions,\n      onResponse: this.options.onResponse,\n      serializerOptions: this.options.serializerOptions,\n    });\n\n    state.createdOn = operationStatus.operationMetdata.createdOn;\n    state.expiresOn = operationStatus.operationMetdata.expiresOn;\n    state.lastModifiedOn = operationStatus.operationMetdata.lastModifiedOn;\n    state.status = operationStatus.operationMetdata.status;\n\n    if (!state.isCompleted && operationStatus.done) {\n      const pagedIterator = this.listHealthcareEntitiesByPage(state.operationId!, {\n        abortSignal: this.options.abortSignal,\n        tracingOptions: this.options.tracingOptions,\n      });\n      state.result = Object.assign(pagedIterator, {\n        statistics: operationStatus.statistics,\n        modelVersion: operationStatus.modelVersion!,\n      });\n      state.isCompleted = true;\n    }\n    if (typeof options.fireProgress === \"function\") {\n      options.fireProgress(state);\n    }\n    return this;\n  }\n\n  async cancel(): Promise<BeginAnalyzeHealthcarePollerOperation> {\n    const state = this.state;\n    if (state.operationId) {\n      await this.client.cancelHealthJob(state.operationId, {\n        abortSignal: this.options.abortSignal,\n        tracingOptions: this.options.tracingOptions,\n      });\n    }\n    state.isCancelled = true;\n    return this;\n  }\n}\n"]}

package/dist-esm/src/lro/health/poller.js

@@ -2,7 +2,7 @@
 // Licensed under the MIT license.
 import { delay } from "../../util";
 import { AnalysisPoller } from "../poller";
-import { BeginAnalyzeHealthcarePollerOperation } from "./operation";
+import { BeginAnalyzeHealthcarePollerOperation, } from "./operation";
 /**
  * Class that represents a poller that waits for the healthcare results.
  * @internal
@@ -10,12 +10,12 @@ import { BeginAnalyzeHealthcarePollerOperation } from "./operation";
 export class BeginAnalyzeHealthcarePoller extends AnalysisPoller {
     // eslint-disable-next-line @azure/azure-sdk/ts-use-interface-parameters
     constructor(inputs) {
-        const { client, documents, options, updateIntervalInMs = 5000, resumeFrom } = inputs;
+        const { client, tracing, documents, options, updateIntervalInMs = 5000, resumeFrom } = inputs;
         let state;
         if (resumeFrom) {
             state = JSON.parse(resumeFrom).state;
         }
-        const operation = new BeginAnalyzeHealthcarePollerOperation((state || {}), client, documents, options);
+        const operation = new BeginAnalyzeHealthcarePollerOperation((state || {}), client, tracing, documents, options);
         super(operation);
         this.updateIntervalInMs = updateIntervalInMs;
     }