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

2024-04-16 17:09:15 +08:00
parent 38dd58e796
commit 9e6d4eeb92
1 changed files with 1 additions and 1 deletions
--- a/api/core/docstore/dataset_docstore.py
+++ b/api/core/docstore/dataset_docstore.py
@@ -84,7 +84,7 @@ class DatasetDocumentStore:
            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: