updated aioption

Former-commit-id: ba671ba0ca

internal/scheduler/database/aiStorage.go

@@ -147,6 +147,7 @@ func (s *AiStorage) SaveAiTask(taskId int64, opt option.Option, adapterName stri
 		aiOpt.Replica = inferOpt.Replica
 		aiOpt.AdapterId = inferOpt.AdapterId
 		aiOpt.TaskType = inferOpt.ModelType
+		aiOpt.ModelName = inferOpt.ModelName
 		aiOpt.StrategyName = inferOpt.Strategy
 	}
 	// 构建主任务结构体
@@ -169,6 +170,7 @@ func (s *AiStorage) SaveAiTask(taskId int64, opt option.Option, adapterName stri
 		Replica:     int64(aiOpt.Replica),
 		JobId:       jobId,
 		TaskType:    aiOpt.TaskType,
+		ModelName:   aiOpt.ModelName,
 		Strategy:    aiOpt.StrategyName,
 		Status:      status,
 		Msg:         msg,

internal/scheduler/schedulers/option/aiOption.go

@@ -15,6 +15,7 @@ type AiOption struct {
 	ComputeCard           string
 	CodeType              string
 	AlgorithmName         string
+	ModelName             string
 
 	ImageId     string
 	SpecId      string

pkg/models/taskaimodel_gen.go

@@ -55,6 +55,7 @@ type (
 		DeletedAt   *time.Time `db:"deleted_at"`
 		Card        string     `db:"card"`
 		InferUrl    string     `db:"infer_url"`
+		ModelName   string     `db:"model_name"`
 	}
 )