search 用于在当前 Index 进行检索,支持向量检索、标量检索、标量过滤检索。
说明
Collection 数据写入/删除后,Index 数据更新时间最长滞后 20s,不能立即在 Index 检索到。
请求参数是 SearchParam,SearchParam 实例包含的参数如下表所示。
参数 | 类型 | 是否必选 | 默认值 | 参数说明 |
---|---|---|---|---|
VectorOrder | map | 否 | VectorOrder:参数 vector(与sparse_vectors组合) 和 id 两者二选一进行配置。
说明 对于使用了 hnsw-hybrid 算法的混合索引,暂时不支持基于 id 进行检索。 | |
ScalarOrder | map | 指定标量 ScalarOrder 对象,参数 fieldName 和 order 需要同时配置。
| ||
filter | map | 否 | 过滤条件,详见 filter 表达式说明。
| |
limit | int | 否 | 10 | 检索结果数量,最大5000个。 |
dense_weight | float | 否 | 0.5 | 对于混合检索,dense_weight 用于控制稠密向量在检索中的权重。范围为[0.2,1]。仅在检索的索引为混合索引时有效。 |
outputFields | list<string> | 否 | 过滤字段,指定要返回的标量或向量字段列表。
如果索引的距离方式为cosine,向量字段返回的向量是归一化后的向量。 | |
partition | string/int | 否 | "default" | 子索引名称,类型与 partitionBy 的 fieldType 一致,字段值对应 partitionBy 的 fieldValue。
|
算子 | 算子说明 | 示例 |
---|---|---|
must | 针对指定字段名生效,语义为必须在 [...] 之中,即 "must in"。 |
|
must_not | 针对指定字段名生效,语义为必须不在 [...] 之中,即 "must not in"。 |
|
range | 针对指定字段名生效,语义为必须在指定范围内。 |
|
range_out | 针对指定字段名生效,语义为必须在指定范围外。配置使用 |
|
and | 逻辑算子,针对逻辑查询需求,对多个条件取交集。 |
|
or | 逻辑算子,针对逻辑查询需求,对多个条件取并集。 |
|
// 获取指定索引,程序初始化时调用即可,无需重复调用 Index index = vikingDBService.getIndex("javaSDKTest", "test_index");
List<String> outputField = new ArrayList<>(); outputField.add("doc_id"); outputField.add("like"); outputField.add("text_vector"); HashMap<String, Object> filter = new HashMap<>(); filter.put("op", "range"); filter.put("field", "price"); filter.put("lt", 4); List<DataObject> datas = index.search(new SearchParam() .setVectorOrder(new VectorOrder().setVector(genRandomVector(12)).setSparseVector({"hello": 0.34, "world": 0.03, "!": 0.11}).build()) .setOutputFields(outputField) .setPartition(3) .setFilter(filter) .setLimit(2) .build()); System.out.println(datas); List<DataObject> datas = index.search(new SearchParam() .setVectorOrder(new VectorOrder().setId("22").build()) .setPartition(3) .build()); System.out.println(datas); List<DataObject> datas = index.search(new SearchParam() .setScalarOrder(new ScalarOrder("price", Order.Asc).build()) .setLimit(2) .setPartition(3) .build()); System.out.println(datas);
Java 调用执行上面的任务,返回 List<DataObject> 。DataObject 实例包含的属性如下表所示。
属性 | 说明 |
---|---|
id | 主键 id。 |
fields | 请求返回中的 fields 字段,是具体的数据,map 类型。 |
score | 表示找到的向量和输入的向量的匹配程度。 |