Update OpenAIChatCompletion.scala to sync with main

microsoft · Oct 19, 2024 · 9567a7b · 9567a7b
1 parent 9cf90b1
commit 9567a7b
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/.../src/main/scala/com/microsoft/azure/synapse/ml/services/openai/OpenAIChatCompletion.scala b/.../src/main/scala/com/microsoft/azure/synapse/ml/services/openai/OpenAIChatCompletion.scala
@@ -37,7 +37,7 @@ class OpenAIChatCompletion(override val uid: String) extends OpenAIServicesBase(
     s"${getUrl}openai/deployments/${getValue(row, deploymentName)}/chat/completions"
   }
 
-  override protected def prepareEntity: Row => Option[AbstractHttpEntity] = {
+  override protected[openai] def prepareEntity: Row => Option[AbstractHttpEntity] = {
     r =>
       lazy val optionalParams: Map[String, Any] = getOptionalParams(r)
       val messages = r.getAs[Seq[Row]](getMessagesCol)
@@ -65,6 +65,3 @@ class OpenAIChatCompletion(override val uid: String) extends OpenAIServicesBase(
   }
 
 }
-
-
-