ee.carlrobert.llm.client.openai.imagegen.OpenAIImageGenerationModel Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of llm-client Show documentation

Java http client wrapped around the OkHttp3 library

There is a newer version: 0.8.28

package ee.carlrobert.llm.client.openai.imagegen;

import ee.carlrobert.llm.imagegen.ImageGenerationModel;
import java.util.Arrays;

public enum OpenAIImageGenerationModel implements ImageGenerationModel {
  DALL_E_2("dall-e-2", "DALL·E 2"),
  DALL_E_3("dall-e-3", "DALL·E 3");
  private final String code;
  private final String description;

  OpenAIImageGenerationModel(String code, String description) {
    this.code = code;
    this.description = description;
  }

  public String getCode() {
    return code;
  }

  public String getDescription() {
    return description;
  }


  @Override
  public String toString() {
    return description;
  }

  public static OpenAIImageGenerationModel findByCode(String code) {
    return Arrays.stream(OpenAIImageGenerationModel.values())
        .filter(item -> item.getCode().equals(code))
        .findFirst().orElseThrow();
  }
}