Apache Cassandra 向量数据库

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

# Apache Cassandra 向量数据库配置
spring.ai.cassandra.vectordb.enabled=true
spring.ai.cassandra.vectordb.contact-points=localhost
spring.ai.cassandra.vectordb.port=9042
spring.ai.cassandra.vectordb.keyspace=vectordb
spring.ai.cassandra.vectordb.table=vectors

@Service
public class CassandraVectorService {
    private final CassandraVectorClient vectorClient;

    public CassandraVectorService(CassandraVectorClient vectorClient) {
        this.vectorClient = vectorClient;
    }

    public void storeVector(String id, float[] vector) {
        vectorClient.store(id, vector);
    }

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

@Configuration
public class CassandraVectorStorageConfig {
    @Bean
    public VectorStorage cassandraVectorStorage(CassandraVectorProperties properties) {
        return new CassandraVectorStorage(properties);
    }
}

@Configuration
public class CassandraSimilaritySearchConfig {
    @Bean
    public SimilaritySearch cassandraSimilaritySearch(CassandraVectorProperties properties) {
        return new CassandraSimilaritySearch(properties);
    }
}

@Service
public class BatchVectorService {
    private final CassandraVectorClient vectorClient;

    public void batchStore(List<VectorDocument> documents) {
        vectorClient.batchStore(documents);
    }

    public List<List<SearchResult>> batchSearch(List<float[]> queryVectors, int k) {
        return vectorClient.batchSearch(queryVectors, k);
    }
}

@Configuration
public class ReplicationStrategyConfig {
    @Bean
    public ReplicationStrategy replicationStrategy() {
        return SimpleStrategy.builder()
            .replicationFactor(3)
            .build();
    }
}

@Configuration
public class ConsistencyLevelConfig {
    @Bean
    public ConsistencyLevel consistencyLevel() {
        return ConsistencyLevel.LOCAL_QUORUM;
    }
}

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