ee.carlrobert.llm.client.you.completion.YouCompletionCustomModel 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.you.completion;

import java.util.Arrays;

public enum YouCompletionCustomModel {
  GPT_4("gpt_4", "GPT-4"),
  GPT_4_TURBO("gpt_4_turbo", "GPT-4 Turbo"),
  CLAUDE_INSTANT("claude_instant", "Claude Instant"),
  CLAUDE_2("claude_2", "Claude 2"),
  GEMINI_PRO("gemini_pro", "Gemini Pro"),
  ZEPHYR("zephyr", "Zephyr (uncensored)");

  private final String model;
  private final String description;

  YouCompletionCustomModel(String model, String description) {
    this.model = model;
    this.description = description;
  }

  public String getModel() {
    return model;
  }

  public String getDescription() {
    return description;
  }

  public String toString() {
    return model;
  }

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