reducing the max token count to enable tests to pass

microsoft · Jan 2, 2025 · 814fda7 · 814fda7
1 parent 9765add
commit 814fda7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...test/scala/com/microsoft/azure/synapse/ml/services/openai/OpenAIChatCompletionSuite.scala b/...test/scala/com/microsoft/azure/synapse/ml/services/openai/OpenAIChatCompletionSuite.scala
@@ -254,7 +254,7 @@ class OpenAIChatCompletionSuite extends TransformerFuzzing[OpenAIChatCompletion]
       .setDeploymentName(deploymentNameGpt4o)
       .setCustomServiceName(openAIServiceName)
       .setApiVersion("2023-05-15")
-      .setMaxTokens(5000)
+      .setMaxTokens(500)
       .setOutputCol("out")
       .setMessagesCol("messages")
       .setTemperature(0)