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

init.sql

-- 安装pgvector扩展
CREATE EXTENSION IF NOT EXISTS vector;-- 创建数据库表
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/pgvectorcontainer_name: postgres-pgvectorenvironment:POSTGRES_DB: aiPOSTGRES_USER: rootPOSTGRES_PASSWORD: 123456ports:- "5432:5432"volumes:- ./pgvector_data:/var/lib/postgresql/data- ./init.sql:/docker-entrypoint-initdb.d/init.sqlrestart: unless-stoppedvolumes:pgvector_data:

命令

# 启动服务
docker-compose up -d# 停止服务
docker-compose stop