@azure/ai-language-text 1.0.0-alpha.20220816.1 → 1.0.0

package/dist-esm/src/generated/models/parameters.js

@@ -46,7 +46,7 @@ export const endpoint = {
 export const apiVersion = {
     parameterPath: "apiVersion",
     mapper: {
-        defaultValue: "2022-04-01-preview",
+        defaultValue: "2022-05-01",
         isConstant: true,
         serializedName: "api-version",
         type: {

package/dist-esm/src/generated/models/parameters.js.map

@@ -1 +1 @@
-{"version":3,"file":"parameters.js","sourceRoot":"","sources":["../../../../src/generated/models/parameters.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAOH,OAAO,EACL,aAAa,IAAI,mBAAmB,EACpC,oBAAoB,IAAI,0BAA0B,EACnD,MAAM,mBAAmB,CAAC;AAE3B,MAAM,CAAC,MAAM,WAAW,GAAuB;IAC7C,aAAa,EAAE,CAAC,SAAS,EAAE,aAAa,CAAC;IACzC,MAAM,EAAE;QACN,YAAY,EAAE,kBAAkB;QAChC,UAAU,EAAE,IAAI;QAChB,cAAc,EAAE,cAAc;QAC9B,IAAI,EAAE;YACJ,IAAI,EAAE,QAAQ;SACf;KACF;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,IAAI,GAAuB;IACtC,aAAa,EAAE,MAAM;IACrB,MAAM,EAAE,mBAAmB;CAC5B,CAAC;AAEF,MAAM,CAAC,MAAM,MAAM,GAAuB;IACxC,aAAa,EAAE,QAAQ;IACvB,MAAM,EAAE;QACN,YAAY,EAAE,kBAAkB;QAChC,UAAU,EAAE,IAAI;QAChB,cAAc,EAAE,QAAQ;QACxB,IAAI,EAAE;YACJ,IAAI,EAAE,QAAQ;SACf;KACF;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,QAAQ,GAA0B;IAC7C,aAAa,EAAE,UAAU;IACzB,MAAM,EAAE;QACN,cAAc,EAAE,UAAU;QAC1B,QAAQ,EAAE,IAAI;QACd,IAAI,EAAE;YACJ,IAAI,EAAE,QAAQ;SACf;KACF;IACD,YAAY,EAAE,IAAI;CACnB,CAAC;AAEF,MAAM,CAAC,MAAM,UAAU,GAA4B;IACjD,aAAa,EAAE,YAAY;IAC3B,MAAM,EAAE;QACN,YAAY,EAAE,oBAAoB;QAClC,UAAU,EAAE,IAAI;QAChB,cAAc,EAAE,aAAa;QAC7B,IAAI,EAAE;YACJ,IAAI,EAAE,QAAQ;SACf;KACF;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,iBAAiB,GAA4B;IACxD,aAAa,EAAE,CAAC,SAAS,EAAE,mBAAmB,CAAC;IAC/C,MAAM,EAAE;QACN,cAAc,EAAE,WAAW;QAC3B,IAAI,EAAE;YACJ,IAAI,EAAE,SAAS;SAChB;KACF;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,KAAK,GAAuB;IACvC,aAAa,EAAE,MAAM;IACrB,MAAM,EAAE,0BAA0B;CACnC,CAAC;AAEF,MAAM,CAAC,MAAM,KAAK,GAA0B;IAC1C,aAAa,EAAE,OAAO;IACtB,MAAM,EAAE;QACN,cAAc,EAAE,OAAO;QACvB,QAAQ,EAAE,IAAI;QACd,IAAI,EAAE;YACJ,IAAI,EAAE,MAAM;SACb;KACF;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,GAAG,GAA4B;IAC1C,aAAa,EAAE,CAAC,SAAS,EAAE,KAAK,CAAC;IACjC,MAAM,EAAE;QACN,cAAc,EAAE,KAAK;QACrB,IAAI,EAAE;YACJ,IAAI,EAAE,QAAQ;SACf;KACF;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,IAAI,GAA4B;IAC3C,aAAa,EAAE,CAAC,SAAS,EAAE,MAAM,CAAC;IAClC,MAAM,EAAE;QACN,cAAc,EAAE,MAAM;QACtB,IAAI,EAAE;YACJ,IAAI,EAAE,QAAQ;SACf;KACF;CACF,CAAC","sourcesContent":["/*\n * Copyright (c) Microsoft Corporation.\n * Licensed under the MIT License.\n *\n * Code generated by Microsoft (R) AutoRest Code Generator.\n * Changes may cause incorrect behavior and will be lost if the code is regenerated.\n */\n\nimport {\n  OperationParameter,\n  OperationURLParameter,\n  OperationQueryParameter\n} from \"@azure/core-client\";\nimport {\n  AnalyzeAction as AnalyzeActionMapper,\n  AnalyzeTextJobsInput as AnalyzeTextJobsInputMapper\n} from \"../models/mappers\";\n\nexport const contentType: OperationParameter = {\n  parameterPath: [\"options\", \"contentType\"],\n  mapper: {\n    defaultValue: \"application/json\",\n    isConstant: true,\n    serializedName: \"Content-Type\",\n    type: {\n      name: \"String\"\n    }\n  }\n};\n\nexport const body: OperationParameter = {\n  parameterPath: \"body\",\n  mapper: AnalyzeActionMapper\n};\n\nexport const accept: OperationParameter = {\n  parameterPath: \"accept\",\n  mapper: {\n    defaultValue: \"application/json\",\n    isConstant: true,\n    serializedName: \"Accept\",\n    type: {\n      name: \"String\"\n    }\n  }\n};\n\nexport const endpoint: OperationURLParameter = {\n  parameterPath: \"endpoint\",\n  mapper: {\n    serializedName: \"Endpoint\",\n    required: true,\n    type: {\n      name: \"String\"\n    }\n  },\n  skipEncoding: true\n};\n\nexport const apiVersion: OperationQueryParameter = {\n  parameterPath: \"apiVersion\",\n  mapper: {\n    defaultValue: \"2022-04-01-preview\",\n    isConstant: true,\n    serializedName: \"api-version\",\n    type: {\n      name: \"String\"\n    }\n  }\n};\n\nexport const includeStatistics: OperationQueryParameter = {\n  parameterPath: [\"options\", \"includeStatistics\"],\n  mapper: {\n    serializedName: \"showStats\",\n    type: {\n      name: \"Boolean\"\n    }\n  }\n};\n\nexport const body1: OperationParameter = {\n  parameterPath: \"body\",\n  mapper: AnalyzeTextJobsInputMapper\n};\n\nexport const jobId: OperationURLParameter = {\n  parameterPath: \"jobId\",\n  mapper: {\n    serializedName: \"jobId\",\n    required: true,\n    type: {\n      name: \"Uuid\"\n    }\n  }\n};\n\nexport const top: OperationQueryParameter = {\n  parameterPath: [\"options\", \"top\"],\n  mapper: {\n    serializedName: \"top\",\n    type: {\n      name: \"Number\"\n    }\n  }\n};\n\nexport const skip: OperationQueryParameter = {\n  parameterPath: [\"options\", \"skip\"],\n  mapper: {\n    serializedName: \"skip\",\n    type: {\n      name: \"Number\"\n    }\n  }\n};\n"]}
+{"version":3,"file":"parameters.js","sourceRoot":"","sources":["../../../../src/generated/models/parameters.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAOH,OAAO,EACL,aAAa,IAAI,mBAAmB,EACpC,oBAAoB,IAAI,0BAA0B,EACnD,MAAM,mBAAmB,CAAC;AAE3B,MAAM,CAAC,MAAM,WAAW,GAAuB;IAC7C,aAAa,EAAE,CAAC,SAAS,EAAE,aAAa,CAAC;IACzC,MAAM,EAAE;QACN,YAAY,EAAE,kBAAkB;QAChC,UAAU,EAAE,IAAI;QAChB,cAAc,EAAE,cAAc;QAC9B,IAAI,EAAE;YACJ,IAAI,EAAE,QAAQ;SACf;KACF;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,IAAI,GAAuB;IACtC,aAAa,EAAE,MAAM;IACrB,MAAM,EAAE,mBAAmB;CAC5B,CAAC;AAEF,MAAM,CAAC,MAAM,MAAM,GAAuB;IACxC,aAAa,EAAE,QAAQ;IACvB,MAAM,EAAE;QACN,YAAY,EAAE,kBAAkB;QAChC,UAAU,EAAE,IAAI;QAChB,cAAc,EAAE,QAAQ;QACxB,IAAI,EAAE;YACJ,IAAI,EAAE,QAAQ;SACf;KACF;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,QAAQ,GAA0B;IAC7C,aAAa,EAAE,UAAU;IACzB,MAAM,EAAE;QACN,cAAc,EAAE,UAAU;QAC1B,QAAQ,EAAE,IAAI;QACd,IAAI,EAAE;YACJ,IAAI,EAAE,QAAQ;SACf;KACF;IACD,YAAY,EAAE,IAAI;CACnB,CAAC;AAEF,MAAM,CAAC,MAAM,UAAU,GAA4B;IACjD,aAAa,EAAE,YAAY;IAC3B,MAAM,EAAE;QACN,YAAY,EAAE,YAAY;QAC1B,UAAU,EAAE,IAAI;QAChB,cAAc,EAAE,aAAa;QAC7B,IAAI,EAAE;YACJ,IAAI,EAAE,QAAQ;SACf;KACF;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,iBAAiB,GAA4B;IACxD,aAAa,EAAE,CAAC,SAAS,EAAE,mBAAmB,CAAC;IAC/C,MAAM,EAAE;QACN,cAAc,EAAE,WAAW;QAC3B,IAAI,EAAE;YACJ,IAAI,EAAE,SAAS;SAChB;KACF;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,KAAK,GAAuB;IACvC,aAAa,EAAE,MAAM;IACrB,MAAM,EAAE,0BAA0B;CACnC,CAAC;AAEF,MAAM,CAAC,MAAM,KAAK,GAA0B;IAC1C,aAAa,EAAE,OAAO;IACtB,MAAM,EAAE;QACN,cAAc,EAAE,OAAO;QACvB,QAAQ,EAAE,IAAI;QACd,IAAI,EAAE;YACJ,IAAI,EAAE,MAAM;SACb;KACF;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,GAAG,GAA4B;IAC1C,aAAa,EAAE,CAAC,SAAS,EAAE,KAAK,CAAC;IACjC,MAAM,EAAE;QACN,cAAc,EAAE,KAAK;QACrB,IAAI,EAAE;YACJ,IAAI,EAAE,QAAQ;SACf;KACF;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,IAAI,GAA4B;IAC3C,aAAa,EAAE,CAAC,SAAS,EAAE,MAAM,CAAC;IAClC,MAAM,EAAE;QACN,cAAc,EAAE,MAAM;QACtB,IAAI,EAAE;YACJ,IAAI,EAAE,QAAQ;SACf;KACF;CACF,CAAC","sourcesContent":["/*\n * Copyright (c) Microsoft Corporation.\n * Licensed under the MIT License.\n *\n * Code generated by Microsoft (R) AutoRest Code Generator.\n * Changes may cause incorrect behavior and will be lost if the code is regenerated.\n */\n\nimport {\n  OperationParameter,\n  OperationURLParameter,\n  OperationQueryParameter\n} from \"@azure/core-client\";\nimport {\n  AnalyzeAction as AnalyzeActionMapper,\n  AnalyzeTextJobsInput as AnalyzeTextJobsInputMapper\n} from \"../models/mappers\";\n\nexport const contentType: OperationParameter = {\n  parameterPath: [\"options\", \"contentType\"],\n  mapper: {\n    defaultValue: \"application/json\",\n    isConstant: true,\n    serializedName: \"Content-Type\",\n    type: {\n      name: \"String\"\n    }\n  }\n};\n\nexport const body: OperationParameter = {\n  parameterPath: \"body\",\n  mapper: AnalyzeActionMapper\n};\n\nexport const accept: OperationParameter = {\n  parameterPath: \"accept\",\n  mapper: {\n    defaultValue: \"application/json\",\n    isConstant: true,\n    serializedName: \"Accept\",\n    type: {\n      name: \"String\"\n    }\n  }\n};\n\nexport const endpoint: OperationURLParameter = {\n  parameterPath: \"endpoint\",\n  mapper: {\n    serializedName: \"Endpoint\",\n    required: true,\n    type: {\n      name: \"String\"\n    }\n  },\n  skipEncoding: true\n};\n\nexport const apiVersion: OperationQueryParameter = {\n  parameterPath: \"apiVersion\",\n  mapper: {\n    defaultValue: \"2022-05-01\",\n    isConstant: true,\n    serializedName: \"api-version\",\n    type: {\n      name: \"String\"\n    }\n  }\n};\n\nexport const includeStatistics: OperationQueryParameter = {\n  parameterPath: [\"options\", \"includeStatistics\"],\n  mapper: {\n    serializedName: \"showStats\",\n    type: {\n      name: \"Boolean\"\n    }\n  }\n};\n\nexport const body1: OperationParameter = {\n  parameterPath: \"body\",\n  mapper: AnalyzeTextJobsInputMapper\n};\n\nexport const jobId: OperationURLParameter = {\n  parameterPath: \"jobId\",\n  mapper: {\n    serializedName: \"jobId\",\n    required: true,\n    type: {\n      name: \"Uuid\"\n    }\n  }\n};\n\nexport const top: OperationQueryParameter = {\n  parameterPath: [\"options\", \"top\"],\n  mapper: {\n    serializedName: \"top\",\n    type: {\n      name: \"Number\"\n    }\n  }\n};\n\nexport const skip: OperationQueryParameter = {\n  parameterPath: [\"options\", \"skip\"],\n  mapper: {\n    serializedName: \"skip\",\n    type: {\n      name: \"Number\"\n    }\n  }\n};\n"]}

package/dist-esm/src/index.js

@@ -14,5 +14,5 @@ export { TextAnalysisClient } from "./textAnalysisClient";
 export * from "./models";
 export { 
 /** orphan exports */
-KnownPiiEntityDomain, KnownPiiEntityCategory, KnownStringIndexType, KnownErrorCode, KnownInnerErrorCode, KnownExtractiveSummarizationOrderingCriteria, } from "./generated/models";
+KnownPiiEntityDomain, KnownPiiEntityCategory, KnownStringIndexType, KnownErrorCode, KnownInnerErrorCode, } from "./generated/models";
 //# sourceMappingURL=index.js.map

package/dist-esm/src/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC;;;;;;;;GAQG;AAEH,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AAEtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,cAAc,UAAU,CAAC;AACzB,OAAO;AA6CL,qBAAqB;AACrB,oBAAoB,EACpB,sBAAsB,EACtB,oBAAoB,EACpB,cAAc,EACd,mBAAmB,EACnB,4CAA4C,GAE7C,MAAM,oBAAoB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\n/**\n * [Azure Cognitive Language Services](https://docs.microsoft.com/azure/cognitive-services/language-service/overview)\n * is a suite of natural language processing (NLP) skills built with\n * best-in-class Microsoft machine learning algorithms used to analyze\n * unstructured text for actions such as sentiment analysis, key phrase\n * extraction, and language detection.\n *\n * @packageDocumentation\n */\n\nexport { AzureKeyCredential } from \"@azure/core-auth\";\n\nexport { TextAnalysisClient } from \"./textAnalysisClient\";\nexport * from \"./models\";\nexport {\n  AssessmentSentiment,\n  SentimentAnalysisAction,\n  EntityLinkingAction,\n  EntityRecognitionAction,\n  KeyPhraseExtractionAction,\n  LanguageDetectionAction,\n  PiiEntityRecognitionAction,\n  StringIndexType,\n  LinkedEntity,\n  Entity,\n  DetectedLanguage,\n  PiiEntityCategory,\n  PiiEntityDomain,\n  SentimentConfidenceScores,\n  SentenceSentimentLabel,\n  DocumentSentimentLabel,\n  TargetConfidenceScores,\n  TokenSentimentLabel,\n  LanguageDetectionInput,\n  TextDocumentInput,\n  TextDocumentStatistics,\n  DocumentWarning,\n  WarningCode,\n  Match,\n  ActionCommon,\n  ActionPrebuilt,\n  HealthcareAction,\n  CustomEntityRecognitionAction,\n  CustomSingleLabelClassificationAction,\n  CustomMultiLabelClassificationAction,\n  ExtractiveSummarizationAction,\n  OperationStatus,\n  ActionCustom,\n  ClassificationCategory,\n  HealthcareAssertion,\n  HealthcareEntityCategory,\n  ExtractiveSummarizationOrderingCriteria,\n  SummarySentence,\n  EntityDataSource,\n  RelationType,\n  EntityAssociation,\n  EntityCertainty,\n  EntityConditionality,\n  FhirVersion,\n  /** orphan exports */\n  KnownPiiEntityDomain,\n  KnownPiiEntityCategory,\n  KnownStringIndexType,\n  KnownErrorCode,\n  KnownInnerErrorCode,\n  KnownExtractiveSummarizationOrderingCriteria,\n  TextDocumentBatchStatistics,\n} from \"./generated/models\";\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC;;;;;;;;GAQG;AAEH,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AAEtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,cAAc,UAAU,CAAC;AACzB,OAAO;AAwCL,qBAAqB;AACrB,oBAAoB,EACpB,sBAAsB,EACtB,oBAAoB,EACpB,cAAc,EACd,mBAAmB,GAEpB,MAAM,oBAAoB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\n/**\n * [Azure Cognitive Language Services](https://docs.microsoft.com/azure/cognitive-services/language-service/overview)\n * is a suite of natural language processing (NLP) skills built with\n * best-in-class Microsoft machine learning algorithms used to analyze\n * unstructured text for actions such as sentiment analysis, key phrase\n * extraction, and language detection.\n *\n * @packageDocumentation\n */\n\nexport { AzureKeyCredential } from \"@azure/core-auth\";\n\nexport { TextAnalysisClient } from \"./textAnalysisClient\";\nexport * from \"./models\";\nexport {\n  AssessmentSentiment,\n  SentimentAnalysisAction,\n  EntityLinkingAction,\n  EntityRecognitionAction,\n  KeyPhraseExtractionAction,\n  LanguageDetectionAction,\n  PiiEntityRecognitionAction,\n  StringIndexType,\n  LinkedEntity,\n  Entity,\n  DetectedLanguage,\n  PiiEntityCategory,\n  PiiEntityDomain,\n  SentimentConfidenceScores,\n  SentenceSentimentLabel,\n  DocumentSentimentLabel,\n  TargetConfidenceScores,\n  TokenSentimentLabel,\n  LanguageDetectionInput,\n  TextDocumentInput,\n  TextDocumentStatistics,\n  DocumentWarning,\n  WarningCode,\n  Match,\n  ActionCommon,\n  ActionPrebuilt,\n  HealthcareAction,\n  CustomEntityRecognitionAction,\n  CustomSingleLabelClassificationAction,\n  CustomMultiLabelClassificationAction,\n  ActionCustom,\n  ClassificationCategory,\n  HealthcareAssertion,\n  HealthcareEntityCategory,\n  EntityDataSource,\n  RelationType,\n  EntityAssociation,\n  EntityCertainty,\n  EntityConditionality,\n  /** orphan exports */\n  KnownPiiEntityDomain,\n  KnownPiiEntityCategory,\n  KnownStringIndexType,\n  KnownErrorCode,\n  KnownInnerErrorCode,\n  TextDocumentBatchStatistics,\n} from \"./generated/models\";\n"]}

package/dist-esm/src/lro.js

@@ -24,7 +24,7 @@ const jobStatusOperationSpec = {
 };
 async function getRawResponse(getResponse, options) {
     const { onResponse } = options || {};
-    let rawResponse = undefined;
+    let rawResponse;
     const flatResponse = await getResponse(Object.assign(Object.assign({}, options), { onResponse: (response, flatResponseParam) => {
             rawResponse = response;
             onResponse === null || onResponse === void 0 ? void 0 : onResponse(response, flatResponseParam);
@@ -46,7 +46,7 @@ async function sendRequest(settings) {
 /**
  * @internal
  */
-export function createSendPollRequest(settings) {
+function createSendPollRequest(settings) {
     const { client, options, tracing, spanStr } = settings;
     return async (path) => {
         return throwError(sendRequest({
@@ -65,8 +65,6 @@ export function createSendPollRequest(settings) {
 export function createAnalyzeBatchLro(settings) {
     const { client, commonOptions, documents, initialRequestOptions, pollRequestOptions, tasks, tracing, } = settings;
     return {
-        requestMethod: "POST",
-        requestPath: "/analyze-text/jobs",
         async sendInitialRequest() {
             return tracing.withSpan(`${clientName}.beginAnalyzeBatch`, Object.assign(Object.assign({}, commonOptions), initialRequestOptions), async (finalOptions) => throwError(getRawResponse((paramOptions) => client.analyzeText.submitJob({
                 tasks,
@@ -87,13 +85,13 @@ export function createAnalyzeBatchLro(settings) {
 /**
  * @internal
  */
-export function getDocsFromState(serializedState) {
+export function getDocIDsFromState(serializedState) {
     try {
-        const { documents } = JSON.parse(serializedState).state;
-        return documents;
+        const { docIds } = JSON.parse(serializedState).state;
+        return docIds;
     }
     catch (e) {
-        logger.error(`Documents are not found in the LRO's state. The results may not be ordered correctly.`);
+        logger.error(`Document IDs are not found in the LRO's state. The results may not be ordered correctly.`);
         return [];
     }
 }
@@ -103,8 +101,6 @@ export function getDocsFromState(serializedState) {
 export function createCreateAnalyzeBatchPollerLro(settings) {
     const { client, options, tracing } = settings;
     return {
-        requestMethod: "POST",
-        requestPath: "/analyze-text/jobs",
         async sendInitialRequest() {
             throw new Error(`The operation has already started`);
         },
@@ -120,9 +116,9 @@ export function createCreateAnalyzeBatchPollerLro(settings) {
  * @internal
  */
 export function processAnalyzeResult(options) {
-    return (_result, state) => {
-        const { client, documents, opOptions, tracing, continuationToken } = options;
-        const pageURL = continuationToken !== null && continuationToken !== void 0 ? continuationToken : state.pollingURL;
+    return () => {
+        const { client, docIds, opOptions, tracing, state } = options;
+        const pageURL = state.continuationToken;
         const pagedResult = {
             firstPageLink: pageURL,
             getPage: async (pageLink, maxPageSize) => {
@@ -138,7 +134,7 @@ export function processAnalyzeResult(options) {
                 });
                 const flatResponse = response.flatResponse;
                 return {
-                    page: transformAnalyzeBatchResults(documents, flatResponse.tasks.items),
+                    page: transformAnalyzeBatchResults(docIds, flatResponse.tasks.items),
                     nextPageLink: flatResponse.nextLink,
                 };
             },
@@ -149,31 +145,30 @@ export function processAnalyzeResult(options) {
 /**
  * @internal
  */
-export function createUpdateAnalyzeState(documents) {
+export function createUpdateAnalyzeState(docIds) {
     return (state, lastResponse) => {
-        const { createdOn, modifiedOn, id, status, displayName, expiresOn, tasks } = lastResponse.body;
+        const { createdOn, modifiedOn, id, displayName, expiresOn, tasks } = lastResponse.flatResponse;
         const mutableState = state;
         mutableState.createdOn = createdOn;
         mutableState.modifiedOn = modifiedOn;
         mutableState.expiresOn = expiresOn;
         mutableState.displayName = displayName;
         mutableState.id = id;
-        mutableState.status = status;
         mutableState.actionSucceededCount = tasks.completed;
         mutableState.actionFailedCount = tasks.failed;
         mutableState.actionInProgressCount = tasks.inProgress;
-        if (mutableState.documents === undefined && documents !== undefined) {
-            mutableState.documents = documents;
+        if (mutableState.docIds === undefined && docIds !== undefined) {
+            mutableState.docIds = docIds;
         }
     };
 }
 /**
  * @internal
  */
-export function createCancelOperation(settings) {
-    return async ({ id }) => {
-        const { client, options, tracing } = settings;
-        await tracing.withSpan(`${clientName}.beginAnalyzeBatch`, options, async (finalOptions) => throwError(getRawResponse((paramOptions) => client.analyzeText.cancelJob(id, paramOptions), finalOptions)));
-    };
+export function createPollerWithCancellation(settings) {
+    const { client, options, poller, id, tracing } = settings;
+    return Object.assign(Object.assign({}, poller), { sendCancellationRequest: async () => {
+            await tracing.withSpan(`${clientName}.beginAnalyzeBatch`, options, async (finalOptions) => throwError(getRawResponse((paramOptions) => client.analyzeText.cancelJob(id, paramOptions), finalOptions)));
+        } });
 }
 //# sourceMappingURL=lro.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"lro.js","sourceRoot":"","sources":["../../src/lro.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,KAAK,OAAO,MAAM,4BAA4B,CAAC;AACtD,OAAO,KAAK,UAAU,MAAM,+BAA+B,CAAC;AAS5D,OAAO,EAIL,gBAAgB,GACjB,MAAM,oBAAoB,CAAC;AAE5B,OAAO,EAAe,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AACxE,OAAO,EAAE,UAAU,EAAE,4BAA4B,EAAE,MAAM,cAAc,CAAC;AAGxE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAElC,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,EAAE,WAAW,CAAC,KAAK,CAAC,CAAC;AAEhE,MAAM,sBAAsB,GAAkB;IAC5C,UAAU,EAAE,KAAK;IACjB,SAAS,EAAE;QACT,GAAG,EAAE;YACH,UAAU,EAAE,OAAO,CAAC,mBAAmB;SACxC;QACD,OAAO,EAAE;YACP,UAAU,EAAE,OAAO,CAAC,aAAa;SAClC;KACF;IACD,gBAAgB,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC;IACrC,eAAe,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,iBAAiB,CAAC;IAChF,UAAU;CACX,CAAC;AAEF,KAAK,UAAU,cAAc,CAC3B,WAAsD,EACtD,OAAiB;IAEjB,MAAM,EAAE,UAAU,EAAE,GAAG,OAAO,IAAI,EAAE,CAAC;IACrC,IAAI,WAAW,GAAsC,SAAS,CAAC;IAC/D,MAAM,YAAY,GAAG,MAAM,WAAW,iCACjC,OAAO,KACV,UAAU,EAAE,CAAC,QAA+B,EAAE,iBAA0B,EAAE,EAAE;YAC1E,WAAW,GAAG,QAAQ,CAAC;YACvB,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAG,QAAQ,EAAE,iBAAiB,CAAC,CAAC;QAC5C,CAAC,IACD,CAAC;IACH,OAAO;QACL,YAAY;QACZ,WAAW,EAAE;YACX,UAAU,EAAE,WAAY,CAAC,MAAM;YAC/B,OAAO,EAAE,WAAY,CAAC,OAAO,CAAC,MAAM,EAAE;YACtC,IAAI,EAAE,WAAY,CAAC,UAAU;SAC9B;KACF,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,WAAW,CAAoC,QAQ7D;IACC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,GAAG,KAAK,EAAE,GAAG,QAAQ,CAAC;IACzF,OAAO,OAAO,CAAC,QAAQ,CAAC,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,YAAsB,EAAE,EAAE,CAC3E,UAAU,CACR,cAAc,CACZ,CAAC,OAAO,EAAE,EAAE,CACV,MAAM,CAAC,oBAAoB,CACzB,EAAE,OAAO,EAAE,kCAEN,IAAI,KACP,IAAI;QACJ,UAAU,IAEb,EACH,YAAY,CACb,CACF,CACF,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,qBAAqB,CAAoC,QAKxE;IACC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,QAAQ,CAAC;IACvD,OAAO,KAAK,EAAE,IAAY,EAAiC,EAAE;QAC3D,OAAO,UAAU,CACf,WAAW,CAAC;YACV,MAAM;YACN,SAAS,EAAE,OAAO;YAClB,IAAI;YACJ,OAAO;YACP,IAAI,EAAE,sBAAsB;YAC5B,OAAO;SACR,CAAC,CACH,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,qBAAqB,CAAC,QAYrC;IACC,MAAM,EACJ,MAAM,EACN,aAAa,EACb,SAAS,EACT,qBAAqB,EACrB,kBAAkB,EAClB,KAAK,EACL,OAAO,GACR,GAAG,QAAQ,CAAC;IACb,OAAO;QACL,aAAa,EAAE,MAAM;QACrB,WAAW,EAAE,oBAAoB;QACjC,KAAK,CAAC,kBAAkB;YACtB,OAAO,OAAO,CAAC,QAAQ,CACrB,GAAG,UAAU,oBAAoB,kCAE5B,aAAa,GACb,qBAAqB,GAE1B,KAAK,EAAE,YAAY,EAAE,EAAE,CACrB,UAAU,CACR,cAAc,CACZ,CAAC,YAAY,EAAE,EAAE,CACf,MAAM,CAAC,WAAW,CAAC,SAAS,CAC1B;gBACE,KAAK;gBACL,aAAa,EAAE;oBACb,SAAS;iBACV;gBACD,WAAW,EAAE,qBAAqB,CAAC,WAAW;aAC/C,EACD,YAAY,CACb,EACH,YAAY,CACb,CACF,CACJ,CAAC;QACJ,CAAC;QACD,eAAe,EAAE,qBAAqB,CAAC;YACrC,MAAM;YACN,OAAO,kCAAO,aAAa,GAAK,kBAAkB,CAAE;YACpD,OAAO,EAAE,GAAG,UAAU,oBAAoB;YAC1C,OAAO;SACR,CAAC;KACH,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAAC,eAAuB;IACtD,IAAI;QACF,MAAM,EAAE,SAAS,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,KAAK,CAAC;QACxD,OAAO,SAAS,CAAC;KAClB;IAAC,OAAO,CAAC,EAAE;QACV,MAAM,CAAC,KAAK,CACV,uFAAuF,CACxF,CAAC;QACF,OAAO,EAAE,CAAC;KACX;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,iCAAiC,CAAoC,QAIpF;IACC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,QAAQ,CAAC;IAC9C,OAAO;QACL,aAAa,EAAE,MAAM;QACrB,WAAW,EAAE,oBAAoB;QACjC,KAAK,CAAC,kBAAkB;YACtB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;QACD,eAAe,EAAE,qBAAqB,CAAC;YACrC,MAAM;YACN,OAAO;YACP,OAAO,EAAE,GAAG,UAAU,oBAAoB;YAC1C,OAAO;SACR,CAAC;KACH,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAAC,OAMpC;IACC,OAAO,CAAC,OAAgB,EAAE,KAAiC,EAA2B,EAAE;QACtF,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,OAAO,EAAE,iBAAiB,EAAE,GAAG,OAAO,CAAC;QAC7E,MAAM,OAAO,GAAG,iBAAiB,aAAjB,iBAAiB,cAAjB,iBAAiB,GAAK,KAAa,CAAC,UAAU,CAAC;QAC/D,MAAM,WAAW,GAAsC;YACrD,aAAa,EAAE,OAAO;YACtB,OAAO,EAAE,KAAK,EAAE,QAAgB,EAAE,WAAoB,EAAE,EAAE;gBACxD,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC;oBACjC,MAAM;oBACN,IAAI,EAAE,sBAAsB;oBAC5B,OAAO,EAAE,GAAG,UAAU,oBAAoB;oBAC1C,qEAAqE;oBACrE,0BAA0B;oBAC1B,SAAS,EAAE,WAAW,CAAC,CAAC,iCAAM,SAAS,KAAE,GAAG,EAAE,WAAW,IAAG,CAAC,CAAC,SAAS;oBACvE,IAAI,EAAE,QAAQ;oBACd,OAAO;iBACR,CAAC,CAAC;gBACH,MAAM,YAAY,GAAG,QAAQ,CAAC,YAA4C,CAAC;gBAC3E,OAAO;oBACL,IAAI,EAAE,4BAA4B,CAAC,SAAS,EAAE,YAAY,CAAC,KAAK,CAAC,KAAK,CAAC;oBACvE,YAAY,EAAE,YAAY,CAAC,QAAQ;iBACpC,CAAC;YACJ,CAAC;SACF,CAAC;QACF,OAAO,qBAAqB,CAAC,WAAW,CAAC,CAAC;IAC5C,CAAC,CAAC;AACJ,CAAC;AAMD;;GAEG;AACH,MAAM,UAAU,wBAAwB,CAAC,SAA+B;IACtE,OAAO,CAAC,KAAiC,EAAE,YAAyB,EAAQ,EAAE;QAC5E,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,SAAS,EAAE,KAAK,EAAE,GACxE,YAAY,CAAC,IAAoC,CAAC;QACpD,MAAM,YAAY,GAAG,KAEpB,CAAC;QACF,YAAY,CAAC,SAAS,GAAG,SAAS,CAAC;QACnC,YAAY,CAAC,UAAU,GAAG,UAAU,CAAC;QACrC,YAAY,CAAC,SAAS,GAAG,SAAS,CAAC;QACnC,YAAY,CAAC,WAAW,GAAG,WAAW,CAAC;QACvC,YAAY,CAAC,EAAE,GAAG,EAAE,CAAC;QACrB,YAAY,CAAC,MAAM,GAAG,MAAM,CAAC;QAC7B,YAAY,CAAC,oBAAoB,GAAG,KAAK,CAAC,SAAS,CAAC;QACpD,YAAY,CAAC,iBAAiB,GAAG,KAAK,CAAC,MAAM,CAAC;QAC9C,YAAY,CAAC,qBAAqB,GAAG,KAAK,CAAC,UAAU,CAAC;QACtD,IAAI,YAAY,CAAC,SAAS,KAAK,SAAS,IAAI,SAAS,KAAK,SAAS,EAAE;YACnE,YAAY,CAAC,SAAS,GAAG,SAAS,CAAC;SACpC;IACH,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,qBAAqB,CAAC,QAIrC;IACC,OAAO,KAAK,EAAE,EAAE,EAAE,EAAE,EAAiB,EAAE;QACrC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,QAAQ,CAAC;QAC9C,MAAM,OAAO,CAAC,QAAQ,CAAC,GAAG,UAAU,oBAAoB,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,CACxF,UAAU,CACR,cAAc,CACZ,CAAC,YAAY,EAAE,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,EAAE,EAAE,YAAY,CAAC,EAChE,YAAY,CACb,CACF,CACF,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\nimport * as Mappers from \"./generated/models/mappers\";\nimport * as Parameters from \"./generated/models/parameters\";\nimport {\n  AnalyzeBatchActionUnion,\n  AnalyzeTextJobStatusOptionalParams,\n  AnalyzeTextJobStatusResponse,\n  GeneratedClient,\n  TextDocumentInput,\n} from \"./generated\";\nimport { AnalyzeBatchOperationState, AnalyzeBatchResult, PagedAnalyzeBatchResult } from \"./models\";\nimport {\n  FullOperationResponse,\n  OperationOptions,\n  OperationSpec,\n  createSerializer,\n} from \"@azure/core-client\";\nimport { LongRunningOperation, LroResponse, RawResponse } from \"@azure/core-lro\";\nimport { PagedResult, getPagedAsyncIterator } from \"@azure/core-paging\";\nimport { throwError, transformAnalyzeBatchResults } from \"./transforms\";\nimport { HttpMethods } from \"@azure/core-rest-pipeline\";\nimport { TracingClient } from \"@azure/core-tracing\";\nimport { clientName } from \"./constants\";\nimport { logger } from \"./logger\";\n\nconst serializer = createSerializer(Mappers, /* isXml */ false);\n\nconst jobStatusOperationSpec: OperationSpec = {\n  httpMethod: \"GET\",\n  responses: {\n    200: {\n      bodyMapper: Mappers.AnalyzeTextJobState,\n    },\n    default: {\n      bodyMapper: Mappers.ErrorResponse,\n    },\n  },\n  headerParameters: [Parameters.accept],\n  queryParameters: [Parameters.top, Parameters.skip, Parameters.includeStatistics],\n  serializer,\n};\n\nasync function getRawResponse<TOptions extends OperationOptions, TResponse>(\n  getResponse: (options: TOptions) => Promise<TResponse>,\n  options: TOptions\n): Promise<LroResponse<TResponse>> {\n  const { onResponse } = options || {};\n  let rawResponse: FullOperationResponse | undefined = undefined;\n  const flatResponse = await getResponse({\n    ...options,\n    onResponse: (response: FullOperationResponse, flatResponseParam: unknown) => {\n      rawResponse = response;\n      onResponse?.(response, flatResponseParam);\n    },\n  });\n  return {\n    flatResponse,\n    rawResponse: {\n      statusCode: rawResponse!.status,\n      headers: rawResponse!.headers.toJSON(),\n      body: rawResponse!.parsedBody,\n    },\n  };\n}\n\nasync function sendRequest<TOptions extends OperationOptions>(settings: {\n  client: GeneratedClient;\n  tracing: TracingClient;\n  spec: OperationSpec;\n  spanStr: string;\n  opOptions: TOptions;\n  path: string;\n  httpMethod?: HttpMethods;\n}): Promise<LroResponse<unknown>> {\n  const { client, opOptions, path, spanStr, spec, tracing, httpMethod = \"GET\" } = settings;\n  return tracing.withSpan(spanStr, opOptions, async (finalOptions: TOptions) =>\n    throwError(\n      getRawResponse(\n        (options) =>\n          client.sendOperationRequest(\n            { options },\n            {\n              ...spec,\n              path,\n              httpMethod,\n            }\n          ),\n        finalOptions\n      )\n    )\n  );\n}\n\n/**\n * @internal\n */\nexport function createSendPollRequest<TOptions extends OperationOptions>(settings: {\n  client: GeneratedClient;\n  tracing: TracingClient;\n  options: TOptions;\n  spanStr: string;\n}): (path: string) => Promise<LroResponse<unknown>> {\n  const { client, options, tracing, spanStr } = settings;\n  return async (path: string): Promise<LroResponse<unknown>> => {\n    return throwError(\n      sendRequest({\n        client,\n        opOptions: options,\n        path,\n        spanStr,\n        spec: jobStatusOperationSpec,\n        tracing,\n      })\n    );\n  };\n}\n\n/**\n * @internal\n */\nexport function createAnalyzeBatchLro(settings: {\n  client: GeneratedClient;\n  tracing: TracingClient;\n  commonOptions: OperationOptions;\n  initialRequestOptions: {\n    displayName?: string;\n  };\n  pollRequestOptions: {\n    includeStatistics?: boolean;\n  };\n  documents: TextDocumentInput[];\n  tasks: AnalyzeBatchActionUnion[];\n}): LongRunningOperation<unknown> {\n  const {\n    client,\n    commonOptions,\n    documents,\n    initialRequestOptions,\n    pollRequestOptions,\n    tasks,\n    tracing,\n  } = settings;\n  return {\n    requestMethod: \"POST\",\n    requestPath: \"/analyze-text/jobs\",\n    async sendInitialRequest(): Promise<LroResponse<unknown>> {\n      return tracing.withSpan(\n        `${clientName}.beginAnalyzeBatch`,\n        {\n          ...commonOptions,\n          ...initialRequestOptions,\n        },\n        async (finalOptions) =>\n          throwError(\n            getRawResponse(\n              (paramOptions) =>\n                client.analyzeText.submitJob(\n                  {\n                    tasks,\n                    analysisInput: {\n                      documents,\n                    },\n                    displayName: initialRequestOptions.displayName,\n                  },\n                  paramOptions\n                ),\n              finalOptions\n            )\n          )\n      );\n    },\n    sendPollRequest: createSendPollRequest({\n      client,\n      options: { ...commonOptions, ...pollRequestOptions },\n      spanStr: `${clientName}.beginAnalyzeBatch`,\n      tracing,\n    }),\n  };\n}\n\n/**\n * @internal\n */\nexport function getDocsFromState(serializedState: string): TextDocumentInput[] {\n  try {\n    const { documents } = JSON.parse(serializedState).state;\n    return documents;\n  } catch (e) {\n    logger.error(\n      `Documents are not found in the LRO's state. The results may not be ordered correctly.`\n    );\n    return [];\n  }\n}\n\n/**\n * @internal\n */\nexport function createCreateAnalyzeBatchPollerLro<OptionsT extends OperationOptions>(settings: {\n  client: GeneratedClient;\n  tracing: TracingClient;\n  options: OptionsT;\n}): LongRunningOperation<unknown> {\n  const { client, options, tracing } = settings;\n  return {\n    requestMethod: \"POST\",\n    requestPath: \"/analyze-text/jobs\",\n    async sendInitialRequest(): Promise<LroResponse<unknown>> {\n      throw new Error(`The operation has already started`);\n    },\n    sendPollRequest: createSendPollRequest({\n      client,\n      options,\n      spanStr: `${clientName}.beginAnalyzeBatch`,\n      tracing,\n    }),\n  };\n}\n\n/**\n * @internal\n */\nexport function processAnalyzeResult(options: {\n  client: GeneratedClient;\n  tracing: TracingClient;\n  documents: TextDocumentInput[];\n  opOptions: AnalyzeTextJobStatusOptionalParams;\n  continuationToken?: string;\n}): (result: unknown, state: AnalyzeBatchOperationState) => PagedAnalyzeBatchResult {\n  return (_result: unknown, state: AnalyzeBatchOperationState): PagedAnalyzeBatchResult => {\n    const { client, documents, opOptions, tracing, continuationToken } = options;\n    const pageURL = continuationToken ?? (state as any).pollingURL;\n    const pagedResult: PagedResult<AnalyzeBatchResult[]> = {\n      firstPageLink: pageURL,\n      getPage: async (pageLink: string, maxPageSize?: number) => {\n        const response = await sendRequest({\n          client,\n          spec: jobStatusOperationSpec,\n          spanStr: `${clientName}.beginAnalyzeBatch`,\n          // if `top` is set to `undefined`, the default value will not be sent\n          // as part of the request.\n          opOptions: maxPageSize ? { ...opOptions, top: maxPageSize } : opOptions,\n          path: pageLink,\n          tracing,\n        });\n        const flatResponse = response.flatResponse as AnalyzeTextJobStatusResponse;\n        return {\n          page: transformAnalyzeBatchResults(documents, flatResponse.tasks.items),\n          nextPageLink: flatResponse.nextLink,\n        };\n      },\n    };\n    return getPagedAsyncIterator(pagedResult);\n  };\n}\n\ntype Writable<T> = {\n  -readonly [P in keyof T]: T[P];\n};\n\n/**\n * @internal\n */\nexport function createUpdateAnalyzeState(documents?: TextDocumentInput[]) {\n  return (state: AnalyzeBatchOperationState, lastResponse: RawResponse): void => {\n    const { createdOn, modifiedOn, id, status, displayName, expiresOn, tasks } =\n      lastResponse.body as AnalyzeTextJobStatusResponse;\n    const mutableState = state as Writable<AnalyzeBatchOperationState> & {\n      documents?: TextDocumentInput[];\n    };\n    mutableState.createdOn = createdOn;\n    mutableState.modifiedOn = modifiedOn;\n    mutableState.expiresOn = expiresOn;\n    mutableState.displayName = displayName;\n    mutableState.id = id;\n    mutableState.status = status;\n    mutableState.actionSucceededCount = tasks.completed;\n    mutableState.actionFailedCount = tasks.failed;\n    mutableState.actionInProgressCount = tasks.inProgress;\n    if (mutableState.documents === undefined && documents !== undefined) {\n      mutableState.documents = documents;\n    }\n  };\n}\n\n/**\n * @internal\n */\nexport function createCancelOperation(settings: {\n  client: GeneratedClient;\n  tracing: TracingClient;\n  options: AnalyzeTextJobStatusOptionalParams;\n}): (state: AnalyzeBatchOperationState) => Promise<void> {\n  return async ({ id }): Promise<void> => {\n    const { client, options, tracing } = settings;\n    await tracing.withSpan(`${clientName}.beginAnalyzeBatch`, options, async (finalOptions) =>\n      throwError(\n        getRawResponse(\n          (paramOptions) => client.analyzeText.cancelJob(id, paramOptions),\n          finalOptions\n        )\n      )\n    );\n  };\n}\n"]}
+{"version":3,"file":"lro.js","sourceRoot":"","sources":["../../src/lro.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,KAAK,OAAO,MAAM,4BAA4B,CAAC;AACtD,OAAO,KAAK,UAAU,MAAM,+BAA+B,CAAC;AAc5D,OAAO,EAIL,gBAAgB,GACjB,MAAM,oBAAoB,CAAC;AAE5B,OAAO,EAAe,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AACxE,OAAO,EAAE,UAAU,EAAE,4BAA4B,EAAE,MAAM,cAAc,CAAC;AAGxE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAElC,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,EAAE,WAAW,CAAC,KAAK,CAAC,CAAC;AAEhE,MAAM,sBAAsB,GAAkB;IAC5C,UAAU,EAAE,KAAK;IACjB,SAAS,EAAE;QACT,GAAG,EAAE;YACH,UAAU,EAAE,OAAO,CAAC,mBAAmB;SACxC;QACD,OAAO,EAAE;YACP,UAAU,EAAE,OAAO,CAAC,aAAa;SAClC;KACF;IACD,gBAAgB,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC;IACrC,eAAe,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,iBAAiB,CAAC;IAChF,UAAU;CACX,CAAC;AAEF,KAAK,UAAU,cAAc,CAC3B,WAAsD,EACtD,OAAiB;IAEjB,MAAM,EAAE,UAAU,EAAE,GAAG,OAAO,IAAI,EAAE,CAAC;IACrC,IAAI,WAAkC,CAAC;IACvC,MAAM,YAAY,GAAG,MAAM,WAAW,iCACjC,OAAO,KACV,UAAU,EAAE,CAAC,QAAQ,EAAE,iBAAiB,EAAE,EAAE;YAC1C,WAAW,GAAG,QAAQ,CAAC;YACvB,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAG,QAAQ,EAAE,iBAAiB,CAAC,CAAC;QAC5C,CAAC,IACD,CAAC;IACH,OAAO;QACL,YAAY;QACZ,WAAW,EAAE;YACX,UAAU,EAAE,WAAY,CAAC,MAAM;YAC/B,OAAO,EAAE,WAAY,CAAC,OAAO,CAAC,MAAM,EAAE;YACtC,IAAI,EAAE,WAAY,CAAC,UAAU;SAC9B;KACF,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,WAAW,CAAoC,QAQ7D;IACC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,GAAG,KAAK,EAAE,GAAG,QAAQ,CAAC;IACzF,OAAO,OAAO,CAAC,QAAQ,CAAC,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,YAAsB,EAAE,EAAE,CAC3E,UAAU,CACR,cAAc,CACZ,CAAC,OAAO,EAAE,EAAE,CACV,MAAM,CAAC,oBAAoB,CACzB,EAAE,OAAO,EAAE,kCAEN,IAAI,KACP,IAAI;QACJ,UAAU,IAEb,EACH,YAAY,CACb,CACF,CACF,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,qBAAqB,CAAoC,QAKjE;IACC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,QAAQ,CAAC;IACvD,OAAO,KAAK,EAAE,IAAY,EAAiC,EAAE;QAC3D,OAAO,UAAU,CACf,WAAW,CAAC;YACV,MAAM;YACN,SAAS,EAAE,OAAO;YAClB,IAAI;YACJ,OAAO;YACP,IAAI,EAAE,sBAAsB;YAC5B,OAAO;SACR,CAAC,CACH,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,qBAAqB,CAAC,QAYrC;IACC,MAAM,EACJ,MAAM,EACN,aAAa,EACb,SAAS,EACT,qBAAqB,EACrB,kBAAkB,EAClB,KAAK,EACL,OAAO,GACR,GAAG,QAAQ,CAAC;IACb,OAAO;QACL,KAAK,CAAC,kBAAkB;YACtB,OAAO,OAAO,CAAC,QAAQ,CACrB,GAAG,UAAU,oBAAoB,kCAE5B,aAAa,GACb,qBAAqB,GAE1B,KAAK,EAAE,YAAY,EAAE,EAAE,CACrB,UAAU,CACR,cAAc,CACZ,CAAC,YAAY,EAAE,EAAE,CACf,MAAM,CAAC,WAAW,CAAC,SAAS,CAC1B;gBACE,KAAK;gBACL,aAAa,EAAE;oBACb,SAAS;iBACV;gBACD,WAAW,EAAE,qBAAqB,CAAC,WAAW;aAC/C,EACD,YAAY,CACb,EACH,YAAY,CACb,CACF,CACJ,CAAC;QACJ,CAAC;QACD,eAAe,EAAE,qBAAqB,CAAC;YACrC,MAAM;YACN,OAAO,kCAAO,aAAa,GAAK,kBAAkB,CAAE;YACpD,OAAO,EAAE,GAAG,UAAU,oBAAoB;YAC1C,OAAO;SACR,CAAC;KACH,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,eAAuB;IACxD,IAAI;QACF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,KAAK,CAAC;QACrD,OAAO,MAAM,CAAC;KACf;IAAC,OAAO,CAAC,EAAE;QACV,MAAM,CAAC,KAAK,CACV,0FAA0F,CAC3F,CAAC;QACF,OAAO,EAAE,CAAC;KACX;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,iCAAiC,CAAoC,QAIpF;IACC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,QAAQ,CAAC;IAC9C,OAAO;QACL,KAAK,CAAC,kBAAkB;YACtB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;QACD,eAAe,EAAE,qBAAqB,CAAC;YACrC,MAAM;YACN,OAAO;YACP,OAAO,EAAE,GAAG,UAAU,oBAAoB;YAC1C,OAAO;SACR,CAAC;KACH,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAAC,OAMpC;IACC,OAAO,GAA4B,EAAE;QACnC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC;QAC9D,MAAM,OAAO,GAAG,KAAK,CAAC,iBAAiB,CAAC;QACxC,MAAM,WAAW,GAAsC;YACrD,aAAa,EAAE,OAAO;YACtB,OAAO,EAAE,KAAK,EAAE,QAAgB,EAAE,WAAoB,EAAE,EAAE;gBACxD,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC;oBACjC,MAAM;oBACN,IAAI,EAAE,sBAAsB;oBAC5B,OAAO,EAAE,GAAG,UAAU,oBAAoB;oBAC1C,qEAAqE;oBACrE,0BAA0B;oBAC1B,SAAS,EAAE,WAAW,CAAC,CAAC,iCAAM,SAAS,KAAE,GAAG,EAAE,WAAW,IAAG,CAAC,CAAC,SAAS;oBACvE,IAAI,EAAE,QAAQ;oBACd,OAAO;iBACR,CAAC,CAAC;gBACH,MAAM,YAAY,GAAG,QAAQ,CAAC,YAA4C,CAAC;gBAC3E,OAAO;oBACL,IAAI,EAAE,4BAA4B,CAAC,MAAM,EAAE,YAAY,CAAC,KAAK,CAAC,KAAK,CAAC;oBACpE,YAAY,EAAE,YAAY,CAAC,QAAQ;iBACpC,CAAC;YACJ,CAAC;SACF,CAAC;QACF,OAAO,qBAAqB,CAAC,WAAW,CAAC,CAAC;IAC5C,CAAC,CAAC;AACJ,CAAC;AAMD;;GAEG;AACH,MAAM,UAAU,wBAAwB,CAAC,MAAiB;IACxD,OAAO,CAAC,KAAiC,EAAE,YAAyB,EAAQ,EAAE;QAC5E,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,EAAE,EAAE,WAAW,EAAE,SAAS,EAAE,KAAK,EAAE,GAChE,YAAY,CAAC,YAA4C,CAAC;QAC5D,MAAM,YAAY,GAAG,KAEpB,CAAC;QACF,YAAY,CAAC,SAAS,GAAG,SAAS,CAAC;QACnC,YAAY,CAAC,UAAU,GAAG,UAAU,CAAC;QACrC,YAAY,CAAC,SAAS,GAAG,SAAS,CAAC;QACnC,YAAY,CAAC,WAAW,GAAG,WAAW,CAAC;QACvC,YAAY,CAAC,EAAE,GAAG,EAAE,CAAC;QACrB,YAAY,CAAC,oBAAoB,GAAG,KAAK,CAAC,SAAS,CAAC;QACpD,YAAY,CAAC,iBAAiB,GAAG,KAAK,CAAC,MAAM,CAAC;QAC9C,YAAY,CAAC,qBAAqB,GAAG,KAAK,CAAC,UAAU,CAAC;QACtD,IAAI,YAAY,CAAC,MAAM,KAAK,SAAS,IAAI,MAAM,KAAK,SAAS,EAAE;YAC7D,YAAY,CAAC,MAAM,GAAG,MAAM,CAAC;SAC9B;IACH,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,4BAA4B,CAAC,QAM5C;IACC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,GAAG,QAAQ,CAAC;IAC1D,uCACK,MAAM,KACT,uBAAuB,EAAE,KAAK,IAAI,EAAE;YAClC,MAAM,OAAO,CAAC,QAAQ,CAAC,GAAG,UAAU,oBAAoB,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,CACxF,UAAU,CACR,cAAc,CACZ,CAAC,YAAY,EAAE,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,EAAE,EAAE,YAAY,CAAC,EAChE,YAAY,CACb,CACF,CACF,CAAC;QACJ,CAAC,IACD;AACJ,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\nimport * as Mappers from \"./generated/models/mappers\";\nimport * as Parameters from \"./generated/models/parameters\";\nimport {\n  AnalyzeBatchActionUnion,\n  AnalyzeTextJobStatusOptionalParams,\n  AnalyzeTextJobStatusResponse,\n  GeneratedClient,\n  TextDocumentInput,\n} from \"./generated\";\nimport {\n  AnalyzeBatchOperationState,\n  AnalyzeBatchResult,\n  PagedAnalyzeBatchResult,\n  PollerLike,\n} from \"./models\";\nimport {\n  FullOperationResponse,\n  OperationOptions,\n  OperationSpec,\n  createSerializer,\n} from \"@azure/core-client\";\nimport { LongRunningOperation, LroResponse, SimplePollerLike } from \"@azure/core-lro\";\nimport { PagedResult, getPagedAsyncIterator } from \"@azure/core-paging\";\nimport { throwError, transformAnalyzeBatchResults } from \"./transforms\";\nimport { HttpMethods } from \"@azure/core-rest-pipeline\";\nimport { TracingClient } from \"@azure/core-tracing\";\nimport { clientName } from \"./constants\";\nimport { logger } from \"./logger\";\n\nconst serializer = createSerializer(Mappers, /* isXml */ false);\n\nconst jobStatusOperationSpec: OperationSpec = {\n  httpMethod: \"GET\",\n  responses: {\n    200: {\n      bodyMapper: Mappers.AnalyzeTextJobState,\n    },\n    default: {\n      bodyMapper: Mappers.ErrorResponse,\n    },\n  },\n  headerParameters: [Parameters.accept],\n  queryParameters: [Parameters.top, Parameters.skip, Parameters.includeStatistics],\n  serializer,\n};\n\nasync function getRawResponse<TOptions extends OperationOptions, TResponse>(\n  getResponse: (options: TOptions) => Promise<TResponse>,\n  options: TOptions\n): Promise<LroResponse<TResponse>> {\n  const { onResponse } = options || {};\n  let rawResponse: FullOperationResponse;\n  const flatResponse = await getResponse({\n    ...options,\n    onResponse: (response, flatResponseParam) => {\n      rawResponse = response;\n      onResponse?.(response, flatResponseParam);\n    },\n  });\n  return {\n    flatResponse,\n    rawResponse: {\n      statusCode: rawResponse!.status,\n      headers: rawResponse!.headers.toJSON(),\n      body: rawResponse!.parsedBody,\n    },\n  };\n}\n\nasync function sendRequest<TOptions extends OperationOptions>(settings: {\n  client: GeneratedClient;\n  tracing: TracingClient;\n  spec: OperationSpec;\n  spanStr: string;\n  opOptions: TOptions;\n  path: string;\n  httpMethod?: HttpMethods;\n}): Promise<LroResponse<unknown>> {\n  const { client, opOptions, path, spanStr, spec, tracing, httpMethod = \"GET\" } = settings;\n  return tracing.withSpan(spanStr, opOptions, async (finalOptions: TOptions) =>\n    throwError(\n      getRawResponse(\n        (options) =>\n          client.sendOperationRequest(\n            { options },\n            {\n              ...spec,\n              path,\n              httpMethod,\n            }\n          ),\n        finalOptions\n      )\n    )\n  );\n}\n\n/**\n * @internal\n */\nfunction createSendPollRequest<TOptions extends OperationOptions>(settings: {\n  client: GeneratedClient;\n  tracing: TracingClient;\n  options: TOptions;\n  spanStr: string;\n}): (path: string) => Promise<LroResponse<unknown>> {\n  const { client, options, tracing, spanStr } = settings;\n  return async (path: string): Promise<LroResponse<unknown>> => {\n    return throwError(\n      sendRequest({\n        client,\n        opOptions: options,\n        path,\n        spanStr,\n        spec: jobStatusOperationSpec,\n        tracing,\n      })\n    );\n  };\n}\n\n/**\n * @internal\n */\nexport function createAnalyzeBatchLro(settings: {\n  client: GeneratedClient;\n  tracing: TracingClient;\n  commonOptions: OperationOptions;\n  initialRequestOptions: {\n    displayName?: string;\n  };\n  pollRequestOptions: {\n    includeStatistics?: boolean;\n  };\n  documents: TextDocumentInput[];\n  tasks: AnalyzeBatchActionUnion[];\n}): LongRunningOperation {\n  const {\n    client,\n    commonOptions,\n    documents,\n    initialRequestOptions,\n    pollRequestOptions,\n    tasks,\n    tracing,\n  } = settings;\n  return {\n    async sendInitialRequest(): Promise<LroResponse<unknown>> {\n      return tracing.withSpan(\n        `${clientName}.beginAnalyzeBatch`,\n        {\n          ...commonOptions,\n          ...initialRequestOptions,\n        },\n        async (finalOptions) =>\n          throwError(\n            getRawResponse(\n              (paramOptions) =>\n                client.analyzeText.submitJob(\n                  {\n                    tasks,\n                    analysisInput: {\n                      documents,\n                    },\n                    displayName: initialRequestOptions.displayName,\n                  },\n                  paramOptions\n                ),\n              finalOptions\n            )\n          )\n      );\n    },\n    sendPollRequest: createSendPollRequest({\n      client,\n      options: { ...commonOptions, ...pollRequestOptions },\n      spanStr: `${clientName}.beginAnalyzeBatch`,\n      tracing,\n    }),\n  };\n}\n\n/**\n * @internal\n */\nexport function getDocIDsFromState(serializedState: string): string[] {\n  try {\n    const { docIds } = JSON.parse(serializedState).state;\n    return docIds;\n  } catch (e) {\n    logger.error(\n      `Document IDs are not found in the LRO's state. The results may not be ordered correctly.`\n    );\n    return [];\n  }\n}\n\n/**\n * @internal\n */\nexport function createCreateAnalyzeBatchPollerLro<OptionsT extends OperationOptions>(settings: {\n  client: GeneratedClient;\n  tracing: TracingClient;\n  options: OptionsT;\n}): LongRunningOperation {\n  const { client, options, tracing } = settings;\n  return {\n    async sendInitialRequest(): Promise<LroResponse<unknown>> {\n      throw new Error(`The operation has already started`);\n    },\n    sendPollRequest: createSendPollRequest({\n      client,\n      options,\n      spanStr: `${clientName}.beginAnalyzeBatch`,\n      tracing,\n    }),\n  };\n}\n\n/**\n * @internal\n */\nexport function processAnalyzeResult(options: {\n  client: GeneratedClient;\n  tracing: TracingClient;\n  docIds: string[];\n  opOptions: AnalyzeTextJobStatusOptionalParams;\n  state: { continuationToken: string };\n}): (result: unknown, state: AnalyzeBatchOperationState) => PagedAnalyzeBatchResult {\n  return (): PagedAnalyzeBatchResult => {\n    const { client, docIds, opOptions, tracing, state } = options;\n    const pageURL = state.continuationToken;\n    const pagedResult: PagedResult<AnalyzeBatchResult[]> = {\n      firstPageLink: pageURL,\n      getPage: async (pageLink: string, maxPageSize?: number) => {\n        const response = await sendRequest({\n          client,\n          spec: jobStatusOperationSpec,\n          spanStr: `${clientName}.beginAnalyzeBatch`,\n          // if `top` is set to `undefined`, the default value will not be sent\n          // as part of the request.\n          opOptions: maxPageSize ? { ...opOptions, top: maxPageSize } : opOptions,\n          path: pageLink,\n          tracing,\n        });\n        const flatResponse = response.flatResponse as AnalyzeTextJobStatusResponse;\n        return {\n          page: transformAnalyzeBatchResults(docIds, flatResponse.tasks.items),\n          nextPageLink: flatResponse.nextLink,\n        };\n      },\n    };\n    return getPagedAsyncIterator(pagedResult);\n  };\n}\n\ntype Writable<T> = {\n  -readonly [P in keyof T]: T[P];\n};\n\n/**\n * @internal\n */\nexport function createUpdateAnalyzeState(docIds?: string[]) {\n  return (state: AnalyzeBatchOperationState, lastResponse: LroResponse): void => {\n    const { createdOn, modifiedOn, id, displayName, expiresOn, tasks } =\n      lastResponse.flatResponse as AnalyzeTextJobStatusResponse;\n    const mutableState = state as Writable<AnalyzeBatchOperationState> & {\n      docIds?: string[];\n    };\n    mutableState.createdOn = createdOn;\n    mutableState.modifiedOn = modifiedOn;\n    mutableState.expiresOn = expiresOn;\n    mutableState.displayName = displayName;\n    mutableState.id = id;\n    mutableState.actionSucceededCount = tasks.completed;\n    mutableState.actionFailedCount = tasks.failed;\n    mutableState.actionInProgressCount = tasks.inProgress;\n    if (mutableState.docIds === undefined && docIds !== undefined) {\n      mutableState.docIds = docIds;\n    }\n  };\n}\n\n/**\n * @internal\n */\nexport function createPollerWithCancellation(settings: {\n  poller: SimplePollerLike<AnalyzeBatchOperationState, PagedAnalyzeBatchResult>;\n  client: GeneratedClient;\n  tracing: TracingClient;\n  options: AnalyzeTextJobStatusOptionalParams;\n  id: string;\n}): PollerLike<AnalyzeBatchOperationState, PagedAnalyzeBatchResult> {\n  const { client, options, poller, id, tracing } = settings;\n  return {\n    ...poller,\n    sendCancellationRequest: async () => {\n      await tracing.withSpan(`${clientName}.beginAnalyzeBatch`, options, async (finalOptions) =>\n        throwError(\n          getRawResponse(\n            (paramOptions) => client.analyzeText.cancelJob(id, paramOptions),\n            finalOptions\n          )\n        )\n      );\n    },\n  };\n}\n"]}

package/dist-esm/src/models.js

@@ -22,19 +22,10 @@ export const AnalyzeBatchActionNames = {
     KeyPhraseExtraction: "KeyPhraseExtraction",
     EntityLinking: "EntityLinking",
     Healthcare: "Healthcare",
-    ExtractiveSummarization: "ExtractiveSummarization",
     CustomEntityRecognition: "CustomEntityRecognition",
     CustomSingleLabelClassification: "CustomSingleLabelClassification",
     CustomMultiLabelClassification: "CustomMultiLabelClassification",
 };
-/**
- * Known values of the {@link HealthcareAction.fhirVersion} parameter.
- */
-export var KnownFhirVersion;
-(function (KnownFhirVersion) {
-    /** 4.0.1 */
-    KnownFhirVersion["4.0.1"] = "4.0.1";
-})(KnownFhirVersion || (KnownFhirVersion = {}));
 /**
  * Enum of possible error codes of a {@link TextAnalysisError}.
  */

package/dist-esm/src/models.js.map

@@ -1 +1 @@
-{"version":3,"file":"models.js","sourceRoot":"","sources":["../../src/models.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAkBL,cAAc,EACd,mBAAmB,GAcpB,MAAM,aAAa,CAAC;AA8DrB;;GAEG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG;IAChC,aAAa,EAAE,eAAe;IAC9B,iBAAiB,EAAE,mBAAmB;IACtC,mBAAmB,EAAE,qBAAqB;IAC1C,oBAAoB,EAAE,sBAAsB;IAC5C,iBAAiB,EAAE,mBAAmB;IACtC,iBAAiB,EAAE,mBAAmB;CAC9B,CAAC;AAEX;;GAEG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG;IACrC,iBAAiB,EAAE,mBAAmB;IACtC,iBAAiB,EAAE,mBAAmB;IACtC,oBAAoB,EAAE,sBAAsB;IAC5C,mBAAmB,EAAE,qBAAqB;IAC1C,aAAa,EAAE,eAAe;IAC9B,UAAU,EAAE,YAAY;IACxB,uBAAuB,EAAE,yBAAyB;IAClD,uBAAuB,EAAE,yBAAyB;IAClD,+BAA+B,EAAE,iCAAiC;IAClE,8BAA8B,EAAE,gCAAgC;CACxD,CAAC;AAmBX;;GAEG;AACH,MAAM,CAAN,IAAY,gBAGX;AAHD,WAAY,gBAAgB;IAC1B,YAAY;IACZ,mCAAiB,CAAA;AACnB,CAAC,EAHW,gBAAgB,KAAhB,gBAAgB,QAG3B;AAcD;;GAEG;AACH,MAAM,CAAC,MAAM,0BAA0B,mCAAQ,cAAc,GAAK,mBAAmB,CAAE,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\nimport {\n  AssessmentSentiment,\n  ClassificationCategory,\n  CustomEntityRecognitionAction,\n  CustomMultiLabelClassificationAction,\n  CustomSingleLabelClassificationAction,\n  DetectedLanguage,\n  DocumentSentimentLabel,\n  DocumentWarning,\n  Entity,\n  EntityDataSource,\n  EntityLinkingAction,\n  EntityRecognitionAction,\n  ExtractiveSummarizationAction,\n  HealthcareAction,\n  HealthcareAssertion,\n  HealthcareEntityCategory,\n  KeyPhraseExtractionAction,\n  KnownErrorCode,\n  KnownInnerErrorCode,\n  LanguageDetectionAction,\n  LinkedEntity,\n  OperationStatus,\n  PiiEntityRecognitionAction,\n  RelationType,\n  SentenceSentimentLabel,\n  SentimentAnalysisAction,\n  SentimentConfidenceScores,\n  SummarySentence,\n  TargetConfidenceScores,\n  TextDocumentBatchStatistics,\n  TextDocumentStatistics,\n  TokenSentimentLabel,\n} from \"./generated\";\nimport { CommonClientOptions, OperationOptions } from \"@azure/core-client\";\nimport { PollOperationState, PollerLike } from \"@azure/core-lro\";\nimport { PagedAsyncIterableIterator } from \"@azure/core-paging\";\n\n/**\n * Configuration options for {@link TextAnalysisClient}.\n */\nexport interface TextAnalysisClientOptions extends CommonClientOptions {\n  /**\n   * The default country hint to use. Defaults to \"us\".\n   */\n  defaultCountryHint?: string;\n\n  /**\n   * The default language to use. Defaults to \"en\".\n   */\n  defaultLanguage?: string;\n  /**\n   * The version of the Cognitive Language Service API to use.\n   */\n  apiVersion?: string;\n}\n\n/**\n * Options common to all operations.\n */\nexport interface TextAnalysisOperationOptions extends OperationOptions {\n  /**\n   * If set to true, response will contain input and document level statistics.\n   */\n  includeStatistics?: boolean;\n  /**\n   * The version of the Cognitive Language Service API to use.\n   */\n  apiVersion?: string;\n}\n\n/**\n * Options for the begin analyze actions operation.\n */\nexport interface BeginAnalyzeBatchOptions extends TextAnalysisOperationOptions {\n  /**\n   * Time delay between poll requests, in milliseconds.\n   */\n  updateIntervalInMs?: number;\n  /**\n   * The operation's display name.\n   */\n  displayName?: string;\n}\n\n/**\n * Options for the begin analyze actions operation.\n */\nexport interface RestoreAnalyzeBatchPollerOptions extends TextAnalysisOperationOptions {\n  /**\n   * Time delay between poll requests, in milliseconds.\n   */\n  updateIntervalInMs?: number;\n}\n\n/**\n * Type of actions supported by the {@link TextAnalysisClient.analyze} method.\n */\nexport const AnalyzeActionNames = {\n  EntityLinking: \"EntityLinking\",\n  EntityRecognition: \"EntityRecognition\",\n  KeyPhraseExtraction: \"KeyPhraseExtraction\",\n  PiiEntityRecognition: \"PiiEntityRecognition\",\n  LanguageDetection: \"LanguageDetection\",\n  SentimentAnalysis: \"SentimentAnalysis\",\n} as const;\n\n/**\n * Type of actions supported by the {@link TextAnalysisClient.beginAnalyzeBatch} method.\n */\nexport const AnalyzeBatchActionNames = {\n  SentimentAnalysis: \"SentimentAnalysis\",\n  EntityRecognition: \"EntityRecognition\",\n  PiiEntityRecognition: \"PiiEntityRecognition\",\n  KeyPhraseExtraction: \"KeyPhraseExtraction\",\n  EntityLinking: \"EntityLinking\",\n  Healthcare: \"Healthcare\",\n  ExtractiveSummarization: \"ExtractiveSummarization\",\n  CustomEntityRecognition: \"CustomEntityRecognition\",\n  CustomSingleLabelClassification: \"CustomSingleLabelClassification\",\n  CustomMultiLabelClassification: \"CustomMultiLabelClassification\",\n} as const;\n\n/**\n * Type of actions supported by the {@link TextAnalysisClient.analyze} method.\n */\nexport type AnalyzeActionName = keyof typeof AnalyzeActionNames;\n\n/**\n * The type of parameters for every action in ${@link AnalyzeActionNames}.\n */\nexport type AnalyzeActionParameters<ActionName extends AnalyzeActionName> = {\n  EntityLinking: EntityLinkingAction;\n  EntityRecognition: EntityRecognitionAction;\n  PiiEntityRecognition: PiiEntityRecognitionAction;\n  KeyPhraseExtraction: KeyPhraseExtractionAction;\n  SentimentAnalysis: SentimentAnalysisAction;\n  LanguageDetection: LanguageDetectionAction;\n}[ActionName];\n\n/**\n * Known values of the {@link HealthcareAction.fhirVersion} parameter.\n */\nexport enum KnownFhirVersion {\n  /** 4.0.1 */\n  \"4.0.1\" = \"4.0.1\",\n}\n\n/**\n * The type of results of every action in ${@link AnalyzeActionNames}.\n */\nexport type AnalyzeResult<ActionName extends AnalyzeActionName> = {\n  EntityLinking: EntityLinkingResult[];\n  EntityRecognition: EntityRecognitionResult[];\n  PiiEntityRecognition: PiiEntityRecognitionResult[];\n  KeyPhraseExtraction: KeyPhraseExtractionResult[];\n  SentimentAnalysis: SentimentAnalysisResult[];\n  LanguageDetection: LanguageDetectionResult[];\n}[ActionName];\n\n/**\n * Enum of possible error codes of a {@link TextAnalysisError}.\n */\nexport const KnownTextAnalysisErrorCode = { ...KnownErrorCode, ...KnownInnerErrorCode };\n\n/**\n * Type describing an API error.\n */\nexport interface TextAnalysisError {\n  /**\n   * A code describing the kind of error produced. See {@link KnownTextAnalysisErrorCode}.\n   */\n  readonly code: string;\n  /**\n   * A message from the service explaining the error\n   */\n  readonly message: string;\n  /**\n   * The target of the particular error (for example, the name of an invalid parameter)\n   */\n  readonly target?: string;\n}\n\n/**\n * Base type for results of an action corresponding to a single input document.\n */\nexport interface TextAnalysisSuccessResult {\n  /**\n   * Unique, non-empty document identifier.\n   */\n  readonly id: string;\n\n  /**\n   * Statistics about the input document and how it was processed by the service.\n   * This property will have a value when includeStatistics is set to true in\n   * the client call.\n   */\n  readonly statistics?: TextDocumentStatistics;\n\n  /**\n   * An array of warning data corresponding to this document.\n   *\n   * If no warnings were returned, this array will be empty.\n   */\n  readonly warnings: DocumentWarning[];\n\n  /**\n   * Discriminant to determine if this is an error result.\n   */\n  readonly error?: undefined;\n}\n\n/**\n * Base type for error results of an action corresponding to a single document.\n */\nexport interface TextAnalysisErrorResult {\n  /**\n   * Unique, non-empty document identifier.\n   */\n  readonly id: string;\n\n  /**\n   * The Error for this document result.\n   */\n  readonly error: TextAnalysisError;\n}\n\n/**\n * The result of an entity recognition action on a single document.\n */\nexport type EntityRecognitionResult = EntityRecognitionSuccessResult | EntityRecognitionErrorResult;\n\n/**\n * The result of an entity recognition action on a single document, containing\n * a collection of {@link Entity} objects identified in that document.\n */\nexport interface EntityRecognitionSuccessResult extends TextAnalysisSuccessResult {\n  /**\n   * The collection of entities identified in the input document.\n   */\n  readonly entities: Entity[];\n}\n\n/**\n * An error result from an entity recognition action on a single document.\n */\nexport type EntityRecognitionErrorResult = TextAnalysisErrorResult;\n\n/**\n * The result of an entity recognition action on a single document.\n */\nexport type PiiEntityRecognitionResult =\n  | PiiEntityRecognitionSuccessResult\n  | PiiEntityRecognitionErrorResult;\n\n/**\n * The result of a pii entity recognition action on a single document,\n * containing the collection of {@link Entity} objects identified in that\n * document.\n */\nexport interface PiiEntityRecognitionSuccessResult extends TextAnalysisSuccessResult {\n  /**\n   * The collection of entities identified in the input document.\n   */\n  readonly entities: Entity[];\n  /**\n   * The text redacted.\n   */\n  readonly redactedText: string;\n}\n\n/**\n * An error result from a pii entity recognition action on a single document.\n */\nexport type PiiEntityRecognitionErrorResult = TextAnalysisErrorResult;\n\n/**\n * The result of an entity linking action on a single document.\n */\nexport type EntityLinkingResult = EntityLinkingSuccessResult | EntityLinkingErrorResult;\n\n/**\n * The result of a entity linking action on a single document, containing a\n * collection of the {@link LinkedEntity} objects identified in that document.\n */\nexport interface EntityLinkingSuccessResult extends TextAnalysisSuccessResult {\n  /**\n   * The collection of entities identified in the input document.\n   */\n  readonly entities: LinkedEntity[];\n}\n\n/**\n * An error result from an entity linking action on a single document.\n */\nexport type EntityLinkingErrorResult = TextAnalysisErrorResult;\n\n/**\n * The result of a language detection action on a single document.\n */\nexport type LanguageDetectionResult = LanguageDetectionSuccessResult | LanguageDetectionErrorResult;\n\n/**\n * The result of a language detection action on a single document,\n * containing a prediction of what language the document is written in.\n */\nexport interface LanguageDetectionSuccessResult extends TextAnalysisSuccessResult {\n  /**\n   * The top detected language by confidence score.\n   */\n  readonly primaryLanguage: DetectedLanguage;\n}\n\n/**\n * An error result from a language detection action on a single document.\n */\nexport type LanguageDetectionErrorResult = TextAnalysisErrorResult;\n\n/**\n * The result of a sentiment analysis action on a single document.\n */\nexport type KeyPhraseExtractionResult =\n  | KeyPhraseExtractionSuccessResult\n  | KeyPhraseExtractionErrorResult;\n\n/**\n * The result of a key phrase extraction action on a single document,\n * containing a collection of the key phrases identified in that document.\n */\nexport interface KeyPhraseExtractionSuccessResult extends TextAnalysisSuccessResult {\n  /**\n   * A list of representative words or phrases. The number of key phrases\n   * returned is proportional to the number of words in the input document.\n   */\n  readonly keyPhrases: string[];\n}\n\n/**\n * An error result from a key phrase extraction action on a single document.\n */\nexport type KeyPhraseExtractionErrorResult = TextAnalysisErrorResult;\n\n/**\n * The result of a sentiment analysis action on a single document.\n */\nexport type SentimentAnalysisResult = SentimentAnalysisSuccessResult | SentimentAnalysisErrorResult;\n\n/**\n * The result of a sentiment analysis action on a single document,\n * containing the predicted sentiment for each sentence as well as for the full\n * document.\n */\nexport interface SentimentAnalysisSuccessResult extends TextAnalysisSuccessResult {\n  /**\n   * Predicted sentiment for document. For a list of possible values, see {@link DocumentSentimentLabel}\n   */\n  readonly sentiment: DocumentSentimentLabel;\n  /**\n   * Document level sentiment confidence scores between 0 and 1 for each\n   * sentiment class.\n   */\n  readonly confidenceScores: SentimentConfidenceScores;\n  /**\n   * The predicted sentiment for each sentence in the corresponding document.\n   */\n  readonly sentences: SentenceSentiment[];\n}\n\n/**\n * The predicted sentiment for a given span of text. For more information\n * regarding text sentiment, see {@link https://docs.microsoft.com//azure/cognitive-services/language-service/sentiment-opinion-mining/overview}.\n */\nexport interface SentenceSentiment {\n  /**\n   * The sentence text.\n   */\n  readonly text: string;\n  /**\n   * The predicted Sentiment for the sentence. For a list of possible values,\n   * see {@link SentenceSentimentLabel}\n   */\n  readonly sentiment: SentenceSentimentLabel;\n  /**\n   * The sentiment confidence score between 0 and 1 for the sentence for all\n   * classes.\n   */\n  readonly confidenceScores: SentimentConfidenceScores;\n  /**\n   * The sentence text offset from the start of the document.\n   */\n  readonly offset: number;\n  /**\n   * The length of the sentence text.\n   */\n  readonly length: number;\n  /**\n   * The list of opinions mined from this sentence. For example in \"The food is\n   * good, but the service is bad\", the following two opinions will be returned:\n   * \"food is good\" and \"service is bad\".\n   *\n   * It is non-empty only returned if {@link includeOpinionMining} was set to\n   * `true`.\n   */\n  readonly opinions: Opinion[];\n}\n\n/**\n * TargetSentiment contains the predicted sentiment, confidence scores and other\n * information about a target of a product. A target of a product/service is a\n * key component of that product/service. For example in \"The food at Hotel Foo\n * is good\", \"food\" is a target of \"Hotel Foo\".\n */\nexport interface TargetSentiment {\n  /**\n   * The sentiment confidence score between 0 and 1 for the target for\n   * 'positive' and 'negative' labels.\n   */\n  readonly confidenceScores: TargetConfidenceScores;\n  /**\n   * The predicted Sentiment for the Target. For a list of possible values,\n   * see {@link TokenSentimentLabel}\n   */\n  readonly sentiment: TokenSentimentLabel;\n  /**\n   * The target text.\n   */\n  readonly text: string;\n  /**\n   * The Target text offset from the start of the sentence.\n   */\n  readonly offset: number;\n  /**\n   * The length of the Target text.\n   */\n  readonly length: number;\n}\n\n/**\n * A mined opinion object represents an opinion we've extracted from a sentence.\n * It consists of both a target that these assessments are about, and the actual\n * assessments themselves.\n */\nexport interface Opinion {\n  /**\n   * The target of a product/service that this assessment is about.\n   */\n  readonly target: TargetSentiment;\n  /**\n   * The actual assessments of the target.\n   */\n  readonly assessments: AssessmentSentiment[];\n}\n\n/**\n * A healthcare entity represented as a node in a directed graph where the edges are\n * a particular type of relationship between the source and target nodes.\n */\nexport interface HealthcareEntity extends Entity {\n  /**\n   * Normalized name for the entity. For example, the normalized text for \"histologically\" is \"histologic\".\n   */\n  readonly normalizedText?: string;\n  /**\n   * Whether the entity is negated.\n   */\n  readonly assertion?: HealthcareAssertion;\n  /**\n   * Entity references in known data sources.\n   */\n  readonly dataSources: EntityDataSource[];\n  /**\n   * Defines values for HealthcareEntityCategory.\n   * {@link KnownHealthcareEntityCategory} can be used interchangeably with HealthcareEntityCategory,\n   *  this enum contains the known values that the service supports.\n   * ### Known values supported by the service\n   * **BODY_STRUCTURE**\n   * **AGE**\n   * **GENDER**\n   * **EXAMINATION_NAME**\n   * **DATE**\n   * **DIRECTION**\n   * **FREQUENCY**\n   * **MEASUREMENT_VALUE**\n   * **MEASUREMENT_UNIT**\n   * **RELATIONAL_OPERATOR**\n   * **TIME**\n   * **GENE_OR_PROTEIN**\n   * **VARIANT**\n   * **ADMINISTRATIVE_EVENT**\n   * **CARE_ENVIRONMENT**\n   * **HEALTHCARE_PROFESSION**\n   * **DIAGNOSIS**\n   * **SYMPTOM_OR_SIGN**\n   * **CONDITION_QUALIFIER**\n   * **MEDICATION_CLASS**\n   * **MEDICATION_NAME**\n   * **DOSAGE**\n   * **MEDICATION_FORM**\n   * **MEDICATION_ROUTE**\n   * **FAMILY_RELATION**\n   * **TREATMENT_NAME**\n   */\n  readonly category: HealthcareEntityCategory;\n}\n\n/**\n * The type of different roles a healthcare entity can play in a relation.\n */\nexport type HealthcareEntityRelationRoleType = string;\n\n/**\n * A healthcare entity that plays a specific role in a relation.\n */\nexport interface HealthcareEntityRelationRole {\n  /**\n   * A healthcare entity\n   */\n  readonly entity: HealthcareEntity;\n  /**\n   * The role of the healthcare entity in a particular relation.\n   */\n  readonly name: HealthcareEntityRelationRoleType;\n}\n\n/**\n * A relationship between two or more healthcare entities.\n */\nexport interface HealthcareEntityRelation {\n  /**\n   * The type of the healthcare relation.\n   */\n  readonly relationType: RelationType;\n  /**\n   * The list of healthcare entities and their roles in the healthcare relation.\n   */\n  readonly roles: HealthcareEntityRelationRole[];\n}\n\n/**\n * The results of a successful healthcare analysis action for a single document.\n */\nexport interface HealthcareSuccessResult extends TextAnalysisSuccessResult {\n  /**\n   * Healthcare entities.\n   */\n  readonly entities: HealthcareEntity[];\n  /**\n   * Relations between healthcare entities.\n   */\n  readonly entityRelations: HealthcareEntityRelation[];\n  /**\n   * JSON bundle containing a FHIR compatible object for consumption in other\n   * Healthcare tools. For additional information see {@link https://www.hl7.org/fhir/overview.html}.\n   */\n  readonly fhirBundle?: Record<string, any>;\n}\n\n/**\n * An error result from the healthcare analysis action on a single document.\n */\nexport type HealthcareErrorResult = TextAnalysisErrorResult;\n\n/**\n * The result of the healthcare analysis action on a single document.\n */\nexport type HealthcareResult = HealthcareSuccessResult | HealthcareErrorResult;\n\n/**\n * The result of the extractive summarization action on a single document.\n */\nexport type SummarizationExtractionResult =\n  | SummarizationExtractionSuccessResult\n  | SummarizationExtractionErrorResult;\n\n/**\n * The result of the extractive summarization action on a single document,\n * containing a collection of the summary identified in that document.\n */\nexport interface SummarizationExtractionSuccessResult extends TextAnalysisSuccessResult {\n  /**\n   * A list of sentences composing a summary of the input document.\n   */\n  readonly sentences: SummarySentence[];\n}\n\n/**\n * An error result from the extractive summarization action on a single document.\n */\nexport type SummarizationExtractionErrorResult = TextAnalysisErrorResult;\n\n/**\n * The result of the custom entity recognition action on a single document.\n */\nexport type CustomEntityRecognitionResult =\n  | CustomEntityRecognitionSuccessResult\n  | CustomEntityRecognitionErrorResult;\n\n/**\n * The result of the custom entity recognition action on a single document,\n * containing a collection of the entities identified in that document.\n */\nexport interface CustomEntityRecognitionSuccessResult extends TextAnalysisSuccessResult {\n  /**\n   * The collection of entities identified in the input document.\n   */\n  readonly entities: Entity[];\n}\n\n/**\n * An error result from the custom entity recognition action on a single document.\n */\nexport type CustomEntityRecognitionErrorResult = TextAnalysisErrorResult;\n\n/**\n * The result of the single-label classification action on a single document.\n */\nexport type CustomSingleLabelClassificationResult =\n  | CustomSingleLabelClassificationSuccessResult\n  | CustomSingleLabelClassificationErrorResult;\n\n/**\n * The result of a successful single-label classification action on a single document,\n * containing the result of the classification.\n */\nexport interface CustomSingleLabelClassificationSuccessResult extends TextAnalysisSuccessResult {\n  /**\n   * The collection of classifications in the input document.\n   */\n  readonly classifications: ClassificationCategory[];\n}\n\n/**\n * An error result from the single-label classification action on a single document.\n */\nexport type CustomSingleLabelClassificationErrorResult = TextAnalysisErrorResult;\n\n/**\n * The result of the multi-label classification action on a multi document.\n */\nexport type CustomMultiLabelClassificationResult =\n  | CustomMultiLabelClassificationSuccessResult\n  | CustomMultiLabelClassificationErrorResult;\n\n/**\n * The result of a successful multi-label classification action on a multi document,\n * containing the result of the classification.\n */\nexport interface CustomMultiLabelClassificationSuccessResult extends TextAnalysisSuccessResult {\n  /**\n   * The collection of classifications in the input document.\n   */\n  readonly classifications: ClassificationCategory[];\n}\n\n/**\n * An error result from the multi-label classification action on a multi document.\n */\nexport type CustomMultiLabelClassificationErrorResult = TextAnalysisErrorResult;\n\n/**\n * Options common to all batch actions.\n */\nexport interface AnalyzeBatchActionCommon {\n  /**\n   * The name of the action.\n   */\n  actionName?: string;\n}\n\n/** Options for an entity linking batch action. */\nexport interface EntityLinkingBatchAction extends AnalyzeBatchActionCommon, EntityLinkingAction {\n  /**\n   * The kind of the action.\n   */\n  kind: \"EntityLinking\";\n}\n\n/** Options for an entity recognition batch action. */\nexport interface EntityRecognitionBatchAction\n  extends AnalyzeBatchActionCommon,\n    EntityRecognitionAction {\n  /**\n   * The kind of the action.\n   */\n  kind: \"EntityRecognition\";\n}\n\n/** Options for an key phrase extraction batch action. */\nexport interface KeyPhraseExtractionBatchAction\n  extends AnalyzeBatchActionCommon,\n    KeyPhraseExtractionAction {\n  /**\n   * The kind of the action.\n   */\n  kind: \"KeyPhraseExtraction\";\n}\n\n/** Options for a pii entity recognition batch action. */\nexport interface PiiEntityRecognitionBatchAction\n  extends AnalyzeBatchActionCommon,\n    PiiEntityRecognitionAction {\n  /**\n   * The kind of the action.\n   */\n  kind: \"PiiEntityRecognition\";\n}\n\n/** Options for a healthcare batch action. */\nexport interface HealthcareBatchAction extends AnalyzeBatchActionCommon, HealthcareAction {\n  /**\n   * The kind of the action.\n   */\n  kind: \"Healthcare\";\n}\n\n/** Options for an extractive summarization batch action. */\nexport interface ExtractiveSummarizationBatchAction\n  extends AnalyzeBatchActionCommon,\n    ExtractiveSummarizationAction {\n  /**\n   * The kind of the action.\n   */\n  kind: \"ExtractiveSummarization\";\n}\n\n/** Options for a sentiment analysis batch action. */\nexport interface SentimentAnalysisBatchAction\n  extends AnalyzeBatchActionCommon,\n    SentimentAnalysisAction {\n  /**\n   * The kind of the action.\n   */\n  kind: \"SentimentAnalysis\";\n}\n\n/** Options for a custom entity recognition batch action. */\nexport interface CustomEntityRecognitionBatchAction\n  extends AnalyzeBatchActionCommon,\n    CustomEntityRecognitionAction {\n  /**\n   * The kind of the action.\n   */\n  kind: \"CustomEntityRecognition\";\n}\n\n/** Options for a custom single-label classification batch action. */\nexport interface CustomSingleLabelClassificationBatchAction\n  extends AnalyzeBatchActionCommon,\n    CustomSingleLabelClassificationAction {\n  /**\n   * The kind of the action.\n   */\n  kind: \"CustomSingleLabelClassification\";\n}\n\n/** Options for a custom multi-label classification batch action. */\nexport interface CustomMultiLabelClassificationBatchAction\n  extends AnalyzeBatchActionCommon,\n    CustomMultiLabelClassificationAction {\n  /**\n   * The kind of the action.\n   */\n  kind: \"CustomMultiLabelClassification\";\n}\n\n/**\n * Batch of actions.\n */\nexport type AnalyzeBatchAction =\n  | EntityLinkingBatchAction\n  | EntityRecognitionBatchAction\n  | KeyPhraseExtractionBatchAction\n  | PiiEntityRecognitionBatchAction\n  | HealthcareBatchAction\n  | ExtractiveSummarizationBatchAction\n  | SentimentAnalysisBatchAction\n  | CustomEntityRecognitionBatchAction\n  | CustomSingleLabelClassificationBatchAction\n  | CustomMultiLabelClassificationBatchAction;\n\n/**\n * Type of actions supported by the {@link TextAnalysisClient.beginAnalyzeBatch} method.\n */\nexport type AnalyzeBatchActionName = keyof typeof AnalyzeBatchActionNames;\n\n/** The State of a batched action */\nexport interface BatchActionState<Kind extends AnalyzeBatchActionName> {\n  /**\n   * The kind of the action results.\n   */\n  readonly kind: Kind;\n  /**\n   * The name of the action.\n   */\n  readonly actionName?: string;\n  /**\n   * Action statistics.\n   */\n  readonly statistics?: TextDocumentBatchStatistics;\n}\n\n/**\n * Action metadata.\n */\nexport interface ActionMetadata {\n  /**\n   * The model version used to perform the action.\n   */\n  readonly modelVersion: string;\n}\n\n/**\n * Custom action metadata.\n */\nexport interface CustomActionMetadata {\n  /**\n   * The name of the project used to perform the action.\n   */\n  readonly projectName: string;\n  /**\n   * The name of the deployment used to perform the action.\n   */\n  readonly deploymentName: string;\n}\n\n/**\n * The state of a succeeded batched action.\n */\nexport interface BatchActionSuccessResult<T, Kind extends AnalyzeBatchActionName>\n  extends BatchActionState<Kind> {\n  /**\n   * The list of document results.\n   */\n  readonly results: T[];\n  /**\n   * When this action was completed by the service.\n   */\n  readonly completedOn: Date;\n  /**\n   * Discriminant to determine if that this is an error result.\n   */\n  readonly error?: undefined;\n}\n\n/**\n * The error of an analyze batch action.\n */\nexport interface BatchActionErrorResult<Kind extends AnalyzeBatchActionName>\n  extends BatchActionState<Kind> {\n  /**\n   * When this action was completed by the service.\n   */\n  readonly failedOn: Date;\n  /**\n   * The Error for this action result.\n   */\n  readonly error: TextAnalysisError;\n}\n\n/**\n * The result of a batched action.\n */\nexport type BatchActionResult<T, Kind extends AnalyzeBatchActionName> =\n  | BatchActionSuccessResult<T, Kind>\n  | BatchActionErrorResult<Kind>;\n\n/**\n * The result of an entity linking batch action.\n */\nexport type EntityLinkingBatchResult = ActionMetadata &\n  BatchActionResult<EntityLinkingResult, \"EntityLinking\">;\n\n/**\n * The result of an entity recognition batch action.\n */\nexport type EntityRecognitionBatchResult = ActionMetadata &\n  BatchActionResult<EntityRecognitionResult, \"EntityRecognition\">;\n\n/**\n * The result of a key phrase extraction batch action.\n */\nexport type KeyPhraseExtractionBatchResult = ActionMetadata &\n  BatchActionResult<KeyPhraseExtractionResult, \"KeyPhraseExtraction\">;\n\n/**\n * The result of a pii entity recognition batch action.\n */\nexport type PiiEntityRecognitionBatchResult = ActionMetadata &\n  BatchActionResult<PiiEntityRecognitionResult, \"PiiEntityRecognition\">;\n\n/**\n * The result of a sentiment analysis batch action.\n */\nexport type SentimentAnalysisBatchResult = ActionMetadata &\n  BatchActionResult<SentimentAnalysisResult, \"SentimentAnalysis\">;\n\n/**\n * The result of a healthcare batch action.\n */\nexport type HealthcareBatchResult = ActionMetadata &\n  BatchActionResult<HealthcareResult, \"Healthcare\">;\n\n/**\n * The result of an extractive summarization batch action.\n */\nexport type ExtractiveSummarizationBatchResult = ActionMetadata &\n  BatchActionResult<SummarizationExtractionResult, \"ExtractiveSummarization\">;\n\n/**\n * The result of a custom entity recognition batch action.\n */\nexport type CustomEntityRecognitionBatchResult = CustomActionMetadata &\n  BatchActionResult<CustomEntityRecognitionResult, \"CustomEntityRecognition\">;\n\n/**\n * The result of a custom single-label classification batch action.\n */\nexport type CustomSingleLabelClassificationBatchResult = CustomActionMetadata &\n  BatchActionResult<CustomSingleLabelClassificationResult, \"CustomSingleLabelClassification\">;\n\n/**\n * The result of a custom multi-label classification batch action.\n */\nexport type CustomMultiLabelClassificationBatchResult = CustomActionMetadata &\n  BatchActionResult<CustomMultiLabelClassificationResult, \"CustomMultiLabelClassification\">;\n/**\n * Results of a batch of actions.\n */\nexport type AnalyzeBatchResult =\n  | EntityLinkingBatchResult\n  | EntityRecognitionBatchResult\n  | KeyPhraseExtractionBatchResult\n  | PiiEntityRecognitionBatchResult\n  | SentimentAnalysisBatchResult\n  | HealthcareBatchResult\n  | ExtractiveSummarizationBatchResult\n  | CustomEntityRecognitionBatchResult\n  | CustomSingleLabelClassificationBatchResult\n  | CustomMultiLabelClassificationBatchResult;\n\n/**\n * An error result from a sentiment analysis action on a single document.\n */\nexport type SentimentAnalysisErrorResult = TextAnalysisErrorResult;\n\n/**\n * Paged results of the {@link TextAnalysisClient.beginAnalyzeBatch} operation.\n */\nexport type PagedAnalyzeBatchResult = PagedAsyncIterableIterator<AnalyzeBatchResult>;\n\n/**\n * A poller that polls long-running operations started by {@link TextAnalysisClient.beginAnalyzeBatch}.\n */\nexport type AnalyzeBatchPoller = PollerLike<AnalyzeBatchOperationState, PagedAnalyzeBatchResult>;\n\n/**\n * The metadata for long-running operations started by {@link TextAnalysisClient.beginAnalyzeBatch}.\n */\nexport interface AnalyzeBatchOperationMetadata {\n  /**\n   * The date and time the operation was created.\n   */\n  readonly createdOn: Date;\n  /**\n   * The date and time when the operation results will expire on the server.\n   */\n  readonly expiresOn?: Date;\n  /**\n   * The operation id.\n   */\n  readonly id: string;\n  /**\n   * The time the operation status was last updated.\n   */\n  readonly modifiedOn: Date;\n  /**\n   * The current status of the operation.\n   */\n  readonly status: OperationStatus;\n  /**\n   * Number of successfully completed actions.\n   */\n  readonly actionSucceededCount: number;\n  /**\n   * Number of failed actions.\n   */\n  readonly actionFailedCount: number;\n  /**\n   * Number of actions still in progress.\n   */\n  readonly actionInProgressCount: number;\n  /**\n   * The operation's display name.\n   */\n  readonly displayName?: string;\n}\n\n/**\n * The state of the begin analyze polling operation.\n */\nexport interface AnalyzeBatchOperationState\n  extends PollOperationState<PagedAnalyzeBatchResult>,\n    AnalyzeBatchOperationMetadata {}\n"]}
+{"version":3,"file":"models.js","sourceRoot":"","sources":["../../src/models.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAiBL,cAAc,EACd,mBAAmB,GAYpB,MAAM,aAAa,CAAC;AA0DrB;;GAEG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG;IAChC,aAAa,EAAE,eAAe;IAC9B,iBAAiB,EAAE,mBAAmB;IACtC,mBAAmB,EAAE,qBAAqB;IAC1C,oBAAoB,EAAE,sBAAsB;IAC5C,iBAAiB,EAAE,mBAAmB;IACtC,iBAAiB,EAAE,mBAAmB;CAC9B,CAAC;AAEX;;GAEG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG;IACrC,iBAAiB,EAAE,mBAAmB;IACtC,iBAAiB,EAAE,mBAAmB;IACtC,oBAAoB,EAAE,sBAAsB;IAC5C,mBAAmB,EAAE,qBAAqB;IAC1C,aAAa,EAAE,eAAe;IAC9B,UAAU,EAAE,YAAY;IACxB,uBAAuB,EAAE,yBAAyB;IAClD,+BAA+B,EAAE,iCAAiC;IAClE,8BAA8B,EAAE,gCAAgC;CACxD,CAAC;AA+BX;;GAEG;AACH,MAAM,CAAC,MAAM,0BAA0B,mCAAQ,cAAc,GAAK,mBAAmB,CAAE,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\nimport {\n  AssessmentSentiment,\n  ClassificationCategory,\n  CustomEntityRecognitionAction,\n  CustomMultiLabelClassificationAction,\n  CustomSingleLabelClassificationAction,\n  DetectedLanguage,\n  DocumentSentimentLabel,\n  DocumentWarning,\n  Entity,\n  EntityDataSource,\n  EntityLinkingAction,\n  EntityRecognitionAction,\n  HealthcareAction,\n  HealthcareAssertion,\n  HealthcareEntityCategory,\n  KeyPhraseExtractionAction,\n  KnownErrorCode,\n  KnownInnerErrorCode,\n  LanguageDetectionAction,\n  LinkedEntity,\n  PiiEntityRecognitionAction,\n  RelationType,\n  SentenceSentimentLabel,\n  SentimentAnalysisAction,\n  SentimentConfidenceScores,\n  TargetConfidenceScores,\n  TextDocumentBatchStatistics,\n  TextDocumentStatistics,\n  TokenSentimentLabel,\n} from \"./generated\";\nimport { CommonClientOptions, OperationOptions } from \"@azure/core-client\";\nimport { OperationState, SimplePollerLike } from \"@azure/core-lro\";\nimport { PagedAsyncIterableIterator } from \"@azure/core-paging\";\n\n/**\n * Configuration options for {@link TextAnalysisClient}.\n */\nexport interface TextAnalysisClientOptions extends CommonClientOptions {\n  /**\n   * The default country hint to use. Defaults to \"us\".\n   */\n  defaultCountryHint?: string;\n\n  /**\n   * The default language to use. Defaults to \"en\".\n   */\n  defaultLanguage?: string;\n  /**\n   * The version of the Cognitive Language Service API to use.\n   */\n  serviceVersion?: string;\n}\n\n/**\n * Options common to all operations.\n */\nexport interface TextAnalysisOperationOptions extends OperationOptions {\n  /**\n   * If set to true, response will contain input and document level statistics.\n   */\n  includeStatistics?: boolean;\n}\n\n/**\n * Options for the begin analyze actions operation.\n */\nexport interface BeginAnalyzeBatchOptions extends TextAnalysisOperationOptions {\n  /**\n   * Time delay between poll requests, in milliseconds.\n   */\n  updateIntervalInMs?: number;\n  /**\n   * The operation's display name.\n   */\n  displayName?: string;\n}\n\n/**\n * Options for the begin analyze actions operation.\n */\nexport interface RestoreAnalyzeBatchPollerOptions extends TextAnalysisOperationOptions {\n  /**\n   * Time delay between poll requests, in milliseconds.\n   */\n  updateIntervalInMs?: number;\n}\n\n/**\n * Type of actions supported by the {@link TextAnalysisClient.analyze} method.\n */\nexport const AnalyzeActionNames = {\n  EntityLinking: \"EntityLinking\",\n  EntityRecognition: \"EntityRecognition\",\n  KeyPhraseExtraction: \"KeyPhraseExtraction\",\n  PiiEntityRecognition: \"PiiEntityRecognition\",\n  LanguageDetection: \"LanguageDetection\",\n  SentimentAnalysis: \"SentimentAnalysis\",\n} as const;\n\n/**\n * Type of actions supported by the {@link TextAnalysisClient.beginAnalyzeBatch} method.\n */\nexport const AnalyzeBatchActionNames = {\n  SentimentAnalysis: \"SentimentAnalysis\",\n  EntityRecognition: \"EntityRecognition\",\n  PiiEntityRecognition: \"PiiEntityRecognition\",\n  KeyPhraseExtraction: \"KeyPhraseExtraction\",\n  EntityLinking: \"EntityLinking\",\n  Healthcare: \"Healthcare\",\n  CustomEntityRecognition: \"CustomEntityRecognition\",\n  CustomSingleLabelClassification: \"CustomSingleLabelClassification\",\n  CustomMultiLabelClassification: \"CustomMultiLabelClassification\",\n} as const;\n\n/**\n * Type of actions supported by the {@link TextAnalysisClient.analyze} method.\n */\nexport type AnalyzeActionName = keyof typeof AnalyzeActionNames;\n\n/**\n * The type of parameters for every action in ${@link AnalyzeActionNames}.\n */\nexport type AnalyzeActionParameters<ActionName extends AnalyzeActionName> = {\n  EntityLinking: EntityLinkingAction;\n  EntityRecognition: EntityRecognitionAction;\n  PiiEntityRecognition: PiiEntityRecognitionAction;\n  KeyPhraseExtraction: KeyPhraseExtractionAction;\n  SentimentAnalysis: SentimentAnalysisAction;\n  LanguageDetection: LanguageDetectionAction;\n}[ActionName];\n\n/**\n * The type of results of every action in ${@link AnalyzeActionNames}.\n */\nexport type AnalyzeResult<ActionName extends AnalyzeActionName> = {\n  EntityLinking: EntityLinkingResult[];\n  EntityRecognition: EntityRecognitionResult[];\n  PiiEntityRecognition: PiiEntityRecognitionResult[];\n  KeyPhraseExtraction: KeyPhraseExtractionResult[];\n  SentimentAnalysis: SentimentAnalysisResult[];\n  LanguageDetection: LanguageDetectionResult[];\n}[ActionName];\n\n/**\n * Enum of possible error codes of a {@link TextAnalysisError}.\n */\nexport const KnownTextAnalysisErrorCode = { ...KnownErrorCode, ...KnownInnerErrorCode };\n\n/**\n * Type describing an API error.\n */\nexport interface TextAnalysisError {\n  /**\n   * A code describing the kind of error produced. See {@link KnownTextAnalysisErrorCode}.\n   */\n  readonly code: string;\n  /**\n   * A message from the service explaining the error\n   */\n  readonly message: string;\n  /**\n   * The target of the particular error (for example, the name of an invalid parameter)\n   */\n  readonly target?: string;\n}\n\n/**\n * Base type for results of an action corresponding to a single input document.\n */\nexport interface TextAnalysisSuccessResult {\n  /**\n   * Unique, non-empty document identifier.\n   */\n  readonly id: string;\n\n  /**\n   * Statistics about the input document and how it was processed by the service.\n   * This property will have a value when includeStatistics is set to true in\n   * the client call.\n   */\n  readonly statistics?: TextDocumentStatistics;\n\n  /**\n   * An array of warning data corresponding to this document.\n   *\n   * If no warnings were returned, this array will be empty.\n   */\n  readonly warnings: DocumentWarning[];\n\n  /**\n   * Discriminant to determine if this is an error result.\n   */\n  readonly error?: undefined;\n}\n\n/**\n * Base type for error results of an action corresponding to a single document.\n */\nexport interface TextAnalysisErrorResult {\n  /**\n   * Unique, non-empty document identifier.\n   */\n  readonly id: string;\n\n  /**\n   * The Error for this document result.\n   */\n  readonly error: TextAnalysisError;\n}\n\n/**\n * The result of an entity recognition action on a single document.\n */\nexport type EntityRecognitionResult = EntityRecognitionSuccessResult | EntityRecognitionErrorResult;\n\n/**\n * The result of an entity recognition action on a single document, containing\n * a collection of {@link Entity} objects identified in that document.\n */\nexport interface EntityRecognitionSuccessResult extends TextAnalysisSuccessResult {\n  /**\n   * The collection of entities identified in the input document.\n   */\n  readonly entities: Entity[];\n}\n\n/**\n * An error result from an entity recognition action on a single document.\n */\nexport type EntityRecognitionErrorResult = TextAnalysisErrorResult;\n\n/**\n * The result of an entity recognition action on a single document.\n */\nexport type PiiEntityRecognitionResult =\n  | PiiEntityRecognitionSuccessResult\n  | PiiEntityRecognitionErrorResult;\n\n/**\n * The result of a pii entity recognition action on a single document,\n * containing the collection of {@link Entity} objects identified in that\n * document.\n */\nexport interface PiiEntityRecognitionSuccessResult extends TextAnalysisSuccessResult {\n  /**\n   * The collection of entities identified in the input document.\n   */\n  readonly entities: Entity[];\n  /**\n   * The text redacted.\n   */\n  readonly redactedText: string;\n}\n\n/**\n * An error result from a pii entity recognition action on a single document.\n */\nexport type PiiEntityRecognitionErrorResult = TextAnalysisErrorResult;\n\n/**\n * The result of an entity linking action on a single document.\n */\nexport type EntityLinkingResult = EntityLinkingSuccessResult | EntityLinkingErrorResult;\n\n/**\n * The result of a entity linking action on a single document, containing a\n * collection of the {@link LinkedEntity} objects identified in that document.\n */\nexport interface EntityLinkingSuccessResult extends TextAnalysisSuccessResult {\n  /**\n   * The collection of entities identified in the input document.\n   */\n  readonly entities: LinkedEntity[];\n}\n\n/**\n * An error result from an entity linking action on a single document.\n */\nexport type EntityLinkingErrorResult = TextAnalysisErrorResult;\n\n/**\n * The result of a language detection action on a single document.\n */\nexport type LanguageDetectionResult = LanguageDetectionSuccessResult | LanguageDetectionErrorResult;\n\n/**\n * The result of a language detection action on a single document,\n * containing a prediction of what language the document is written in.\n */\nexport interface LanguageDetectionSuccessResult extends TextAnalysisSuccessResult {\n  /**\n   * The top detected language by confidence score.\n   */\n  readonly primaryLanguage: DetectedLanguage;\n}\n\n/**\n * An error result from a language detection action on a single document.\n */\nexport type LanguageDetectionErrorResult = TextAnalysisErrorResult;\n\n/**\n * The result of a sentiment analysis action on a single document.\n */\nexport type KeyPhraseExtractionResult =\n  | KeyPhraseExtractionSuccessResult\n  | KeyPhraseExtractionErrorResult;\n\n/**\n * The result of a key phrase extraction action on a single document,\n * containing a collection of the key phrases identified in that document.\n */\nexport interface KeyPhraseExtractionSuccessResult extends TextAnalysisSuccessResult {\n  /**\n   * A list of representative words or phrases. The number of key phrases\n   * returned is proportional to the number of words in the input document.\n   */\n  readonly keyPhrases: string[];\n}\n\n/**\n * An error result from a key phrase extraction action on a single document.\n */\nexport type KeyPhraseExtractionErrorResult = TextAnalysisErrorResult;\n\n/**\n * The result of a sentiment analysis action on a single document.\n */\nexport type SentimentAnalysisResult = SentimentAnalysisSuccessResult | SentimentAnalysisErrorResult;\n\n/**\n * The result of a sentiment analysis action on a single document,\n * containing the predicted sentiment for each sentence as well as for the full\n * document.\n */\nexport interface SentimentAnalysisSuccessResult extends TextAnalysisSuccessResult {\n  /**\n   * Predicted sentiment for document. For a list of possible values, see {@link DocumentSentimentLabel}\n   */\n  readonly sentiment: DocumentSentimentLabel;\n  /**\n   * Document level sentiment confidence scores between 0 and 1 for each\n   * sentiment class.\n   */\n  readonly confidenceScores: SentimentConfidenceScores;\n  /**\n   * The predicted sentiment for each sentence in the corresponding document.\n   */\n  readonly sentences: SentenceSentiment[];\n}\n\n/**\n * The predicted sentiment for a given span of text. For more information\n * regarding text sentiment, see {@link https://docs.microsoft.com//azure/cognitive-services/language-service/sentiment-opinion-mining/overview}.\n */\nexport interface SentenceSentiment {\n  /**\n   * The sentence text.\n   */\n  readonly text: string;\n  /**\n   * The predicted Sentiment for the sentence. For a list of possible values,\n   * see {@link SentenceSentimentLabel}\n   */\n  readonly sentiment: SentenceSentimentLabel;\n  /**\n   * The sentiment confidence score between 0 and 1 for the sentence for all\n   * classes.\n   */\n  readonly confidenceScores: SentimentConfidenceScores;\n  /**\n   * The sentence text offset from the start of the document.\n   */\n  readonly offset: number;\n  /**\n   * The length of the sentence text.\n   */\n  readonly length: number;\n  /**\n   * The list of opinions mined from this sentence. For example in \"The food is\n   * good, but the service is bad\", the following two opinions will be returned:\n   * \"food is good\" and \"service is bad\".\n   *\n   * It is non-empty only returned if {@link includeOpinionMining} was set to\n   * `true`.\n   */\n  readonly opinions: Opinion[];\n}\n\n/**\n * TargetSentiment contains the predicted sentiment, confidence scores and other\n * information about a target of a product. A target of a product/service is a\n * key component of that product/service. For example in \"The food at Hotel Foo\n * is good\", \"food\" is a target of \"Hotel Foo\".\n */\nexport interface TargetSentiment {\n  /**\n   * The sentiment confidence score between 0 and 1 for the target for\n   * 'positive' and 'negative' labels.\n   */\n  readonly confidenceScores: TargetConfidenceScores;\n  /**\n   * The predicted Sentiment for the Target. For a list of possible values,\n   * see {@link TokenSentimentLabel}\n   */\n  readonly sentiment: TokenSentimentLabel;\n  /**\n   * The target text.\n   */\n  readonly text: string;\n  /**\n   * The Target text offset from the start of the sentence.\n   */\n  readonly offset: number;\n  /**\n   * The length of the Target text.\n   */\n  readonly length: number;\n}\n\n/**\n * A mined opinion object represents an opinion we've extracted from a sentence.\n * It consists of both a target that these assessments are about, and the actual\n * assessments themselves.\n */\nexport interface Opinion {\n  /**\n   * The target of a product/service that this assessment is about.\n   */\n  readonly target: TargetSentiment;\n  /**\n   * The actual assessments of the target.\n   */\n  readonly assessments: AssessmentSentiment[];\n}\n\n/**\n * A healthcare entity represented as a node in a directed graph where the edges are\n * a particular type of relationship between the source and target nodes.\n */\nexport interface HealthcareEntity extends Entity {\n  /**\n   * Normalized name for the entity. For example, the normalized text for \"histologically\" is \"histologic\".\n   */\n  readonly normalizedText?: string;\n  /**\n   * Whether the entity is negated.\n   */\n  readonly assertion?: HealthcareAssertion;\n  /**\n   * Entity references in known data sources.\n   */\n  readonly dataSources: EntityDataSource[];\n  /**\n   * Defines values for HealthcareEntityCategory.\n   * {@link KnownHealthcareEntityCategory} can be used interchangeably with HealthcareEntityCategory,\n   *  this enum contains the known values that the service supports.\n   * ### Known values supported by the service\n   * **BODY_STRUCTURE**\n   * **AGE**\n   * **GENDER**\n   * **EXAMINATION_NAME**\n   * **DATE**\n   * **DIRECTION**\n   * **FREQUENCY**\n   * **MEASUREMENT_VALUE**\n   * **MEASUREMENT_UNIT**\n   * **RELATIONAL_OPERATOR**\n   * **TIME**\n   * **GENE_OR_PROTEIN**\n   * **VARIANT**\n   * **ADMINISTRATIVE_EVENT**\n   * **CARE_ENVIRONMENT**\n   * **HEALTHCARE_PROFESSION**\n   * **DIAGNOSIS**\n   * **SYMPTOM_OR_SIGN**\n   * **CONDITION_QUALIFIER**\n   * **MEDICATION_CLASS**\n   * **MEDICATION_NAME**\n   * **DOSAGE**\n   * **MEDICATION_FORM**\n   * **MEDICATION_ROUTE**\n   * **FAMILY_RELATION**\n   * **TREATMENT_NAME**\n   */\n  readonly category: HealthcareEntityCategory;\n}\n\n/**\n * The type of different roles a healthcare entity can play in a relation.\n */\nexport type HealthcareEntityRelationRoleType = string;\n\n/**\n * A healthcare entity that plays a specific role in a relation.\n */\nexport interface HealthcareEntityRelationRole {\n  /**\n   * A healthcare entity\n   */\n  readonly entity: HealthcareEntity;\n  /**\n   * The role of the healthcare entity in a particular relation.\n   */\n  readonly name: HealthcareEntityRelationRoleType;\n}\n\n/**\n * A relationship between two or more healthcare entities.\n */\nexport interface HealthcareEntityRelation {\n  /**\n   * The type of the healthcare relation.\n   */\n  readonly relationType: RelationType;\n  /**\n   * The list of healthcare entities and their roles in the healthcare relation.\n   */\n  readonly roles: HealthcareEntityRelationRole[];\n}\n\n/**\n * The results of a successful healthcare analysis action for a single document.\n */\nexport interface HealthcareSuccessResult extends TextAnalysisSuccessResult {\n  /**\n   * Healthcare entities.\n   */\n  readonly entities: HealthcareEntity[];\n  /**\n   * Relations between healthcare entities.\n   */\n  readonly entityRelations: HealthcareEntityRelation[];\n}\n\n/**\n * An error result from the healthcare analysis action on a single document.\n */\nexport type HealthcareErrorResult = TextAnalysisErrorResult;\n\n/**\n * The result of the healthcare analysis action on a single document.\n */\nexport type HealthcareResult = HealthcareSuccessResult | HealthcareErrorResult;\n\n/**\n * The result of the custom entity recognition action on a single document.\n */\nexport type CustomEntityRecognitionResult =\n  | CustomEntityRecognitionSuccessResult\n  | CustomEntityRecognitionErrorResult;\n\n/**\n * The result of the custom entity recognition action on a single document,\n * containing a collection of the entities identified in that document.\n */\nexport interface CustomEntityRecognitionSuccessResult extends TextAnalysisSuccessResult {\n  /**\n   * The collection of entities identified in the input document.\n   */\n  readonly entities: Entity[];\n}\n\n/**\n * An error result from the custom entity recognition action on a single document.\n */\nexport type CustomEntityRecognitionErrorResult = TextAnalysisErrorResult;\n\n/**\n * The result of the single-label classification action on a single document.\n */\nexport type CustomSingleLabelClassificationResult =\n  | CustomSingleLabelClassificationSuccessResult\n  | CustomSingleLabelClassificationErrorResult;\n\n/**\n * The result of a successful single-label classification action on a single document,\n * containing the result of the classification.\n */\nexport interface CustomSingleLabelClassificationSuccessResult extends TextAnalysisSuccessResult {\n  /**\n   * The collection of classifications in the input document.\n   */\n  readonly classifications: ClassificationCategory[];\n}\n\n/**\n * An error result from the single-label classification action on a single document.\n */\nexport type CustomSingleLabelClassificationErrorResult = TextAnalysisErrorResult;\n\n/**\n * The result of the multi-label classification action on a multi document.\n */\nexport type CustomMultiLabelClassificationResult =\n  | CustomMultiLabelClassificationSuccessResult\n  | CustomMultiLabelClassificationErrorResult;\n\n/**\n * The result of a successful multi-label classification action on a multi document,\n * containing the result of the classification.\n */\nexport interface CustomMultiLabelClassificationSuccessResult extends TextAnalysisSuccessResult {\n  /**\n   * The collection of classifications in the input document.\n   */\n  readonly classifications: ClassificationCategory[];\n}\n\n/**\n * An error result from the multi-label classification action on a multi document.\n */\nexport type CustomMultiLabelClassificationErrorResult = TextAnalysisErrorResult;\n\n/**\n * Options common to all batch actions.\n */\nexport interface AnalyzeBatchActionCommon {\n  /**\n   * The name of the action.\n   */\n  actionName?: string;\n}\n\n/** Options for an entity linking batch action. */\nexport interface EntityLinkingBatchAction extends AnalyzeBatchActionCommon, EntityLinkingAction {\n  /**\n   * The kind of the action.\n   */\n  kind: \"EntityLinking\";\n}\n\n/** Options for an entity recognition batch action. */\nexport interface EntityRecognitionBatchAction\n  extends AnalyzeBatchActionCommon,\n    EntityRecognitionAction {\n  /**\n   * The kind of the action.\n   */\n  kind: \"EntityRecognition\";\n}\n\n/** Options for an key phrase extraction batch action. */\nexport interface KeyPhraseExtractionBatchAction\n  extends AnalyzeBatchActionCommon,\n    KeyPhraseExtractionAction {\n  /**\n   * The kind of the action.\n   */\n  kind: \"KeyPhraseExtraction\";\n}\n\n/** Options for a pii entity recognition batch action. */\nexport interface PiiEntityRecognitionBatchAction\n  extends AnalyzeBatchActionCommon,\n    PiiEntityRecognitionAction {\n  /**\n   * The kind of the action.\n   */\n  kind: \"PiiEntityRecognition\";\n}\n\n/** Options for a healthcare batch action. */\nexport interface HealthcareBatchAction extends AnalyzeBatchActionCommon, HealthcareAction {\n  /**\n   * The kind of the action.\n   */\n  kind: \"Healthcare\";\n}\n\n/** Options for a sentiment analysis batch action. */\nexport interface SentimentAnalysisBatchAction\n  extends AnalyzeBatchActionCommon,\n    SentimentAnalysisAction {\n  /**\n   * The kind of the action.\n   */\n  kind: \"SentimentAnalysis\";\n}\n\n/** Options for a custom entity recognition batch action. */\nexport interface CustomEntityRecognitionBatchAction\n  extends AnalyzeBatchActionCommon,\n    CustomEntityRecognitionAction {\n  /**\n   * The kind of the action.\n   */\n  kind: \"CustomEntityRecognition\";\n}\n\n/** Options for a custom single-label classification batch action. */\nexport interface CustomSingleLabelClassificationBatchAction\n  extends AnalyzeBatchActionCommon,\n    CustomSingleLabelClassificationAction {\n  /**\n   * The kind of the action.\n   */\n  kind: \"CustomSingleLabelClassification\";\n}\n\n/** Options for a custom multi-label classification batch action. */\nexport interface CustomMultiLabelClassificationBatchAction\n  extends AnalyzeBatchActionCommon,\n    CustomMultiLabelClassificationAction {\n  /**\n   * The kind of the action.\n   */\n  kind: \"CustomMultiLabelClassification\";\n}\n\n/**\n * Batch of actions.\n */\nexport type AnalyzeBatchAction =\n  | EntityLinkingBatchAction\n  | EntityRecognitionBatchAction\n  | KeyPhraseExtractionBatchAction\n  | PiiEntityRecognitionBatchAction\n  | HealthcareBatchAction\n  | SentimentAnalysisBatchAction\n  | CustomEntityRecognitionBatchAction\n  | CustomSingleLabelClassificationBatchAction\n  | CustomMultiLabelClassificationBatchAction;\n\n/**\n * Type of actions supported by the {@link TextAnalysisClient.beginAnalyzeBatch} method.\n */\nexport type AnalyzeBatchActionName = keyof typeof AnalyzeBatchActionNames;\n\n/** The State of a batched action */\nexport interface BatchActionState<Kind extends AnalyzeBatchActionName> {\n  /**\n   * The kind of the action results.\n   */\n  readonly kind: Kind;\n  /**\n   * The name of the action.\n   */\n  readonly actionName?: string;\n  /**\n   * Action statistics.\n   */\n  readonly statistics?: TextDocumentBatchStatistics;\n}\n\n/**\n * Action metadata.\n */\nexport interface ActionMetadata {\n  /**\n   * The model version used to perform the action.\n   */\n  readonly modelVersion: string;\n}\n\n/**\n * Custom action metadata.\n */\nexport interface CustomActionMetadata {\n  /**\n   * The name of the project used to perform the action.\n   */\n  readonly projectName: string;\n  /**\n   * The name of the deployment used to perform the action.\n   */\n  readonly deploymentName: string;\n}\n\n/**\n * The state of a succeeded batched action.\n */\nexport interface BatchActionSuccessResult<T, Kind extends AnalyzeBatchActionName>\n  extends BatchActionState<Kind> {\n  /**\n   * The list of document results.\n   */\n  readonly results: T[];\n  /**\n   * When this action was completed by the service.\n   */\n  readonly completedOn: Date;\n  /**\n   * Discriminant to determine if that this is an error result.\n   */\n  readonly error?: undefined;\n}\n\n/**\n * The error of an analyze batch action.\n */\nexport interface BatchActionErrorResult<Kind extends AnalyzeBatchActionName>\n  extends BatchActionState<Kind> {\n  /**\n   * When this action was completed by the service.\n   */\n  readonly failedOn: Date;\n  /**\n   * The Error for this action result.\n   */\n  readonly error: TextAnalysisError;\n}\n\n/**\n * The result of a batched action.\n */\nexport type BatchActionResult<T, Kind extends AnalyzeBatchActionName> =\n  | BatchActionSuccessResult<T, Kind>\n  | BatchActionErrorResult<Kind>;\n\n/**\n * The result of an entity linking batch action.\n */\nexport type EntityLinkingBatchResult = ActionMetadata &\n  BatchActionResult<EntityLinkingResult, \"EntityLinking\">;\n\n/**\n * The result of an entity recognition batch action.\n */\nexport type EntityRecognitionBatchResult = ActionMetadata &\n  BatchActionResult<EntityRecognitionResult, \"EntityRecognition\">;\n\n/**\n * The result of a key phrase extraction batch action.\n */\nexport type KeyPhraseExtractionBatchResult = ActionMetadata &\n  BatchActionResult<KeyPhraseExtractionResult, \"KeyPhraseExtraction\">;\n\n/**\n * The result of a pii entity recognition batch action.\n */\nexport type PiiEntityRecognitionBatchResult = ActionMetadata &\n  BatchActionResult<PiiEntityRecognitionResult, \"PiiEntityRecognition\">;\n\n/**\n * The result of a sentiment analysis batch action.\n */\nexport type SentimentAnalysisBatchResult = ActionMetadata &\n  BatchActionResult<SentimentAnalysisResult, \"SentimentAnalysis\">;\n\n/**\n * The result of a healthcare batch action.\n */\nexport type HealthcareBatchResult = ActionMetadata &\n  BatchActionResult<HealthcareResult, \"Healthcare\">;\n\n/**\n * The result of a custom entity recognition batch action.\n */\nexport type CustomEntityRecognitionBatchResult = CustomActionMetadata &\n  BatchActionResult<CustomEntityRecognitionResult, \"CustomEntityRecognition\">;\n\n/**\n * The result of a custom single-label classification batch action.\n */\nexport type CustomSingleLabelClassificationBatchResult = CustomActionMetadata &\n  BatchActionResult<CustomSingleLabelClassificationResult, \"CustomSingleLabelClassification\">;\n\n/**\n * The result of a custom multi-label classification batch action.\n */\nexport type CustomMultiLabelClassificationBatchResult = CustomActionMetadata &\n  BatchActionResult<CustomMultiLabelClassificationResult, \"CustomMultiLabelClassification\">;\n/**\n * Results of a batch of actions.\n */\nexport type AnalyzeBatchResult =\n  | EntityLinkingBatchResult\n  | EntityRecognitionBatchResult\n  | KeyPhraseExtractionBatchResult\n  | PiiEntityRecognitionBatchResult\n  | SentimentAnalysisBatchResult\n  | HealthcareBatchResult\n  | CustomEntityRecognitionBatchResult\n  | CustomSingleLabelClassificationBatchResult\n  | CustomMultiLabelClassificationBatchResult;\n\n/**\n * An error result from a sentiment analysis action on a single document.\n */\nexport type SentimentAnalysisErrorResult = TextAnalysisErrorResult;\n\n/**\n * Paged results of the {@link TextAnalysisClient.beginAnalyzeBatch} operation.\n */\nexport type PagedAnalyzeBatchResult = PagedAsyncIterableIterator<AnalyzeBatchResult>;\n\n/**\n * A poller that polls long-running operations started by {@link TextAnalysisClient.beginAnalyzeBatch}.\n */\nexport type AnalyzeBatchPoller = PollerLike<AnalyzeBatchOperationState, PagedAnalyzeBatchResult>;\n\n/**\n * The metadata for long-running operations started by {@link TextAnalysisClient.beginAnalyzeBatch}.\n */\nexport interface AnalyzeBatchOperationMetadata {\n  /**\n   * The date and time the operation was created.\n   */\n  readonly createdOn: Date;\n  /**\n   * The date and time when the operation results will expire on the server.\n   */\n  readonly expiresOn?: Date;\n  /**\n   * The operation id.\n   */\n  readonly id: string;\n  /**\n   * The time the operation status was last updated.\n   */\n  readonly modifiedOn: Date;\n  /**\n   * Number of successfully completed actions.\n   */\n  readonly actionSucceededCount: number;\n  /**\n   * Number of failed actions.\n   */\n  readonly actionFailedCount: number;\n  /**\n   * Number of actions still in progress.\n   */\n  readonly actionInProgressCount: number;\n  /**\n   * The operation's display name.\n   */\n  readonly displayName?: string;\n}\n\n/**\n * The state of the begin analyze polling operation.\n */\nexport interface AnalyzeBatchOperationState\n  extends OperationState<PagedAnalyzeBatchResult>,\n    AnalyzeBatchOperationMetadata {}\n\n/**\n * Abstract representation of a poller, intended to expose just the minimal API that the user needs to work with.\n */\nexport interface PollerLike<TState extends OperationState<TResult>, TResult>\n  extends SimplePollerLike<TState, TResult> {\n  /**\n   * sends a cancellation request.\n   */\n  sendCancellationRequest: () => Promise<void>;\n}\n"]}

package/dist-esm/src/textAnalysisClient.js

@@ -2,14 +2,14 @@
 // Licensed under the MIT license.
 import { __rest } from "tslib";
 import { DEFAULT_COGNITIVE_SCOPE, SDK_VERSION } from "./constants";
-import { bearerTokenAuthenticationPolicy, } from "@azure/core-rest-pipeline";
 import { isTokenCredential } from "@azure/core-auth";
-import { LroEngine } from "@azure/core-lro";
 import { createTracingClient } from "@azure/core-tracing";
 import { convertToLanguageDetectionInput, convertToTextDocumentInput, getOperationOptions, isStringArray, } from "./util";
-import { createAnalyzeBatchLro, createCancelOperation, createCreateAnalyzeBatchPollerLro, createUpdateAnalyzeState, getDocsFromState, processAnalyzeResult, } from "./lro";
+import { createAnalyzeBatchLro, createCreateAnalyzeBatchPollerLro, createPollerWithCancellation, createUpdateAnalyzeState, getDocIDsFromState, processAnalyzeResult, } from "./lro";
 import { throwError, transformActionResult } from "./transforms";
 import { GeneratedClient } from "./generated/generatedClient";
+import { bearerTokenAuthenticationPolicy } from "@azure/core-rest-pipeline";
+import { createHttpPoller } from "@azure/core-lro";
 import { logger } from "./logger";
 import { textAnalyticsAzureKeyCredentialPolicy } from "./azureKeyCredentialPolicy";
 /**
@@ -51,15 +51,15 @@ import { textAnalyticsAzureKeyCredentialPolicy } from "./azureKeyCredentialPolic
  */
 export class TextAnalysisClient {
     constructor(endpointUrl, credential, options = {}) {
-        const { defaultCountryHint = "us", defaultLanguage = "en" } = options, pipelineOptions = __rest(options, ["defaultCountryHint", "defaultLanguage"]);
+        const { defaultCountryHint = "us", defaultLanguage = "en", serviceVersion } = options, pipelineOptions = __rest(options, ["defaultCountryHint", "defaultLanguage", "serviceVersion"]);
         this.defaultCountryHint = defaultCountryHint;
         this.defaultLanguage = defaultLanguage;
-        const internalPipelineOptions = Object.assign(Object.assign({}, pipelineOptions), {
+        const internalPipelineOptions = Object.assign(Object.assign(Object.assign({}, pipelineOptions), {
             loggingOptions: {
                 logger: logger.info,
                 additionalAllowedHeaderNames: ["x-ms-correlation-request-id", "x-ms-request-id"],
             },
-        });
+        }), { apiVersion: serviceVersion });
         this._client = new GeneratedClient(endpointUrl, internalPipelineOptions);
         const authPolicy = isTokenCredential(credential)
             ? bearerTokenAuthenticationPolicy({ credential, scopes: DEFAULT_COGNITIVE_SCOPE })
@@ -105,7 +105,7 @@ export class TextAnalysisClient {
             },
             parameters: action,
         }, updatedOptions)
-            .then((result) => transformActionResult(actionName, realInputs, result))));
+            .then((result) => transformActionResult(actionName, realInputs.map(({ id }) => id), result))));
     }
     // implementation
     async beginAnalyzeBatch(actions, documents, languageOrOptions, options = {}) {
@@ -141,51 +141,66 @@ export class TextAnalysisClient {
             tasks: realActions,
             tracing: this._tracing,
         });
-        const poller = new LroEngine(lro, {
+        const docIds = realInputs.map(({ id }) => id);
+        const state = { continuationToken: "" };
+        const poller = await createHttpPoller(lro, {
             intervalInMs: updateIntervalInMs,
             processResult: processAnalyzeResult({
                 client: this._client,
                 tracing: this._tracing,
-                documents: realInputs,
+                docIds,
                 opOptions: Object.assign(Object.assign({}, rest), { includeStatistics }),
+                state,
             }),
-            updateState: createUpdateAnalyzeState(realInputs),
-            cancel: createCancelOperation({
-                client: this._client,
-                tracing: this._tracing,
-                options: rest,
-            }),
+            updateState: createUpdateAnalyzeState(docIds),
+            withOperationLocation(operationLocation) {
+                state.continuationToken = operationLocation;
+            },
         });
         await poller.poll();
-        return poller;
+        const id = poller.getOperationState().id;
+        return createPollerWithCancellation({
+            id,
+            client: this._client,
+            options,
+            poller,
+            tracing: this._tracing,
+        });
     }
     // implementation
     async restoreAnalyzeBatchPoller(serializedState, options = {}) {
         const { includeStatistics, updateIntervalInMs } = options, rest = __rest(options, ["includeStatistics", "updateIntervalInMs"]);
-        const documents = getDocsFromState(serializedState);
+        const docIds = getDocIDsFromState(serializedState);
         const lro = createCreateAnalyzeBatchPollerLro({
             client: this._client,
             options: Object.assign(Object.assign({}, rest), { includeStatistics }),
             tracing: this._tracing,
         });
-        const poller = new LroEngine(lro, {
+        const state = { continuationToken: "" };
+        const poller = await createHttpPoller(lro, {
             intervalInMs: updateIntervalInMs,
-            resumeFrom: serializedState,
+            restoreFrom: serializedState,
             processResult: processAnalyzeResult({
                 client: this._client,
                 tracing: this._tracing,
-                documents,
+                docIds,
                 opOptions: Object.assign(Object.assign({}, rest), { includeStatistics }),
+                state,
             }),
             updateState: createUpdateAnalyzeState(),
-            cancel: createCancelOperation({
-                client: this._client,
-                tracing: this._tracing,
-                options: rest,
-            }),
+            withOperationLocation(operationLocation) {
+                state.continuationToken = operationLocation;
+            },
         });
         await poller.poll();
-        return poller;
+        const id = poller.getOperationState().id;
+        return createPollerWithCancellation({
+            id,
+            client: this._client,
+            options,
+            poller,
+            tracing: this._tracing,
+        });
     }
 }
 //# sourceMappingURL=textAnalysisClient.js.map