昇腾社区首页
中文
注册

search

功能描述

在数据库中检索与传入的向量相似的向量。

函数原型

@abstractmethod
def search(embeddings, k, filter_dict)

参数说明

参数名

数据类型

是否必选

说明

embeddings

Union[List[List[float], List[Dict[int, float]]]

必选

欲添加的向量对象,可以为稠密向量或者稀疏向量。

k

int

可选

返回的相似向量的个数。

filter_dict

Dict

可选

由检索条件组成的字典,当前只支持对document_id进行过滤,过滤的文档id以列表形式传入,例如,需要在document_id为1,2,4这几篇文档中过滤,则传入的字典为{"document_id": [1, 2, 4]}。