昇腾社区首页
中文
注册

add

功能描述

向量数据库添加文本索引。先将文本块embed得到向量,然后把向量存入向量数据库。

函数原型

def add(ids: List[int], embeddings: np.ndarray, document_id, docs, metadatas)

参数说明

参数名

数据类型

可选/必选

说明

ids

List[int]

必选

待添加向量的索引ID。

embeddings

ndarray

必选

numpy的数组对象。

document_id

int

可选

待添加向量所属文档的id

docs

List[str]

可选

待添加向量的文本。

metadatas

List[dict]

可选

待添加向量的文本元数据信息。

embeddings的shape必须等于2,embeddings包含的向量数量需要等于ids的长度,docs包含的文档数需要等于ids的长度,单次添加向量的总数小于1000亿。