OpenSearch 向量数据库

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

# OpenSearch 向量数据库配置
spring.ai.opensearch.vectordb.enabled=true
spring.ai.opensearch.vectordb.uri=http://localhost:9200
spring.ai.opensearch.vectordb.username=admin
spring.ai.opensearch.vectordb.password=admin
spring.ai.opensearch.vectordb.index-name=vectors

@Service
public class OpenSearchVectorService {
    private final OpenSearchVectorClient vectorClient;

    public OpenSearchVectorService(OpenSearchVectorClient 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 OpenSearchVectorStorageConfig {
    @Bean
    public VectorStorage openSearchVectorStorage(OpenSearchVectorProperties properties) {
        return new OpenSearchVectorStorage(properties);
    }
}

@Configuration
public class OpenSearchSimilaritySearchConfig {
    @Bean
    public SimilaritySearch openSearchSimilaritySearch(OpenSearchVectorProperties properties) {
        return new OpenSearchSimilaritySearch(properties);
    }
}

@Service
public class IndexService {
    private final OpenSearchVectorClient vectorClient;

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

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

@Configuration
public class IndexConfig {
    @Bean
    public IndexConfig indexConfig() {
        return IndexConfig.builder()
            .name("custom-vectors")
            .dimension(1536)
            .indexType(IndexType.HNSW)
            .metricType(MetricType.COSINE)
            .build();
    }
}

@Configuration
public class MappingConfig {
    @Bean
    public MappingConfig mappingConfig() {
        return MappingConfig.builder()
            .vectorField("vector")
            .metadataFields(Arrays.asList("title", "content"))
            .build();
    }
}

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