add conditon task kind

nm/start.go

@@ -202,7 +202,6 @@ func handlerStandardTask(nodeManager *models.NodeManagerClient,
 				}
 				if !taskMsgWorker.IsExecAiTask {
 					msgRespWorker.RegisterMsgResp(nodeManager, worker, FetchStandardTaskResp, nil)
-					taskMsgWorker.IsExecStandardTask = true
 					break
 				}
 			}

nm/task_msg.go

@@ -105,6 +105,7 @@ func (t *TaskHandler) ComputeTaskHandler(taskMsg *nodeManagerV1.PushTaskMessage)
 		return
 	}
 	log.Info("received task cmd :", taskCmd)
+
 	if taskMsg.TaskKind == baseV1.TaskKind_ComputeTask {
 		t.IsExecAiTask = true
 		if t.IsExecStandardTask {
@@ -117,6 +118,8 @@ func (t *TaskHandler) ComputeTaskHandler(taskMsg *nodeManagerV1.PushTaskMessage)
 			}
 			t.IsExecStandardTask = false
 		}
+	} else if taskMsg.TaskKind == baseV1.TaskKind_StandardTask {
+		t.IsExecStandardTask = true
 	}
 	images, err := t.DockerOp.PsImages()
 	if err != nil {
@@ -215,6 +218,8 @@ func (t *TaskHandler) ComputeTaskHandler(taskMsg *nodeManagerV1.PushTaskMessage)
 	}
 	if taskMsg.TaskKind == baseV1.TaskKind_ComputeTask {
 		t.IsExecAiTask = false
+	} else if taskMsg.TaskKind == baseV1.TaskKind_StandardTask {
+		t.IsExecStandardTask = false
 	}
 	log.Info("received computeTask--------------------------------")
 }