io.quarkiverse.langchain4j.chroma.runtime.QueryRequest Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of quarkus-langchain4j-chroma Show documentation

There is a newer version: 0.21.0

Show newest version

package io.quarkiverse.langchain4j.chroma.runtime;

import static java.util.Arrays.asList;
import static java.util.Collections.singletonList;

import java.util.List;

import io.quarkus.runtime.annotations.RegisterForReflection;

@RegisterForReflection
public class QueryRequest {

    private final List> queryEmbeddings;
    private final int nResults;
    private final List include = asList("metadatas", "documents", "distances", "embeddings");

    public QueryRequest(List queryEmbedding, int nResults) {
        this.queryEmbeddings = singletonList(queryEmbedding);
        this.nResults = nResults;
    }

    public List> getQueryEmbeddings() {
        return queryEmbeddings;
    }

    public int getnResults() {
        return nResults;
    }

    public List getInclude() {
        return include;
    }
}