@azure/ai-language-text 1.1.0-alpha.20230228.2 → 1.1.0-alpha.20230301.2

package/dist/index.js

@@ -166,6 +166,21 @@ function getOperationOptions(options) {
         rest,
     };
 }
+/**
+ *
+ * @param error - error with the target in the JSON error pointer format "#/items/0
+ * @returns number: the position of the task with error
+ */
+function extractErrorPointerIndex(error) {
+    if (!error.target) {
+        throw new Error("Unexpected response from service - no target present");
+    }
+    const position = parseInt(error.target.split("/").pop());
+    if (isNaN(position)) {
+        throw new Error(`Unexpected response from service - action pointer "${error.target}" is not a valid action pointer.`);
+    }
+    return position;
+}
 
 /*
  * Copyright (c) Microsoft Corporation.
@@ -4076,74 +4091,172 @@ function toHealthcareResult(docIds, results) {
 /**
  * @internal
  */
-function transformAnalyzeBatchResults(docIds, response = []) {
-    return response.map((actionData) => {
-        const { lastUpdateDateTime: completedOn, actionName, kind } = actionData;
-        switch (kind) {
+function transformAnalyzeBatchResults(docIds, response = [], errors = []) {
+    const errorMap = toIndexErrorMap(errors);
+    return response.map((actionData, idx) => {
+        const { lastUpdateDateTime: completedOn, actionName, kind: resultKind } = actionData;
+        const error = errorMap.get(idx);
+        switch (resultKind) {
             case "SentimentAnalysisLROResults": {
+                const kind = "SentimentAnalysis";
+                if (actionData.status === "failed") {
+                    return returnErrorTask(kind, error, completedOn);
+                }
                 const { results } = actionData;
                 const { modelVersion, statistics } = results;
-                return Object.assign(Object.assign(Object.assign({ kind: "SentimentAnalysis", results: toSentimentAnalysisResult(docIds, results), completedOn }, (actionName ? { actionName } : {})), (statistics ? { statistics } : {})), { modelVersion });
+                return Object.assign(Object.assign(Object.assign({ kind, results: toSentimentAnalysisResult(docIds, results), completedOn }, (actionName ? { actionName } : {})), (statistics ? { statistics } : {})), { modelVersion });
             }
             case "EntityRecognitionLROResults": {
+                const kind = "EntityRecognition";
+                if (actionData.status === "failed") {
+                    return returnErrorTask(kind, error, completedOn);
+                }
                 const { results } = actionData;
                 const { modelVersion, statistics } = results;
                 return Object.assign(Object.assign(Object.assign({ kind: "EntityRecognition", results: toEntityRecognitionResult(docIds, results), completedOn }, (actionName ? { actionName } : {})), (statistics ? { statistics } : {})), { modelVersion });
             }
             case "PiiEntityRecognitionLROResults": {
+                const kind = "PiiEntityRecognition";
+                if (actionData.status === "failed") {
+                    return returnErrorTask(kind, error, completedOn);
+                }
                 const { results } = actionData;
                 const { modelVersion, statistics } = results;
-                return Object.assign(Object.assign(Object.assign({ kind: "PiiEntityRecognition", results: toPiiEntityRecognitionResult(docIds, results), completedOn }, (actionName ? { actionName } : {})), (statistics ? { statistics } : {})), { modelVersion });
+                return Object.assign(Object.assign(Object.assign({ kind, results: toPiiEntityRecognitionResult(docIds, results), completedOn }, (actionName ? { actionName } : {})), (statistics ? { statistics } : {})), { modelVersion });
             }
             case "KeyPhraseExtractionLROResults": {
+                const kind = "KeyPhraseExtraction";
+                if (actionData.status === "failed") {
+                    return returnErrorTask(kind, error, completedOn);
+                }
                 const { results } = actionData;
                 const { modelVersion, statistics } = results;
-                return Object.assign(Object.assign(Object.assign({ kind: "KeyPhraseExtraction", results: toKeyPhraseExtractionResult(docIds, results), completedOn }, (actionName ? { actionName } : {})), (statistics ? { statistics } : {})), { modelVersion });
+                return Object.assign(Object.assign(Object.assign({ kind, results: toKeyPhraseExtractionResult(docIds, results), completedOn }, (actionName ? { actionName } : {})), (statistics ? { statistics } : {})), { modelVersion });
             }
             case "EntityLinkingLROResults": {
+                const kind = "EntityLinking";
+                if (actionData.status === "failed") {
+                    return returnErrorTask(kind, error, completedOn);
+                }
                 const { results } = actionData;
                 const { modelVersion, statistics } = results;
-                return Object.assign(Object.assign(Object.assign({ kind: "EntityLinking", results: toEntityLinkingResult(docIds, results), completedOn }, (actionName ? { actionName } : {})), (statistics ? { statistics } : {})), { modelVersion });
+                return Object.assign(Object.assign(Object.assign({ kind, results: toEntityLinkingResult(docIds, results), completedOn }, (actionName ? { actionName } : {})), (statistics ? { statistics } : {})), { modelVersion });
             }
             case "HealthcareLROResults": {
+                const kind = "Healthcare";
+                if (actionData.status === "failed") {
+                    return returnErrorTask(kind, error, completedOn);
+                }
                 const { results } = actionData;
                 const { modelVersion, statistics } = results;
-                return Object.assign(Object.assign(Object.assign({ kind: "Healthcare", results: toHealthcareResult(docIds, results), completedOn }, (actionName ? { actionName } : {})), (statistics ? { statistics } : {})), { modelVersion });
+                return Object.assign(Object.assign(Object.assign({ kind, results: toHealthcareResult(docIds, results), completedOn }, (actionName ? { actionName } : {})), (statistics ? { statistics } : {})), { modelVersion });
             }
             case "CustomEntityRecognitionLROResults": {
+                const kind = "CustomEntityRecognition";
+                if (actionData.status === "failed") {
+                    return returnErrorCustomTask(kind, error, completedOn);
+                }
                 const { results } = actionData;
                 const { deploymentName, projectName, statistics } = results;
-                return Object.assign(Object.assign(Object.assign({ kind: "CustomEntityRecognition", results: transformDocumentResults(docIds, results), completedOn }, (actionName ? { actionName } : {})), (statistics ? { statistics } : {})), { deploymentName,
+                return Object.assign(Object.assign(Object.assign({ kind, results: transformDocumentResults(docIds, results), completedOn }, (actionName ? { actionName } : {})), (statistics ? { statistics } : {})), { deploymentName,
                     projectName });
             }
             case "CustomSingleLabelClassificationLROResults": {
+                const kind = "CustomSingleLabelClassification";
+                if (actionData.status === "failed") {
+                    return returnErrorCustomTask(kind, error, completedOn);
+                }
                 const { results } = actionData;
                 const { deploymentName, projectName, statistics } = results;
-                return Object.assign(Object.assign(Object.assign({ kind: "CustomSingleLabelClassification", results: transformDocumentResults(docIds, results), completedOn }, (actionName ? { actionName } : {})), (statistics ? { statistics } : {})), { deploymentName,
+                return Object.assign(Object.assign(Object.assign({ kind, results: transformDocumentResults(docIds, results), completedOn }, (actionName ? { actionName } : {})), (statistics ? { statistics } : {})), { deploymentName,
                     projectName });
             }
             case "CustomMultiLabelClassificationLROResults": {
+                const kind = "CustomMultiLabelClassification";
+                if (actionData.status === "failed") {
+                    return returnErrorCustomTask(kind, error, completedOn);
+                }
                 const { results } = actionData;
                 const { deploymentName, projectName, statistics } = results;
-                return Object.assign(Object.assign(Object.assign({ kind: "CustomMultiLabelClassification", results: transformDocumentResults(docIds, results), completedOn }, (actionName ? { actionName } : {})), (statistics ? { statistics } : {})), { deploymentName,
+                return Object.assign(Object.assign(Object.assign({ kind, results: transformDocumentResults(docIds, results), completedOn }, (actionName ? { actionName } : {})), (statistics ? { statistics } : {})), { deploymentName,
                     projectName });
             }
             case "ExtractiveSummarizationLROResults": {
+                const kind = "ExtractiveSummarization";
+                if (actionData.status === "failed") {
+                    return returnErrorTask(kind, error, completedOn);
+                }
                 const { results } = actionData;
                 const { modelVersion, statistics } = results;
-                return Object.assign(Object.assign(Object.assign({ kind: "ExtractiveSummarization", results: transformDocumentResults(docIds, results), completedOn }, (actionName ? { actionName } : {})), (statistics ? { statistics } : {})), { modelVersion });
+                return Object.assign(Object.assign(Object.assign({ kind, results: transformDocumentResults(docIds, results), completedOn }, (actionName ? { actionName } : {})), (statistics ? { statistics } : {})), { modelVersion });
             }
             case "AbstractiveSummarizationLROResults": {
+                const kind = "AbstractiveSummarization";
+                if (actionData.status === "failed") {
+                    return returnErrorTask(kind, error, completedOn);
+                }
                 const { results } = actionData;
                 const { modelVersion, statistics } = results;
-                return Object.assign(Object.assign(Object.assign({ kind: "AbstractiveSummarization", results: transformDocumentResults(docIds, results), completedOn }, (actionName ? { actionName } : {})), (statistics ? { statistics } : {})), { modelVersion });
+                return Object.assign(Object.assign(Object.assign({ kind, results: transformDocumentResults(docIds, results), completedOn }, (actionName ? { actionName } : {})), (statistics ? { statistics } : {})), { modelVersion });
             }
             default: {
-                throw new Error(`Unsupported results kind: ${kind}`);
+                throw new Error(`Unsupported results kind: ${resultKind}`);
             }
         }
     });
 }
+/**
+ * @internal
+ * Transform a list of error into index and error Map
+ */
+function toIndexErrorMap(errors) {
+    const errorMap = new Map();
+    for (const error of errors) {
+        const position = extractErrorPointerIndex(error);
+        const errorWithoutTarget = tslib.__rest(error, ["target"]);
+        errorMap.set(position, toTextAnalysisError(errorWithoutTarget));
+    }
+    return errorMap;
+}
+/**
+ * Return the error for non-custom task
+ *
+ * @param kind - non custom task kind
+ * @param error - error returned from the service
+ * @param failedOn - the LastUpdateDateTime from the service
+ * @returns - AnalyzeBatchResult with error
+ */
+function returnErrorTask(kind, error, failedOn) {
+    if (!error) {
+        throw new Error("Unexpected response from service - no errors for missing action results.");
+    }
+    return {
+        kind,
+        modelVersion: "",
+        failedOn,
+        error,
+    };
+}
+/**
+ * Return the error for non-custom task
+ *
+ * @param kind - non custom task kind
+ * @param error - error returned from the service
+ * @param failedOn - the LastUpdateDateTime from the service
+ * @returns AnalyzeBatchResult for custom task with error
+ */
+function returnErrorCustomTask(kind, error, failedOn) {
+    if (!error) {
+        throw new Error("Unexpected response from service - no errors for missing action results.");
+    }
+    return {
+        kind,
+        projectName: "",
+        deploymentName: "",
+        failedOn,
+        error,
+    };
+}
 
 // Copyright (c) Microsoft Corporation.
 const serializer$2 = coreClient.createSerializer(Mappers, /* isXml */ false);
@@ -4204,7 +4317,7 @@ function createSendPollRequest(settings) {
             opOptions: addOnResponse(options, (_, response) => {
                 const castResponse = response;
                 if (castResponse.status.toLowerCase() === "partiallysucceeded") {
-                    castResponse.status = "failed";
+                    castResponse.status = "succeeded";
                 }
             }),
             path,
@@ -4289,7 +4402,7 @@ function processAnalyzeResult(options) {
                 });
                 const flatResponse = response.flatResponse;
                 return {
-                    page: transformAnalyzeBatchResults(docIds, flatResponse.tasks.items),
+                    page: transformAnalyzeBatchResults(docIds, flatResponse.tasks.items, flatResponse.errors),
                     nextPageLink: flatResponse.nextLink,
                 };
             },