mirror of
				https://github.com/songquanpeng/one-api.git
				synced 2025-10-31 22:03:41 +08:00 
			
		
		
		
	Compare commits
	
		
			10 Commits
		
	
	
		
			v0.4.4
			...
			v0.4.5-alp
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
|  | c5837c3bb7 | ||
|  | eb70b84665 | ||
|  | a909972313 | ||
|  | 6855d0dc39 | ||
|  | a43b1e2add | ||
|  | 46c43396d8 | ||
|  | 6dcffca065 | ||
|  | d754620ef7 | ||
|  | 21111126a2 | ||
|  | d91e7dcfdc | 
							
								
								
									
										36
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										36
									
								
								README.md
									
									
									
									
									
								
							| @@ -55,10 +55,11 @@ _✨ All in one 的 OpenAI 接口,整合各种 API 访问方式,开箱即用 | ||||
|    + [x] [API2D](https://api2d.com/r/197971) | ||||
|    + [x] [OhMyGPT](https://aigptx.top?aff=uFpUl2Kf) | ||||
|    + [x] [AI Proxy](https://aiproxy.io/?i=OneAPI) (邀请码:`OneAPI`) | ||||
|    + [x] [OpenAI-SB](https://openai-sb.com) | ||||
|    + [x] [API2GPT](http://console.api2gpt.com/m/00002S) | ||||
|    + [x] [CloseAI](https://console.closeai-asia.com/r/2412) | ||||
|    + [x] [AI.LS](https://ai.ls) | ||||
|    + [x] [OpenAI Max](https://openaimax.com) | ||||
|    + [x] [OpenAI-SB](https://openai-sb.com) | ||||
|    + [x] [CloseAI](https://console.openai-asia.com/r/2412) | ||||
|    + [x] 自定义渠道:例如各种未收录的第三方代理服务 | ||||
| 2. 支持通过**负载均衡**的方式访问多个渠道。 | ||||
| 3. 支持 **stream 模式**,可以通过流式传输实现打字机效果。 | ||||
| @@ -74,15 +75,18 @@ _✨ All in one 的 OpenAI 接口,整合各种 API 访问方式,开箱即用 | ||||
|     1. 支持自定义系统名称,logo 以及页脚。 | ||||
|     2. 支持自定义首页和关于页面,可以选择使用 HTML & Markdown 代码进行自定义,或者使用一个单独的网页通过 iframe 嵌入。 | ||||
| 13. 支持通过系统访问令牌访问管理 API。 | ||||
| 14. 支持用户管理,支持**多种用户登录注册方式**: | ||||
| 14. 支持 Cloudflare Turnstile 用户校验。 | ||||
| 15. 支持用户管理,支持**多种用户登录注册方式**: | ||||
|     + 邮箱登录注册以及通过邮箱进行密码重置。 | ||||
|     + [GitHub 开放授权](https://github.com/settings/applications/new)。 | ||||
|     + 微信公众号授权(需要额外部署 [WeChat Server](https://github.com/songquanpeng/wechat-server))。 | ||||
| 15. 未来其他大模型开放 API 后,将第一时间支持,并将其封装成同样的 API 访问方式。 | ||||
| 16. 未来其他大模型开放 API 后,将第一时间支持,并将其封装成同样的 API 访问方式。 | ||||
|  | ||||
| ## 部署 | ||||
| ### 基于 Docker 进行部署 | ||||
| 执行:`docker run -d --restart always -p 3000:3000 -v /home/ubuntu/data/one-api:/data justsong/one-api` | ||||
| 部署命令:`docker run --name one-api -d --restart always -p 3000:3000 -v /home/ubuntu/data/one-api:/data justsong/one-api` | ||||
|  | ||||
| 更新命令:`docker run --rm -v /var/run/docker.sock:/var/run/docker.sock containrrr/watchtower -cR` | ||||
|  | ||||
| `-p 3000:3000` 中的第一个 `3000` 是宿主机的端口,可以根据需要进行修改。 | ||||
|  | ||||
| @@ -151,6 +155,26 @@ sudo service nginx restart | ||||
|  | ||||
| 环境变量的具体使用方法详见[此处](#环境变量)。 | ||||
|  | ||||
| ### 部署第三方服务配合 One API 使用 | ||||
| > 欢迎 PR 添加更多示例。 | ||||
|  | ||||
| #### ChatGPT Next Web | ||||
| 项目主页:https://github.com/Yidadaa/ChatGPT-Next-Web | ||||
|  | ||||
| ```bash | ||||
| docker run --name chat-next-web -d -p 3001:3000 -e BASE_URL=https://openai.justsong.cn yidadaa/chatgpt-next-web | ||||
| ``` | ||||
|  | ||||
| 注意修改端口号和 `BASE_URL`。 | ||||
|  | ||||
| #### ChatGPT Web | ||||
| 项目主页:https://github.com/Chanzhaoyu/chatgpt-web | ||||
|  | ||||
| ```bash | ||||
| docker run --name chatgpt-web -d -p 3002:3002 -e OPENAI_API_BASE_URL=https://openai.justsong.cn -e OPENAI_API_KEY=sk-xxx chenzhaoyu94/chatgpt-web | ||||
| ``` | ||||
|  | ||||
| 注意修改端口号、`OPENAI_API_BASE_URL` 和 `OPENAI_API_KEY`。 | ||||
|  | ||||
| ### 部署到第三方平台 | ||||
| <details> | ||||
| @@ -207,7 +231,7 @@ graph LR | ||||
|    + 例子:`REDIS_CONN_STRING=redis://default:redispw@localhost:49153` | ||||
| 2. `SESSION_SECRET`:设置之后将使用固定的会话密钥,这样系统重新启动后已登录用户的 cookie 将依旧有效。 | ||||
|    + 例子:`SESSION_SECRET=random_string` | ||||
| 3. `SQL_DSN`:设置之后将使用指定数据库而非 SQLite。 | ||||
| 3. `SQL_DSN`:设置之后将使用指定数据库而非 SQLite,请使用 MySQL 8.0 版本。 | ||||
|    + 例子:`SQL_DSN=root:123456@tcp(localhost:3306)/one-api` | ||||
| 4. `FRONTEND_BASE_URL`:设置之后将使用指定的前端地址,而非后端地址。 | ||||
|    + 例子:`FRONTEND_BASE_URL=https://openai.justsong.cn` | ||||
|   | ||||
| @@ -1,9 +1,10 @@ | ||||
| package common | ||||
|  | ||||
| import ( | ||||
| 	"github.com/google/uuid" | ||||
| 	"sync" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/google/uuid" | ||||
| ) | ||||
|  | ||||
| var StartTime = time.Now().Unix() // unit: second | ||||
| @@ -54,6 +55,8 @@ var TurnstileSiteKey = "" | ||||
| var TurnstileSecretKey = "" | ||||
|  | ||||
| var QuotaForNewUser = 0 | ||||
| var QuotaForInviter = 0 | ||||
| var QuotaForInvitee = 0 | ||||
| var ChannelDisableThreshold = 5.0 | ||||
| var AutomaticDisableChannelEnabled = false | ||||
| var QuotaRemindThreshold = 1000 | ||||
| @@ -133,19 +136,21 @@ const ( | ||||
| 	ChannelTypeAILS      = 9 | ||||
| 	ChannelTypeAIProxy   = 10 | ||||
| 	ChannelTypePaLM      = 11 | ||||
| 	ChannelTypeAPI2GPT   = 12 | ||||
| ) | ||||
|  | ||||
| var ChannelBaseURLs = []string{ | ||||
| 	"",                            // 0 | ||||
| 	"https://api.openai.com",      // 1 | ||||
| 	"https://oa.api2d.net",        // 2 | ||||
| 	"",                            // 3 | ||||
| 	"https://api.openai-asia.com", // 4 | ||||
| 	"https://api.openai-sb.com",   // 5 | ||||
| 	"https://api.openaimax.com",   // 6 | ||||
| 	"https://api.ohmygpt.com",     // 7 | ||||
| 	"",                            // 8 | ||||
| 	"https://api.caipacity.com",   // 9 | ||||
| 	"https://api.aiproxy.io",      // 10 | ||||
| 	"",                            // 11 | ||||
| 	"",                             // 0 | ||||
| 	"https://api.openai.com",       // 1 | ||||
| 	"https://oa.api2d.net",         // 2 | ||||
| 	"",                             // 3 | ||||
| 	"https://api.openai-proxy.org", // 4 | ||||
| 	"https://api.openai-sb.com",    // 5 | ||||
| 	"https://api.openaimax.com",    // 6 | ||||
| 	"https://api.ohmygpt.com",      // 7 | ||||
| 	"",                             // 8 | ||||
| 	"https://api.caipacity.com",    // 9 | ||||
| 	"https://api.aiproxy.io",       // 10 | ||||
| 	"",                             // 11 | ||||
| 	"https://api.api2gpt.com",      // 12 | ||||
| } | ||||
|   | ||||
| @@ -157,6 +157,15 @@ func GenerateKey() string { | ||||
| 	return string(key) | ||||
| } | ||||
|  | ||||
| func GetRandomString(length int) string { | ||||
| 	rand.Seed(time.Now().UnixNano()) | ||||
| 	key := make([]byte, length) | ||||
| 	for i := 0; i < length; i++ { | ||||
| 		key[i] = keyChars[rand.Intn(len(keyChars))] | ||||
| 	} | ||||
| 	return string(key) | ||||
| } | ||||
|  | ||||
| func GetTimestamp() int64 { | ||||
| 	return time.Now().Unix() | ||||
| } | ||||
|   | ||||
| @@ -54,6 +54,13 @@ type AIProxyUserOverviewResponse struct { | ||||
| 	} `json:"data"` | ||||
| } | ||||
|  | ||||
| type API2GPTUsageResponse struct { | ||||
| 	Object         string  `json:"object"` | ||||
| 	TotalGranted   float64 `json:"total_granted"` | ||||
| 	TotalUsed      float64 `json:"total_used"` | ||||
| 	TotalRemaining float64 `json:"total_remaining"` | ||||
| } | ||||
|  | ||||
| // GetAuthHeader get auth header | ||||
| func GetAuthHeader(token string) http.Header { | ||||
| 	h := http.Header{} | ||||
| @@ -127,6 +134,23 @@ func updateChannelAIProxyBalance(channel *model.Channel) (float64, error) { | ||||
| 	return response.Data.TotalPoints, nil | ||||
| } | ||||
|  | ||||
| func updateChannelAPI2GPTBalance(channel *model.Channel) (float64, error) { | ||||
| 	url := "https://api.api2gpt.com/dashboard/billing/credit_grants" | ||||
| 	body, err := GetResponseBody("GET", url, channel, GetAuthHeader(channel.Key)) | ||||
|  | ||||
| 	if err != nil { | ||||
| 		return 0, err | ||||
| 	} | ||||
| 	response := API2GPTUsageResponse{} | ||||
| 	err = json.Unmarshal(body, &response) | ||||
| 	fmt.Print(response) | ||||
| 	if err != nil { | ||||
| 		return 0, err | ||||
| 	} | ||||
| 	channel.UpdateBalance(response.TotalRemaining) | ||||
| 	return response.TotalRemaining, nil | ||||
| } | ||||
|  | ||||
| func updateChannelBalance(channel *model.Channel) (float64, error) { | ||||
| 	baseURL := common.ChannelBaseURLs[channel.Type] | ||||
| 	switch channel.Type { | ||||
| @@ -142,6 +166,8 @@ func updateChannelBalance(channel *model.Channel) (float64, error) { | ||||
| 		return updateChannelOpenAISBBalance(channel) | ||||
| 	case common.ChannelTypeAIProxy: | ||||
| 		return updateChannelAIProxyBalance(channel) | ||||
| 	case common.ChannelTypeAPI2GPT: | ||||
| 		return updateChannelAPI2GPTBalance(channel) | ||||
| 	default: | ||||
| 		return 0, errors.New("尚未实现") | ||||
| 	} | ||||
|   | ||||
| @@ -125,7 +125,7 @@ func GitHubOAuth(c *gin.Context) { | ||||
| 			user.Role = common.RoleCommonUser | ||||
| 			user.Status = common.UserStatusEnabled | ||||
|  | ||||
| 			if err := user.Insert(); err != nil { | ||||
| 			if err := user.Insert(0); err != nil { | ||||
| 				c.JSON(http.StatusOK, gin.H{ | ||||
| 					"success": false, | ||||
| 					"message": err.Error(), | ||||
|   | ||||
| @@ -259,8 +259,9 @@ func relayHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode { | ||||
| 			if err != nil { | ||||
| 				common.SysError("Error consuming token remain quota: " + err.Error()) | ||||
| 			} | ||||
| 			tokenName := c.GetString("token_name") | ||||
| 			userId := c.GetInt("id") | ||||
| 			model.RecordLog(userId, model.LogTypeConsume, fmt.Sprintf("使用模型 %s 消耗 %d 点额度(模型倍率 %.2f,分组倍率 %.2f)", textRequest.Model, quota, modelRatio, groupRatio)) | ||||
| 			model.RecordLog(userId, model.LogTypeConsume, fmt.Sprintf("通过令牌「%s」使用模型 %s 消耗 %d 点额度(模型倍率 %.2f,分组倍率 %.2f)", tokenName, textRequest.Model, quota, modelRatio, groupRatio)) | ||||
| 			model.UpdateUserUsedQuotaAndRequestCount(userId, quota) | ||||
| 			channelId := c.GetInt("channel_id") | ||||
| 			model.UpdateChannelUsedQuota(channelId, quota) | ||||
|   | ||||
| @@ -150,15 +150,18 @@ func Register(c *gin.Context) { | ||||
| 			return | ||||
| 		} | ||||
| 	} | ||||
| 	affCode := user.AffCode // this code is the inviter's code, not the user's own code | ||||
| 	inviterId, _ := model.GetUserIdByAffCode(affCode) | ||||
| 	cleanUser := model.User{ | ||||
| 		Username:    user.Username, | ||||
| 		Password:    user.Password, | ||||
| 		DisplayName: user.Username, | ||||
| 		InviterId:   inviterId, | ||||
| 	} | ||||
| 	if common.EmailVerificationEnabled { | ||||
| 		cleanUser.Email = user.Email | ||||
| 	} | ||||
| 	if err := cleanUser.Insert(); err != nil { | ||||
| 	if err := cleanUser.Insert(inviterId); err != nil { | ||||
| 		c.JSON(http.StatusOK, gin.H{ | ||||
| 			"success": false, | ||||
| 			"message": err.Error(), | ||||
| @@ -280,6 +283,34 @@ func GenerateAccessToken(c *gin.Context) { | ||||
| 	return | ||||
| } | ||||
|  | ||||
| func GetAffCode(c *gin.Context) { | ||||
| 	id := c.GetInt("id") | ||||
| 	user, err := model.GetUserById(id, true) | ||||
| 	if err != nil { | ||||
| 		c.JSON(http.StatusOK, gin.H{ | ||||
| 			"success": false, | ||||
| 			"message": err.Error(), | ||||
| 		}) | ||||
| 		return | ||||
| 	} | ||||
| 	if user.AffCode == "" { | ||||
| 		user.AffCode = common.GetRandomString(4) | ||||
| 		if err := user.Update(false); err != nil { | ||||
| 			c.JSON(http.StatusOK, gin.H{ | ||||
| 				"success": false, | ||||
| 				"message": err.Error(), | ||||
| 			}) | ||||
| 			return | ||||
| 		} | ||||
| 	} | ||||
| 	c.JSON(http.StatusOK, gin.H{ | ||||
| 		"success": true, | ||||
| 		"message": "", | ||||
| 		"data":    user.AffCode, | ||||
| 	}) | ||||
| 	return | ||||
| } | ||||
|  | ||||
| func GetSelf(c *gin.Context) { | ||||
| 	id := c.GetInt("id") | ||||
| 	user, err := model.GetUserById(id, false) | ||||
| @@ -495,7 +526,7 @@ func CreateUser(c *gin.Context) { | ||||
| 		Password:    user.Password, | ||||
| 		DisplayName: user.DisplayName, | ||||
| 	} | ||||
| 	if err := cleanUser.Insert(); err != nil { | ||||
| 	if err := cleanUser.Insert(0); err != nil { | ||||
| 		c.JSON(http.StatusOK, gin.H{ | ||||
| 			"success": false, | ||||
| 			"message": err.Error(), | ||||
|   | ||||
| @@ -85,7 +85,7 @@ func WeChatAuth(c *gin.Context) { | ||||
| 			user.Role = common.RoleCommonUser | ||||
| 			user.Status = common.UserStatusEnabled | ||||
|  | ||||
| 			if err := user.Insert(); err != nil { | ||||
| 			if err := user.Insert(0); err != nil { | ||||
| 				c.JSON(http.StatusOK, gin.H{ | ||||
| 					"success": false, | ||||
| 					"message": err.Error(), | ||||
|   | ||||
| @@ -112,6 +112,7 @@ func TokenAuth() func(c *gin.Context) { | ||||
| 		} | ||||
| 		c.Set("id", token.UserId) | ||||
| 		c.Set("token_id", token.Id) | ||||
| 		c.Set("token_name", token.Name) | ||||
| 		requestURL := c.Request.URL.String() | ||||
| 		consumeQuota := true | ||||
| 		if strings.HasPrefix(requestURL, "/v1/models") { | ||||
|   | ||||
| @@ -10,6 +10,6 @@ func CORS() gin.HandlerFunc { | ||||
| 	config.AllowAllOrigins = true | ||||
| 	config.AllowCredentials = true | ||||
| 	config.AllowMethods = []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"} | ||||
| 	config.AllowHeaders = []string{"Origin", "Content-Length", "Content-Type", "Authorization", "Accept", "Connection"} | ||||
| 	config.AllowHeaders = []string{"Origin", "Content-Length", "Content-Type", "Authorization", "Accept", "Connection", "x-requested-with"} | ||||
| 	return cors.New(config) | ||||
| } | ||||
|   | ||||
| @@ -56,6 +56,8 @@ func InitOptionMap() { | ||||
| 	common.OptionMap["TurnstileSiteKey"] = "" | ||||
| 	common.OptionMap["TurnstileSecretKey"] = "" | ||||
| 	common.OptionMap["QuotaForNewUser"] = strconv.Itoa(common.QuotaForNewUser) | ||||
| 	common.OptionMap["QuotaForInviter"] = strconv.Itoa(common.QuotaForInviter) | ||||
| 	common.OptionMap["QuotaForInvitee"] = strconv.Itoa(common.QuotaForInvitee) | ||||
| 	common.OptionMap["QuotaRemindThreshold"] = strconv.Itoa(common.QuotaRemindThreshold) | ||||
| 	common.OptionMap["PreConsumedQuota"] = strconv.Itoa(common.PreConsumedQuota) | ||||
| 	common.OptionMap["ModelRatio"] = common.ModelRatio2JSONString() | ||||
| @@ -175,6 +177,10 @@ func updateOptionMap(key string, value string) (err error) { | ||||
| 		common.TurnstileSecretKey = value | ||||
| 	case "QuotaForNewUser": | ||||
| 		common.QuotaForNewUser, _ = strconv.Atoi(value) | ||||
| 	case "QuotaForInviter": | ||||
| 		common.QuotaForInviter, _ = strconv.Atoi(value) | ||||
| 	case "QuotaForInvitee": | ||||
| 		common.QuotaForInvitee, _ = strconv.Atoi(value) | ||||
| 	case "QuotaRemindThreshold": | ||||
| 		common.QuotaRemindThreshold, _ = strconv.Atoi(value) | ||||
| 	case "PreConsumedQuota": | ||||
|   | ||||
| @@ -26,6 +26,8 @@ type User struct { | ||||
| 	UsedQuota        int    `json:"used_quota" gorm:"type:int;default:0;column:used_quota"` // used quota | ||||
| 	RequestCount     int    `json:"request_count" gorm:"type:int;default:0;"`               // request number | ||||
| 	Group            string `json:"group" gorm:"type:varchar(32);default:'default'"` | ||||
| 	AffCode          string `json:"aff_code" gorm:"type:varchar(32);column:aff_code;uniqueIndex"` | ||||
| 	InviterId        int    `json:"inviter_id" gorm:"type:int;column:inviter_id;index"` | ||||
| } | ||||
|  | ||||
| func GetMaxUserId() int { | ||||
| @@ -58,6 +60,15 @@ func GetUserById(id int, selectAll bool) (*User, error) { | ||||
| 	return &user, err | ||||
| } | ||||
|  | ||||
| func GetUserIdByAffCode(affCode string) (int, error) { | ||||
| 	if affCode == "" { | ||||
| 		return 0, errors.New("affCode 为空!") | ||||
| 	} | ||||
| 	var user User | ||||
| 	err := DB.Select("id").First(&user, "aff_code = ?", affCode).Error | ||||
| 	return user.Id, err | ||||
| } | ||||
|  | ||||
| func DeleteUserById(id int) (err error) { | ||||
| 	if id == 0 { | ||||
| 		return errors.New("id 为空!") | ||||
| @@ -66,7 +77,7 @@ func DeleteUserById(id int) (err error) { | ||||
| 	return user.Delete() | ||||
| } | ||||
|  | ||||
| func (user *User) Insert() error { | ||||
| func (user *User) Insert(inviterId int) error { | ||||
| 	var err error | ||||
| 	if user.Password != "" { | ||||
| 		user.Password, err = common.Password2Hash(user.Password) | ||||
| @@ -76,6 +87,7 @@ func (user *User) Insert() error { | ||||
| 	} | ||||
| 	user.Quota = common.QuotaForNewUser | ||||
| 	user.AccessToken = common.GetUUID() | ||||
| 	user.AffCode = common.GetRandomString(4) | ||||
| 	result := DB.Create(user) | ||||
| 	if result.Error != nil { | ||||
| 		return result.Error | ||||
| @@ -83,6 +95,16 @@ func (user *User) Insert() error { | ||||
| 	if common.QuotaForNewUser > 0 { | ||||
| 		RecordLog(user.Id, LogTypeSystem, fmt.Sprintf("新用户注册赠送 %d 点额度", common.QuotaForNewUser)) | ||||
| 	} | ||||
| 	if inviterId != 0 { | ||||
| 		if common.QuotaForInvitee > 0 { | ||||
| 			_ = IncreaseUserQuota(user.Id, common.QuotaForInvitee) | ||||
| 			RecordLog(user.Id, LogTypeSystem, fmt.Sprintf("使用邀请码赠送 %d 点额度", common.QuotaForInvitee)) | ||||
| 		} | ||||
| 		if common.QuotaForInviter > 0 { | ||||
| 			_ = IncreaseUserQuota(inviterId, common.QuotaForInviter) | ||||
| 			RecordLog(inviterId, LogTypeSystem, fmt.Sprintf("邀请用户赠送 %d 点额度", common.QuotaForInviter)) | ||||
| 		} | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
|   | ||||
| @@ -37,6 +37,7 @@ func SetApiRouter(router *gin.Engine) { | ||||
| 				selfRoute.PUT("/self", controller.UpdateSelf) | ||||
| 				selfRoute.DELETE("/self", controller.DeleteSelf) | ||||
| 				selfRoute.GET("/token", controller.GenerateAccessToken) | ||||
| 				selfRoute.GET("/aff", controller.GetAffCode) | ||||
| 				selfRoute.POST("/topup", controller.TopUp) | ||||
| 			} | ||||
|  | ||||
|   | ||||
| @@ -36,6 +36,8 @@ function renderBalance(type, balance) { | ||||
|       return <span>¥{(balance / 10000).toFixed(2)}</span>; | ||||
|     case 10: // AI Proxy | ||||
|       return <span>{renderNumber(balance)}</span>; | ||||
|     case 12: // API2GPT | ||||
|       return <span>¥{balance.toFixed(2)}</span>; | ||||
|     default: | ||||
|       return <span>不支持</span>; | ||||
|   } | ||||
|   | ||||
| @@ -1,7 +1,7 @@ | ||||
| import React, { useEffect, useState } from 'react'; | ||||
| import { Button, Divider, Form, Header, Image, Message, Modal } from 'semantic-ui-react'; | ||||
| import { Link } from 'react-router-dom'; | ||||
| import { API, copy, showError, showInfo, showSuccess } from '../helpers'; | ||||
| import { API, copy, showError, showInfo, showNotice, showSuccess } from '../helpers'; | ||||
| import Turnstile from 'react-turnstile'; | ||||
|  | ||||
| const PersonalSetting = () => { | ||||
| @@ -45,6 +45,18 @@ const PersonalSetting = () => { | ||||
|     } | ||||
|   }; | ||||
|  | ||||
|   const getAffLink = async () => { | ||||
|     const res = await API.get('/api/user/aff'); | ||||
|     const { success, message, data } = res.data; | ||||
|     if (success) { | ||||
|       let link = `${window.location.origin}/register?aff=${data}`; | ||||
|       await copy(link); | ||||
|       showNotice(`邀请链接已复制到剪切板:${link}`); | ||||
|     } else { | ||||
|       showError(message); | ||||
|     } | ||||
|   }; | ||||
|  | ||||
|   const bindWeChat = async () => { | ||||
|     if (inputs.wechat_verification_code === '') return; | ||||
|     const res = await API.get( | ||||
| @@ -110,6 +122,7 @@ const PersonalSetting = () => { | ||||
|         更新个人信息 | ||||
|       </Button> | ||||
|       <Button onClick={generateAccessToken}>生成系统访问令牌</Button> | ||||
|       <Button onClick={getAffLink}>复制邀请链接</Button> | ||||
|       <Divider /> | ||||
|       <Header as='h3'>账号绑定</Header> | ||||
|       { | ||||
|   | ||||
| @@ -27,6 +27,10 @@ const RegisterForm = () => { | ||||
|   const [turnstileToken, setTurnstileToken] = useState(''); | ||||
|   const [loading, setLoading] = useState(false); | ||||
|   const logo = getLogo(); | ||||
|   let affCode = new URLSearchParams(window.location.search).get('aff'); | ||||
|   if (affCode) { | ||||
|     localStorage.setItem('aff', affCode); | ||||
|   } | ||||
|  | ||||
|   useEffect(() => { | ||||
|     let status = localStorage.getItem('status'); | ||||
| @@ -63,6 +67,10 @@ const RegisterForm = () => { | ||||
|         return; | ||||
|       } | ||||
|       setLoading(true); | ||||
|       if (!affCode) { | ||||
|         affCode = localStorage.getItem('aff'); | ||||
|       } | ||||
|       inputs.aff_code = affCode; | ||||
|       const res = await API.post( | ||||
|         `/api/user/register?turnstile=${turnstileToken}`, | ||||
|         inputs | ||||
|   | ||||
| @@ -27,6 +27,8 @@ const SystemSetting = () => { | ||||
|     TurnstileSecretKey: '', | ||||
|     RegisterEnabled: '', | ||||
|     QuotaForNewUser: 0, | ||||
|     QuotaForInviter: 0, | ||||
|     QuotaForInvitee: 0, | ||||
|     QuotaRemindThreshold: 0, | ||||
|     PreConsumedQuota: 0, | ||||
|     ModelRatio: '', | ||||
| @@ -34,7 +36,7 @@ const SystemSetting = () => { | ||||
|     TopUpLink: '', | ||||
|     AutomaticDisableChannelEnabled: '', | ||||
|     ChannelDisableThreshold: 0, | ||||
|     LogConsumeEnabled: '', | ||||
|     LogConsumeEnabled: '' | ||||
|   }); | ||||
|   const [originInputs, setOriginInputs] = useState({}); | ||||
|   let [loading, setLoading] = useState(false); | ||||
| @@ -101,6 +103,8 @@ const SystemSetting = () => { | ||||
|       name === 'TurnstileSiteKey' || | ||||
|       name === 'TurnstileSecretKey' || | ||||
|       name === 'QuotaForNewUser' || | ||||
|       name === 'QuotaForInviter' || | ||||
|       name === 'QuotaForInvitee' || | ||||
|       name === 'QuotaRemindThreshold' || | ||||
|       name === 'PreConsumedQuota' || | ||||
|       name === 'ModelRatio' || | ||||
| @@ -122,6 +126,12 @@ const SystemSetting = () => { | ||||
|     if (originInputs['QuotaForNewUser'] !== inputs.QuotaForNewUser) { | ||||
|       await updateOption('QuotaForNewUser', inputs.QuotaForNewUser); | ||||
|     } | ||||
|     if (originInputs['QuotaForInvitee'] !== inputs.QuotaForInvitee) { | ||||
|       await updateOption('QuotaForInvitee', inputs.QuotaForInvitee); | ||||
|     } | ||||
|     if (originInputs['QuotaForInviter'] !== inputs.QuotaForInviter) { | ||||
|       await updateOption('QuotaForInviter', inputs.QuotaForInviter); | ||||
|     } | ||||
|     if (originInputs['QuotaRemindThreshold'] !== inputs.QuotaRemindThreshold) { | ||||
|       await updateOption('QuotaRemindThreshold', inputs.QuotaRemindThreshold); | ||||
|     } | ||||
| @@ -329,6 +339,28 @@ const SystemSetting = () => { | ||||
|               placeholder='请求结束后多退少补' | ||||
|             /> | ||||
|           </Form.Group> | ||||
|           <Form.Group widths={4}> | ||||
|             <Form.Input | ||||
|               label='邀请新用户奖励配额' | ||||
|               name='QuotaForInviter' | ||||
|               onChange={handleInputChange} | ||||
|               autoComplete='new-password' | ||||
|               value={inputs.QuotaForInviter} | ||||
|               type='number' | ||||
|               min='0' | ||||
|               placeholder='例如:100' | ||||
|             /> | ||||
|             <Form.Input | ||||
|               label='新用户使用邀请码奖励配额' | ||||
|               name='QuotaForInvitee' | ||||
|               onChange={handleInputChange} | ||||
|               autoComplete='new-password' | ||||
|               value={inputs.QuotaForInvitee} | ||||
|               type='number' | ||||
|               min='0' | ||||
|               placeholder='例如:100' | ||||
|             /> | ||||
|           </Form.Group> | ||||
|           <Form.Group widths='equal'> | ||||
|             <Form.TextArea | ||||
|               label='模型倍率' | ||||
|   | ||||
| @@ -240,7 +240,7 @@ const UsersTable = () => { | ||||
|                     /> | ||||
|                   </Table.Cell> | ||||
|                   <Table.Cell>{renderGroup(user.group)}</Table.Cell> | ||||
|                   <Table.Cell>{user.email ? renderText(user.email, 30) : '无'}</Table.Cell> | ||||
|                   <Table.Cell>{user.email ? renderText(user.email, 20) : '无'}</Table.Cell> | ||||
|                   <Table.Cell> | ||||
|                     <Popup content='剩余额度' trigger={<Label>{renderNumber(user.quota)}</Label>} /> | ||||
|                     <Popup content='已用额度' trigger={<Label>{renderNumber(user.used_quota)}</Label>} /> | ||||
|   | ||||
| @@ -8,5 +8,6 @@ export const CHANNEL_OPTIONS = [ | ||||
|   { key: 6, text: 'OpenAI Max', value: 6, color: 'violet' }, | ||||
|   { key: 7, text: 'OhMyGPT', value: 7, color: 'purple' }, | ||||
|   { key: 9, text: 'AI.LS', value: 9, color: 'yellow' }, | ||||
|   { key: 10, text: 'AI Proxy', value: 10, color: 'purple' } | ||||
|   { key: 10, text: 'AI Proxy', value: 10, color: 'purple' }, | ||||
|   { key: 12, text: 'API2GPT', value: 12, color: 'blue' } | ||||
| ]; | ||||
|   | ||||
		Reference in New Issue
	
	Block a user