Commit 7d3ed1fef37863419d75d33783d3a34717c0786e
1 parent
ca002498
【新增】增加open ai image 模型类型
Showing
1 changed file
with
35 additions
and
0 deletions
yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/imageopenai/OpenAiImageModelEnum.java
0 → 100644
| 1 | +package cn.iocoder.yudao.framework.ai.imageopenai; | ||
| 2 | + | ||
| 3 | +import lombok.AllArgsConstructor; | ||
| 4 | +import lombok.Getter; | ||
| 5 | + | ||
| 6 | +/** | ||
| 7 | + * open ai | ||
| 8 | + * | ||
| 9 | + * @author fansili | ||
| 10 | + * @time 2024/4/28 14:21 | ||
| 11 | + * @since 1.0 | ||
| 12 | + */ | ||
| 13 | +@AllArgsConstructor | ||
| 14 | +@Getter | ||
| 15 | +public enum OpenAiImageModelEnum { | ||
| 16 | + | ||
| 17 | + DALL_E_2("dall-e-2", "dall-e-2"), | ||
| 18 | + | ||
| 19 | + DALL_E_3("dall-e-3", "dall-e-3") | ||
| 20 | + | ||
| 21 | + ; | ||
| 22 | + | ||
| 23 | + private String model; | ||
| 24 | + | ||
| 25 | + private String name; | ||
| 26 | + | ||
| 27 | + public static OpenAiImageModelEnum valueOfModel(String model) { | ||
| 28 | + for (OpenAiImageModelEnum itemEnum : OpenAiImageModelEnum.values()) { | ||
| 29 | + if (itemEnum.getModel().equals(model)) { | ||
| 30 | + return itemEnum; | ||
| 31 | + } | ||
| 32 | + } | ||
| 33 | + throw new IllegalArgumentException("Invalid MessageType value: " + model); | ||
| 34 | + } | ||
| 35 | +} |