PostgreSQL基本使用
1. 开启向量数据库
常用的RAG框架/系统都会需要用到向量数据库,来存储和检索向量数据,来提供相似度搜索等功能。PG也支持向量数据库,通过安装pgvector扩展来开启。在需要向量的数据库中执行以下SQL句即可开启pgvector扩展:
sql
## 比如当前在postgres数据库中开启pgvector扩展
CREATE EXTENSION IF NOT EXISTS pgvector;
测试pgvector扩展是否开启成功
sql
CREATE TABLE items (
id SERIAL PRIMARY KEY,
content TEXT,
embedding VECTOR(4) -- 4 维向量,按你使用的嵌入模型调整
);
INSERT INTO items (content, embedding) VALUES
('这是一只猫', '[0.12, 0.34, 0.56, 0.78]'),
('这是一条狗', '[0.11, 0.35, 0.55, 0.79]');
SELECT id, content
FROM items
ORDER BY embedding <=> '[0.1, 0.3, 0.5, 0.7]'
LIMIT 5;