batch status methods

cequence-io · Nov 5, 2024 · fc22785 · fc22785
1 parent 3ffa6fb
commit fc22785
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/openai-core/src/main/scala/io/cequence/openaiscala/domain/Batch.scala b/openai-core/src/main/scala/io/cequence/openaiscala/domain/Batch.scala
@@ -86,7 +86,17 @@ object Batch {
     cancelled_at: Option[Long],
     request_counts: Map[String, Int],
     metadata: Option[Map[String, String]]
-  )
+  ) {
+    def isRunning =
+      List("in_progress", "validating", "finalizing", "cancelling").contains(status)
+
+    // "failed", "completed", "expired", "cancelled"
+    def isFinished = !isRunning
+
+    def isSuccess = status == "completed"
+
+    def isFailedOrCancelledOrExpired = isFinished && !isSuccess
+  }
 
   case class BatchProcessingErrors(
     `object`: String,