mirror of
https://github.com/stulzq/azure-openai-proxy.git
synced 2025-12-18 23:04:19 +01:00
feat: add version output
This commit is contained in:
3
.gitignore
vendored
3
.gitignore
vendored
@@ -17,4 +17,5 @@
|
||||
.idea/
|
||||
bin/
|
||||
release/
|
||||
docker-compose.yml
|
||||
docker-compose.yml
|
||||
dist/
|
||||
|
||||
@@ -25,7 +25,7 @@ var (
|
||||
fallbackModelMapper = regexp.MustCompile(`[.:]`)
|
||||
)
|
||||
|
||||
func init() {
|
||||
func Init() {
|
||||
AzureOpenAIAPIVer = os.Getenv(constant.ENV_AZURE_OPENAI_API_VER)
|
||||
AzureOpenAIEndpoint = os.Getenv(constant.ENV_AZURE_OPENAI_ENDPOINT)
|
||||
|
||||
|
||||
23
cmd/main.go
23
cmd/main.go
@@ -2,6 +2,9 @@ package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"flag"
|
||||
"fmt"
|
||||
"github.com/stulzq/azure-openai-proxy/azure"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
@@ -12,7 +15,16 @@ import (
|
||||
"github.com/pkg/errors"
|
||||
)
|
||||
|
||||
var (
|
||||
version = ""
|
||||
buildDate = ""
|
||||
gitCommit = ""
|
||||
)
|
||||
|
||||
func main() {
|
||||
parseFlag()
|
||||
|
||||
azure.Init()
|
||||
gin.SetMode(gin.ReleaseMode)
|
||||
r := gin.Default()
|
||||
registerRoute(r)
|
||||
@@ -43,3 +55,14 @@ func runServer(srv *http.Server) {
|
||||
}
|
||||
log.Println("Server exiting")
|
||||
}
|
||||
|
||||
func parseFlag() {
|
||||
ver := flag.Bool("v", false, "version")
|
||||
flag.Parse()
|
||||
if *ver {
|
||||
fmt.Println("version:", version)
|
||||
fmt.Println("buildDate:", buildDate)
|
||||
fmt.Println("gitCommit:", gitCommit)
|
||||
os.Exit(0)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user