API: Embedding Models

public class BedrockCohereEmbeddingModel extends DimensionAwareEmbeddingModel {
    public BedrockCohereEmbeddingModel(Builder builder);

    // Core methods
    public Response<List<Embedding>> embedAll(List<TextSegment> textSegments);

    // Builder
    public static Builder builder();
}

public static class Builder {
    public Builder model(Model model);
    public Builder model(String model);
    public Builder inputType(InputType inputType);
    public Builder inputType(String inputType);
    public Builder truncate(Truncate truncate);
    public Builder truncate(String truncate);
    public Builder client(BedrockRuntimeClient client);
    public Builder region(Region region);
    public Builder credentialsProvider(AwsCredentialsProvider credentialsProvider);
    public Builder maxRetries(Integer maxRetries);
    public Builder maxSegmentsPerBatch(Integer maxSegmentsPerBatch);

    public BedrockCohereEmbeddingModel build();
}

public enum Model {
    COHERE_EMBED_ENGLISH_V3("cohere.embed-english-v3"),
    COHERE_EMBED_MULTILINGUAL_V3("cohere.embed-multilingual-v3");

    public String getValue();
}

public enum InputType {
    SEARCH_DOCUMENT("search_document"),
    SEARCH_QUERY("search_query"),
    CLASSIFICATION("classification"),
    CLUSTERING("clustering");

    public String getValue();
}

public enum Truncate {
    NONE("NONE"),
    START("START"),
    END("END");

    public String getValue();
}

public class BedrockTitanEmbeddingModel extends AbstractBedrockEmbeddingModel<BedrockTitanEmbeddingResponse> {
    // Core methods
    public Response<List<Embedding>> embedAll(List<TextSegment> textSegments);
    public String getModel();
    public Integer getDimensions();
    public Boolean getNormalize();
    public Region getRegion();
    public AwsCredentialsProvider getCredentialsProvider();
    public Integer getMaxRetries();

    // Builder
    public static BedrockTitanEmbeddingModelBuilder<?, ?> builder();
}

public abstract static class BedrockTitanEmbeddingModelBuilder<B extends BedrockTitanEmbeddingModelBuilder<B, C>, C> {
    public B model(String model);
    public B dimensions(Integer dimensions);
    public B normalize(Boolean normalize);
    public B client(BedrockRuntimeClient client);
    public B region(Region region);
    public B credentialsProvider(AwsCredentialsProvider credentialsProvider);
    public B maxRetries(Integer maxRetries);

    public abstract C build();
}

public enum Types {
    TitanEmbedTextV1("amazon.titan-embed-text-v1"),
    TitanEmbedTextV2("amazon.titan-embed-text-v2:0");

    public String getValue();
}