121 lines
4.3 KiB
Go
121 lines
4.3 KiB
Go
package common
|
||
|
||
import (
|
||
"flag"
|
||
"fmt"
|
||
"log"
|
||
"one-api/constant"
|
||
"os"
|
||
"path/filepath"
|
||
"strconv"
|
||
"time"
|
||
)
|
||
|
||
var (
|
||
Port = flag.Int("port", 3000, "the listening port")
|
||
PrintVersion = flag.Bool("version", false, "print version and exit")
|
||
PrintHelp = flag.Bool("help", false, "print help and exit")
|
||
LogDir = flag.String("log-dir", "./logs", "specify the log directory")
|
||
)
|
||
|
||
func printHelp() {
|
||
fmt.Println("New API " + Version + " - All in one API service for OpenAI API.")
|
||
fmt.Println("Copyright (C) 2023 JustSong. All rights reserved.")
|
||
fmt.Println("GitHub: https://github.com/songquanpeng/one-api")
|
||
fmt.Println("Usage: one-api [--port <port>] [--log-dir <log directory>] [--version] [--help]")
|
||
}
|
||
|
||
func InitEnv() {
|
||
flag.Parse()
|
||
|
||
if *PrintVersion {
|
||
fmt.Println(Version)
|
||
os.Exit(0)
|
||
}
|
||
|
||
if *PrintHelp {
|
||
printHelp()
|
||
os.Exit(0)
|
||
}
|
||
|
||
if os.Getenv("SESSION_SECRET") != "" {
|
||
ss := os.Getenv("SESSION_SECRET")
|
||
if ss == "random_string" {
|
||
log.Println("WARNING: SESSION_SECRET is set to the default value 'random_string', please change it to a random string.")
|
||
log.Println("警告：SESSION_SECRET被设置为默认值'random_string'，请修改为随机字符串。")
|
||
log.Fatal("Please set SESSION_SECRET to a random string.")
|
||
} else {
|
||
SessionSecret = ss
|
||
}
|
||
}
|
||
if os.Getenv("CRYPTO_SECRET") != "" {
|
||
CryptoSecret = os.Getenv("CRYPTO_SECRET")
|
||
} else {
|
||
CryptoSecret = SessionSecret
|
||
}
|
||
if os.Getenv("SQLITE_PATH") != "" {
|
||
SQLitePath = os.Getenv("SQLITE_PATH")
|
||
}
|
||
if *LogDir != "" {
|
||
var err error
|
||
*LogDir, err = filepath.Abs(*LogDir)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
if _, err := os.Stat(*LogDir); os.IsNotExist(err) {
|
||
err = os.Mkdir(*LogDir, 0777)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
}
|
||
}
|
||
|
||
// Initialize variables from constants.go that were using environment variables
|
||
DebugEnabled = os.Getenv("DEBUG") == "true"
|
||
MemoryCacheEnabled = os.Getenv("MEMORY_CACHE_ENABLED") == "true"
|
||
IsMasterNode = os.Getenv("NODE_TYPE") != "slave"
|
||
|
||
// Parse requestInterval and set RequestInterval
|
||
requestInterval, _ = strconv.Atoi(os.Getenv("POLLING_INTERVAL"))
|
||
RequestInterval = time.Duration(requestInterval) * time.Second
|
||
|
||
// Initialize variables with GetEnvOrDefault
|
||
SyncFrequency = GetEnvOrDefault("SYNC_FREQUENCY", 60)
|
||
BatchUpdateInterval = GetEnvOrDefault("BATCH_UPDATE_INTERVAL", 5)
|
||
RelayTimeout = GetEnvOrDefault("RELAY_TIMEOUT", 0)
|
||
|
||
// Initialize string variables with GetEnvOrDefaultString
|
||
GeminiSafetySetting = GetEnvOrDefaultString("GEMINI_SAFETY_SETTING", "BLOCK_NONE")
|
||
CohereSafetySetting = GetEnvOrDefaultString("COHERE_SAFETY_SETTING", "NONE")
|
||
|
||
// Initialize rate limit variables
|
||
GlobalApiRateLimitEnable = GetEnvOrDefaultBool("GLOBAL_API_RATE_LIMIT_ENABLE", true)
|
||
GlobalApiRateLimitNum = GetEnvOrDefault("GLOBAL_API_RATE_LIMIT", 180)
|
||
GlobalApiRateLimitDuration = int64(GetEnvOrDefault("GLOBAL_API_RATE_LIMIT_DURATION", 180))
|
||
|
||
GlobalWebRateLimitEnable = GetEnvOrDefaultBool("GLOBAL_WEB_RATE_LIMIT_ENABLE", true)
|
||
GlobalWebRateLimitNum = GetEnvOrDefault("GLOBAL_WEB_RATE_LIMIT", 60)
|
||
GlobalWebRateLimitDuration = int64(GetEnvOrDefault("GLOBAL_WEB_RATE_LIMIT_DURATION", 180))
|
||
|
||
initConstantEnv()
|
||
}
|
||
|
||
func initConstantEnv() {
|
||
constant.StreamingTimeout = GetEnvOrDefault("STREAMING_TIMEOUT", 120)
|
||
constant.DifyDebug = GetEnvOrDefaultBool("DIFY_DEBUG", true)
|
||
constant.MaxFileDownloadMB = GetEnvOrDefault("MAX_FILE_DOWNLOAD_MB", 20)
|
||
// ForceStreamOption 覆盖请求参数，强制返回usage信息
|
||
constant.ForceStreamOption = GetEnvOrDefaultBool("FORCE_STREAM_OPTION", true)
|
||
constant.GetMediaToken = GetEnvOrDefaultBool("GET_MEDIA_TOKEN", true)
|
||
constant.GetMediaTokenNotStream = GetEnvOrDefaultBool("GET_MEDIA_TOKEN_NOT_STREAM", true)
|
||
constant.UpdateTask = GetEnvOrDefaultBool("UPDATE_TASK", true)
|
||
constant.AzureDefaultAPIVersion = GetEnvOrDefaultString("AZURE_DEFAULT_API_VERSION", "2025-04-01-preview")
|
||
constant.GeminiVisionMaxImageNum = GetEnvOrDefault("GEMINI_VISION_MAX_IMAGE_NUM", 16)
|
||
constant.NotifyLimitCount = GetEnvOrDefault("NOTIFY_LIMIT_COUNT", 2)
|
||
constant.NotificationLimitDurationMinute = GetEnvOrDefault("NOTIFICATION_LIMIT_DURATION_MINUTE", 10)
|
||
// GenerateDefaultToken 是否生成初始令牌，默认关闭。
|
||
constant.GenerateDefaultToken = GetEnvOrDefaultBool("GENERATE_DEFAULT_TOKEN", false)
|
||
// 是否启用错误日志
|
||
constant.ErrorLogEnabled = GetEnvOrDefaultBool("ERROR_LOG_ENABLED", false)
|
||
}
|