com.azure.cosmos.models.CosmosVectorEmbeddingPolicy Maven / Gradle / Ivy

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.

package com.azure.cosmos.models;

import com.azure.cosmos.implementation.Constants;
import com.azure.cosmos.implementation.JsonSerializable;
import com.fasterxml.jackson.annotation.JsonProperty;

import java.util.List;

import static com.azure.cosmos.implementation.guava25.base.Preconditions.checkNotNull;

/**
 * Vector Embedding Policy
 */
public final class CosmosVectorEmbeddingPolicy {

    private JsonSerializable jsonSerializable;
    /**
     * Paths for embeddings along with path-specific settings for the item.
     */
    @JsonProperty(Constants.Properties.VECTOR_EMBEDDINGS)
    private List cosmosVectorEmbeddings;

    /**
     * Constructor
     */
    public CosmosVectorEmbeddingPolicy() {
        this.jsonSerializable = new JsonSerializable();
    }

    /**
     * Gets the paths for embeddings along with path-specific settings for the item.
     *
     * @return the paths for embeddings along with path-specific settings for the item.
     */
    public List getVectorEmbeddings() {
        return this.cosmosVectorEmbeddings;
    }

    /**
     * Sets the paths for embeddings along with path-specific settings for the item.
     *
     * @param cosmosVectorEmbeddings paths for embeddings along with path-specific settings for the item.
     */
    public void setCosmosVectorEmbeddings(List cosmosVectorEmbeddings) {
        cosmosVectorEmbeddings.forEach(embedding -> {
            checkNotNull(embedding, "Null values are not allowed in cosmosVectorEmbeddings list.");
        });
        this.cosmosVectorEmbeddings = cosmosVectorEmbeddings;
    }

}