update ReplicateImageNameSuffix

conf/config.go

@@ -28,6 +28,7 @@ type Config struct {
 	OssUrl                   string  `json:"oss_url"`
 	WaitLastTaskExecTime     int64   `json:"wait_last_task_exec_time"`
 	OpSys                    string  `json:"op_sys"`
+	ReplicateImageNameSuffix string  `json:"replicate_image_name_suffix"`
 }
 
 var _cfg *Config = nil

config.json

@@ -9,5 +9,6 @@
   "validator_url": "172.31.35.130:20011",
   "oss_url": "https://tmp-file.aigic.ai/api/v1/upload",
   "wait_last_task_exec_time": 60,
-  "op_sys": "linux"
+  "op_sys": "linux",
+  "replicate_image_name_suffix": "docker.aigic.ai/ai"
 }
\ No newline at end of file

models/const.go

@@ -15,7 +15,6 @@ const (
 	Async                 = "respond-async"
 	MaxExecTime           = "MaxExecTime"
 	HealthCheckAPI        = "/health-check"
-	ReplicateImageNameSuffix = "docker.agicoin.ai/agicoin"
 	READY                 = "READY"
 	ZeroHost              = "0.0.0.0"
 	ModelPublishStatusYes = 1

nm/task_handler.go

@@ -476,7 +476,7 @@ func (op *TaskOp) waitContainerRunning(handler *TaskWorker, imageId string) erro
 			if !running {
 				continue
 			}
-			if isMatch := strings.HasPrefix(op.taskCmd.ImageName, models.ReplicateImageNameSuffix); isMatch {
+			if isMatch := strings.HasPrefix(op.taskCmd.ImageName, conf.GetConfig().ReplicateImageNameSuffix); isMatch {
 				if isReqSuccess, err := op.checkContainerHealthy(internalIp, internalPort); err != nil {
 					log.WithField("err", err).Errorf("check container healthy failed")
 					return fmt.Errorf("%s-%s", "check container healthy failed", err.Error())