From 5d54d9c220906ec5fb7c7684197ae3a43883f6d7 Mon Sep 17 00:00:00 2001
From: Farrukh Masud <farrukhmasud@microsoft.com>
Date: Thu, 30 Jan 2025 18:42:14 -0800
Subject: [PATCH] fixing minor problems and documentation

---
 .../services/language/AnalyzeTextLongRunningOperations.scala | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/cognitive/src/main/scala/com/microsoft/azure/synapse/ml/services/language/AnalyzeTextLongRunningOperations.scala b/cognitive/src/main/scala/com/microsoft/azure/synapse/ml/services/language/AnalyzeTextLongRunningOperations.scala
index 1d1e412ee4..cc5d0f1ed4 100644
--- a/cognitive/src/main/scala/com/microsoft/azure/synapse/ml/services/language/AnalyzeTextLongRunningOperations.scala
+++ b/cognitive/src/main/scala/com/microsoft/azure/synapse/ml/services/language/AnalyzeTextLongRunningOperations.scala
@@ -55,6 +55,8 @@ object AnalyzeTextLongRunningOperations extends ComplexParamsReadable[AnalyzeTex
  *   <li>EntityLinking</li>
  *   <li>EntityRecognition</li>
  *   <li>CustomEntityRecognition</li>
+ *   <li>CustomSingleLabelClassification</li>
+ *   <li>CustomMultiLabelClassification</li>
  * </ul>
  * Each task has its own set of parameters that can be set to control the behavior of the service and response
  * schema.
@@ -62,7 +64,7 @@ object AnalyzeTextLongRunningOperations extends ComplexParamsReadable[AnalyzeTex
  */
 class AnalyzeTextLongRunningOperations(override val uid: String) extends CognitiveServicesBaseNoHandler(uid)
                                                                          with HasAPIVersion
-                                                                         with BasicAsyncReply
+                                                                         with ModifiableAsyncReply
                                                                          with HasCognitiveServiceInput
                                                                          with HasInternalJsonOutputParser
                                                                          with HasSetLocation
@@ -79,7 +81,6 @@ class AnalyzeTextLongRunningOperations(override val uid: String) extends Cogniti
                                                                          with HandleEntityLinking
                                                                          with HandleEntityRecognition
                                                                          with HandleCustomEntityRecognition
-                                                                         with ModifiableAsyncReply
                                                                          with HandleCustomLabelClassification {
   logClass(FeatureNames.AiServices.Language)