searchById--向量数据库VikingDB-火山引擎

文档中心

立即注册

导航

searchById

最近更新时间：2025.04.12 14:52:49首次发布时间：2024.04.17 14:21:08

概述

searchById 用于主键 id 检索。根据主键 id，搜索与其距离最近的 limit 个向量。

说明

对于使用了 hnsw-hybrid 算法的混合索引，暂时不支持基于 id 进行检索。
Collection 数据写入/删除后，Index 数据更新时间预计20s，不能立即在 Index 检索到。
当请求参数 filter 配置时，表示混合检索；当请求参数 filter 没有配置时，表示纯向量检索。

前提条件

通过 createCollection 接口创建数据集时，定义字段 fields 已添加 vector 字段。
通过 upsertData 接口写入数据时，已写入 vector 类型的字段名称和字段值。
通过 createIndex 接口创建索引时，已创建 vectorIndex 向量索引。

请求参数

请求参数是 SearchByIdParam，SearchByIdParam 实例包含的参数如下表所示。

参数	类型	是否必选	默认值	参数说明
id	int64	是		主键 id。
filter	map	否		过滤条件，详见标量过滤默认为空，不做过滤。过滤条件包含 must、must_not、range、range_out四类查询算子，包含 and 和 or 两种对查询算子的组合。
limit	int	否	10	检索结果数量，最大5000个。
outputFields	list<string>	否		过滤字段，指定要返回的标量或向量字段列表。 outputFields 不传时，返回所有的标量字段，不返回向量字段。 outputFields 为空列表时，不返回 fields 字段。 outputFields 格式错误或者过滤字段不是 collection 里的字段时, 接口返回错误。如果索引的距离方式为cosine，向量字段返回的向量是归一化后的向量。
denseWeight	float	否	0.5	对于混合检索，dense_weight 用于控制稠密向量在检索中的权重。范围为[0.2，1]。仅在检索的索引为混合索引时有效。
partition	string/int	否	"default"	子索引名称，类型与 partitionBy 的 fieldType 一致，字段值对应 partition_by 的 fieldValue。 fieldType 为 int64，list<int64> 时，partition 输入类型为 int64。 fieldType 为 string，list<string> 时，partition 输入类型为 string，格式要求 "^[a-zA-Z0-9._]+$"。

示例

请求参数

//获取指定索引，程序初始化时调用即可，无需重复调用
Index index = vikingDBService.getIndex("javaSDKTest", "test_index");

List<String> outputField = new ArrayList<>();
outputField.add("doc_id");
outputField.add("price");
outputField.add("text_vector");
HashMap<String, Object> filter = new HashMap<>();
filter.put("op", "range");
filter.put("field", "price");
filter.put("lt", 4);
SearchByIdParam searchByIdParam = new SearchByIdParam()
                                    .setOutputFields(outputField)
                                    .setId("22")
                                    .setPartition("default");
                                    .setFilter(filter)
                                    .setLimit(2)
                                    .build();

List<DataObject> datas = index.searchById(searchByIdParam);
System.out.println(datas);

返回值

Java 调用执行上面的任务，返回 List<DataObject> 。DataObject 实例包含的属性如下表所示。

属性	说明
id	主键 id。
fields	请求返回中的 fields 字段，是具体的数据，map 类型。
score	表示找到的向量和输入的向量的匹配程度。