BJ_ZTEITS-Developers / parking-cloud-yudao

Commit 8f2ce2faae8818542bd806cc8a07fdb630381028

Authored by cherishsince
1 parent 32bc6329

修改mj message type 类型

Inline Side-by-side
Showing 1 changed file with 3 additions and 3 deletions
yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/midjourney/webSocket/listener/MidjourneyMessageListener.java
  View file @8f2ce2f
@@ -63,11 +63,11 @@ public class MidjourneyMessageListener { @@ -63,11 +63,11 @@ public class MidjourneyMessageListener {
63 63
64 private void convertGenerateStatus(MidjourneyMessage mjMessage) { 64 private void convertGenerateStatus(MidjourneyMessage mjMessage) {
65 if (mjMessage.getType() == 20 && mjMessage.getContent().getStatus().contains("Waiting")) { 65 if (mjMessage.getType() == 20 && mjMessage.getContent().getStatus().contains("Waiting")) {
66 - mjMessage.setGenerateStatus(MidjourneyGennerateStatusEnum.WAITING.getValue()); 66 + mjMessage.setGenerateStatus(MidjourneyGennerateStatusEnum.WAITING.getStatus());
67 } else if (mjMessage.getType() == 20 && !StrUtil.isBlank(mjMessage.getContent().getProgress())) { 67 } else if (mjMessage.getType() == 20 && !StrUtil.isBlank(mjMessage.getContent().getProgress())) {
68 - mjMessage.setGenerateStatus(MidjourneyGennerateStatusEnum.IN_PROGRESS.getValue()); 68 + mjMessage.setGenerateStatus(MidjourneyGennerateStatusEnum.IN_PROGRESS.getStatus());
69 } else if (mjMessage.getType() == 0 && !CollUtil.isEmpty(mjMessage.getComponents())) { 69 } else if (mjMessage.getType() == 0 && !CollUtil.isEmpty(mjMessage.getComponents())) {
70 - mjMessage.setGenerateStatus(MidjourneyGennerateStatusEnum.COMPLETED.getValue()); 70 + mjMessage.setGenerateStatus(MidjourneyGennerateStatusEnum.COMPLETED.getStatus());
71 } 71 }
72 } 72 }
73 73