Update main.go

main.go

@@ -21,12 +21,230 @@ import (
 	"sync"
 	"syscall"
 	"time"
-
+    
+    "strconv"
+    "strings"
+    "io/ioutil"
+    
+    "gopkg.in/yaml.v2"
 	"github.com/google/uuid"
 	"github.com/spf13/viper"
 	"golang.org/x/net/proxy"
 )
 
+
+
+// 配置结构体定义
+type Config struct {
+    ThinkingServices []ThinkingService `yaml:"thinking_services"`
+    Channels         map[string]Channel `yaml:"channels"`
+    Global           GlobalConfig      `yaml:"global"`
+}
+
+type ThinkingService struct {
+    ID                   int     `yaml:"id"`
+    Name                 string  `yaml:"name"`
+    Mode                 string  `yaml:"mode"`
+    Model                string  `yaml:"model"`
+    BaseURL              string  `yaml:"base_url"`
+    APIPath              string  `yaml:"api_path"`
+    APIKey               string  `yaml:"api_key"`
+    Timeout              int     `yaml:"timeout"`
+    Weight               int     `yaml:"weight"`
+    Proxy                string  `yaml:"proxy"`
+    ReasoningEffort      string  `yaml:"reasoning_effort"`
+    ReasoningFormat      string  `yaml:"reasoning_format"`
+    Temperature          float64 `yaml:"temperature"`
+    ForceStopDeepThinking bool   `yaml:"force_stop_deep_thinking"`
+}
+
+type Channel struct {
+    Name     string `yaml:"name"`
+    BaseURL  string `yaml:"base_url"`
+    APIPath  string `yaml:"api_path"`
+    Timeout  int    `yaml:"timeout"`
+    Proxy    string `yaml:"proxy"`
+}
+
+type GlobalConfig struct {
+    Log    LogConfig    `yaml:"log"`
+    Server ServerConfig `yaml:"server"`
+}
+
+type LogConfig struct {
+    Level    string     `yaml:"level"`
+    Format   string     `yaml:"format"`
+    Output   string     `yaml:"output"`
+    FilePath string     `yaml:"file_path"`
+    Debug    DebugConfig `yaml:"debug"`
+}
+
+type DebugConfig struct {
+    Enabled         bool `yaml:"enabled"`
+    PrintRequest    bool `yaml:"print_request"`
+    PrintResponse   bool `yaml:"print_response"`
+    MaxContentLength int  `yaml:"max_content_length"`
+}
+
+type ServerConfig struct {
+    Port         int    `yaml:"port"`
+    Host         string `yaml:"host"`
+    ReadTimeout  int    `yaml:"read_timeout"`
+    WriteTimeout int    `yaml:"write_timeout"`
+    IdleTimeout  int    `yaml:"idle_timeout"`
+}
+
+// 从环境变量获取字符串值，如果环境变量不存在则返回默认值
+func getEnvString(key, defaultValue string) string {
+    value := os.Getenv(key)
+    if value == "" {
+        return defaultValue
+    }
+    return value
+}
+
+// 从环境变量获取整数值，如果环境变量不存在或无法解析则返回默认值
+func getEnvInt(key string, defaultValue int) int {
+    valueStr := os.Getenv(key)
+    if valueStr == "" {
+        return defaultValue
+    }
+    
+    value, err := strconv.Atoi(valueStr)
+    if err != nil {
+        fmt.Printf("警告: 环境变量 %s 的值 '%s' 不是有效的整数，使用默认值 %d\n", 
+            key, valueStr, defaultValue)
+        return defaultValue
+    }
+    
+    return value
+}
+
+// 从环境变量获取浮点数值，如果环境变量不存在或无法解析则返回默认值
+func getEnvFloat(key string, defaultValue float64) float64 {
+    valueStr := os.Getenv(key)
+    if valueStr == "" {
+        return defaultValue
+    }
+    
+    value, err := strconv.ParseFloat(valueStr, 64)
+    if err != nil {
+        fmt.Printf("警告: 环境变量 %s 的值 '%s' 不是有效的浮点数，使用默认值 %f\n", 
+            key, valueStr, defaultValue)
+        return defaultValue
+    }
+    
+    return value
+}
+
+// 从环境变量获取布尔值，如果环境变量不存在或无法解析则返回默认值
+func getEnvBool(key string, defaultValue bool) bool {
+    valueStr := os.Getenv(key)
+    if valueStr == "" {
+        return defaultValue
+    }
+    
+    // 将字符串转换为小写以进行比较
+    valueStr = strings.ToLower(valueStr)
+    
+    // 检查常见的布尔值表示
+    if valueStr == "true" || valueStr == "1" || valueStr == "yes" || valueStr == "y" {
+        return true
+    } else if valueStr == "false" || valueStr == "0" || valueStr == "no" || valueStr == "n" {
+        return false
+    }
+    
+    fmt.Printf("警告: 环境变量 %s 的值 '%s' 不是有效的布尔值，使用默认值 %t\n", 
+        key, valueStr, defaultValue)
+    return defaultValue
+}
+
+// 加载配置文件并应用环境变量
+func LoadConfig(configPath string) (*Config, error) {
+    // 读取配置文件
+    data, err := ioutil.ReadFile(configPath)
+    if err != nil {
+        return nil, fmt.Errorf("读取配置文件错误: %v", err)
+    }
+    
+    // 解析YAML
+    var config Config
+    if err := yaml.Unmarshal(data, &config); err != nil {
+        return nil, fmt.Errorf("解析配置文件错误: %v", err)
+    }
+    
+    // 应用环境变量覆盖配置
+    applyEnvironmentVariables(&config)
+    
+    return &config, nil
+}
+
+// 应用环境变量覆盖配置
+func applyEnvironmentVariables(config *Config) {
+    // 确保至少有一个思考服务
+    if len(config.ThinkingServices) > 0 {
+        // 应用环境变量到第一个思考服务
+        ts := &config.ThinkingServices[0]
+        ts.Name = getEnvString("THINKING_SERVICE_NAME", ts.Name)
+        ts.Mode = getEnvString("THINKING_SERVICE_MODE", ts.Mode)
+        ts.Model = getEnvString("THINKING_SERVICE_MODEL", ts.Model)
+        ts.BaseURL = getEnvString("THINKING_SERVICE_BASE_URL", ts.BaseURL)
+        ts.APIKey = getEnvString("THINKING_SERVICE_API_KEY", ts.APIKey)
+        ts.Proxy = getEnvString("THINKING_SERVICE_PROXY", ts.Proxy)
+        ts.ReasoningEffort = getEnvString("THINKING_SERVICE_REASONING_EFFORT", ts.ReasoningEffort)
+        ts.ReasoningFormat = getEnvString("THINKING_SERVICE_REASONING_FORMAT", ts.ReasoningFormat)
+        ts.Timeout = getEnvInt("THINKING_SERVICE_TIMEOUT", ts.Timeout)
+        ts.Weight = getEnvInt("THINKING_SERVICE_WEIGHT", ts.Weight)
+        ts.Temperature = getEnvFloat("THINKING_SERVICE_TEMPERATURE", ts.Temperature)
+        ts.ForceStopDeepThinking = getEnvBool("THINKING_SERVICE_FORCE_STOP", ts.ForceStopDeepThinking)
+    }
+    
+    // 应用环境变量到通道配置
+    if channel, exists := config.Channels["1"]; exists {
+        channel.Name = getEnvString("CHANNEL_NAME", channel.Name)
+        channel.BaseURL = getEnvString("CHANNEL_BASE_URL", channel.BaseURL)
+        channel.Proxy = getEnvString("CHANNEL_PROXY", channel.Proxy)
+        channel.Timeout = getEnvInt("CHANNEL_TIMEOUT", channel.Timeout)
+        config.Channels["1"] = channel
+    }
+    
+    // 应用环境变量到全局配置
+    config.Global.Log.Level = getEnvString("LOG_LEVEL", config.Global.Log.Level)
+    config.Global.Log.Format = getEnvString("LOG_FORMAT", config.Global.Log.Format)
+    config.Global.Log.Output = getEnvString("LOG_OUTPUT", config.Global.Log.Output)
+    config.Global.Log.FilePath = getEnvString("LOG_FILE_PATH", config.Global.Log.FilePath)
+    
+    config.Global.Log.Debug.Enabled = getEnvBool("DEBUG_ENABLED", config.Global.Log.Debug.Enabled)
+    config.Global.Log.Debug.PrintRequest = getEnvBool("DEBUG_PRINT_REQUEST", config.Global.Log.Debug.PrintRequest)
+    config.Global.Log.Debug.PrintResponse = getEnvBool("DEBUG_PRINT_RESPONSE", config.Global.Log.Debug.PrintResponse)
+    config.Global.Log.Debug.MaxContentLength = getEnvInt("DEBUG_MAX_CONTENT_LENGTH", config.Global.Log.Debug.MaxContentLength)
+    
+    config.Global.Server.Port = getEnvInt("SERVER_PORT", config.Global.Server.Port)
+    config.Global.Server.Host = getEnvString("SERVER_HOST", config.Global.Server.Host)
+    config.Global.Server.ReadTimeout = getEnvInt("SERVER_READ_TIMEOUT", config.Global.Server.ReadTimeout)
+    config.Global.Server.WriteTimeout = getEnvInt("SERVER_WRITE_TIMEOUT", config.Global.Server.WriteTimeout)
+    config.Global.Server.IdleTimeout = getEnvInt("SERVER_IDLE_TIMEOUT", config.Global.Server.IdleTimeout)
+}
+
+// 示例主函数
+func main() {
+    // 加载配置
+    config, err := LoadConfig("config.yaml")
+    if err != nil {
+        fmt.Printf("加载配置失败: %v\n", err)
+        os.Exit(1)
+    }
+    
+    // 打印配置信息（仅用于调试）
+    fmt.Printf("已加载配置:\n")
+    fmt.Printf("思考服务名称: %s\n", config.ThinkingServices[0].Name)
+    fmt.Printf("思考服务基础URL: %s\n", config.ThinkingServices[0].BaseURL)
+    fmt.Printf("通道名称: %s\n", config.Channels["1"].Name)
+    fmt.Printf("通道基础URL: %s\n", config.Channels["1"].BaseURL)
+    
+    // 这里继续你的应用程序逻辑...
+
+
 // ---------------------- 配置结构 ----------------------
 
 type Config struct {