grail/gf-lt
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Enha: option to pass api token through env

Browse Source
This commit is contained in:
Grail Finder
2025-12-13 12:26:57 +03:00
parent 6fd6e77415
commit e481661799
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
config/config.go
View File

@@ -1,6 +1,8 @@
package config package config
import ( import (
"os"
"github.com/BurntSushi/toml" "github.com/BurntSushi/toml"
) )
@@ -84,6 +86,13 @@ func LoadConfig(fn string) (*Config, error) {
config.OpenRouterCompletionAPI: config.OpenRouterChatAPI, config.OpenRouterCompletionAPI: config.OpenRouterChatAPI,
config.OpenRouterChatAPI: config.ChatAPI, 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 // Build ApiLinks slice with only non-empty API links
// Only include DeepSeek APIs if DeepSeekToken is provided // Only include DeepSeek APIs if DeepSeekToken is provided
if config.DeepSeekToken != "" { if config.DeepSeekToken != "" {
@@ -94,7 +103,6 @@ func LoadConfig(fn string) (*Config, error) {
config.ApiLinks = append(config.ApiLinks, config.DeepSeekCompletionAPI) config.ApiLinks = append(config.ApiLinks, config.DeepSeekCompletionAPI)
} }
} }
// Only include OpenRouter APIs if OpenRouterToken is provided // Only include OpenRouter APIs if OpenRouterToken is provided
if config.OpenRouterToken != "" { if config.OpenRouterToken != "" {
if config.OpenRouterChatAPI != "" { if config.OpenRouterChatAPI != "" {
@@ -104,7 +112,6 @@ func LoadConfig(fn string) (*Config, error) {
config.ApiLinks = append(config.ApiLinks, config.OpenRouterCompletionAPI) config.ApiLinks = append(config.ApiLinks, config.OpenRouterCompletionAPI)
} }
} }
// Always include basic APIs // Always include basic APIs
if config.ChatAPI != "" { if config.ChatAPI != "" {
config.ApiLinks = append(config.ApiLinks, config.ChatAPI) config.ApiLinks = append(config.ApiLinks, config.ChatAPI)