diff options
Diffstat (limited to 'config/config.go')
| -rw-r--r-- | config/config.go | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/config/config.go b/config/config.go index 681757d..eef8035 100644 --- a/config/config.go +++ b/config/config.go @@ -1,6 +1,8 @@ package config import ( + "os" + "github.com/BurntSushi/toml" ) @@ -84,6 +86,13 @@ func LoadConfig(fn string) (*Config, error) { config.OpenRouterCompletionAPI: config.OpenRouterChatAPI, config.OpenRouterChatAPI: config.ChatAPI, } + // check env if keys not in config + if config.OpenRouterToken == "" { + config.OpenRouterToken = os.Getenv("OPENROUTER_API_KEY") + } + if config.DeepSeekToken == "" { + config.DeepSeekToken = os.Getenv("DEEPSEEK_API_KEY") + } // Build ApiLinks slice with only non-empty API links // Only include DeepSeek APIs if DeepSeekToken is provided if config.DeepSeekToken != "" { @@ -94,7 +103,6 @@ func LoadConfig(fn string) (*Config, error) { config.ApiLinks = append(config.ApiLinks, config.DeepSeekCompletionAPI) } } - // Only include OpenRouter APIs if OpenRouterToken is provided if config.OpenRouterToken != "" { if config.OpenRouterChatAPI != "" { @@ -104,7 +112,6 @@ func LoadConfig(fn string) (*Config, error) { config.ApiLinks = append(config.ApiLinks, config.OpenRouterCompletionAPI) } } - // Always include basic APIs if config.ChatAPI != "" { config.ApiLinks = append(config.ApiLinks, config.ChatAPI) |