소스 검색

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

Jyong 1 년 전
부모
커밋
9e6d4eeb92
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      api/core/docstore/dataset_docstore.py

+ 1 - 1
api/core/docstore/dataset_docstore.py

@@ -84,7 +84,7 @@ class DatasetDocumentStore:
             if not isinstance(doc, Document):
             if not isinstance(doc, Document):
                 raise ValueError("doc must be a 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
             # NOTE: doc could already exist in the store, but we overwrite it
             if not allow_update and segment_document:
             if not allow_update and segment_document: