com.dtsx.astra.sdk.cassio.SimilaritySearchQuery Maven / Gradle / Ivy

Go to download

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

There is a newer version: 1.2.7

package com.dtsx.astra.sdk.cassio;

import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.List;
import java.util.Map;

/**
 * Wrap query parameters as a Bean.
 */
@Data @Builder
public class SimilaritySearchQuery {

    /**
     * Maximum number of item returned
     */
    private int recordCount;

    /**
     * Minimum distance computation
     */
    private double threshold = 0.0;

    /**
     * Embeddings to be searched.
     */
    private List embeddings;

    /**
     * Default distance is cosine
     */
    private SimilarityMetric distance = SimilarityMetric.DOT_PRODUCT;

    /**
     * If provided search on metadata
     */
    private Map metaData;

}