1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14

# Use postgres/example user/password credentials
version: '3.0'

services:
  pgvector:
    image: pgvector/pgvector:pg16
    restart: always
    environment:
      POSTGRES_PASSWORD: 123456
    volumes:
      - ./data:/var/lib/postgresql/data
    ports:
      - 5432:5432
  

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11

# 初始化插件
CREATE EXTENSION vector;

# 新建表
CREATE TABLE items (id bigserial PRIMARY KEY, embedding vector(3));

# 插入数据
INSERT INTO items (embedding) VALUES ('[1,2,3]'), ('[4,5,6]');

# 查询
SELECT * FROM items ORDER BY embedding <-> '[3,1,2]' LIMIT 5;