hhyykk/ipms-sjy
1
1
Fork 0
mirror of https://gitee.com/hhyykk/ipms-sjy.git synced 2025-07-13 10:35:07 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

【新增】AI：进一步统一 DALL、SD 的绘制实现

Browse Source
This commit is contained in:
YunaiV
2024-06-01 14:16:37 +08:00
parent 0563503102
commit 6856f5f192
34 changed files with 220 additions and 587 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
yudao-module-ai/yudao-module-ai-api/src/main/java/cn/iocoder/yudao/module/ai/AiCommonConstants.java
View File

@ -1,21 +0,0 @@
package cn.iocoder.yudao.module.ai;
/**
* ai 常用的常量
*
* @author fansili
* @time 2024/5/7 09:29
* @since 1.0
*/
public class AiCommonConstants {
/**
* 绘画 request - style
*/
public static final String DRAW_REQ_KEY_STYLE = "style";
/**
* dall size - 模板(1024x1024)
*/
public static final String DALL_SIZE_TEMPLATE = "%sx%s";
}

4
yudao-module-ai/yudao-module-ai-api/src/main/java/cn/iocoder/yudao/module/ai/api/package-info.java Normal file
View File

@ -0,0 +1,4 @@
/**
* 占位，没有特别的作用
*/
package cn.iocoder.yudao.module.ai.api;

37
yudao-module-ai/yudao-module-ai-api/src/main/java/cn/iocoder/yudao/module/ai/enums/AiChatConversationTypeEnum.java
View File

@ -1,37 +0,0 @@
package cn.iocoder.yudao.module.ai.enums;
import lombok.AllArgsConstructor;
import lombok.Getter;
/**
* 对话类型
* 创建对话、继续对话
*
* @author fansili
* @time 2024/4/14 18:15
* @since 1.0
*/
@AllArgsConstructor
@Getter
public enum AiChatConversationTypeEnum {
// roleChat、userChat
ROLE_CHAT("roleChat", "角色对话"),
USER_CHAT("userChat", "用户对话"),
;
private String type;
private String name;
public static AiChatConversationTypeEnum valueOfType(String type) {
for (AiChatConversationTypeEnum itemEnum : AiChatConversationTypeEnum.values()) {
if (itemEnum.getType().equals(type)) {
return itemEnum;
}
}
throw new IllegalArgumentException("Invalid MessageType value: " + type);
}
}

37
yudao-module-ai/yudao-module-ai-api/src/main/java/cn/iocoder/yudao/module/ai/enums/AiChatRoleCategoryEnum.java
View File

@ -1,37 +0,0 @@
package cn.iocoder.yudao.module.ai.enums;
import lombok.AllArgsConstructor;
import lombok.Getter;
/**
* 聊天role 分类
*
* @author fansili
* @time 2024/4/24 16:41
* @since 1.0
*/
@AllArgsConstructor
@Getter
public enum AiChatRoleCategoryEnum {
WRITING("writing", "写作"),
ENTERTAINMENT("entertainment", "娱乐"),
;
private String category;
private String name;
public static AiChatRoleCategoryEnum valueOfCategory(String category) {
for (AiChatRoleCategoryEnum itemEnum : AiChatRoleCategoryEnum.values()) {
if (itemEnum.getCategory().equals(category)) {
return itemEnum;
}
}
throw new IllegalArgumentException("Invalid MessageType value: " + category);
}
}

36
yudao-module-ai/yudao-module-ai-api/src/main/java/cn/iocoder/yudao/module/ai/enums/AiImagePublicStatusEnum.java
View File

@ -1,36 +0,0 @@
package cn.iocoder.yudao.module.ai.enums;
import lombok.AllArgsConstructor;
import lombok.Getter;
/**
* ai绘画 public 状态
*
* @author fansili
* @time 2024/4/28 17:05
* @since 1.0
*/
@AllArgsConstructor
@Getter
public enum AiImagePublicStatusEnum {
PRIVATE("private", "私有"),
PUBLIC("public", "公开"),
;
// TODO @fan：final 一下
private final String status;
private final String name;
public static AiImagePublicStatusEnum valueOfStatus(String status) {
for (AiImagePublicStatusEnum itemEnum : AiImagePublicStatusEnum.values()) {
if (itemEnum.getStatus().equals(status)) {
return itemEnum;
}
}
throw new IllegalArgumentException("Invalid MessageType value: " + status);
}
}

2
yudao-module-ai/yudao-module-ai-api/src/main/java/cn/iocoder/yudao/module/ai/ErrorCodeConstants.java → yudao-module-ai/yudao-module-ai-api/src/main/java/cn/iocoder/yudao/module/ai/enums/ErrorCodeConstants.java
View File

@ -1,4 +1,4 @@
package cn.iocoder.yudao.module.ai;
package cn.iocoder.yudao.module.ai.enums;
import cn.iocoder.yudao.framework.common.exception.ErrorCode;

4
yudao-module-ai/yudao-module-ai-api/src/main/java/cn/iocoder/yudao/module/ai/enums/AiImageStatusEnum.java → yudao-module-ai/yudao-module-ai-api/src/main/java/cn/iocoder/yudao/module/ai/enums/image/AiImageStatusEnum.java
View File

@ -1,4 +1,4 @@
package cn.iocoder.yudao.module.ai.enums;
package cn.iocoder.yudao.module.ai.enums.image;
import lombok.AllArgsConstructor;
import lombok.Getter;
@ -13,7 +13,7 @@ import lombok.Getter;
public enum AiImageStatusEnum {
IN_PROGRESS("10", "进行中"),
COMPLETE("20", "完成"),
SUCCESS("20", "完成"),
FAIL("30", "失败");
/**

2
yudao-module-ai/yudao-module-ai-api/src/main/java/cn/iocoder/yudao/module/ai/enums/AiModelEnum.java → yudao-module-ai/yudao-module-ai-api/src/main/java/cn/iocoder/yudao/module/ai/enums/model/AiModelEnum.java
View File

@ -1,4 +1,4 @@
package cn.iocoder.yudao.module.ai.enums;
package cn.iocoder.yudao.module.ai.enums.model;
import lombok.AllArgsConstructor;
import lombok.Getter;