feat 存档

backend/model.go

@@ -0,0 +1,43 @@
+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
+	}
+}