services/ai_task_service/task/task_base.go

@@ -289,7 +289,7 @@ func (g DefaultAITaskTemplate) Update(cloudbrainId int64) *response.BizError {
 	//二是处于PREPARING的时间超过了配置的等待时间，此时意味着异步创建任务时间过长或者出现了未知异常
 	if cloudbrain.NeedActiveStop() {
 		log.Info("AI task should active stop.cloudbrainId=%d", cloudbrainId)
-		if g.JobType == models.JobTypeDebug {
+		if g.JobType == models.JobTypeDebug || g.JobType == models.JobTypeOnlineInference {
 			err = StopAITaskByJobNameFromRemote(cloudbrain, c.QueryNoteBookByJobName, c.StopNoteBook)
 		} else {
 			err = StopAITaskByJobNameFromRemote(cloudbrain, c.QueryTrainJobByJobName, c.StopTrainJob)
@@ -307,7 +307,7 @@ func (g DefaultAITaskTemplate) Update(cloudbrainId int64) *response.BizError {
 		log.Info("AI task is preparing.No need to update from remote.cloudbrainId=%d", cloudbrainId)
 		return nil
 	}
-	if g.JobType == models.JobTypeDebug {
+	if g.JobType == models.JobTypeDebug || g.JobType == models.JobTypeOnlineInference {
 		err = UpdateAITaskFromRemote(cloudbrain, c.QueryNoteBook)
 	} else {
 		err = UpdateAITaskFromRemote(cloudbrain, c.QueryTrainJob)
@@ -435,7 +435,7 @@ func (g DefaultAITaskTemplate) GetOperationProfile(cloudbrainId int64) (*entity.
 	}
 	var s *entity.OperationProfile
 	var err error
-	if g.JobType == models.JobTypeDebug {
+	if g.JobType == models.JobTypeDebug || g.JobType == models.JobTypeOnlineInference {
 		s, err = GetOperationProfile(cloudbrainId, c.GetNoteBookOperationProfile)
 	} else {
 		s, err = GetOperationProfile(cloudbrainId, c.GetTrainJobOperationProfile)