From 8d318d2a411d1a2a3e673e50e63de020b2e9f24f Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <shaohu.zhang@fit2cloud.com>
Date: Tue, 26 Mar 2024 16:55:41 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=B8=8B=E8=BD=BD=E6=A8=A1=E5=9E=8B?=
 =?UTF-8?q?=E5=87=BA=E9=94=99=E5=A4=84=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../serializers/provider_serializers.py       | 43 +++++++++++--------
 1 file changed, 24 insertions(+), 19 deletions(-)

diff --git a/apps/setting/serializers/provider_serializers.py b/apps/setting/serializers/provider_serializers.py
index ad4c7d265..2ac28343a 100644
--- a/apps/setting/serializers/provider_serializers.py
+++ b/apps/setting/serializers/provider_serializers.py
@@ -28,25 +28,30 @@ class ModelPullManage:
 
     @staticmethod
     def pull(model: Model, credential: Dict):
-        response = ModelProvideConstants[model.provider].value.down_model(model.model_type, model.model_name,
-                                                                          credential)
-        down_model_chunk = {}
-        timestamp = time.time()
-        for chunk in response:
-            down_model_chunk[chunk.digest] = chunk.to_dict()
-            if time.time() - timestamp > 5:
-                QuerySet(Model).filter(id=model.id).update(meta={"down_model_chunk": list(down_model_chunk.values())})
-                timestamp = time.time()
-        status = Status.ERROR
-        message = ""
-        down_model_chunk_list = list(down_model_chunk.values())
-        for chunk in down_model_chunk_list:
-            if chunk.get('status') == DownModelChunkStatus.success.value:
-                status = Status.SUCCESS
-            if chunk.get('status') == DownModelChunkStatus.error.value:
-                message = chunk.get("digest")
-        QuerySet(Model).filter(id=model.id).update(meta={"down_model_chunk": [], "message": message},
-                                                   status=status)
+        try:
+            response = ModelProvideConstants[model.provider].value.down_model(model.model_type, model.model_name,
+                                                                              credential)
+            down_model_chunk = {}
+            timestamp = time.time()
+            for chunk in response:
+                down_model_chunk[chunk.digest] = chunk.to_dict()
+                if time.time() - timestamp > 5:
+                    QuerySet(Model).filter(id=model.id).update(
+                        meta={"down_model_chunk": list(down_model_chunk.values())})
+                    timestamp = time.time()
+            status = Status.ERROR
+            message = ""
+            down_model_chunk_list = list(down_model_chunk.values())
+            for chunk in down_model_chunk_list:
+                if chunk.get('status') == DownModelChunkStatus.success.value:
+                    status = Status.SUCCESS
+                if chunk.get('status') == DownModelChunkStatus.error.value:
+                    message = chunk.get("digest")
+            QuerySet(Model).filter(id=model.id).update(meta={"down_model_chunk": [], "message": message},
+                                                       status=status)
+        except Exception as e:
+            QuerySet(Model).filter(id=model.id).update(meta={"down_model_chunk": [], "message": str(e)},
+                                                       status=Status.ERROR)
 
 
 class ModelSerializer(serializers.Serializer):