zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 下载模型出错处理

Browse Source
This commit is contained in:
shaohuzhang1 2024-03-26 16:55:41 +08:00
parent b891d0612f
commit 8d318d2a41
1 changed files with 24 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
apps/setting/serializers/provider_serializers.py
View File

@ -28,25 +28,30 @@ class ModelPullManage:
@staticmethod @staticmethod
def pull(model: Model, credential: Dict): def pull(model: Model, credential: Dict):
response = ModelProvideConstants[model.provider].value.down_model(model.model_type, model.model_name, try:
credential) response = ModelProvideConstants[model.provider].value.down_model(model.model_type, model.model_name,
down_model_chunk = {} credential)
timestamp = time.time() down_model_chunk = {}
for chunk in response: timestamp = time.time()
down_model_chunk[chunk.digest] = chunk.to_dict() for chunk in response:
if time.time() - timestamp > 5: down_model_chunk[chunk.digest] = chunk.to_dict()
QuerySet(Model).filter(id=model.id).update(meta={"down_model_chunk": list(down_model_chunk.values())}) if time.time() - timestamp > 5:
timestamp = time.time() QuerySet(Model).filter(id=model.id).update(
status = Status.ERROR meta={"down_model_chunk": list(down_model_chunk.values())})
message = "" timestamp = time.time()
down_model_chunk_list = list(down_model_chunk.values()) status = Status.ERROR
for chunk in down_model_chunk_list: message = ""
if chunk.get('status') == DownModelChunkStatus.success.value: down_model_chunk_list = list(down_model_chunk.values())
status = Status.SUCCESS for chunk in down_model_chunk_list:
if chunk.get('status') == DownModelChunkStatus.error.value: if chunk.get('status') == DownModelChunkStatus.success.value:
message = chunk.get("digest") status = Status.SUCCESS
QuerySet(Model).filter(id=model.id).update(meta={"down_model_chunk": [], "message": message}, if chunk.get('status') == DownModelChunkStatus.error.value:
status=status) 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): class ModelSerializer(serializers.Serializer):