blob: 9eee8e2abe96fb4d5cd3e65048444e751e797422 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
package config
import (
"log"
"os"
"strconv"
"thunderbird-ai-compose-server/crypto"
"thunderbird-ai-compose-server/types"
"github.com/joho/godotenv"
)
var Config types.ServerConfig
func init() {
godotenv.Load()
Config = types.ServerConfig{
AuthorizationKey: crypto.GenerateSecretKey(),
Port: getEnvAsInt("PORT"),
Provider: types.Provider(getEnv("PROVIDER")),
Model: getEnv("MODEL"),
APIKey: getEnv("API_KEY"),
}
if Config.Port == 0 {
Config.Port = 3000
}
if Config.Provider == "" {
Config.Provider = types.Gemini
}
if Config.Model == "" {
Config.Model = "gemini-2.5-flash"
}
if Config.APIKey == "" {
log.Fatal("API_KEY environment variable is required")
}
}
func getEnv(key string) string {
if value, exists := os.LookupEnv(key); exists {
return value
}
return ""
}
func getEnvAsInt(key string) int {
valueStr := getEnv(key)
if value, err := strconv.Atoi(valueStr); err == nil {
return value
}
return 0
}
|
