[TA] Enable Pii Categories in actions

sdk/textanalytics/Azure.AI.TextAnalytics/CHANGELOG.md

@@ -6,6 +6,7 @@
 - Added value `None` to enum `PiiEntityDomainType` to allow user to specify no domain.
 - Added property `ActionName` to all `xxActions` input types so user can specify a name per action. If not provided, service will generate a name.
 - Added property `ActionName` to all `xxActionResult` output types that displays the name of each action. 
+- The parameter `CategoriesFilter` in `RecognizePiiEntitiesActions` has been enabled for `StartAnalyzeActions` methods.
 
 ### Breaking changes
 - Renamed `StartAnalyzeBatchActions` to `StartAnalyzeActions`.

sdk/textanalytics/Azure.AI.TextAnalytics/README.md

@@ -724,7 +724,6 @@ This functionality allows running multiple actions in one or more documents. Act
 
 ### Known Issues
 - `StartAnalyzeHealthcareEntities` is in gated preview and can not be used with AAD credentials. For more information, see [the Text Analytics for Health documentation](https://docs.microsoft.com/azure/cognitive-services/text-analytics/how-tos/text-analytics-for-health?tabs=ner#request-access-to-the-public-preview).
-- The parameter `CategoriesFilter` in `RecognizePiiEntitiesOptions` is currently not working when used in `StartAnalyzeBatchActions`. [19237](https://github.com/Azure/azure-sdk-for-net/issues/19237).
 - At time of this SDK release, the `ModelVersion` option to `StartAnalyzeHealthcareEntities` is ignored by the service. The service always processes the operation using the `latest` model.
 
 ## Troubleshooting

sdk/textanalytics/Azure.AI.TextAnalytics/src/Internal/PiiTaskParameters.cs

@@ -1,6 +1,7 @@
 // Copyright (c) Microsoft Corporation. All rights reserved.
 // Licensed under the MIT License.
 
+using System.Collections.Generic;
 using Azure.Core;
 
 namespace Azure.AI.TextAnalytics.Models
@@ -9,5 +10,9 @@ namespace Azure.AI.TextAnalytics.Models
     /// PiiTaskParameters class.
     /// </summary>
     [CodeGenModel("PiiTaskParameters")]
-    internal partial class PiiTaskParameters { }
+    internal partial class PiiTaskParameters
+    {
+        /// <summary> (Optional) describes the PII categories to return. </summary>
+        public IList<PiiEntityCategory> PiiCategories { get; set; }
+    }
 }

sdk/textanalytics/Azure.AI.TextAnalytics/src/Transforms.cs

@@ -329,8 +329,8 @@ internal static PiiTask ConvertToPiiTask(RecognizePiiEntitiesAction action)
                     Domain = action.DomainFilter.GetString() ?? (PiiTaskParametersDomain?)null,
                     ModelVersion = action.ModelVersion,
                     StringIndexType = Constants.DefaultStringIndexType,
-                    LoggingOptOut = action.DisableServiceLogs
-                    // Categories are not enabled because of https://github.com/Azure/azure-sdk-for-net/issues/19237
+                    LoggingOptOut = action.DisableServiceLogs,
+                    PiiCategories = action.CategoriesFilter
                 },
                 TaskName = action.ActionName
             };

sdk/textanalytics/Azure.AI.TextAnalytics/tests/AnalyzeOperationTests.cs

@@ -446,6 +446,42 @@ public async Task AnalyzeOperationWithPHIDomain()
             Assert.AreEqual(2, piiDocumentsResults[0].Entities.Count);
         }
 
+        [RecordedTest]
+        public async Task AnalyzeOperationWithPiiCategories()
+        {
+            TextAnalyticsClient client = GetClient();
+
+            var documents = new List<string>
+            {
+                "A developer with SSN 859-98-0987 whose phone number is 800-102-1100 is building tools with our APIs. They work at Microsoft.",
+            };
+
+            TextAnalyticsActions batchActions = new TextAnalyticsActions()
+            {
+                RecognizePiiEntitiesActions = new List<RecognizePiiEntitiesAction>() { new RecognizePiiEntitiesAction() { CategoriesFilter = { PiiEntityCategory.USSocialSecurityNumber } } },
+            };
+
+            AnalyzeActionsOperation operation = await client.StartAnalyzeActionsAsync(documents, batchActions, "en");
+
+            await operation.WaitForCompletionAsync();
+
+            //Take the first page
+            AnalyzeActionsResult resultCollection = operation.Value.ToEnumerableAsync().Result.FirstOrDefault();
+
+            IReadOnlyCollection<RecognizePiiEntitiesActionResult> piiActionsResults = resultCollection.RecognizePiiEntitiesResults;
+
+            Assert.IsNotNull(piiActionsResults);
+
+            RecognizePiiEntitiesResultCollection piiDocumentsResults = piiActionsResults.FirstOrDefault().DocumentsResults;
+            Assert.AreEqual(1, piiDocumentsResults.Count);
+
+            Assert.IsNotEmpty(piiDocumentsResults[0].Entities.RedactedText);
+
+            Assert.IsFalse(piiDocumentsResults[0].HasError);
+            Assert.AreEqual(1, piiDocumentsResults[0].Entities.Count);
+            Assert.AreEqual(PiiEntityCategory.USSocialSecurityNumber, piiDocumentsResults[0].Entities.FirstOrDefault().Category);
+        }
+
         [RecordedTest]
         public async Task AnalyzeOperationWithStatisticsTest()
         {