This commit is contained in:
2023-04-28 06:40:35 +08:00
parent 8ae72dd983
commit 1c08718840
2 changed files with 11 additions and 2 deletions

View File

@@ -45,6 +45,15 @@ WS | /api/{name}/{item_id} | Websocket 連接對象
對象模型: 對象模型:
```go ```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 { type Model struct {
ID int `json:"id"` ID int `json:"id"`
Name string `json:"name"` Name string `json:"name"`

View File

@@ -26,13 +26,13 @@ func (model *Model) Create() error {
return err return err
} }
defer db.Close() 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 { if err != nil {
log.Println(err) log.Println(err)
return err return err
} }
defer stmt.Close() 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 { if err != nil {
log.Println(err) log.Println(err)
return err return err