From 7d3ed1fef37863419d75d33783d3a34717c0786e Mon Sep 17 00:00:00 2001
From: cherishsince <cherishsince@aliyun.com>
Date: Sun, 28 Apr 2024 14:38:07 +0800
Subject: [PATCH] 【新增】增加open ai image 模型类型

---
 yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/imageopenai/OpenAiImageModelEnum.java | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+), 0 deletions(-)
 create mode 100644 yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/imageopenai/OpenAiImageModelEnum.java

diff --git a/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/imageopenai/OpenAiImageModelEnum.java b/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/imageopenai/OpenAiImageModelEnum.java
new file mode 100644
index 0000000..919dd11
--- /dev/null
+++ b/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/imageopenai/OpenAiImageModelEnum.java
@@ -0,0 +1,35 @@
+package cn.iocoder.yudao.framework.ai.imageopenai;
+
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+
+/**
+ * open ai
+ *
+ * @author fansili
+ * @time 2024/4/28 14:21
+ * @since 1.0
+ */
+@AllArgsConstructor
+@Getter
+public enum OpenAiImageModelEnum {
+
+    DALL_E_2("dall-e-2", "dall-e-2"),
+
+    DALL_E_3("dall-e-3", "dall-e-3")
+
+    ;
+
+    private String model;
+
+    private String name;
+
+    public static OpenAiImageModelEnum valueOfModel(String model) {
+        for (OpenAiImageModelEnum itemEnum : OpenAiImageModelEnum.values()) {
+            if (itemEnum.getModel().equals(model)) {
+                return itemEnum;
+            }
+        }
+        throw new IllegalArgumentException("Invalid MessageType value: " + model);
+    }
+}
--
libgit2 0.21.4