clarifai2.dto.model.EmbeddingModel Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of core Show documentation

Clarifai Java API Client

The newest version!

package clarifai2.dto.model;

import clarifai2.dto.model.output_info.EmbeddingOutputInfo;
import clarifai2.dto.prediction.Embedding;
import com.google.auto.value.AutoValue;
import org.jetbrains.annotations.NotNull;

@SuppressWarnings("NullableProblems")
@AutoValue
public abstract class EmbeddingModel extends Model {

  EmbeddingModel() {} // AutoValue instances only

  @NotNull @Override public final ModelType modelType() { return modelTypeStatic(); }
  public static ModelType modelTypeStatic() { return ModelType.EMBEDDING; }

  @SuppressWarnings("ConstantConditions")
  @NotNull
  @Override
  public final EmbeddingOutputInfo outputInfo() {
    return (EmbeddingOutputInfo) super.outputInfo();
  }

  @AutoValue.Builder
  public interface Builder extends Model.Builder {
    @NotNull @Override EmbeddingModel build();
  }
}