commit

adan1324 · Sep 30, 2024 · c46a02a · c46a02a
1 parent 39e80f0
commit c46a02a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/scripts/benchmarks/pysdk.py b/scripts/benchmarks/pysdk.py
@@ -203,6 +203,7 @@ def process_upsert_data(items: tuple):
     if rs.code != 0:
         logger.info(rs.msg)
     if len(rs.get_document_ids()) != size:
+        logger.info(rs.msg)
         logger.info(rs.get_document_ids())
     assert len(rs.get_document_ids()) == size
 

diff --git a/sdk/python/vearch/core/space.py b/sdk/python/vearch/core/space.py
@@ -90,8 +90,10 @@ def upsert(self, data: Union[List, pd.DataFrame]) -> UpsertResult:
                 documents.append(record)
         else:
             return UpsertResult(CodeType.UPSERT_DOC, "data type has error: " + err_msg)
-
-        return self.client._upsert(self.database_name, self.name, documents)
+        logger.info(f"docs:{len(documents)}")
+        result =  self.client._upsert(self.database_name, self.name, documents)
+        logger.info(f"success:{result.is_success()}, len:{len(result.document_ids)}")
+        return result
 
     def _check_data_type(
         self, data: Union[List, pd.DataFrame]