Bläddra i källkod

update document segment (#950)

Co-authored-by: jyong <jyong@dify.ai>
Jyong 1 år sedan
förälder
incheckning
5623839c71
2 ändrade filer med 2 tillägg och 0 borttagningar
  1. 1 0
      api/core/docstore/dataset_docstore.py
  2. 1 0
      api/core/indexing_runner.py

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

@@ -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']:

+ 1 - 0
api/core/indexing_runner.py

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