services/ai_task_service/schedule/model_schedule.go

@@ -205,7 +205,22 @@ func LocalMigrateOperate(jobName, computeSource string, r *models.ModelMigrateRe
 	}
 	if computeSource == models.NPUResource {
 		//因为NPU的输出会被压缩，因此需要解压+移桶
-		decompress(r.DestBucket+"/"+r.DestObjectKey, setting.Bucket+"/"+strings.TrimSuffix(r.DestObjectKey, models.ModelSuffix))
+		if strings.Contains(r.DestObjectKey, ".") {
+			decompress(r.DestBucket+"/"+r.DestObjectKey, setting.Bucket+"/"+strings.TrimSuffix(r.DestObjectKey, models.ModelSuffix))
+
+		} else { //如果是文件夹，遍历文件
+			fileInfos, err := storage.GetOneLevelObjectsUnderDir(r.DestBucket, "", r.DestObjectKey)
+			if err != nil {
+				log.Error("UpdateModelMigrateStatusByStep err. r.ID=%d step=%d err=%v", r.ID, models.BucketMoveFailed, err)
+				return err
+			}
+
+			for _, fileInfo := range fileInfos {
+				decompress(r.DestBucket+"/"+r.DestObjectKey+"/"+fileInfo.FileName, setting.Bucket+"/"+strings.TrimSuffix(r.DestObjectKey, models.ModelSuffix))
+			}
+
+		}
+
 	} else {
 		//因为调度无法指定桶，所以调度成功后我们还需要移桶
 		if setting.UseLocalMinioMigrate {