mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-09-29 14:46:38 +08:00
71171c63f5
* ♻️ refactor: move file directory * ♻️ refactor: move file directory * ♻️ refactor: support multiple config methods * 🔥 del: remove unused code * 💩 refactor: Refactor channel management and synchronization * 💄 improve: add channel website * ✨ feat: allow recording 0 consumption
50 lines
1.1 KiB
Go
50 lines
1.1 KiB
Go
package config
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"one-api/common"
|
|
"os"
|
|
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
var (
|
|
port = flag.Int("port", 0, "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", "", "specify the log directory")
|
|
config = flag.String("config", "config.yaml", "specify the config.yaml path")
|
|
)
|
|
|
|
func flagConfig() {
|
|
flag.Parse()
|
|
|
|
if *printVersion {
|
|
fmt.Println(common.Version)
|
|
os.Exit(0)
|
|
}
|
|
|
|
if *printHelp {
|
|
help()
|
|
os.Exit(0)
|
|
}
|
|
|
|
if *port != 0 {
|
|
viper.Set("port", *port)
|
|
}
|
|
|
|
if *logDir != "" {
|
|
viper.Set("log_dir", *logDir)
|
|
}
|
|
|
|
}
|
|
|
|
func help() {
|
|
fmt.Println("One API " + common.Version + " - All in one API service for OpenAI API.")
|
|
fmt.Println("Copyright (C) 2024 MartialBE. All rights reserved.")
|
|
fmt.Println("Original copyright holder: JustSong")
|
|
fmt.Println("GitHub: https://github.com/MartialBE/one-api")
|
|
fmt.Println("Usage: one-api [--port <port>] [--log-dir <log directory>] [--config <config.yaml path>] [--version] [--help]")
|
|
}
|