diff --git a/common/config/config.go b/common/config/config.go index 4f1c25b6..4aa247b4 100644 --- a/common/config/config.go +++ b/common/config/config.go @@ -143,6 +143,8 @@ var MetricFailChanSize = env.Int("METRIC_FAIL_CHAN_SIZE", 128) var InitialRootToken = os.Getenv("INITIAL_ROOT_TOKEN") +var InitialRootAccessToken = os.Getenv("INITIAL_ROOT_ACCESS_TOKEN") + var GeminiVersion = env.String("GEMINI_VERSION", "v1") var RelayProxy = env.String("RELAY_PROXY", "") diff --git a/model/main.go b/model/main.go index 4b5323c4..3a21cfc1 100644 --- a/model/main.go +++ b/model/main.go @@ -29,13 +29,17 @@ func CreateRootAccountIfNeed() error { if err != nil { return err } + accessToken := random.GetUUID() + if config.InitialRootAccessToken != "" { + accessToken = config.InitialRootAccessToken + } rootUser := User{ Username: "root", Password: hashedPassword, Role: RoleRootUser, Status: UserStatusEnabled, DisplayName: "Root User", - AccessToken: random.GetUUID(), + AccessToken: accessToken, Quota: 500000000000000, } DB.Create(&rootUser)