BJ_ZTEITS-Developers / parking-cloud-yudao

Commit 1b725fd517d729f34544f6330f6898d25efabab0

Authored by cherishsince
1 parent c7ee3dc2

【优化】调整 AI Image 增肌 draw rquest 和 response

Inline Side-by-side
Showing 1 changed file with 31 additions and 35 deletions
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/image/AiImageDO.java
  View file @1b725fd
... ... @@ -6,19 +6,16 @@ import com.baomidou.mybatisplus.annotation.TableId;
6 6 import com.baomidou.mybatisplus.annotation.TableName;
7 7 import io.swagger.v3.oas.annotations.media.Schema;
8 8 import lombok.Data;
9   -import lombok.experimental.Accessors;
10 9  
11   -// TODO @fan：@time、@sine 注解可以不加哈
  10 +import java.util.Map;
  11 +
12 12 /**
13 13 * AI 绘画 DO
14 14 *
15 15 * @author fansili
16   - * @time 2024/4/25 15:53
17   - * @since 1.0
18 16 */
  17 +@TableName("ai_image")
19 18 @Data
20   -@Accessors(chain = true) // TODO @fan：这个不用添加哈，全局的 lombok.config 搞啦
21   -@TableName("ai_image") // TODO @fan：建议放在 @Data 前面，因为它是关键注解，@Data 本质可悲替代
22 19 public class AiImageDO extends BaseDO {
23 20  
24 21 @TableId(type = IdType.AUTO)
... ... @@ -30,50 +27,49 @@ public class AiImageDO extends BaseDO {
30 27 @Schema(description = "提示词")
31 28 private String prompt;
32 29  
33   - // TODO @fan：要加一个平台字段；platform；因为一个平台，会有多种 model 模型；
  30 + @Schema(description = "平台")
  31 + private String platform;
34 32  
35 33 @Schema(description = "模型")
36 34 private String model;
37 35  
38   - // TODO @fan：这个记录到 db，要不改成 width、height？更明确的尺寸
39   - @Schema(description = "尺寸大小")
40   - private String size;
  36 + @Schema(description = "图片宽度")
  37 + private String width;
41 38  
42   - // TODO @fan：style 是所有模型都有么？如果不是，是不是靠 drawRequest 记录下就好了；
43   - @Schema(description = "风格")
44   - private String style;
45   -
46   - @Schema(description = "图片地址(自己服务器)")
47   - private String picUrl;
  39 + @Schema(description = "图片高度")
  40 + private String height;
48 41  
49 42 @Schema(description = "绘画状态：提交、排队、绘画中、绘画完成、绘画失败")
50 43 private String status;
51 44  
52   - @Schema(description = "绘画图片地址(绘画好的服务器)")
53   - private String originalPicUrl;
54   -
55   - @Schema(description = "绘画错误信息")
56   - private String errorMessage;
57   -
58 45 @Schema(description = "是否发布")
59 46 private String publicStatus;
60 47  
61   - // TODO @fan：增加一个 Map<String, Object> 字段，drawRequest；用于记录请求模型的字段；对应的就是 ImageOptions
62   - // TODO @fan：增加一个 Map<String, Object> 字段，记录 ImageResponseMetadata？这样 mjNonceId、mjOperationId、mjOperationName、mjOperations 这些字段，貌似可以收掉
63   -
64   - // ============ mj 需要字段
65   -
66   - @Schema(description = "用户操作的Nonce编号(MJ返回)")
67   - private String mjNonceId;
  48 + @Schema(description = "图片地址(自己服务器)")
  49 + private String picUrl;
68 50  
69   - @Schema(description = "用户操作的操作编号(MJ返回)")
70   - private String mjOperationId;
  51 + @Schema(description = "绘画图片地址(绘画好的服务器)")
  52 + private String originalPicUrl;
71 53  
72   - @Schema(description = "用户操作的操作名字(MJ返回)")
73   - private String mjOperationName;
  54 + @Schema(description = "绘画错误信息")
  55 + private String errorMessage;
74 56  
75   - @Schema(description = "mj图片生产成功保存的 components json 数组")
76   - private String mjOperations;
  57 + // ============ 绘画请求参数
  58 +
  59 + /**
  60 + * - style
  61 + */
  62 + @Schema(description = "绘画请求参数")
  63 + private Map<String, Object> drawRequest;
  64 +
  65 + /**
  66 + * - mjNonceId
  67 + * - mjOperationId
  68 + * - mjOperationName
  69 + * - mjOperations
  70 + */
  71 + @Schema(description = "绘画请求响应参数")
  72 + private Map<String, Object> drawResponse;
77 73  
78 74 }
79 75  
... ...