-- 启用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);

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