rename

microsoft · Sep 27, 2024 · ec12ada · ec12ada
1 parent 55eaefa
commit ec12ada
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/cognitive/src/main/scala/com/microsoft/azure/synapse/ml/services/CognitiveServiceBase.scala b/cognitive/src/main/scala/com/microsoft/azure/synapse/ml/services/CognitiveServiceBase.scala
@@ -330,7 +330,7 @@ trait HasCognitiveServiceInput extends HasURL with HasSubscriptionKey with HasAA
     val providedCustomAuthHeader = getValueOpt(row, CustomAuthHeader)
     if (providedCustomAuthHeader .isEmpty && PlatformDetails.runningOnFabric()) {
       logInfo("Using Default AAD Token On Fabric")
-      Option(FabricClient.getCognitiveAuthHeader)
+      Option(FabricClient.getCognitiveMWCTokenAuthHeader)
     } else {
       providedCustomAuthHeader
     }

diff --git a/core/src/main/scala/com/microsoft/azure/synapse/ml/fabric/FabricClient.scala b/core/src/main/scala/com/microsoft/azure/synapse/ml/fabric/FabricClient.scala
@@ -122,7 +122,7 @@ object FabricClient extends RESTUtils {
 
   private def getHeaders: Map[String, String] = {
     Map(
-      "Authorization" -> s"${getMLWorkloadAuthHeader}",
+      "Authorization" -> s"${getMLWorkloadAADAuthHeader}",
       "RequestId" -> UUID.randomUUID().toString,
       "Content-Type" -> "application/json",
       "x-ms-workload-resource-moniker" -> UUID.randomUUID().toString
@@ -144,9 +144,9 @@ object FabricClient extends RESTUtils {
     usagePost(url, body, getHeaders);
   }
 
-  def getMLWorkloadAuthHeader: String = TokenLibrary.getMLWorkloadAuthHeader
+  def getMLWorkloadAADAuthHeader: String = TokenLibrary.getMLWorkloadAADAuthHeader
 
-  def getCognitiveAuthHeader: String = {
-    TokenLibrary.getCognitiveAuthHeader(WorkspaceID.getOrElse(""), ArtifactID.getOrElse(""))
+  def getCognitiveMWCTokenAuthHeader: String = {
+    TokenLibrary.getCognitiveMwcTokenAuthHeader(WorkspaceID.getOrElse(""), ArtifactID.getOrElse(""))
   }
 }
diff --git a/core/src/main/scala/com/microsoft/azure/synapse/ml/fabric/TokenLibrary.scala b/core/src/main/scala/com/microsoft/azure/synapse/ml/fabric/TokenLibrary.scala
@@ -44,8 +44,8 @@ object TokenLibrary {
   }
 
 
-  def getMLWorkloadAuthHeader: String = "Bearer " + getAccessToken
+  def getMLWorkloadAADAuthHeader: String = "Bearer " + getAccessToken
 
-  def getCognitiveAuthHeader(workspaceId: String, artifactId: String): String = "MwcToken " +
+  def getCognitiveMwcTokenAuthHeader(workspaceId: String, artifactId: String): String = "MwcToken " +
     getSparkMwcToken(workspaceId, artifactId)
 }