Milvus 向量数据库

<dependency>
    <groupId>org.springframework.ai</groupId>
    <artifactId>spring-ai-milvus-vectordb</artifactId>
    <version>${spring-ai.version}</version>
</dependency>

# Milvus 向量数据库配置
spring.ai.milvus.vectordb.enabled=true
spring.ai.milvus.vectordb.host=localhost
spring.ai.milvus.vectordb.port=19530
spring.ai.milvus.vectordb.collection-name=vectors

@Service
public class MilvusVectorService {
    private final MilvusVectorClient vectorClient;

    public MilvusVectorService(MilvusVectorClient vectorClient) {
        this.vectorClient = vectorClient;
    }

    public void storeVector(String id, float[] vector, Map<String, Object> metadata) {
        vectorClient.store(id, vector, metadata);
    }

    public List<SearchResult> searchSimilar(float[] queryVector, int k) {
        return vectorClient.search(queryVector, k);
    }
}

@Configuration
public class MilvusVectorStorageConfig {
    @Bean
    public VectorStorage milvusVectorStorage(MilvusVectorProperties properties) {
        return new MilvusVectorStorage(properties);
    }
}

@Configuration
public class MilvusSimilaritySearchConfig {
    @Bean
    public SimilaritySearch milvusSimilaritySearch(MilvusVectorProperties properties) {
        return new MilvusSimilaritySearch(properties);
    }
}

@Service
public class CollectionService {
    private final MilvusVectorClient vectorClient;

    public void createCollection(String name, int dimension) {
        vectorClient.createCollection(name, dimension);
    }

    public void dropCollection(String name) {
        vectorClient.dropCollection(name);
    }
}

@Configuration
public class CollectionConfig {
    @Bean
    public CollectionConfig collectionConfig() {
        return CollectionConfig.builder()
            .name("custom-collection")
            .dimension(1536)
            .indexType(IndexType.IVF_FLAT)
            .metricType(MetricType.COSINE)
            .build();
    }
}

@Configuration
public class PartitionConfig {
    @Bean
    public PartitionConfig partitionConfig() {
        return PartitionConfig.builder()
            .name("custom-partition")
            .build();
    }
}

management.endpoints.web.exposure.include=milvus-vectordb
management.endpoint.milvus-vectordb.enabled=true