perf: Refine the Model Manager code (#3099)

This commit is contained in:
shaohuzhang1 2025-05-16 12:37:52 +08:00 committed by GitHub
parent b0630b3ddd
commit fa4f7e99fd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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(