Supabase 向量检索配置:pgvector 表结构与 RPC 查询

Supabase 的向量检索依赖 PostgreSQL 的 pgvector 扩展。
你不需要额外安装数据库,只要在 Supabase 项目的 SQL Editor 中启用扩展、建好带 vector 列的表和索引,再通过 RPC 函数或直接 SQL 做相似度查询即可。
下面按从零配置的顺序展开,适用于 Supabase 免费版和付费版项目。

先确认扩展和表结构

登录 Supabase 控制台,进入目标项目,打开左侧 SQL Editor。
先执行以下语句启用 pgvector:

create extension if not exists vector;

如果没有报错,说明扩展可用。
接着创建一张用于存储文档向量的表,例如 documents:

create table documents (
  id bigint primary key generated always as identity,
  content text not null,
  embedding vector(1536)
);

这里 vector(1536) 表示向量维度为 1536,必须与你实际使用的嵌入模型输出维度一致。
例如 OpenAI 的 text-embedding-3-small 默认输出 1536 维,更换模型时要同步修改列定义。

建立向量索引

数据量较小时可以不建索引直接查询,但数据超过几千行后建议创建 HNSW 索引来加速近似最近邻搜索:

create index on documents
using hnsw (embedding vector_cosine_ops);

vector_cosine_ops 对应余弦距离。
如果你在查询时使用内积或欧氏距离,索引操作符类也要相应换成 vector_ip_ops 或 vector_l2_ops,否则查询可能用不上索引。

建完索引后,可以通过以下语句确认索引存在:

select indexname, indexdef
from pg_indexes
where tablename = 'documents';

编写 RPC 检索函数

Supabase 推荐把检索逻辑封装成数据库函数,然后通过客户端 RPC 调用。
在 SQL Editor 中执行:

create or replace function match_documents (
  query_embedding vector(1536),
  match_threshold float default 0.7,
  match_count int default 5
)
returns table (
  id bigint,
  content text,
  similarity float
)
language sql stable
as $$
  select
    documents.id,
    documents.content,
    1 - (documents.embedding <=> query_embedding) as similarity
  from documents
  where 1 - (documents.embedding <=> query_embedding) > match_threshold
  order by documents.embedding <=> query_embedding
  limit match_count;
$$;

函数参数 query_embedding 的维度必须和表列一致。<=> 是余弦距离操作符,1 - 距离 得到相似度,值越接近 1 表示越相似。
函数创建后,在 Supabase 客户端的 JavaScript 中这样调用:

const { data, error } = await supabase.rpc('match_documents', {
  query_embedding: embedding,
  match_threshold: 0.7,
  match_count: 5
});

注意:RPC 函数默认遵循表的行级安全策略(RLS)。
如果表启用了 RLS 且没有给当前用户配置 select 权限,函数会返回空结果或权限错误。
测试阶段可以先在表上关闭 RLS,或者为 anon / authenticated 角色创建允许读取的策略。

常见配置错误

  • 向量维度不匹配:插入数据或调用函数时报 expected 1536 dimensions, not 768,说明嵌入模型输出维度与表定义不一致,需要统一修改列类型和函数参数。
  • 索引未被使用:查询时使用 order by embedding <-> query_embedding 但索引建的是 vector_cosine_ops,操作符类不一致会导致全表扫描,建议查询和索引使用同一种距离度量。
  • RPC 调用返回空数组:先检查表里是否有数据,再确认 RLS 策略是否允许当前角色读取,最后检查 match_threshold 是否设得过高。

验证检索是否生效

在 SQL Editor 中直接插入两条测试向量并查询:

insert into documents (content, embedding)
values
  ('测试文档一', array_fill(0.1, array[1536])::vector),
  ('测试文档二', array_fill(0.2, array[1536])::vector);

select * from match_documents(
  array_fill(0.1, array[1536])::vector,
  0.5,
  2
);

预期返回两条记录,且第一条相似度更高。
如果查询结果为空,优先检查向量维度是否一致以及函数内阈值条件。

整个配置的核心在于三处对齐:表列维度、函数参数维度、实际嵌入模型输出维度。
索引操作符类与查询距离度量保持一致,再确认 RLS 权限,Supabase 向量检索就能正常跑通。

分享到:
上一篇
PostgreSQL 向量检索配置:pgvector 扩展安装与相似度查询
下一篇
Neon 向量检索怎么配:pgvector 扩展与相似度查询实操
1
系统公告

泽御云中秋国庆双节活动上线:新购8折,拼团3.99元起

尊敬的用户:
泽御云“月满中秋·礼贺国庆”双节活动现已开启,活动时间为2026年9月23日至10月10日。 活动期间可享以下福利:
1. 常规云服务器新购使用优惠码“泽御中秋国庆同乐”,符合条件的订单享8折优惠。
2. 香港精品云服务器5人拼团低至3.99元,部分4核4G套餐3人拼团年付388元,续费同价。
3. 新用户购买年付云服务器,符合活动规则可赠送2个月使用时长。
4. 老用户续费季度赠15天,续费年度赠2个月;活动期间升级配置免收配置迁移手续费。
5. 推荐好友成功下单,符合条件的推荐人可获赠7天服务器使用时长。
6. 活动期间享宕机补偿标准翻倍、简单网站迁移协助及技术工单优先处理权益。
温馨提示:优惠码不适用于拼团套餐、活动轻量产品、年付订单及续费订单;拼团套餐为独立特价活动,不与赠时类福利叠加。赠送时长不可折现、退款或跨账户转移,具体规则以活动页面说明为准。
服务中心
客服
在线客服
24小时为您服务
咨询
联系我们
联系我们,为您的业务提供专属服务。
24/7 技术支持
如果您遇到寻求进一步的帮助,请过工单与我们进行联系。
24/7 即时支持
泽御云
售前客服
泽御云
泽御云
售后客服
泽御云
技术支持
评价
您对当前页面的整体感受是否满意?
😞
非常不满意
😕
不满意
😐
一般
🙂
满意
😊
非常满意