pgvector:PostgreSQL 原生向量搜索扩展

文章目录

pgvector:PostgreSQL 原生向量搜索扩展

pgvector 在 GitHub 上已经拿到 21.2K Star 了。

这个开源工具专门给 PostgreSQL 加了向量相似性搜索能力,向量可以跟业务数据存在同一个库,不需要额外搭向量数据库。

1、这玩意儿是干嘛的

就是在 PostgreSQL 里加向量存储和搜索的能力,不需要单独部署额外的数据库组件。

支持精确和近似最近邻搜索,覆盖单精度、半精度、二进制、稀疏四种向量类型,内置L2距离、内积、余弦距离、L1距离、汉明距离、Jaccard距离六种算法。

所有带PostgreSQL客户端的语言都能直接用,还继承PostgreSQL本身的ACID合规、时间点恢复、联表查询等所有特性。向量规模大的场景可以用量化功能做水平扩展。

2、为什么要用它

做过RAG或者向量检索的人都知道,单独搭向量数据库的麻烦:业务数据存在关系库,向量存在向量库,两边要做数据同步,一致性很难保证,查询的时候还要跨库做关联,性能损耗大。

pgvector把向量存储和检索直接集成在PostgreSQL里,向量跟业务表存在同一个库,不需要做数据同步,支持直接跟业务表做联表查询,不用额外维护一套数据库组件,运维成本低很多。

支持HNSW和IVFFlat两种近似索引,内存占用可控,查询性能满足绝大多数生产场景需求。

3、怎么安装使用

Linux和Mac环境直接编译安装:

sh 复制代码
cd /tmp
git clone --branch v0.8.2 https://github.com/pgvector/pgvector.git
cd pgvector
make
make install

也可以用Docker、Homebrew、APT、Yum等方式一键安装,Windows环境可以用nmake编译或者Docker部署。

启用扩展:

sql 复制代码
CREATE EXTENSION vector;

建表存向量:

sql 复制代码
CREATE TABLE items (id bigserial PRIMARY KEY, embedding vector(3));

插入向量:

sql 复制代码
INSERT INTO items (embedding) VALUES ('[1,2,3]'), ('[4,5,6]');

查询最近邻:

sql 复制代码
SELECT * FROM items ORDER BY embedding <-> '[3,1,2]' LIMIT 5;

4、适合哪些场景

  • 做RAG系统、需要存向量同时关联业务数据的开发者
  • 不想单独维护向量数据库、希望降低架构复杂度的团队
  • 已经在用PostgreSQL、需要加向量检索能力的业务场景

开源地址:https://github.com/pgvector/pgvector

L、需要加向量检索能力的业务场景

开源地址:https://github.com/pgvector/pgvector

相关推荐
凌虚19 小时前
面向 MySQL 用户的 PostgreSQL 快速上手指南
数据库·后端·架构
五阿哥永琪19 小时前
MySQL中操作json的函数!
数据库·mysql·json
来者皆善19 小时前
了解Mysql优化吗?如何优化索引?
数据库·mysql
赤壁小虾21 小时前
【渲染流水线】[逐片元阶段]-[透明度测试]以UnityURP为例
java·前端·数据库
MC皮蛋侠客1 天前
SQLAlchemy 系列(七):高级建模与高效写入——批量 DML、方言与扩展
数据库·python
奈斯先生Vector1 天前
把 Midjourney 二次编辑做成生产系统:customId 能力令牌、Action Graph 与 WebUI 精修工作台
数据库·人工智能·架构·aigc·音视频·midjourney
深圳帝王星科技1 天前
SE8408 Sync Buck 100V 30A
其他
Lethehong1 天前
双擎并驱·全链路并行:KFS让TB级异构增量同步秒级到达
数据库
MC皮蛋侠客1 天前
SQLAlchemy 系列(八):AsyncIO、并发与 Web 生命周期——让每个并发任务持有自己的 Session
数据库·python
一水1 天前
Redis实战:一个AI工作流系统里的五个应用场景,从传参到限流的完整链路
数据库·redis·wpf