Update main.go

main.go

@@ -22,9 +22,6 @@ import (
 	"syscall"
 	"time"
     
-    "strconv"
-    "strings"
-    "io/ioutil"
     
     "gopkg.in/yaml.v2"
 	"github.com/google/uuid"
@@ -34,217 +31,6 @@ import (
 
 
 
-// 配置结构体定义
-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 {