gerardgajda/pgvector

CloudNativePG PostgreSQL 镜像（含pgvector扩展）

镜像概述

本镜像基于CloudNativePG构建，是一个集成了pgvector扩展的PostgreSQL数据库镜像。pgvector是PostgreSQL的开源向量扩展，提供向量数据类型和相似度搜索功能，使PostgreSQL能够高效存储和查询向量数据，适用于AI应用开发和向量数据库场景。

核心功能与特性

• 云原生支持：基于CloudNativePG构建，具备容器化部署、高可用性和可扩展性，适合云环境和Kubernetes集群
• 向量数据支持：集成pgvector扩展，提供vector数据类型，支持向量存储
• 相似度计算：支持多种向量相似度算法，包括余弦相似度、欧氏距离、内积等
• SQL兼容：可通过标准SQL语法操作向量数据，与PostgreSQL生态系统完全兼容
• 性能优化：支持向量索引（如IVFFlat、HNSW），提升大规模向量数据查询性能

使用场景与适用范围

• AI应用开发：语义搜索、推荐系统、问答系统等需要向量相似度匹配的场景
• 向量数据库构建：作为向量存储后端，存储文本嵌入、图像特征、音频特征等向量数据
• 多模态数据处理：处理包含向量特征的多模态数据，实现跨模态检索
• 科研与数据分析：需要向量计算和相似度分析的学术研究或数据分析任务

使用方法与配置说明

基本使用（Docker Run）

bash
docker run -d \
  --name cnpg-pgvector \
  -e POSTGRES_USER=myuser \
  -e POSTGRES_PASSWORD=mypassword \
  -e POSTGRES_DB=mydb \
  -p 5432:5432 \
  docker.xuanyuan.run/cloudnative-pg/pgvector:latest

Docker Compose 配置

yaml
version: '3.8'
services:
  postgres:
    image: docker.xuanyuan.run/cloudnative-pg/pgvector:latest
    container_name: cnpg-pgvector
    environment:
      POSTGRES_USER: myuser
      POSTGRES_PASSWORD: mypassword
      POSTGRES_DB: mydb
    ports:
      - "5432:5432"
    volumes:
      - postgres-data:/var/lib/postgresql/data
    restart: unless-stopped

volumes:
  postgres-data:

启用pgvector扩展

连接数据库后，执行以下SQL命令启用pgvector扩展：

sql
CREATE EXTENSION vector;

向量数据操作示例

创建含向量字段的表：

sql
CREATE TABLE items (
  id SERIAL PRIMARY KEY,
  embedding vector(3)  -- 3维向量
);

插入向量数据：

sql
INSERT INTO items (embedding) VALUES ('[1,2,3]'), ('[4,5,6]');

查询余弦相似度：

sql
SELECT id, embedding <-> '[3,2,1]' AS cosine_distance 
FROM docker.xuanyuan.run/items 
ORDER BY cosine_distance;

环境变量配置

注意事项

• 向量维度需在创建表时指定，建议根据实际需求（如文本嵌入模型输出维度）设置
• 对于大规模向量数据（百万级以上），建议创建向量索引提升查询性能
• 镜像版本需与CloudNativePG和pgvector版本兼容，建议使用官方推荐的最新稳定版