【增加】AI Image 列表兼容 Midjourney 操作按钮

This commit is contained in:
cherishsince 2024-06-04 17:39:59 +08:00
parent 541f6b9e6b
commit 4c20593211
2 changed files with 29 additions and 0 deletions

View File

@ -1,8 +1,10 @@
package cn.iocoder.yudao.module.ai.controller.admin.image.vo;
import cn.iocoder.yudao.module.ai.client.vo.MidjourneyNotifyReqVO;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;
import java.util.List;
import java.util.Map;
@Schema(description = "管理后台 - 绘画 Response VO")
@ -45,4 +47,12 @@ public class AiImageRespVO {
@Schema(description = "绘制参数")
private Map<String, String> options;
@Schema(description = "绘画 response")
private MidjourneyNotifyReqVO response;
@Schema(description = "mj 进度")
private String progress;
@Schema(description = "mj buttons 按钮")
private List<MidjourneyNotifyReqVO.Button> buttons;
}

View File

@ -1,6 +1,7 @@
package cn.iocoder.yudao.module.ai.dal.dataobject.image;
import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
import cn.iocoder.yudao.module.ai.client.vo.MidjourneyNotifyReqVO;
import cn.iocoder.yudao.module.ai.dal.dataobject.model.AiChatModelDO;
import cn.iocoder.yudao.module.ai.enums.image.AiImageStatusEnum;
import cn.iocoder.yudao.module.system.api.user.dto.AdminUserRespDTO;
@ -13,6 +14,7 @@ import lombok.Data;
import org.springframework.ai.openai.OpenAiImageOptions;
import org.springframework.ai.stabilityai.api.StabilityAiImageOptions;
import java.util.List;
import java.util.Map;
/**
@ -92,6 +94,23 @@ public class AiImageDO extends BaseDO {
@TableField(typeHandler = JacksonTypeHandler.class)
private Map<String, String> options;
/**
* 绘画 response
*/
@TableField(typeHandler = JacksonTypeHandler.class)
private MidjourneyNotifyReqVO response;
/**
* mj 进度
*/
private String progress;
/**
* mj buttons 按钮
*/
@TableField(typeHandler = JacksonTypeHandler.class)
private List<MidjourneyNotifyReqVO.Button> buttons;
// TODO @芋艿再瞅瞅
/**
* midjourney proxy 关联的 job id