package com.example.snippets_amalgamation;

import io.qdrant.client.QdrantClient;
import io.qdrant.client.QdrantGrpcClient;
import io.qdrant.client.grpc.Collections.BinaryQuantization;
import io.qdrant.client.grpc.Collections.BinaryQuantizationEncoding;
import io.qdrant.client.grpc.Collections.CreateCollection;
import io.qdrant.client.grpc.Collections.Distance;
import io.qdrant.client.grpc.Collections.QuantizationConfig;
import io.qdrant.client.grpc.Collections.VectorParams;
import io.qdrant.client.grpc.Collections.VectorsConfig;

public class Snippet {
        public static void run() throws Exception {
                QdrantClient client =
                    new QdrantClient(QdrantGrpcClient.newBuilder("localhost", 6334, false).build());

                client
                    .createCollectionAsync(
                        CreateCollection.newBuilder()
                            .setCollectionName("{collection_name}")
                            .setVectorsConfig(
                                VectorsConfig.newBuilder()
                                    .setParams(
                                        VectorParams.newBuilder()
                                            .setSize(1536)
                                            .setDistance(Distance.Cosine)
                                            .build())
                                    .build())
                            .setQuantizationConfig(
                                QuantizationConfig.newBuilder()
                                    .setBinary(BinaryQuantization
                                        .newBuilder()
                                        .setEncoding(BinaryQuantizationEncoding.TwoBits)
                                        .setAlwaysRam(true)
                                        .build())
                                    .build())
                            .build())
                    .get();
        }
}