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