From 1c08718840544aef38a9fd5588a0ed10e0bcaa58 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=A1=9C=E8=8F=AF?= <huan0016@gmail.com>
Date: Fri, 28 Apr 2023 06:40:35 +0800
Subject: [PATCH] model

---
 README.md       | 9 +++++++++
 models/Model.go | 4 ++--
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 0f72185..55f3e3a 100644
--- a/README.md
+++ b/README.md
@@ -45,6 +45,15 @@ WS     | /api/{name}/{item_id}          | Websocket 連接對象
 
 對象模型:
 ```go
+type ListView struct {
+	Code     int           `json:"code"`
+	Page     int           `json:"page"`
+	PageSize int           `json:"page_size"`
+	Total    int           `json:"total"`
+	Next     bool          `json:"next"`
+	List     []interface{} `json:"list"`
+}
+
 type Model struct {
 	ID           int    `json:"id"`
 	Name         string `json:"name"`
diff --git a/models/Model.go b/models/Model.go
index 94a649f..95d082d 100644
--- a/models/Model.go
+++ b/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