From ddb6a18d55d1a8004107de8e20847aa6bb200d93 Mon Sep 17 00:00:00 2001
From: Krista Pratico <krpratic@microsoft.com>
Date: Thu, 29 Apr 2021 08:26:18 -0700
Subject: [PATCH] add entity enums (#18410)

---
 sdk/textanalytics/azure-ai-textanalytics/CHANGELOG.md     | 2 ++
 .../azure/ai/textanalytics/__init__.py                    | 8 +++++++-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/sdk/textanalytics/azure-ai-textanalytics/CHANGELOG.md b/sdk/textanalytics/azure-ai-textanalytics/CHANGELOG.md
index db49a5bf7188..2f709587138c 100644
--- a/sdk/textanalytics/azure-ai-textanalytics/CHANGELOG.md
+++ b/sdk/textanalytics/azure-ai-textanalytics/CHANGELOG.md
@@ -2,6 +2,8 @@
 
 ## 5.1.0b7 (Unreleased)
 
+**New Features**
+- Added enums `EntityConditionality`, `EntityCertainty`, and `EntityAssociation`.
 
 ## 5.1.0b6 (2021-03-09)
 
diff --git a/sdk/textanalytics/azure-ai-textanalytics/azure/ai/textanalytics/__init__.py b/sdk/textanalytics/azure-ai-textanalytics/azure/ai/textanalytics/__init__.py
index 7cff8c13a27c..691e774f4d18 100644
--- a/sdk/textanalytics/azure-ai-textanalytics/azure/ai/textanalytics/__init__.py
+++ b/sdk/textanalytics/azure-ai-textanalytics/azure/ai/textanalytics/__init__.py
@@ -51,7 +51,10 @@
 from ._paging import AnalyzeHealthcareEntitiesResult
 from ._generated.v3_1_preview_5.models import (
     PiiCategory as PiiEntityCategoryType,
-    RelationType as HealthcareEntityRelationType
+    RelationType as HealthcareEntityRelationType,
+    Conditionality as EntityConditionality,
+    Certainty as EntityCertainty,
+    Association as EntityAssociation
 )
 
 __all__ = [
@@ -99,6 +102,9 @@
     "HealthcareRelation",
     "HealthcareRelationRole",
     "HealthcareEntityAssertion",
+    "EntityConditionality",
+    "EntityCertainty",
+    "EntityAssociation"
 ]
 
 __version__ = VERSION