fix: 程序启动将正在下载的模型设置为失败状态

1Panel-dev · Mar 22, 2024 · 2d91a0f · 2d91a0f
1 parent 8ba67f0
commit 2d91a0f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/apps/common/event/__init__.py b/apps/common/event/__init__.py
@@ -6,9 +6,13 @@
     @date：2023/11/10 10:43
     @desc:
 """
+import setting.models
+from setting.models import Model
 from .listener_manage import *
 
 
 def run():
     listener_manage.ListenerManagement().run()
     QuerySet(Document).filter(status=Status.embedding).update(**{'status': Status.error})
+    QuerySet(Model).filter(status=setting.models.Status.DOWNLOAD).update(status=setting.models.Status.ERROR,
+                                                                         meta={'message': "下载程序被中断,请重试"})