update standard task handler

nm/task_msg.go

@@ -219,35 +219,37 @@ func (t *TaskHandler) ComputeTaskHandler(taskMsg *nodeManagerV1.PushTaskMessage)
 			log.Error("received error: ", err)
 			return
 		}
-		isUseFileCache := true
-		for key, value := range taskParam.Headers {
-			if key == models.UseFileCache {
-				if value[0] == "0" {
-					isUseFileCache = false
-					break
+		if taskMsg.TaskKind == baseV1.TaskKind_StandardTask {
+			isUseFileCache := true
+			for key, value := range taskParam.Headers {
+				if key == models.UseFileCache {
+					if value[0] == "0" {
+						isUseFileCache = false
+						break
+					}
 				}
 			}
-		}
-		if isUseFileCache && readBody != nil {
-			containerResp := &models.ModelResponse{}
-			err = json.Unmarshal(readBody, &containerResp)
-			if err != nil {
-				log.WithError(err).Error("Error unmarshalling oss resp body failed")
-				return
-			}
-			if len(containerResp.Output) == 1 {
-				if utils.IsBase64ImageStr(containerResp.Output[0]) {
-					imageStr := strings.SplitN(containerResp.Output[0], ",", 2)[1]
-					queryString := utils.MatchFileCacheQueryString(taskParam.Headers, taskCmd.ImageName, t.DockerOp.ModelsInfo)
-					ossUri, err := t.uploadOSS(taskMsg.TaskId, queryString, imageStr)
-					if err != nil {
-						log.WithError(err).Error("upload image into file cache failed")
-						return
-					}
-					log.WithField("uri", ossUri).Info("upload image OSS  successful")
-					if ossUri != "" {
-						taskExecResult.TaskHttpStatusCode = models.RedirectCode
-						post.Header.Set("Location", ossUri)
+			if isUseFileCache && readBody != nil {
+				containerResp := &models.ModelResponse{}
+				err = json.Unmarshal(readBody, &containerResp)
+				if err != nil {
+					log.WithError(err).Error("Error unmarshalling oss resp body failed")
+					return
+				}
+				if len(containerResp.Output) == 1 {
+					if utils.IsBase64ImageStr(containerResp.Output[0]) {
+						imageStr := strings.SplitN(containerResp.Output[0], ",", 2)[1]
+						queryString := utils.MatchFileCacheQueryString(taskParam.Headers, taskCmd.ImageName, t.DockerOp.ModelsInfo)
+						ossUri, err := t.uploadOSS(taskMsg.TaskId, queryString, imageStr)
+						if err != nil {
+							log.WithError(err).Error("upload image into file cache failed")
+							return
+						}
+						log.WithField("uri", ossUri).Info("upload image OSS  successful")
+						if ossUri != "" {
+							taskExecResult.TaskHttpStatusCode = models.RedirectCode
+							post.Header.Set("Location", ossUri)
+						}
 					}
 				}
 			}