pgvector 向量数据库

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

# pgvector 向量数据库配置
spring.ai.pgvector.vectordb.enabled=true
spring.ai.pgvector.vectordb.url=jdbc:postgresql://localhost:5432/vectordb
spring.ai.pgvector.vectordb.username=postgres
spring.ai.pgvector.vectordb.password=postgres
spring.ai.pgvector.vectordb.table-name=vectors

@Service
public class PgVectorService {
    private final PgVectorClient vectorClient;

    public PgVectorService(PgVectorClient 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 PgVectorStorageConfig {
    @Bean
    public VectorStorage pgVectorStorage(PgVectorProperties properties) {
        return new PgVectorStorage(properties);
    }
}

@Configuration
public class PgVectorSimilaritySearchConfig {
    @Bean
    public SimilaritySearch pgVectorSimilaritySearch(PgVectorProperties properties) {
        return new PgVectorSimilaritySearch(properties);
    }
}

@Service
public class TableService {
    private final PgVectorClient vectorClient;

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

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

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

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

@Configuration
public class ExtensionConfig {
    @Bean
    public ExtensionConfig extensionConfig() {
        return ExtensionConfig.builder()
            .name("vector")
            .version("0.5.1")
            .build();
    }
}

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