io.quarkiverse.langchain4j.chroma.runtime.QueryResponse 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 java.util.List;
import java.util.Map;

import com.fasterxml.jackson.annotation.JsonCreator;

import io.quarkus.runtime.annotations.RegisterForReflection;

@RegisterForReflection
public class QueryResponse {

    private final List> ids;
    private final List>> embeddings;
    private final List> documents;
    private final List>> metadatas;
    private final List> distances;

    @JsonCreator
    public QueryResponse(List> ids, List>> embeddings, List> documents,
            List>> metadatas, List> distances) {
        this.ids = ids;
        this.embeddings = embeddings;
        this.documents = documents;
        this.metadatas = metadatas;
        this.distances = distances;
    }

    public List> getIds() {
        return ids;
    }

    public List>> getEmbeddings() {
        return embeddings;
    }

    public List> getDocuments() {
        return documents;
    }

    public List>> getMetadatas() {
        return metadatas;
    }

    public List> getDistances() {
        return distances;
    }
}