services/ai_task_service/task/task_service.go

@@ -84,6 +84,9 @@ func buildAITaskInfo(task *models.Cloudbrain, creator *models.User, config *enti
 	if imageName == "" {
 		imageName = task.EngineName
 	}
+	if imageUrl == "" {
+		imageUrl = task.EngineName
+	}
 	if imageId == "" && task.EngineID > 0 {
 		imageId = fmt.Sprint(task.EngineID)
 	}