Files
ai/routers/params.go
2023-06-05 10:22:37 +08:00

35 lines
929 B
Go

package routers
import (
"main/models"
"main/utils"
"net/http"
)
func ParamsListGet(w http.ResponseWriter, r *http.Request) {
var listview models.ListView
var item = make(map[string]interface{})
item["id"] = "model"
item["name"] = "模型"
listview.List = append(listview.List, item)
listview.Total = 1
listview.Page = 1
listview.PageSize = 10
listview.Next = false
w.Header().Set("Content-Type", "application/json; charset=utf-8")
w.Write(utils.ToJSON(listview))
}
func ParamsModelsGet(w http.ResponseWriter, r *http.Request) {
params := make(map[string]interface{})
params["type"] = []string{"lora", "ckp", "hyper", "ti"}
params["status"] = []string{"pending", "running", "finished", "failed"}
params["base_model"] = []string{"SD1.5", "SD2"}
params["size"] = []string{"512x512", "768x768", "1024x768"}
w.Header().Set("Content-Type", "application/json; charset=utf-8")
w.Write(utils.ToJSON(params))
}