fix service to use for testing embedding dimensions

microsoft · Apr 30, 2024 · d59f417 · d59f417
1 parent 6cdf750
commit d59f417
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/...src/test/scala/com/microsoft/azure/synapse/ml/services/openai/OpenAIEmbeddingsSuite.scala b/...src/test/scala/com/microsoft/azure/synapse/ml/services/openai/OpenAIEmbeddingsSuite.scala
@@ -34,11 +34,12 @@ class OpenAIEmbeddingsSuite extends TransformerFuzzing[OpenAIEmbedding] with Ope
   }
 
   lazy val embeddingExtra: OpenAIEmbedding = new OpenAIEmbedding()
-    .setSubscriptionKey(openAIAPIKey)
+    .setSubscriptionKey(openAIAPIKeyGpt4)
     .setDeploymentName("text-embedding-3-small")
     .setApiVersion("2024-03-01-preview")
     .setDimensions(100)
-    .setCustomServiceName(openAIServiceName)
+    .setUser("testUser")
+    .setCustomServiceName(openAIServiceNameGpt4)
     .setTextCol("text")
     .setOutputCol("out")