Skip to content

Commit 9de7821

Browse files
committed
perf: Refine the Model Manager code
1 parent b0630b3 commit 9de7821

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

apps/common/event/listener_manage.py

+1-5
Original file line numberDiff line numberDiff line change
@@ -238,11 +238,8 @@ def update_status(query_set: QuerySet, taskType: TaskType, state: State):
238238
for key in params_dict:
239239
_value_ = params_dict[key]
240240
exec_sql = exec_sql.replace(key, str(_value_))
241-
lock.acquire()
242-
try:
241+
with lock:
243242
native_update(query_set, exec_sql)
244-
finally:
245-
lock.release()
246243

247244
@staticmethod
248245
def embedding_by_document(document_id, embedding_model: Embeddings, state_list=None):
@@ -272,7 +269,6 @@ def is_the_task_interrupted():
272269
ListenerManagement.update_status(QuerySet(Document).filter(id=document_id), TaskType.EMBEDDING,
273270
State.STARTED)
274271

275-
276272
# 根据段落进行向量化处理
277273
page_desc(QuerySet(Paragraph)
278274
.annotate(

0 commit comments

Comments
 (0)