update OpenAI endpoints' default timeout

microsoft · Dec 5, 2023 · a86c03f · a86c03f
1 parent 5d13792
commit a86c03f
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/...tive/src/main/scala/com/microsoft/azure/synapse/ml/services/openai/OpenAICompletion.scala b/...tive/src/main/scala/com/microsoft/azure/synapse/ml/services/openai/OpenAICompletion.scala
@@ -24,6 +24,8 @@ class OpenAICompletion(override val uid: String) extends CognitiveServicesBase(u
   with HasInternalJsonOutputParser with SynapseMLLogging {
   logClass(FeatureNames.AiServices.OpenAI)
 
+  setDefault(timeout -> 360.0)
+
   def this() = this(Identifiable.randomUID("OpenAICompletion"))
 
   def urlPath: String = ""

diff --git a/...itive/src/main/scala/com/microsoft/azure/synapse/ml/services/openai/OpenAIEmbedding.scala b/...itive/src/main/scala/com/microsoft/azure/synapse/ml/services/openai/OpenAIEmbedding.scala
@@ -26,6 +26,8 @@ class OpenAIEmbedding (override val uid: String) extends CognitiveServicesBase(u
   with HasOpenAISharedParams with HasCognitiveServiceInput  with SynapseMLLogging {
   logClass(FeatureNames.AiServices.OpenAI)
 
+  setDefault(timeout -> 360.0)
+
   def this() = this(Identifiable.randomUID("OpenAIEmbedding"))
 
   def urlPath: String = ""

diff --git a/cognitive/src/main/scala/com/microsoft/azure/synapse/ml/services/openai/OpenAIPrompt.scala b/cognitive/src/main/scala/com/microsoft/azure/synapse/ml/services/openai/OpenAIPrompt.scala
@@ -26,6 +26,8 @@ class OpenAIPrompt(override val uid: String) extends Transformer
   with HasSubscriptionKey with HasAADToken with HasCustomAuthHeader
   with ComplexParamsWritable with SynapseMLLogging {
 
+  setDefault(timeout -> 360.0)
+
   logClass(FeatureNames.AiServices.OpenAI)
 
   def this() = this(Identifiable.randomUID("OpenAIPrompt"))