【增加】autoconfig自动配置增加 style 适配。

cherishsince
1 parent 7467c241
Showing 2 changed files with 37 additions and 0 deletions
yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/config/YudaoAiAutoConfiguration.java
yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/imageopenai/OpenAiImageStyleEnum.java
@@ -100,6 +100,7 @@ public class YudaoAiAutoConfiguration {
                 new OpenAiImageOptions()
                         .setModel(openAiImageProperties.getModel())
                         .setResponseFormat(OpenAiImageOptions.ResponseFormatEnum.URL.getValue())
+                        .setStyle(openAiImageProperties.getStyle())
         );
     }
 }
 \ No newline at end of file
+package cn.iocoder.yudao.framework.ai.imageopenai;
+
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+
+/**
+ * open ai image style
+ *
+ * @author fansili
+ * @time 2024/4/28 16:15
+ * @since 1.0
+ */
+@AllArgsConstructor
+@Getter
+public enum OpenAiImageStyleEnum {
+
+    // 图像生成的风格。可为vivid（生动）或 natural（自然）。vivid会使模型偏向生成超现实和戏剧性的图像，而natural则会让模型产出更自然、不那么超现实的图像。该参数仅对dall-e-3模型有效。
+
+    VIVID("vivid", "生动"),
+    NATURAL("natural", "自然"),
+
+    ;
+
+    private String style;
+
+    private String name;
+
+    public static OpenAiImageStyleEnum valueOfStyle(String style) {
+        for (OpenAiImageStyleEnum itemEnum : OpenAiImageStyleEnum.values()) {
+            if (itemEnum.getStyle().equals(style)) {
+                return itemEnum;
+            }
+        }
+        throw new IllegalArgumentException("Invalid MessageType value: " + style);
+    }
+}