search
功能描述
在数据库中检索与传入的向量相似的向量。
函数原型
def search(embeddings, k, filter_dict)
参数说明
参数名 |
数据类型 |
可选/必选 |
说明 |
---|---|---|---|
embeddings |
Union[List[List[float]], List[Dict[int, float]]] |
必选 |
稠密向量或稀疏向量,如果为前者则其类型为List[List[float]],如果是后者则其类型为List[Dict[int, float]]。 |
k |
int |
可选 |
返回的相似向量的个数,取值大于0,默认为“3”,取值范围(0, 10000]。 |
filter_dict |
Dict |
可选 |
由检索条件组成的字典,filter_dict包含元素的个数需小于等于32,当前只支持对document_id进行过滤,过滤的文档id以列表形式传入,例如,需要在document_id为1,2,4这几篇文档中过滤,则传入的字典为{"document_id": [1, 2, 4]} |
返回值说明
数据类型 |
说明 |
---|---|
Tuple[List[List[float]], List[List[int]]] |
返回2个数据,第一个数据表示相似向量的得分列表,第二个数据表示相似向量的ID列表。 |
父主题: OpenGaussDB类