entity/container.go

@@ -19,6 +19,7 @@ type ContainerData struct {
 	GetBackEndpoint string `json:"getBackEndpoint"`
 	S3DownloadUrl   string `json:"s3DownloadUrl"`
 	Size            int64  `json:"size"`
+	IsOverwrite     bool   `json:"isOverwrite"`
 	StorageType     StorageType
 }
 

models/cloudbrain.go

@@ -2051,6 +2051,7 @@ type GrampusDataset struct {
 	ReadOnly        bool   `json:"readOnly"`
 	GetBackEndpoint string `json:"getBackEndpoint"`
 	Size            int64  `json:"size"`
+	IsOverwrite     bool   `json:"isOverwrite"`
 }
 
 type CreateGrampusJobRequest struct {

services/ai_task_service/cluster/c2net.go

@@ -247,6 +247,7 @@ func convertContainer2Grampus(d entity.ContainerData) models.GrampusDataset {
 		ReadOnly:        d.ReadOnly,
 		GetBackEndpoint: d.GetBackEndpoint,
 		Size:            d.Size,
+		IsOverwrite:     d.IsOverwrite,
 	}
 }
 

services/ai_task_service/container_builder/pre_model_builder.go

@@ -104,6 +104,7 @@ func (b *PretrainModelBuilder) Build(ctx *context.CreationContext) ([]entity.Con
 			S3DownloadUrl: uploader.GetS3DownloadUrl(preTrainModelPath),
 			IsDir:         false,
 			Size:          size,
+			IsOverwrite:   true,
 		}
 		preTrainModelEntity = append(preTrainModelEntity, modelData)
 	}