146 lines
3.7 KiB
Go
146 lines
3.7 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type Config struct {
|
|
HTTPHost string
|
|
HTTPPort int
|
|
DatabaseURL string
|
|
MigrateOnStart bool
|
|
APIAuthToken string
|
|
|
|
LLMBaseURL string
|
|
LLMAPIKey string
|
|
LLMModel string
|
|
LLMTimeout time.Duration
|
|
AudioBaseURL string
|
|
AudioAPIKey string
|
|
AudioModel string
|
|
AudioTimeout time.Duration
|
|
AudioPrompt string
|
|
AIStatsSidecarURL string
|
|
AIStatsTimeout time.Duration
|
|
|
|
WorkerID string
|
|
WorkerHTTPHost string
|
|
WorkerHTTPPort int
|
|
WorkerPollInterval time.Duration
|
|
WorkerClaimLimit int
|
|
WorkerLeaseTimeout time.Duration
|
|
WorkerTaskTypes []string
|
|
WorkerModelProfiles []string
|
|
}
|
|
|
|
func Load() Config {
|
|
return Config{
|
|
HTTPHost: envString("HTTP_HOST", "0.0.0.0"),
|
|
HTTPPort: envInt("HTTP_PORT", 8080),
|
|
DatabaseURL: envString("DATABASE_URL", ""),
|
|
MigrateOnStart: envBool("MIGRATE_ON_START", true),
|
|
APIAuthToken: envString("AI_SERVICE_TOKEN", ""),
|
|
|
|
LLMBaseURL: envString("LLM_BASE_URL", ""),
|
|
LLMAPIKey: envString("LLM_API_KEY", ""),
|
|
LLMModel: envString("LLM_MODEL", "qwen2.5-14b"),
|
|
LLMTimeout: envDuration("LLM_TIMEOUT", 5*time.Minute),
|
|
AudioBaseURL: envString("AUDIO_TRANSCRIPTION_BASE_URL", envString("AUDIO_LLM_BASE_URL", "")),
|
|
AudioAPIKey: envString("AUDIO_TRANSCRIPTION_API_KEY", envString("AUDIO_LLM_API_KEY", envString("LLM_API_KEY", ""))),
|
|
AudioModel: envString("AUDIO_TRANSCRIPTION_MODEL", "openai/whisper-large-v3"),
|
|
AudioTimeout: envDuration("AUDIO_TRANSCRIPTION_TIMEOUT", envDuration("AUDIO_LLM_TIMEOUT", 10*time.Minute)),
|
|
AudioPrompt: envString("AUDIO_TRANSCRIPTION_PROMPT", envString("AUDIO_LLM_PROMPT", defaultAudioPrompt())),
|
|
AIStatsSidecarURL: envString("AI_STATS_SIDECAR_URL", ""),
|
|
AIStatsTimeout: envDuration("AI_STATS_TIMEOUT", 8*time.Second),
|
|
|
|
WorkerID: envString("WORKER_ID", hostname()),
|
|
WorkerHTTPHost: envString("WORKER_HTTP_HOST", "0.0.0.0"),
|
|
WorkerHTTPPort: envInt("WORKER_HTTP_PORT", 8081),
|
|
WorkerPollInterval: envDuration("WORKER_POLL_INTERVAL", 2*time.Second),
|
|
WorkerClaimLimit: envInt("WORKER_CLAIM_LIMIT", 4),
|
|
WorkerLeaseTimeout: envDuration("WORKER_LEASE_TIMEOUT", 15*time.Minute),
|
|
WorkerTaskTypes: envCSV("WORKER_TASK_TYPES"),
|
|
WorkerModelProfiles: envCSV("WORKER_MODEL_PROFILES"),
|
|
}
|
|
}
|
|
|
|
func envString(key, fallback string) string {
|
|
if v := os.Getenv(key); v != "" {
|
|
return v
|
|
}
|
|
return fallback
|
|
}
|
|
|
|
func envInt(key string, fallback int) int {
|
|
raw := os.Getenv(key)
|
|
if raw == "" {
|
|
return fallback
|
|
}
|
|
v, err := strconv.Atoi(raw)
|
|
if err != nil {
|
|
return fallback
|
|
}
|
|
return v
|
|
}
|
|
|
|
func envBool(key string, fallback bool) bool {
|
|
raw := os.Getenv(key)
|
|
if raw == "" {
|
|
return fallback
|
|
}
|
|
v, err := strconv.ParseBool(raw)
|
|
if err != nil {
|
|
return fallback
|
|
}
|
|
return v
|
|
}
|
|
|
|
func envDuration(key string, fallback time.Duration) time.Duration {
|
|
raw := os.Getenv(key)
|
|
if raw == "" {
|
|
return fallback
|
|
}
|
|
v, err := time.ParseDuration(raw)
|
|
if err != nil {
|
|
return fallback
|
|
}
|
|
return v
|
|
}
|
|
|
|
func envCSV(key string) []string {
|
|
raw := strings.TrimSpace(os.Getenv(key))
|
|
if raw == "" {
|
|
return nil
|
|
}
|
|
parts := strings.Split(raw, ",")
|
|
out := make([]string, 0, len(parts))
|
|
for _, part := range parts {
|
|
if v := strings.TrimSpace(part); v != "" {
|
|
out = append(out, v)
|
|
}
|
|
}
|
|
return out
|
|
}
|
|
|
|
func envCSVDefault(key string, fallback []string) []string {
|
|
if values := envCSV(key); len(values) > 0 {
|
|
return values
|
|
}
|
|
return fallback
|
|
}
|
|
|
|
func defaultAudioPrompt() string {
|
|
return ""
|
|
}
|
|
|
|
func hostname() string {
|
|
h, err := os.Hostname()
|
|
if err != nil || h == "" {
|
|
return "ai-service-worker"
|
|
}
|
|
return h
|
|
}
|