From fa4f7e99fd5ecf23db83a5f65c20ae89fdfe0a52 Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com>
Date: Fri, 16 May 2025 12:37:52 +0800
Subject: [PATCH] perf: Refine the Model Manager code (#3099)

---
 apps/common/event/listener_manage.py | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/apps/common/event/listener_manage.py b/apps/common/event/listener_manage.py
index 72d16ebb5..dd2a54a0c 100644
--- a/apps/common/event/listener_manage.py
+++ b/apps/common/event/listener_manage.py
@@ -238,11 +238,8 @@ class ListenerManagement:
         for key in params_dict:
             _value_ = params_dict[key]
             exec_sql = exec_sql.replace(key, str(_value_))
-        lock.acquire()
-        try:
+        with lock:
             native_update(query_set, exec_sql)
-        finally:
-            lock.release()
 
     @staticmethod
     def embedding_by_document(document_id, embedding_model: Embeddings, state_list=None):
@@ -272,7 +269,6 @@ class ListenerManagement:
             ListenerManagement.update_status(QuerySet(Document).filter(id=document_id), TaskType.EMBEDDING,
                                              State.STARTED)
 
-
             # 根据段落进行向量化处理
             page_desc(QuerySet(Paragraph)
                       .annotate(