freedakgmail
136 lines
3.2 KiB
Go
136 lines
3.2 KiB
Go
package config
|
||
|
||
import (
|
||
"os"
|
||
"time"
|
||
)
|
||
|
||
type Config struct {
|
||
Server ServerConfig
|
||
Database DatabaseConfig
|
||
Redis RedisConfig
|
||
JWT JWTConfig
|
||
Dify DifyConfig
|
||
LLM LLMConfig
|
||
Embedding EmbeddingConfig
|
||
Gateway GatewayConfig
|
||
MinIO MinIOConfig
|
||
PPTWorker PPTWorkerConfig
|
||
}
|
||
|
||
type PPTWorkerConfig struct {
|
||
URL string // PPT Worker 微服务地址
|
||
}
|
||
|
||
type LLMConfig struct {
|
||
Provider string // "openai" or "anthropic"
|
||
OpenAIKey string
|
||
OpenAIBaseURL string
|
||
OpenAIModel string
|
||
AnthropicKey string
|
||
AnthropicBaseURL string
|
||
AnthropicModel string
|
||
}
|
||
|
||
type EmbeddingConfig struct {
|
||
APIKey string // Embedding API 密钥
|
||
BaseURL string // Embedding API 基础 URL（OpenAI 兼容格式）
|
||
Model string // 向量模型名称
|
||
Dimensions int // 向量维度
|
||
}
|
||
|
||
type ServerConfig struct {
|
||
Host string
|
||
Port string
|
||
}
|
||
|
||
type DatabaseConfig struct {
|
||
URL string
|
||
}
|
||
|
||
type RedisConfig struct {
|
||
URL string
|
||
}
|
||
|
||
type JWTConfig struct {
|
||
Secret string
|
||
AccessExpiry time.Duration
|
||
RefreshExpiry time.Duration
|
||
}
|
||
|
||
type DifyConfig struct {
|
||
APIURL string
|
||
APIKey string
|
||
}
|
||
|
||
type GatewayConfig struct {
|
||
URL string
|
||
}
|
||
|
||
type MinIOConfig struct {
|
||
Endpoint string
|
||
AccessKey string
|
||
SecretKey string
|
||
Bucket string
|
||
UseSSL bool
|
||
}
|
||
|
||
func Load() *Config {
|
||
return &Config{
|
||
Server: ServerConfig{
|
||
Host: getEnv("SERVER_HOST", "0.0.0.0"),
|
||
Port: getEnv("SERVER_PORT", "8080"),
|
||
},
|
||
Database: DatabaseConfig{
|
||
URL: getEnv("DATABASE_URL", "postgres://localhost:5432/govai_portal?sslmode=disable"),
|
||
},
|
||
Redis: RedisConfig{
|
||
URL: getEnv("REDIS_URL", "redis://localhost:6379"),
|
||
},
|
||
JWT: JWTConfig{
|
||
Secret: getEnv("JWT_SECRET", "dev-secret-change-in-production"),
|
||
AccessExpiry: 24 * time.Hour,
|
||
RefreshExpiry: 7 * 24 * time.Hour,
|
||
},
|
||
Dify: DifyConfig{
|
||
APIURL: getEnv("DIFY_API_URL", "http://localhost:5001/v1"),
|
||
APIKey: getEnv("DIFY_API_KEY", ""),
|
||
},
|
||
LLM: LLMConfig{
|
||
Provider: getEnv("LLM_PROVIDER", "openai"),
|
||
OpenAIKey: getEnv("OPENAI_API_KEY", ""),
|
||
OpenAIBaseURL: getEnv("OPENAI_BASE_URL", "https://dashscope.aliyuncs.com/compatible-mode/v1"),
|
||
OpenAIModel: getEnv("OPENAI_MODEL", "qwen-plus"),
|
||
AnthropicKey: getEnv("ANTHROPIC_API_KEY", ""),
|
||
AnthropicBaseURL: getEnv("ANTHROPIC_BASE_URL", "https://api.anthropic.com"),
|
||
AnthropicModel: getEnv("ANTHROPIC_MODEL", "claude-sonnet-4-20250514"),
|
||
},
|
||
Embedding: EmbeddingConfig{
|
||
APIKey: getEnv("EMBEDDING_API_KEY", ""),
|
||
BaseURL: getEnv("EMBEDDING_BASE_URL", "https://dashscope.aliyuncs.com/compatible-mode/v1"),
|
||
Model: getEnv("EMBEDDING_MODEL", "text-embedding-v3"),
|
||
Dimensions: 1024,
|
||
},
|
||
Gateway: GatewayConfig{
|
||
URL: getEnv("MODEL_GATEWAY_URL", "http://localhost:8081"),
|
||
},
|
||
MinIO: MinIOConfig{
|
||
Endpoint: getEnv("MINIO_ENDPOINT", "localhost:9000"),
|
||
AccessKey: getEnv("MINIO_ACCESS_KEY", "minioadmin"),
|
||
SecretKey: getEnv("MINIO_SECRET_KEY", "minioadmin"),
|
||
Bucket: getEnv("MINIO_BUCKET", "aily-files"),
|
||
UseSSL: false,
|
||
},
|
||
PPTWorker: PPTWorkerConfig{
|
||
URL: getEnv("PPT_WORKER_URL", "http://localhost:8090"),
|
||
},
|
||
}
|
||
}
|
||
|
||
func getEnv(key, fallback string) string {
|
||
if v := os.Getenv(key); v != "" {
|
||
return v
|
||
}
|
||
return fallback
|
||
}
|