package com.example.snippets_amalgamation;

import static io.qdrant.client.QueryFactory.nearest;
import static io.qdrant.client.ValueFactory.value;

import io.qdrant.client.QdrantClient;
import io.qdrant.client.QdrantGrpcClient;
import io.qdrant.client.grpc.Points.Document;
import java.util.Map;

public class Snippet {
        public static void run() throws Exception {
                QdrantClient client =
                    new QdrantClient(
                        QdrantGrpcClient.newBuilder("xyz-example.qdrant.io", 6334, true)
                            .withApiKey("<your-api-key")
                            .build());

                client
                    .queryAsync(
                        Points.QueryPoints.newBuilder()
                            .setCollectionName("{collection_name}")
                            .setQuery(
                                nearest(
                                    Document.newBuilder()
                                        .setModel("cohere/embed-v4.0")
                                        .setText("a green square")
                                        .putAllOptions(
                                            Map.of(
                                                "cohere-api-key",
                                                value("<YOUR_COHERE_API_KEY>"),
                                                "output_dimension",
                                                value(512)))
                                        .build()))
                            .build())
                    .get();
        }
}