From 606ed0c8abf3b92d6ef625ed1abeed439b47304b Mon Sep 17 00:00:00 2001
From: Kevin Hu <kevinhu.sh@gmail.com>
Date: Mon, 3 Mar 2025 12:22:36 +0800
Subject: [PATCH] Fix: in case running KG repeatly. (#5538)

### What problem does this PR solve?

#5512

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
---
 api/db/services/document_service.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/api/db/services/document_service.py b/api/db/services/document_service.py
index 4c59271ec..70b24b8a0 100644
--- a/api/db/services/document_service.py
+++ b/api/db/services/document_service.py
@@ -382,9 +382,9 @@ class DocumentService(CommonService):
     def update_progress(cls):
         MSG = {
             "raptor": "Start RAPTOR (Recursive Abstractive Processing for Tree-Organized Retrieval).",
-            "graphrag": "Entities extraction progress",
-            "graph_resolution": "Start Graph Resolution",
-            "graph_community": "Start Graph Community Reports Generation"
+            "graphrag": "Entities",
+            "graph_resolution": "Resolution",
+            "graph_community": "Communities"
         }
         docs = cls.get_unfinished_docs()
         for d in docs: