BJ_ZTEITS-Developers / parking-cloud-yudao

Commit 7d3ed1fef37863419d75d33783d3a34717c0786e

Authored by cherishsince
1 parent ca002498

【新增】增加open ai image 模型类型

Inline Side-by-side
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
  View file @7d3ed1f
  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 +}
... ...