fix the return with wrong datatype of segment (#3525)

langgenius · Apr 16, 2024 · 9e6d4ee · 9e6d4ee
1 parent 38dd58e
commit 9e6d4ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/core/docstore/dataset_docstore.py b/api/core/docstore/dataset_docstore.py
@@ -84,7 +84,7 @@ def add_documents(
             if not isinstance(doc, Document):
                 raise ValueError("doc must be a Document")
 
-            segment_document = self.get_document(doc_id=doc.metadata['doc_id'], raise_error=False)
+            segment_document = self.get_document_segment(doc_id=doc.metadata['doc_id'])
 
             # NOTE: doc could already exist in the store, but we overwrite it
             if not allow_update and segment_document: