feat :: update generating status

API-Team-Organization · Oct 22, 2024 · fcd9996 · fcd9996
1 parent fe47b32
commit fcd9996
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/kotlin/com/teamapi/palette/service/infra/GenerativeImageService.kt b/src/main/kotlin/com/teamapi/palette/service/infra/GenerativeImageService.kt
@@ -78,9 +78,10 @@ class GenerativeImageService(
             generated.collect {
                 when (it) {
                     is QueueInfoMessage -> {
-                        actor.addQueue(room.id!!, it.position, true)
+                        actor.setGenerating(room.id!!, true, it.position)
                     }
                     is GenerateMessage -> {
+                        actor.setGenerating(room.id!!, false, -1)
                         if (it.result) {
                             guaranteed = it.image!!
                         }