fix: 模型对话异常,模型编辑时状态未改正

apps/application/serializers/chat_message_serializers.py

@@ -162,9 +162,9 @@ class ChatMessageSerializer(serializers.Serializer):
         model = QuerySet(Model).filter(id=model.id).first()
         if model is None:
             raise AppApiException(500, "模型不存在")
-            if model == Status.ERROR:
+        if model.status == Status.ERROR:
             raise AppApiException(500, "当前模型不可用")
-            if model == Status.DOWNLOAD:
+        if model.status == Status.DOWNLOAD:
             raise AppApiException(500, "模型正在下载中,请稍后再发起对话")
         return chat_info
 

apps/setting/serializers/provider_serializers.py

@@ -241,6 +241,7 @@ class ModelSerializer(serializers.Serializer):
                         model.status = Status.DOWNLOAD
                     else:
                         raise e
+                model.status = Status.SUCCESS
                 update_keys = ['credential', 'name', 'model_type', 'model_name']
                 for update_key in update_keys:
                     if update_key in instance and instance.get(update_key) is not None: