entity/container.go

@@ -43,6 +43,7 @@ type ContainerBuildOpts struct {
 	AcceptStorageType   []StorageType
 	Uncompressed        bool
 	MKDIR               bool
+	VolumeFolder          bool
 }
 
 func (opts ContainerBuildOpts) IsStorageTypeIn(storageType StorageType) bool {

services/ai_task_service/cluster/c2net.go

@@ -90,7 +90,7 @@ func convertNoteBookReq2Grampus(req entity.CreateNoteBookTaskRequest) models.Cre
 	codePath := "/code"
 	if len(req.Tasks[0].Code) > 0 {
 		codePath = req.Tasks[0].Code[0].ContainerPath
-		if strings.Contains(codePath, "/") {
+		if strings.HasSuffix(codePath, ".zip") {
 			codePath = codePath[0:strings.LastIndex(codePath, "/")]
 		}
 	}

services/ai_task_service/container_builder/code_builder.go

@@ -59,9 +59,12 @@ func (b *CodeBuilder) Build(ctx *context.CreationContext) ([]entity.ContainerDat
 	}
 
 	var codeArchiveName, objectKey string
-	//无论代码是压缩包还是文件夹形式，挂载的都是文件夹，以默认分支命名压缩包
-
-	objectKey = remoteDir + "/"
+	if !b.Opts.Uncompressed && !b.Opts.VolumeFolder {
+		codeArchiveName = cloudbrain.DefaultBranchName + ".zip"
+		objectKey = path.Join(remoteDir, codeArchiveName)
+	} else {
+		objectKey = remoteDir + "/"
+	}
 
 	containerPath := ""
 	if opts.ContainerPath != "" {

services/ai_task_service/task/grampus_notebook_task.go

@@ -39,6 +39,7 @@ func GetGrampusNoteBookConfig(opts entity.AITaskConfigKey) *entity.AITaskBaseCon
 				ContainerPath:     codePath,
 				ReadOnly:          false,
 				AcceptStorageType: []entity.StorageType{entity.MINIO, entity.OBS},
+                VolumeFolder: true,
 			},
 			entity.ContainerDataset: {
 				ContainerPath:     datasetPath,