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

相关推荐
agilearchitect2 小时前
LLaVA:开源多模态大模型的主流选择
其他
Java面试题总结2 小时前
PostgreSQL 性能调优:内存、I/O 与连接管理
数据库·postgresql
北秋,2 小时前
Oracle 数据库基础用法
数据库·oracle
CLX05052 小时前
C#怎么实现全局异常过滤器_C#如何捕获控制器报错【核心】
jvm·数据库·python
多加点辣也没关系2 小时前
设计模式-享元模式
数据库·设计模式·享元模式
牢七2 小时前
Mailvelope
数据库
阿里云瑶池数据库3 小时前
如何破解多租户隔离与性能难题?PolarDB PostgreSQL实战指南
数据库·postgresql
倔强的石头_3 小时前
异构操作系统架构下的数据库表空间高级管理:路径兼容与自动化运维实战
数据库
倒霉熊dd3 小时前
Python 学习(第二部分:函数、模块与面向对象编程)
前端·数据库·python