44 lines
870 B
Go
44 lines
870 B
Go
package backend
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
type ModelConfig struct {
|
|
Key string `json:"key"`
|
|
Model string `json:"model"`
|
|
In float64 `json:"in"`
|
|
Out float64 `json:"out"`
|
|
}
|
|
|
|
func (a *App) GetModelConfig() *ModelConfig {
|
|
content, err := os.ReadFile(filepath.Join("./config", "model.json"))
|
|
if err != nil {
|
|
return &ModelConfig{}
|
|
}
|
|
resp := &ModelConfig{}
|
|
if err = json.Unmarshal(content, &resp); err != nil {
|
|
return &ModelConfig{}
|
|
}
|
|
|
|
return resp
|
|
}
|
|
|
|
func (a *App) SetModelConfig(req *ModelConfig) {
|
|
data, err := json.Marshal(req)
|
|
if err != nil {
|
|
fmt.Printf("保存模型配置失败: %v\n", err)
|
|
return
|
|
}
|
|
|
|
// 写入配置文件
|
|
configPath := filepath.Join("./config", "model.json")
|
|
if err := os.WriteFile(configPath, data, 0644); err != nil {
|
|
fmt.Printf("保存模型配置失败: %v\n", err)
|
|
return
|
|
}
|
|
}
|