mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-09-18 01:26:37 +08:00
8b477d896d
- Added required packages to go.mod: `github.com/Laisky/errors/v2 v2.0.1`, `github.com/stretchr/testify v1.8.3`, `github.com/davecgh/go-spew v1.1.1`, `github.com/pmezard/go-difflib v1.0.0` - Increased the number of returned recordings to 100 in `relay-utils.go` - Refactored and simplified code in `relay-aiproxy.go` for improved message retrieval and error handling - Added new test file `relay_test.go` and various test cases for different message types - Modified functions in `group.go` and `relay.go` for improved functionality
58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
package common
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
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("One 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 init() {
|
|
// flag.Parse()
|
|
|
|
// if *PrintVersion {
|
|
// fmt.Println(Version)
|
|
// os.Exit(0)
|
|
// }
|
|
|
|
// if *PrintHelp {
|
|
// printHelp()
|
|
// os.Exit(0)
|
|
// }
|
|
|
|
if os.Getenv("SESSION_SECRET") != "" {
|
|
SessionSecret = os.Getenv("SESSION_SECRET")
|
|
}
|
|
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)
|
|
}
|
|
}
|
|
}
|
|
}
|