update document segment (#950)

Co-authored-by: jyong <jyong@dify.ai>
This commit is contained in:
Jyong 2023-08-22 17:59:24 +08:00 committed by GitHub
parent 78d3aa5fcd
commit 5623839c71
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 0 deletions

View File

@ -103,6 +103,7 @@ class DatesetDocumentStore:
content=doc.page_content,
word_count=len(doc.page_content),
tokens=tokens,
enabled=False,
created_by=self._user_id,
)
if 'answer' in doc.metadata and doc.metadata['answer']:

View File

@ -691,6 +691,7 @@ class IndexingRunner:
DocumentSegment.status == "indexing"
).update({
DocumentSegment.status: "completed",
DocumentSegment.enabled: True,
DocumentSegment.completed_at: datetime.datetime.utcnow()
})