From db5563d4bc5abb54c4a3820a6bd345e448306e07 Mon Sep 17 00:00:00 2001 From: Bowen Liang Date: Wed, 25 Jun 2025 22:53:40 +0800 Subject: [PATCH] single batch insert --- .../vdb/oceanbase/oceanbase_vector.py | 21 +++++++++++-------- 1 file changed, 12 insertions(+), 9 deletions(-) diff --git a/api/core/rag/datasource/vdb/oceanbase/oceanbase_vector.py b/api/core/rag/datasource/vdb/oceanbase/oceanbase_vector.py index 667f8d8bbe..d63d901874 100644 --- a/api/core/rag/datasource/vdb/oceanbase/oceanbase_vector.py +++ b/api/core/rag/datasource/vdb/oceanbase/oceanbase_vector.py @@ -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)