package com.example.snippets_amalgamation;

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

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

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

                Context ctx = RequestHeaders.withHeader(
                    Context.current(), "openai-api-key", "<YOUR_OPENAI_API_KEY>");

                ctx.call(() -> client
                    .queryAsync(
                        QueryPoints.newBuilder()
                            .setCollectionName("{collection_name}")
                            .setQuery(
                                nearest(
                                    Document.newBuilder()
                                        .setModel("openai/text-embedding-3-large")
                                        .setText("How to bake cookies?")
                                        .putAllOptions(
                                            Map.of(
                                                "dimensions",
                                                value(512)))
                                        .build()))
                            .build())
                    .get());
        }
}