model

models/Model.go

@@ -26,13 +26,13 @@ func (model *Model) Create() error {
 		return err
 	}
 	defer db.Close()
-	stmt, err := db.Prepare("INSERT INTO models(name, type, created_at, updated_at) values(?, ?, ?, ?)")
+	stmt, err := db.Prepare("INSERT INTO models(name, type, trigger_words, base_model, model_path, status, progress, tags, created_at, updated_at) values(?, ?, ?, ?, ?, ?, ?, ?, ?, ?)")
 	if err != nil {
 		log.Println(err)
 		return err
 	}
 	defer stmt.Close()
-	result, err := stmt.Exec(model.Name, model.Type, model.CreatedAt, model.UpdatedAt)
+	result, err := stmt.Exec(model.Name, model.Type, model.TriggerWords, model.BaseModel, model.ModelPath, model.Status, model.Progress, model.Tags, model.CreatedAt, model.UpdatedAt)
 	if err != nil {
 		log.Println(err)
 		return err