Fine tune event and fine tune hyperparams response holders adapted + …

…EitherIntString JSON format employed
cequence-io · Sep 27, 2023 · 1221abd · 1221abd
1 parent 22e9dfa
commit 1221abd
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/openai-client/src/main/scala/io/cequence/openaiscala/JsonFormats.scala b/openai-client/src/main/scala/io/cequence/openaiscala/JsonFormats.scala
@@ -92,6 +92,7 @@ object JsonFormats {
 
   implicit val fineTuneEventFormat: Format[FineTuneEvent] =
     Json.format[FineTuneEvent]
+  implicit val eitherIntStringFormat = JsonUtil.eitherFormat[Int, String]
   implicit val fineTuneHyperparamsFormat: Format[FineTuneHyperparams] =
     Json.format[FineTuneHyperparams]
   implicit val fineTuneFormat: Format[FineTuneJob] = Json.format[FineTuneJob]

diff --git a/openai-core/src/main/scala/io/cequence/openaiscala/domain/response/FineTuneJob.scala b/openai-core/src/main/scala/io/cequence/openaiscala/domain/response/FineTuneJob.scala
@@ -37,15 +37,14 @@ case class FineTuneJob(
 }
 
 case class FineTuneEvent(
+  id: String,
   created_at: ju.Date,
   level: String,
-  message: String
+  message: String,
+  data: Option[String]
 )
 
-// TODO: adapt
 case class FineTuneHyperparams(
-  batch_size: Option[Int],
-  learning_rate_multiplier: Option[Double],
-  n_epochs: Int,
-  prompt_loss_weight: Double
+  // the number of epochs or auto (if not specified initially)
+  n_epochs: Either[Int, String]
 )