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]}。 |
父主题: VectorStore