docker-compose安装PostgreSQL和pgvector向量数据库

init.sql

-- 启用pgvector扩展
CREATE EXTENSION IF NOT EXISTS vector;

-- 安装uuid扩展
CREATE EXTENSION IF NOT EXISTS "uuid-ossp";

-- 创建数据库表
CREATE TABLE IF NOT EXISTS vector_store (
id uuid DEFAULT uuid_generate_v4() PRIMARY KEY,
content text,
metadata json,
embedding vector(1536)
);

-- 创建向量索引（可选）
CREATE INDEX ON vector_store USING HNSW (embedding vector_cosine_ops);

docker-compose.yaml

version: '3.8'

services:
  postgres:
    image: ankane/pgvector
    container_name: postgres-pgvector
    environment:
      POSTGRES_DB: ai
      POSTGRES_USER: root
      POSTGRES_PASSWORD: 123456
    ports:
      - "5432:5432"
    volumes:
      - ./pgvector_data:/var/lib/postgresql/data
    restart: unless-stopped

命令

# 启动服务
docker-compose up -d

# 停止服务
docker-compose stop