package com.ai.service;

import io.milvus.client.MilvusClient;
import io.milvus.grpc.DataType;
import io.milvus.param.collection.CreateCollectionParam;
import io.milvus.param.collection.FieldType;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

@Service
public class SimilarService {

    @Autowired
    private MilvusClient milvusClient;

    public void createCollection(){
        //定义字段
        FieldType spuId = FieldType.newBuilder()
                .withName("spu_id")
                .withDataType(DataType.Int64)
                .withPrimaryKey(true)
                .build();

        FieldType vectorField = FieldType.newBuilder()
                .withName("type_vector")
                .withDataType(DataType.FloatVector)
                .withDimension(1536)
                .build();

        CreateCollectionParam createParam = CreateCollectionParam.newBuilder()
                .withCollectionName("similar_collection")
                .addFieldType(spuId)
                .addFieldType(vectorField)
                .build();

        milvusClient.createCollection(createParam);
    }
}