diff --git a/.gitignore b/.gitignore index f884f1a..f910715 100644 --- a/.gitignore +++ b/.gitignore @@ -17,4 +17,5 @@ .idea/ bin/ release/ -docker-compose.yml \ No newline at end of file +docker-compose.yml +dist/ diff --git a/azure/init.go b/azure/init.go index 013428b..c9157c9 100644 --- a/azure/init.go +++ b/azure/init.go @@ -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) diff --git a/cmd/main.go b/cmd/main.go index 87c6b89..607c4db 100644 --- a/cmd/main.go +++ b/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) + } +}