update task handler

models/const.go

@@ -13,9 +13,13 @@ const (
 	UseRedirect              = "Use-Redirect"
 	Prefer                   = "Prefer"
 	Async                    = "respond-async"
+	MaxExecTime              = "MaxExecTime"
 	HealthCheckAPI           = "/health-check"
 	ReplicateImageNameSuffix = "docker.agicoin.ai/agicoin"
 	READY                    = "READY"
+	ZeroHost                 = "0.0.0.0"
 	ModelPublishStatusYes    = 1
 	ModelPublishStatusNo     = 2
+	DefaultMaxExecTime       = 300
+	DefaultTaskTimer         = 2
 )

test/task_msg_test.go

@@ -111,12 +111,11 @@ func TestTaskHandler_computeTaskHandler(t1 *testing.T) {
 	}
 	for _, tt := range tests {
 		t1.Run(tt.name, func(t1 *testing.T) {
-			t := &nm.TaskHandler{
-				Wg:         tt.fields.wg,
-				LruCache:   tt.fields.lruCache,
-				DockerOp:   tt.fields.DockerOp,
-				TaskMsg:    tt.fields.TaskMsg,
-				HttpClient: tt.fields.HttpClient,
+			t := &nm.TaskWorker{
+				Wg:       tt.fields.wg,
+				LruCache: tt.fields.lruCache,
+				DockerOp: tt.fields.DockerOp,
+				TaskMsg:  tt.fields.TaskMsg,
 			}
 			tt.fields.wg.Add(1)
 			t.ComputeTaskHandler(tt.args.taskMsg)