_write_async should only return number of documents written

Author: davidsbatista

integrations/weaviate/src/haystack_integrations/document_stores/weaviate/document_store.py

@@ -1044,6 +1044,7 @@ async def _write_async(self, documents: list[Document], policy: DuplicatePolicy)
         """
         collection = await self.async_collection
 
+        written = 0
         duplicate_errors_ids = []
         for doc in documents:
             if not isinstance(doc, Document):
@@ -1063,14 +1064,15 @@ async def _write_async(self, documents: list[Document], policy: DuplicatePolicy)
                     vector=doc.embedding,
                 )
 
+                written += 1
             except weaviate.exceptions.UnexpectedStatusCodeError:
                 if policy == DuplicatePolicy.FAIL:
                     duplicate_errors_ids.append(doc.id)
 
         if duplicate_errors_ids:
             msg = f"IDs '{', '.join(duplicate_errors_ids)}' already exist in the document store."
             raise DuplicateDocumentError(msg)
-        return len(documents)
+        return written
 
     def write_documents(self, documents: list[Document], policy: DuplicatePolicy = DuplicatePolicy.NONE) -> int:
         """