single batch insert

pull/21617/head
Bowen Liang 11 months ago
parent cf50a7436a
commit db5563d4bc

@ -152,16 +152,19 @@ class OceanBaseVector(BaseVector):
def add_texts(self, documents: list[Document], embeddings: list[list[float]], **kwargs):
ids = self._get_uuids(documents)
data_list = []
for id, doc, emb in zip(ids, documents, embeddings):
self._client.insert(
table_name=self._collection_name,
data={
"id": id,
"vector": emb,
"text": doc.page_content,
"metadata": doc.metadata,
},
)
data_list.append({
"id": id,
"vector": emb,
"text": doc.page_content,
"metadata": doc.metadata,
})
self._client.insert(
table_name=self._collection_name,
data=data_list,
)
def text_exists(self, id: str) -> bool:
cur = self._client.get(table_name=self._collection_name, ids=id)

Loading…
Cancel
Save