single batch insert

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

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

Loading…
Cancel
Save