Add init_user_credits config

cmd/httpd/httpd.go

@@ -87,7 +87,8 @@ func NewCmdHttpd() *cobra.Command {
 			indexes := index.New(ctx, milvusClient)
 
 			userz := userServ.New(userServ.Config{
-				ExtraRate: cfg.Sys.ExtraRate,
+				ExtraRate:       cfg.Sys.ExtraRate,
+				InitUserCredits: cfg.Sys.InitUserCredits,
 			}, client, users)
 			indexService := indexServ.NewService(ctx, gptHandler, indexes, userz)
 

config.example.yaml

@@ -9,6 +9,7 @@ openai:
 sys:
   secret_key: ""
   extra_rate: 0.1
+  init_user_credits: 0.1
 
 milvus:
   address: "localhost:19530"

config/config.go

@@ -49,8 +49,9 @@ type DBConfig struct {
 }
 
 type System struct {
-	ExtraRate float64 `yaml:"extra_rate"`
-	SecretKey string  `yaml:"secret_key"`
+	ExtraRate       float64 `yaml:"extra_rate"`
+	InitUserCredits float64 `yaml:"init_user_credits"`
+	SecretKey       string  `yaml:"secret_key"`
 }
 
 type Auth struct {

core/app.go

@@ -12,8 +12,8 @@ type (
 	App struct {
 		ID                 uint64 `json:"id"`
 		AppID              string `json:"app_id"`
-		AppSecret          string `json:"app_secret"`
-		SecureAppSecret    string `json:"-"`
+		AppSecret          string `gorm:"-" json:"app_secret"`
+		SecureAppSecret    string `gorm:"-" json:"-"`
 		AppSecretEncrypted string `json:"-"`
 		UserID             uint64 `json:"user_id"`
 		Name               string `json:"name"`

core/user.go

@@ -57,12 +57,12 @@ type (
 		// (
 		//   @fullName, @avatarURL,
 		//   @mixinUserID, @mixinIdentityNumber,
-		//   @lang, 1,
+		//   @lang, @credits,
 		//   @mvmPublicKey,
 		//   NOW(), NOW()
 		// )
 		// RETURNING "id"
-		CreateUser(ctx context.Context, fullName, avatarURL, mixinUserID, mixinIdentityNumber, lang, mvmPublicKey string) (uint64, error)
+		CreateUser(ctx context.Context, fullName, avatarURL, mixinUserID, mixinIdentityNumber, lang, mvmPublicKey string, credits decimal.Decimal) (uint64, error)
 
 		// UPDATE @@table
 		// 	{{set}}

service/user/user.go

@@ -29,7 +29,8 @@ func New(
 }
 
 type Config struct {
-	ExtraRate float64
+	ExtraRate       float64
+	InitUserCredits float64
 }
 
 type UserService struct {
@@ -110,7 +111,7 @@ func (s *UserService) LoginWithMixin(ctx context.Context, token, pubkey, lang st
 	// create
 	if errors.Is(err, gorm.ErrRecordNotFound) {
 		newUserId, err := s.users.CreateUser(ctx, user.FullName, user.AvatarURL, user.MixinUserID, user.MixinIdentityNumber,
-			user.Lang, user.MvmPublicKey)
+			user.Lang, user.MvmPublicKey, decimal.NewFromFloat(s.cfg.InitUserCredits))
 		if err != nil {
 			fmt.Printf("err users.Create: %v\n", err)
 			return nil, err