From a6e4b74d944269c2ecfc99925fca67a34eae7c21 Mon Sep 17 00:00:00 2001
From: KevinHuSh <kevinhu.sh@gmail.com>
Date: Tue, 7 May 2024 19:46:17 +0800
Subject: [PATCH] remove unused dependency (#664)

### What problem does this PR solve?

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
---
 rag/svr/task_executor.py | 2 +-
 requirements.txt         | 1 -
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/rag/svr/task_executor.py b/rag/svr/task_executor.py
index a1db2106..467ccafc 100644
--- a/rag/svr/task_executor.py
+++ b/rag/svr/task_executor.py
@@ -89,6 +89,7 @@ def set_progress(task_id, from_page=0, to_page=-1,
     except Exception as e:
         cron_logger.error("set_progress:({}), {}".format(task_id, str(e)))
 
+    close_connection()
     if cancel:
         sys.exit()
 
@@ -312,4 +313,3 @@ if __name__ == "__main__":
 
     while True:
         main()
-        close_connection()
diff --git a/requirements.txt b/requirements.txt
index 14914f38..cc57273f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -50,7 +50,6 @@ joblib==1.3.2
 lxml==5.1.0
 MarkupSafe==2.1.5
 minio==7.2.4
-mpi4py==3.1.5
 mpmath==1.3.0
 multidict==6.0.5
 multiprocess==0.70.16