MariaDB 向量数据库

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

# MariaDB 向量数据库配置
spring.ai.mariadb.vectordb.enabled=true
spring.ai.mariadb.vectordb.url=jdbc:mariadb://localhost:3306/vectordb
spring.ai.mariadb.vectordb.username=root
spring.ai.mariadb.vectordb.password=password
spring.ai.mariadb.vectordb.table-name=vectors

@Service
public class MariaDBVectorService {
    private final MariaDBVectorClient vectorClient;

    public MariaDBVectorService(MariaDBVectorClient 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 MariaDBVectorStorageConfig {
    @Bean
    public VectorStorage mariaDBVectorStorage(MariaDBVectorProperties properties) {
        return new MariaDBVectorStorage(properties);
    }
}

@Configuration
public class MariaDBSimilaritySearchConfig {
    @Bean
    public SimilaritySearch mariaDBSimilaritySearch(MariaDBVectorProperties properties) {
        return new MariaDBSimilaritySearch(properties);
    }
}

@Service
@Transactional
public class TransactionalVectorService {
    private final MariaDBVectorClient vectorClient;

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

@Configuration
public class TableConfig {
    @Bean
    public TableConfig tableConfig() {
        return TableConfig.builder()
            .name("vector_table")
            .vectorColumn("vector")
            .metadataColumns(Arrays.asList("title", "content"))
            .build();
    }
}

@Configuration
public class IndexConfig {
    @Bean
    public IndexConfig indexConfig() {
        return IndexConfig.builder()
            .name("vector_index")
            .type(IndexType.VECTOR)
            .dimension(1536)
            .build();
    }
}

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