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): 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, "id": id,
data={ "vector": emb,
"id": id, "text": doc.page_content,
"vector": emb, "metadata": doc.metadata,
"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: def text_exists(self, id: str) -> bool:
cur = self._client.get(table_name=self._collection_name, ids=id) cur = self._client.get(table_name=self._collection_name, ids=id)

Loading…
Cancel
Save