Feat/vdb migrate command (#2562)

Co-authored-by: jyong <jyong@dify.ai>
2024-02-26 19:47:29 +08:00 · 2024-02-26 19:47:29 +08:00 · 0620fa3094
commit 0620fa3094
parent d93288f711
5 changed files with 134 additions and 57 deletions
--- a/api/core/indexing_runner.py
+++ b/api/core/indexing_runner.py
@ -664,6 +664,7 @@ class IndexingRunner:
                )
            # load index
            index_processor.load(dataset, chunk_documents)
+            db.session.add(dataset)

            document_ids = [document.metadata['doc_id'] for document in chunk_documents]
            db.session.query(DocumentSegment).filter(