mirror of
				https://github.com/songquanpeng/one-api.git
				synced 2025-10-31 13:53:41 +08:00 
			
		
		
		
	Compare commits
	
		
			46 Commits
		
	
	
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
|  | d79289ccdd | ||
|  | f89f6c7fa6 | ||
|  | b7d71b4f0a | ||
|  | 70ed126ccb | ||
|  | 57b213a035 | ||
|  | 549e944b95 | ||
|  | 0cdab80a6e | ||
|  | 760183a970 | ||
|  | 58fb18aace | ||
|  | 630156dc0a | ||
|  | 5f23f59d1c | ||
|  | 538a5d7a9b | ||
|  | 593e1926e9 | ||
|  | e87ad1f402 | ||
|  | 07cccdc8c0 | ||
|  | f71f01662c | ||
|  | 54d7a1c2e8 | ||
|  | f426f31bd7 | ||
|  | 2930577cd6 | ||
|  | e09512177a | ||
|  | d6dbaff3c2 | ||
|  | 7f9577a386 | ||
|  | 38668e7331 | ||
|  | 323f3d263a | ||
|  | 0c34ed4c61 | ||
|  | 7c7eb6b7ec | ||
|  | 8b2ef666ef | ||
|  | 955d5f8707 | ||
|  | 47ca449e32 | ||
|  | 39481eb6c0 | ||
|  | 69153e7231 | ||
|  | cdef10cad8 | ||
|  | 077853416d | ||
|  | 596446dba4 | ||
|  | 9d0bec83df | ||
|  | f97a9ce597 | ||
|  | 4339f45f74 | ||
|  | e398e0756b | ||
|  | 64db39320a | ||
|  | 0b4bf30908 | ||
|  | d29c273073 | ||
|  | 74f508e847 | ||
|  | 145bb14cb2 | ||
|  | 8901f03864 | ||
|  | 813bf0bd66 | ||
|  | 45e9fd66e7 | 
							
								
								
									
										23
									
								
								.github/ISSUE_TEMPLATE/bug_report.md
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										23
									
								
								.github/ISSUE_TEMPLATE/bug_report.md
									
									
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,23 @@ | ||||
| --- | ||||
| name: 报告问题 | ||||
| about: 使用简练详细的语言描述你遇到的问题 | ||||
| title: '' | ||||
| labels: bug | ||||
| assignees: '' | ||||
|  | ||||
| --- | ||||
|  | ||||
| **例行检查** | ||||
| + [ ] 我已确认目前没有类似 issue | ||||
| + [ ] 我已确认我已升级到最新版本 | ||||
| + [ ] 我理解并愿意跟进此 issue,协助测试和提供反馈  | ||||
| + [ ] 我理解并认可上述内容,并理解项目维护者精力有限,不遵循规则的 issue 可能会被无视或直接关闭 | ||||
|  | ||||
| **问题描述** | ||||
|  | ||||
| **复现步骤** | ||||
|  | ||||
| **预期结果** | ||||
|  | ||||
| **相关截图** | ||||
| 如果没有的话,请删除此节。 | ||||
							
								
								
									
										11
									
								
								.github/ISSUE_TEMPLATE/config.yml
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								.github/ISSUE_TEMPLATE/config.yml
									
									
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,11 @@ | ||||
| blank_issues_enabled: false | ||||
| contact_links: | ||||
|   - name: 项目群聊 | ||||
|     url: https://openai.justsong.cn/ | ||||
|     about: QQ 群:828520184,自动审核,备注 One API | ||||
|   - name: 赞赏支持 | ||||
|     url: https://iamazing.cn/page/reward | ||||
|     about: 请作者喝杯咖啡,以激励作者持续开发 | ||||
|   - name: 付费部署或定制功能 | ||||
|     url: https://openai.justsong.cn/ | ||||
|     about: 加群后联系群主 | ||||
							
								
								
									
										18
									
								
								.github/ISSUE_TEMPLATE/feature_request.md
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										18
									
								
								.github/ISSUE_TEMPLATE/feature_request.md
									
									
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,18 @@ | ||||
| --- | ||||
| name: 功能请求 | ||||
| about: 使用简练详细的语言描述希望加入的新功能 | ||||
| title: '' | ||||
| labels: enhancement | ||||
| assignees: '' | ||||
|  | ||||
| --- | ||||
|  | ||||
| **例行检查** | ||||
| + [ ] 我已确认目前没有类似 issue | ||||
| + [ ] 我已确认我已升级到最新版本 | ||||
| + [ ] 我理解并愿意跟进此 issue,协助测试和提供反馈 | ||||
| + [ ] 我理解并认可上述内容,并理解项目维护者精力有限,不遵循规则的 issue 可能会被无视或直接关闭 | ||||
|  | ||||
| **功能描述** | ||||
|  | ||||
| **应用场景** | ||||
							
								
								
									
										75
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										75
									
								
								README.md
									
									
									
									
									
								
							| @@ -29,10 +29,10 @@ _✨ All in one 的 OpenAI 接口,整合各种 API 访问方式,开箱即用 | ||||
| </p> | ||||
|  | ||||
| <p align="center"> | ||||
|   <a href="https://github.com/songquanpeng/one-api/releases">程序下载</a> | ||||
|   · | ||||
|   <a href="https://github.com/songquanpeng/one-api#部署">部署教程</a> | ||||
|   · | ||||
|   <a href="https://github.com/songquanpeng/one-api#使用方法">使用方法</a> | ||||
|   · | ||||
|   <a href="https://github.com/songquanpeng/one-api/issues">意见反馈</a> | ||||
|   · | ||||
|   <a href="https://github.com/songquanpeng/one-api#截图展示">截图展示</a> | ||||
| @@ -44,11 +44,13 @@ _✨ All in one 的 OpenAI 接口,整合各种 API 访问方式,开箱即用 | ||||
|   <a href="https://iamazing.cn/page/reward">赞赏支持</a> | ||||
| </p> | ||||
|  | ||||
| > **Warning**:使用 Docker 拉取的最新镜像可能是 `alpha` 版本,如果追求稳定性请手动指定版本。 | ||||
| > **Note**:使用 Docker 拉取的最新镜像可能是 `alpha` 版本,如果追求稳定性请手动指定版本。 | ||||
|  | ||||
| > **Warning**:从 `v0.3` 版本升级到 `v0.4` 版本需要手动迁移数据库,请手动执行[数据库迁移脚本](./bin/migration_v0.3-v0.4.sql)。 | ||||
|  | ||||
| ## 功能 | ||||
| 1. 支持多种 API 访问渠道,欢迎 PR 或提 issue 添加更多渠道: | ||||
|    + [x] OpenAI 官方通道 | ||||
|    + [x] OpenAI 官方通道(支持配置代理) | ||||
|    + [x] **Azure OpenAI API** | ||||
|    + [x] [API2D](https://api2d.com/r/197971) | ||||
|    + [x] [OhMyGPT](https://aigptx.top?aff=uFpUl2Kf) | ||||
| @@ -57,25 +59,26 @@ _✨ All in one 的 OpenAI 接口,整合各种 API 访问方式,开箱即用 | ||||
|    + [x] [OpenAI Max](https://openaimax.com) | ||||
|    + [x] [OpenAI-SB](https://openai-sb.com) | ||||
|    + [x] [CloseAI](https://console.openai-asia.com/r/2412) | ||||
|    + [x] 自定义渠道:例如使用自行搭建的 OpenAI 代理 | ||||
|    + [x] 自定义渠道:例如各种未收录的第三方代理服务 | ||||
| 2. 支持通过**负载均衡**的方式访问多个渠道。 | ||||
| 3. 支持 **stream 模式**,可以通过流式传输实现打字机效果。 | ||||
| 4. 支持**多机部署**,[详见此处](#多机部署)。 | ||||
| 5. 支持**令牌管理**,设置令牌的过期时间和使用次数。 | ||||
| 6. 支持**兑换码管理**,支持批量生成和导出兑换码,可使用兑换码为账户进行充值。 | ||||
| 7. 支持**通道管理**,批量创建通道。 | ||||
| 8. 支持**用户分组**以及**渠道分组**。 | ||||
| 8. 支持**用户分组**以及**渠道分组**,支持为不同分组设置不同的倍率。 | ||||
| 9. 支持渠道**设置模型列表**。 | ||||
| 10. 支持发布公告,设置充值链接,设置新用户初始额度。 | ||||
| 11. 支持丰富的**自定义**设置, | ||||
| 10. 支持**查看额度明细**。 | ||||
| 11. 支持发布公告,设置充值链接,设置新用户初始额度。 | ||||
| 12. 支持丰富的**自定义**设置, | ||||
|     1. 支持自定义系统名称,logo 以及页脚。 | ||||
|     2. 支持自定义首页和关于页面,可以选择使用 HTML & Markdown 代码进行自定义,或者使用一个单独的网页通过 iframe 嵌入。 | ||||
| 12. 支持通过系统访问令牌访问管理 API。 | ||||
| 13. 支持用户管理,支持**多种用户登录注册方式**: | ||||
| 13. 支持通过系统访问令牌访问管理 API。 | ||||
| 14. 支持用户管理,支持**多种用户登录注册方式**: | ||||
|     + 邮箱登录注册以及通过邮箱进行密码重置。 | ||||
|     + [GitHub 开放授权](https://github.com/settings/applications/new)。 | ||||
|     + 微信公众号授权(需要额外部署 [WeChat Server](https://github.com/songquanpeng/wechat-server))。 | ||||
| 14. 未来其他大模型开放 API 后,将第一时间支持,并将其封装成同样的 API 访问方式。 | ||||
| 15. 未来其他大模型开放 API 后,将第一时间支持,并将其封装成同样的 API 访问方式。 | ||||
|  | ||||
| ## 部署 | ||||
| ### 基于 Docker 进行部署 | ||||
| @@ -114,6 +117,8 @@ sudo certbot --nginx | ||||
| sudo service nginx restart | ||||
| ``` | ||||
|  | ||||
| 初始账号用户名为 `root`,密码为 `123456`。 | ||||
|  | ||||
| ### 手动部署 | ||||
| 1. 从 [GitHub Releases](https://github.com/songquanpeng/one-api/releases/latest) 下载可执行文件或者从源码编译: | ||||
|    ```shell | ||||
| @@ -146,6 +151,27 @@ sudo service nginx restart | ||||
|  | ||||
| 环境变量的具体使用方法详见[此处](#环境变量)。 | ||||
|  | ||||
|  | ||||
| ### 部署到第三方平台 | ||||
| <details> | ||||
| <summary><strong>部署到 Zeabur</strong></summary> | ||||
| <div> | ||||
|  | ||||
| > Zeabur 的服务器在国外,自动解决了网络的问题,同时免费的额度也足够个人使用。 | ||||
|  | ||||
| 1. 首先 fork 一份代码。 | ||||
| 2. 进入 [Zeabur](https://zeabur.com/),登录,进入控制台。 | ||||
| 3. 新建一个 Project,在 Service -> Add Service 选择 Marketplace,选择 MySQL,并记下连接参数(用户名、密码、地址、端口)。 | ||||
| 4. 复制链接参数,运行 ```create database `one-api` ``` 创建数据库。 | ||||
| 5. 然后在 Service -> Add Service,选择 Git(第一次使用需要先授权),选择你 fork 的仓库。 | ||||
| 6. Deploy 会自动开始,先取消。进入下方 Variable,添加一个 `PORT`,值为 `3000`,再添加一个 `SQL_DSN`,值为 `<username>:<password>@tcp(<addr>:<port>)/one-api` ,然后保存。 注意如果不填写 `SQL_DSN`,数据将无法持久化,重新部署后数据会丢失。 | ||||
| 7. 选择 Redeploy。 | ||||
| 8. 进入下方 Domains,选择一个合适的域名前缀,如 "my-one-api",最终域名为 "my-one-api.zeabur.app",也可以 CNAME 自己的域名。 | ||||
| 9. 等待部署完成,点击生成的域名进入 One API。 | ||||
|  | ||||
| </div> | ||||
| </details> | ||||
|  | ||||
| ## 配置 | ||||
| 系统本身开箱即用。 | ||||
|  | ||||
| @@ -153,11 +179,24 @@ sudo service nginx restart | ||||
|  | ||||
| 等到系统启动后,使用 `root` 用户登录系统并做进一步的配置。 | ||||
|  | ||||
| ## 使用方式 | ||||
| 在`渠道`页面中添加你的 API Key,之后在`令牌`页面中新增一个访问令牌。 | ||||
| ## 使用方法 | ||||
| 在`渠道`页面中添加你的 API Key,之后在`令牌`页面中新增访问令牌。 | ||||
|  | ||||
| 之后就可以使用你的令牌访问 One API 了,使用方式与 [OpenAI API](https://platform.openai.com/docs/api-reference/introduction) 一致。 | ||||
|  | ||||
| 你需要在各种用到 OpenAI API 的地方设置 API Base 为你的 One API 的部署地址,例如:`https://openai.justsong.cn`,API Key 则为你在 One API 中生成的令牌。 | ||||
|  | ||||
| 注意,具体的 API Base 的格式取决于你所使用的客户端。 | ||||
|  | ||||
| ```mermaid | ||||
| graph LR | ||||
|     A(用户) | ||||
|     A --->|请求| B(One API) | ||||
|     B -->|中继请求| C(OpenAI) | ||||
|     B -->|中继请求| D(Azure) | ||||
|     B -->|中继请求| E(其他下游渠道) | ||||
| ``` | ||||
|  | ||||
| 可以通过在令牌后面添加渠道 ID 的方式指定使用哪一个渠道处理本次请求,例如:`Authorization: Bearer ONE_API_KEY-CHANNEL_ID`。 | ||||
| 注意,需要是管理员用户创建的令牌才能指定渠道 ID。 | ||||
|  | ||||
| @@ -198,3 +237,13 @@ https://openai.justsong.cn | ||||
|    + 令牌额度仅供用户设置最大使用量,用户可自由设置。 | ||||
| 2. 宝塔部署后访问出现空白页面? | ||||
|    + 自动配置的问题,详见[#97](https://github.com/songquanpeng/one-api/issues/97)。 | ||||
| 3. 提示无可用渠道? | ||||
|    + 请检查的用户分组和渠道分组设置。 | ||||
|    + 以及渠道的模型设置。 | ||||
|  | ||||
| ## 注意 | ||||
| 本项目为开源项目,请在遵循 OpenAI 的[使用条款](https://openai.com/policies/terms-of-use)以及法律法规的情况下使用,不得用于非法用途。 | ||||
|  | ||||
| 本项目使用 MIT 协议进行开源,请以某种方式保留 One API 的版权信息。 | ||||
|  | ||||
| 依据 MIT 协议,使用者需自行承担使用本项目的风险与责任,本开源项目开发者与此无关。 | ||||
							
								
								
									
										17
									
								
								bin/migration_v0.3-v0.4.sql
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										17
									
								
								bin/migration_v0.3-v0.4.sql
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,17 @@ | ||||
| INSERT INTO abilities (`group`, model, channel_id, enabled) | ||||
| SELECT c.`group`, m.model, c.id, 1 | ||||
| FROM channels c | ||||
| CROSS JOIN ( | ||||
|     SELECT 'gpt-3.5-turbo' AS model UNION ALL | ||||
|     SELECT 'gpt-3.5-turbo-0301' AS model UNION ALL | ||||
|     SELECT 'gpt-4' AS model UNION ALL | ||||
|     SELECT 'gpt-4-0314' AS model | ||||
| ) AS m | ||||
| WHERE c.status = 1 | ||||
|   AND NOT EXISTS ( | ||||
|     SELECT 1 | ||||
|     FROM abilities a | ||||
|     WHERE a.`group` = c.`group` | ||||
|       AND a.model = m.model | ||||
|       AND a.channel_id = c.id | ||||
| ); | ||||
| @@ -25,6 +25,7 @@ var OptionMap map[string]string | ||||
| var OptionMapRWMutex sync.RWMutex | ||||
|  | ||||
| var ItemsPerPage = 10 | ||||
| var MaxRecentItems = 100 | ||||
|  | ||||
| var PasswordLoginEnabled = true | ||||
| var PasswordRegisterEnabled = true | ||||
| @@ -34,6 +35,8 @@ var WeChatAuthEnabled = false | ||||
| var TurnstileCheckEnabled = false | ||||
| var RegisterEnabled = true | ||||
|  | ||||
| var LogConsumeEnabled = true | ||||
|  | ||||
| var SMTPServer = "" | ||||
| var SMTPPort = 587 | ||||
| var SMTPAccount = "" | ||||
|   | ||||
							
								
								
									
										31
									
								
								common/group-ratio.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										31
									
								
								common/group-ratio.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,31 @@ | ||||
| package common | ||||
|  | ||||
| import "encoding/json" | ||||
|  | ||||
| var GroupRatio = map[string]float64{ | ||||
| 	"default": 1, | ||||
| 	"vip":     1, | ||||
| 	"svip":    1, | ||||
| } | ||||
|  | ||||
| func GroupRatio2JSONString() string { | ||||
| 	jsonBytes, err := json.Marshal(GroupRatio) | ||||
| 	if err != nil { | ||||
| 		SysError("Error marshalling model ratio: " + err.Error()) | ||||
| 	} | ||||
| 	return string(jsonBytes) | ||||
| } | ||||
|  | ||||
| func UpdateGroupRatioByJSONString(jsonStr string) error { | ||||
| 	GroupRatio = make(map[string]float64) | ||||
| 	return json.Unmarshal([]byte(jsonStr), &GroupRatio) | ||||
| } | ||||
|  | ||||
| func GetGroupRatio(name string) float64 { | ||||
| 	ratio, ok := GroupRatio[name] | ||||
| 	if !ok { | ||||
| 		SysError("Group ratio not found: " + name) | ||||
| 		return 1 | ||||
| 	} | ||||
| 	return ratio | ||||
| } | ||||
| @@ -2,16 +2,23 @@ package common | ||||
|  | ||||
| import "encoding/json" | ||||
|  | ||||
| // ModelRatio | ||||
| // https://platform.openai.com/docs/models/model-endpoint-compatibility | ||||
| // https://openai.com/pricing | ||||
| // TODO: when a new api is enabled, check the pricing here | ||||
| // 1 === $0.002 / 1K tokens | ||||
| var ModelRatio = map[string]float64{ | ||||
| 	"gpt-4":                   15, | ||||
| 	"gpt-4-0314":              15, | ||||
| 	"gpt-4-0613":              15, | ||||
| 	"gpt-4-32k":               30, | ||||
| 	"gpt-4-32k-0314":          30, | ||||
| 	"gpt-3.5-turbo":           1, // $0.002 / 1K tokens | ||||
| 	"gpt-3.5-turbo-0301":      1, | ||||
| 	"gpt-4-32k-0613":          30, | ||||
| 	"gpt-3.5-turbo":           0.75, // $0.0015 / 1K tokens | ||||
| 	"gpt-3.5-turbo-0301":      0.75, | ||||
| 	"gpt-3.5-turbo-0613":      0.75, | ||||
| 	"gpt-3.5-turbo-16k":       1.5, // $0.003 / 1K tokens | ||||
| 	"gpt-3.5-turbo-16k-0613":  1.5, | ||||
| 	"text-ada-001":            0.2, | ||||
| 	"text-babbage-001":        0.25, | ||||
| 	"text-curie-001":          1, | ||||
| @@ -26,8 +33,8 @@ var ModelRatio = map[string]float64{ | ||||
| 	"ada":                     10, | ||||
| 	"text-embedding-ada-002":  0.2, | ||||
| 	"text-search-ada-doc-001": 10, | ||||
| 	"text-moderation-stable":  10, | ||||
| 	"text-moderation-latest":  10, | ||||
| 	"text-moderation-stable":  0.1, | ||||
| 	"text-moderation-latest":  0.1, | ||||
| } | ||||
|  | ||||
| func ModelRatio2JSONString() string { | ||||
| @@ -39,6 +46,7 @@ func ModelRatio2JSONString() string { | ||||
| } | ||||
|  | ||||
| func UpdateModelRatioByJSONString(jsonStr string) error { | ||||
| 	ModelRatio = make(map[string]float64) | ||||
| 	return json.Unmarshal([]byte(jsonStr), &ModelRatio) | ||||
| } | ||||
|  | ||||
|   | ||||
| @@ -4,13 +4,14 @@ import ( | ||||
| 	"encoding/json" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"github.com/gin-gonic/gin" | ||||
| 	"io" | ||||
| 	"net/http" | ||||
| 	"one-api/common" | ||||
| 	"one-api/model" | ||||
| 	"strconv" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/gin-gonic/gin" | ||||
| ) | ||||
|  | ||||
| // https://github.com/songquanpeng/one-api/issues/79 | ||||
| @@ -37,36 +38,116 @@ type OpenAIUsageResponse struct { | ||||
| 	TotalUsage float64 `json:"total_usage"` // unit: 0.01 dollar | ||||
| } | ||||
|  | ||||
| type OpenAISBUsageResponse struct { | ||||
| 	Msg  string `json:"msg"` | ||||
| 	Data *struct { | ||||
| 		Credit string `json:"credit"` | ||||
| 	} `json:"data"` | ||||
| } | ||||
|  | ||||
| type AIProxyUserOverviewResponse struct { | ||||
| 	Success   bool   `json:"success"` | ||||
| 	Message   string `json:"message"` | ||||
| 	ErrorCode int    `json:"error_code"` | ||||
| 	Data      struct { | ||||
| 		TotalPoints float64 `json:"totalPoints"` | ||||
| 	} `json:"data"` | ||||
| } | ||||
|  | ||||
| // GetAuthHeader get auth header | ||||
| func GetAuthHeader(token string) http.Header { | ||||
| 	h := http.Header{} | ||||
| 	h.Add("Authorization", fmt.Sprintf("Bearer %s", token)) | ||||
| 	return h | ||||
| } | ||||
|  | ||||
| func GetResponseBody(method, url string, channel *model.Channel, headers http.Header) ([]byte, error) { | ||||
| 	client := &http.Client{} | ||||
| 	req, err := http.NewRequest(method, url, nil) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	for k := range headers { | ||||
| 		req.Header.Add(k, headers.Get(k)) | ||||
| 	} | ||||
| 	res, err := client.Do(req) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	body, err := io.ReadAll(res.Body) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	err = res.Body.Close() | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	return body, nil | ||||
| } | ||||
|  | ||||
| func updateChannelOpenAISBBalance(channel *model.Channel) (float64, error) { | ||||
| 	url := fmt.Sprintf("https://api.openai-sb.com/sb-api/user/status?api_key=%s", channel.Key) | ||||
| 	body, err := GetResponseBody("GET", url, channel, GetAuthHeader(channel.Key)) | ||||
| 	if err != nil { | ||||
| 		return 0, err | ||||
| 	} | ||||
| 	response := OpenAISBUsageResponse{} | ||||
| 	err = json.Unmarshal(body, &response) | ||||
| 	if err != nil { | ||||
| 		return 0, err | ||||
| 	} | ||||
| 	if response.Data == nil { | ||||
| 		return 0, errors.New(response.Msg) | ||||
| 	} | ||||
| 	balance, err := strconv.ParseFloat(response.Data.Credit, 64) | ||||
| 	if err != nil { | ||||
| 		return 0, err | ||||
| 	} | ||||
| 	channel.UpdateBalance(balance) | ||||
| 	return balance, nil | ||||
| } | ||||
|  | ||||
| func updateChannelAIProxyBalance(channel *model.Channel) (float64, error) { | ||||
| 	url := "https://aiproxy.io/api/report/getUserOverview" | ||||
| 	headers := http.Header{} | ||||
| 	headers.Add("Api-Key", channel.Key) | ||||
| 	body, err := GetResponseBody("GET", url, channel, headers) | ||||
| 	if err != nil { | ||||
| 		return 0, err | ||||
| 	} | ||||
| 	response := AIProxyUserOverviewResponse{} | ||||
| 	err = json.Unmarshal(body, &response) | ||||
| 	if err != nil { | ||||
| 		return 0, err | ||||
| 	} | ||||
| 	if !response.Success { | ||||
| 		return 0, fmt.Errorf("code: %d, message: %s", response.ErrorCode, response.Message) | ||||
| 	} | ||||
| 	channel.UpdateBalance(response.Data.TotalPoints) | ||||
| 	return response.Data.TotalPoints, nil | ||||
| } | ||||
|  | ||||
| func updateChannelBalance(channel *model.Channel) (float64, error) { | ||||
| 	baseURL := common.ChannelBaseURLs[channel.Type] | ||||
| 	switch channel.Type { | ||||
| 	case common.ChannelTypeOpenAI: | ||||
| 		// do nothing | ||||
| 		if channel.BaseURL != "" { | ||||
| 			baseURL = channel.BaseURL | ||||
| 		} | ||||
| 	case common.ChannelTypeAzure: | ||||
| 		return 0, errors.New("尚未实现") | ||||
| 	case common.ChannelTypeCustom: | ||||
| 		baseURL = channel.BaseURL | ||||
| 	case common.ChannelTypeOpenAISB: | ||||
| 		return updateChannelOpenAISBBalance(channel) | ||||
| 	case common.ChannelTypeAIProxy: | ||||
| 		return updateChannelAIProxyBalance(channel) | ||||
| 	default: | ||||
| 		return 0, errors.New("尚未实现") | ||||
| 	} | ||||
| 	url := fmt.Sprintf("%s/v1/dashboard/billing/subscription", baseURL) | ||||
|  | ||||
| 	client := &http.Client{} | ||||
| 	req, err := http.NewRequest("GET", url, nil) | ||||
| 	if err != nil { | ||||
| 		return 0, err | ||||
| 	} | ||||
| 	auth := fmt.Sprintf("Bearer %s", channel.Key) | ||||
| 	req.Header.Add("Authorization", auth) | ||||
| 	res, err := client.Do(req) | ||||
| 	if err != nil { | ||||
| 		return 0, err | ||||
| 	} | ||||
| 	body, err := io.ReadAll(res.Body) | ||||
| 	if err != nil { | ||||
| 		return 0, err | ||||
| 	} | ||||
| 	err = res.Body.Close() | ||||
| 	body, err := GetResponseBody("GET", url, channel, GetAuthHeader(channel.Key)) | ||||
| 	if err != nil { | ||||
| 		return 0, err | ||||
| 	} | ||||
| @@ -82,20 +163,7 @@ func updateChannelBalance(channel *model.Channel) (float64, error) { | ||||
| 		startDate = now.AddDate(0, 0, -100).Format("2006-01-02") | ||||
| 	} | ||||
| 	url = fmt.Sprintf("%s/v1/dashboard/billing/usage?start_date=%s&end_date=%s", baseURL, startDate, endDate) | ||||
| 	req, err = http.NewRequest("GET", url, nil) | ||||
| 	if err != nil { | ||||
| 		return 0, err | ||||
| 	} | ||||
| 	req.Header.Add("Authorization", auth) | ||||
| 	res, err = client.Do(req) | ||||
| 	if err != nil { | ||||
| 		return 0, err | ||||
| 	} | ||||
| 	body, err = io.ReadAll(res.Body) | ||||
| 	if err != nil { | ||||
| 		return 0, err | ||||
| 	} | ||||
| 	err = res.Body.Close() | ||||
| 	body, err = GetResponseBody("GET", url, channel, GetAuthHeader(channel.Key)) | ||||
| 	if err != nil { | ||||
| 		return 0, err | ||||
| 	} | ||||
|   | ||||
| @@ -27,6 +27,8 @@ func testChannel(channel *model.Channel, request *ChatRequest) error { | ||||
| 	} else { | ||||
| 		if channel.Type == common.ChannelTypeCustom { | ||||
| 			requestURL = channel.BaseURL | ||||
| 		} else if channel.Type == common.ChannelTypeOpenAI && channel.BaseURL != "" { | ||||
| 			requestURL = channel.BaseURL | ||||
| 		} | ||||
| 		requestURL += "/v1/chat/completions" | ||||
| 	} | ||||
| @@ -57,7 +59,7 @@ func testChannel(channel *model.Channel, request *ChatRequest) error { | ||||
| 		return err | ||||
| 	} | ||||
| 	if response.Usage.CompletionTokens == 0 { | ||||
| 		return errors.New(fmt.Sprintf("type %s, code %s, message %s", response.Error.Type, response.Error.Code, response.Error.Message)) | ||||
| 		return errors.New(fmt.Sprintf("type %s, code %v, message %s", response.Error.Type, response.Error.Code, response.Error.Message)) | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|   | ||||
							
								
								
									
										19
									
								
								controller/group.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										19
									
								
								controller/group.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,19 @@ | ||||
| package controller | ||||
|  | ||||
| import ( | ||||
| 	"github.com/gin-gonic/gin" | ||||
| 	"net/http" | ||||
| 	"one-api/common" | ||||
| ) | ||||
|  | ||||
| func GetGroups(c *gin.Context) { | ||||
| 	groupNames := make([]string, 0) | ||||
| 	for groupName, _ := range common.GroupRatio { | ||||
| 		groupNames = append(groupNames, groupName) | ||||
| 	} | ||||
| 	c.JSON(http.StatusOK, gin.H{ | ||||
| 		"success": true, | ||||
| 		"message": "", | ||||
| 		"data":    groupNames, | ||||
| 	}) | ||||
| } | ||||
							
								
								
									
										86
									
								
								controller/log.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										86
									
								
								controller/log.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,86 @@ | ||||
| package controller | ||||
|  | ||||
| import ( | ||||
| 	"github.com/gin-gonic/gin" | ||||
| 	"one-api/common" | ||||
| 	"one-api/model" | ||||
| 	"strconv" | ||||
| ) | ||||
|  | ||||
| func GetAllLogs(c *gin.Context) { | ||||
| 	p, _ := strconv.Atoi(c.Query("p")) | ||||
| 	if p < 0 { | ||||
| 		p = 0 | ||||
| 	} | ||||
| 	logType, _ := strconv.Atoi(c.Query("type")) | ||||
| 	logs, err := model.GetAllLogs(logType, p*common.ItemsPerPage, common.ItemsPerPage) | ||||
| 	if err != nil { | ||||
| 		c.JSON(200, gin.H{ | ||||
| 			"success": false, | ||||
| 			"message": err.Error(), | ||||
| 		}) | ||||
| 		return | ||||
| 	} | ||||
| 	c.JSON(200, gin.H{ | ||||
| 		"success": true, | ||||
| 		"message": "", | ||||
| 		"data":    logs, | ||||
| 	}) | ||||
| } | ||||
|  | ||||
| func GetUserLogs(c *gin.Context) { | ||||
| 	p, _ := strconv.Atoi(c.Query("p")) | ||||
| 	if p < 0 { | ||||
| 		p = 0 | ||||
| 	} | ||||
| 	userId := c.GetInt("id") | ||||
| 	logType, _ := strconv.Atoi(c.Query("type")) | ||||
| 	logs, err := model.GetUserLogs(userId, logType, p*common.ItemsPerPage, common.ItemsPerPage) | ||||
| 	if err != nil { | ||||
| 		c.JSON(200, gin.H{ | ||||
| 			"success": false, | ||||
| 			"message": err.Error(), | ||||
| 		}) | ||||
| 		return | ||||
| 	} | ||||
| 	c.JSON(200, gin.H{ | ||||
| 		"success": true, | ||||
| 		"message": "", | ||||
| 		"data":    logs, | ||||
| 	}) | ||||
| } | ||||
|  | ||||
| func SearchAllLogs(c *gin.Context) { | ||||
| 	keyword := c.Query("keyword") | ||||
| 	logs, err := model.SearchAllLogs(keyword) | ||||
| 	if err != nil { | ||||
| 		c.JSON(200, gin.H{ | ||||
| 			"success": false, | ||||
| 			"message": err.Error(), | ||||
| 		}) | ||||
| 		return | ||||
| 	} | ||||
| 	c.JSON(200, gin.H{ | ||||
| 		"success": true, | ||||
| 		"message": "", | ||||
| 		"data":    logs, | ||||
| 	}) | ||||
| } | ||||
|  | ||||
| func SearchUserLogs(c *gin.Context) { | ||||
| 	keyword := c.Query("keyword") | ||||
| 	userId := c.GetInt("id") | ||||
| 	logs, err := model.SearchUserLogs(userId, keyword) | ||||
| 	if err != nil { | ||||
| 		c.JSON(200, gin.H{ | ||||
| 			"success": false, | ||||
| 			"message": err.Error(), | ||||
| 		}) | ||||
| 		return | ||||
| 	} | ||||
| 	c.JSON(200, gin.H{ | ||||
| 		"success": true, | ||||
| 		"message": "", | ||||
| 		"data":    logs, | ||||
| 	}) | ||||
| } | ||||
| @@ -71,6 +71,33 @@ func init() { | ||||
| 			Root:       "gpt-3.5-turbo-0301", | ||||
| 			Parent:     nil, | ||||
| 		}, | ||||
| 		{ | ||||
| 			Id:         "gpt-3.5-turbo-0613", | ||||
| 			Object:     "model", | ||||
| 			Created:    1677649963, | ||||
| 			OwnedBy:    "openai", | ||||
| 			Permission: permission, | ||||
| 			Root:       "gpt-3.5-turbo-0613", | ||||
| 			Parent:     nil, | ||||
| 		}, | ||||
| 		{ | ||||
| 			Id:         "gpt-3.5-turbo-16k", | ||||
| 			Object:     "model", | ||||
| 			Created:    1677649963, | ||||
| 			OwnedBy:    "openai", | ||||
| 			Permission: permission, | ||||
| 			Root:       "gpt-3.5-turbo-16k", | ||||
| 			Parent:     nil, | ||||
| 		}, | ||||
| 		{ | ||||
| 			Id:         "gpt-3.5-turbo-16k-0613", | ||||
| 			Object:     "model", | ||||
| 			Created:    1677649963, | ||||
| 			OwnedBy:    "openai", | ||||
| 			Permission: permission, | ||||
| 			Root:       "gpt-3.5-turbo-16k-0613", | ||||
| 			Parent:     nil, | ||||
| 		}, | ||||
| 		{ | ||||
| 			Id:         "gpt-4", | ||||
| 			Object:     "model", | ||||
| @@ -89,6 +116,15 @@ func init() { | ||||
| 			Root:       "gpt-4-0314", | ||||
| 			Parent:     nil, | ||||
| 		}, | ||||
| 		{ | ||||
| 			Id:         "gpt-4-0613", | ||||
| 			Object:     "model", | ||||
| 			Created:    1677649963, | ||||
| 			OwnedBy:    "openai", | ||||
| 			Permission: permission, | ||||
| 			Root:       "gpt-4-0613", | ||||
| 			Parent:     nil, | ||||
| 		}, | ||||
| 		{ | ||||
| 			Id:         "gpt-4-32k", | ||||
| 			Object:     "model", | ||||
| @@ -107,6 +143,15 @@ func init() { | ||||
| 			Root:       "gpt-4-32k-0314", | ||||
| 			Parent:     nil, | ||||
| 		}, | ||||
| 		{ | ||||
| 			Id:         "gpt-4-32k-0613", | ||||
| 			Object:     "model", | ||||
| 			Created:    1677649963, | ||||
| 			OwnedBy:    "openai", | ||||
| 			Permission: permission, | ||||
| 			Root:       "gpt-4-32k-0613", | ||||
| 			Parent:     nil, | ||||
| 		}, | ||||
| 		{ | ||||
| 			Id:         "text-embedding-ada-002", | ||||
| 			Object:     "model", | ||||
| @@ -161,6 +206,24 @@ func init() { | ||||
| 			Root:       "text-ada-001", | ||||
| 			Parent:     nil, | ||||
| 		}, | ||||
| 		{ | ||||
| 			Id:         "text-moderation-latest", | ||||
| 			Object:     "model", | ||||
| 			Created:    1677649963, | ||||
| 			OwnedBy:    "openai", | ||||
| 			Permission: permission, | ||||
| 			Root:       "text-moderation-latest", | ||||
| 			Parent:     nil, | ||||
| 		}, | ||||
| 		{ | ||||
| 			Id:         "text-moderation-stable", | ||||
| 			Object:     "model", | ||||
| 			Created:    1677649963, | ||||
| 			OwnedBy:    "openai", | ||||
| 			Permission: permission, | ||||
| 			Root:       "text-moderation-stable", | ||||
| 			Parent:     nil, | ||||
| 		}, | ||||
| 	} | ||||
| 	openAIModelsMap = make(map[string]OpenAIModels) | ||||
| 	for _, model := range openAIModels { | ||||
|   | ||||
| @@ -58,6 +58,20 @@ func countTokenMessages(messages []Message, model string) int { | ||||
| 	return tokenNum | ||||
| } | ||||
|  | ||||
| func countTokenInput(input any, model string) int { | ||||
| 	switch input.(type) { | ||||
| 	case string: | ||||
| 		return countTokenText(input.(string), model) | ||||
| 	case []string: | ||||
| 		text := "" | ||||
| 		for _, s := range input.([]string) { | ||||
| 			text += s | ||||
| 		} | ||||
| 		return countTokenText(text, model) | ||||
| 	} | ||||
| 	return 0 | ||||
| } | ||||
|  | ||||
| func countTokenText(text string, model string) int { | ||||
| 	tokenEncoder := getTokenEncoder(model) | ||||
| 	token := tokenEncoder.Encode(text, nil, nil) | ||||
|   | ||||
| @@ -24,6 +24,7 @@ const ( | ||||
| 	RelayModeChatCompletions | ||||
| 	RelayModeCompletions | ||||
| 	RelayModeEmbeddings | ||||
| 	RelayModeModeration | ||||
| ) | ||||
|  | ||||
| // https://platform.openai.com/docs/api-reference/chat | ||||
| @@ -31,12 +32,13 @@ const ( | ||||
| type GeneralOpenAIRequest struct { | ||||
| 	Model       string    `json:"model"` | ||||
| 	Messages    []Message `json:"messages"` | ||||
| 	Prompt      string    `json:"prompt"` | ||||
| 	Prompt      any       `json:"prompt"` | ||||
| 	Stream      bool      `json:"stream"` | ||||
| 	MaxTokens   int       `json:"max_tokens"` | ||||
| 	Temperature float64   `json:"temperature"` | ||||
| 	TopP        float64   `json:"top_p"` | ||||
| 	N           int       `json:"n"` | ||||
| 	Input       any       `json:"input"` | ||||
| } | ||||
|  | ||||
| type ChatRequest struct { | ||||
| @@ -63,7 +65,7 @@ type OpenAIError struct { | ||||
| 	Message string `json:"message"` | ||||
| 	Type    string `json:"type"` | ||||
| 	Param   string `json:"param"` | ||||
| 	Code    string `json:"code"` | ||||
| 	Code    any    `json:"code"` | ||||
| } | ||||
|  | ||||
| type OpenAIErrorWithStatusCode struct { | ||||
| @@ -100,11 +102,13 @@ func Relay(c *gin.Context) { | ||||
| 		relayMode = RelayModeCompletions | ||||
| 	} else if strings.HasPrefix(c.Request.URL.Path, "/v1/embeddings") { | ||||
| 		relayMode = RelayModeEmbeddings | ||||
| 	} else if strings.HasPrefix(c.Request.URL.Path, "/v1/moderations") { | ||||
| 		relayMode = RelayModeModeration | ||||
| 	} | ||||
| 	err := relayHelper(c, relayMode) | ||||
| 	if err != nil { | ||||
| 		if err.StatusCode == http.StatusTooManyRequests { | ||||
| 			err.OpenAIError.Message = "负载已满,请稍后再试,或升级账户以提升服务质量。" | ||||
| 			err.OpenAIError.Message = "当前分组负载已饱和,请稍后再试,或升级账户以提升服务质量。" | ||||
| 		} | ||||
| 		c.JSON(err.StatusCode, gin.H{ | ||||
| 			"error": err.OpenAIError, | ||||
| @@ -136,6 +140,7 @@ func relayHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode { | ||||
| 	channelType := c.GetInt("channel") | ||||
| 	tokenId := c.GetInt("token_id") | ||||
| 	consumeQuota := c.GetBool("consume_quota") | ||||
| 	group := c.GetString("group") | ||||
| 	var textRequest GeneralOpenAIRequest | ||||
| 	if consumeQuota || channelType == common.ChannelTypeAzure || channelType == common.ChannelTypePaLM { | ||||
| 		err := common.UnmarshalBodyReusable(c, &textRequest) | ||||
| @@ -143,10 +148,17 @@ func relayHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode { | ||||
| 			return errorWrapper(err, "bind_request_body_failed", http.StatusBadRequest) | ||||
| 		} | ||||
| 	} | ||||
| 	if relayMode == RelayModeModeration && textRequest.Model == "" { | ||||
| 		textRequest.Model = "text-moderation-latest" | ||||
| 	} | ||||
| 	baseURL := common.ChannelBaseURLs[channelType] | ||||
| 	requestURL := c.Request.URL.String() | ||||
| 	if channelType == common.ChannelTypeCustom { | ||||
| 		baseURL = c.GetString("base_url") | ||||
| 	} else if channelType == common.ChannelTypeOpenAI { | ||||
| 		if c.GetString("base_url") != "" { | ||||
| 			baseURL = c.GetString("base_url") | ||||
| 		} | ||||
| 	} | ||||
| 	fullRequestURL := fmt.Sprintf("%s%s", baseURL, requestURL) | ||||
| 	if channelType == common.ChannelTypeAzure { | ||||
| @@ -165,6 +177,7 @@ func relayHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode { | ||||
| 		// https://github.com/songquanpeng/one-api/issues/67 | ||||
| 		model_ = strings.TrimSuffix(model_, "-0301") | ||||
| 		model_ = strings.TrimSuffix(model_, "-0314") | ||||
| 		model_ = strings.TrimSuffix(model_, "-0613") | ||||
| 		fullRequestURL = fmt.Sprintf("%s/openai/deployments/%s/%s", baseURL, model_, task) | ||||
| 	} else if channelType == common.ChannelTypePaLM { | ||||
| 		err := relayPaLM(textRequest, c) | ||||
| @@ -175,13 +188,17 @@ func relayHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode { | ||||
| 	case RelayModeChatCompletions: | ||||
| 		promptTokens = countTokenMessages(textRequest.Messages, textRequest.Model) | ||||
| 	case RelayModeCompletions: | ||||
| 		promptTokens = countTokenText(textRequest.Prompt, textRequest.Model) | ||||
| 		promptTokens = countTokenInput(textRequest.Prompt, textRequest.Model) | ||||
| 	case RelayModeModeration: | ||||
| 		promptTokens = countTokenInput(textRequest.Input, textRequest.Model) | ||||
| 	} | ||||
| 	preConsumedTokens := common.PreConsumedQuota | ||||
| 	if textRequest.MaxTokens != 0 { | ||||
| 		preConsumedTokens = promptTokens + textRequest.MaxTokens | ||||
| 	} | ||||
| 	ratio := common.GetModelRatio(textRequest.Model) | ||||
| 	modelRatio := common.GetModelRatio(textRequest.Model) | ||||
| 	groupRatio := common.GetGroupRatio(group) | ||||
| 	ratio := modelRatio * groupRatio | ||||
| 	preConsumedQuota := int(float64(preConsumedTokens) * ratio) | ||||
| 	if consumeQuota { | ||||
| 		err := model.PreConsumeTokenQuota(tokenId, preConsumedQuota) | ||||
| @@ -223,23 +240,30 @@ func relayHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode { | ||||
| 	defer func() { | ||||
| 		if consumeQuota { | ||||
| 			quota := 0 | ||||
| 			usingGPT4 := strings.HasPrefix(textRequest.Model, "gpt-4") | ||||
| 			completionRatio := 1 | ||||
| 			if usingGPT4 { | ||||
| 			completionRatio := 1.34 // default for gpt-3 | ||||
| 			if strings.HasPrefix(textRequest.Model, "gpt-4") { | ||||
| 				completionRatio = 2 | ||||
| 			} | ||||
| 			if isStream { | ||||
| 				responseTokens := countTokenText(streamResponseText, textRequest.Model) | ||||
| 				quota = promptTokens + responseTokens*completionRatio | ||||
| 				quota = promptTokens + int(float64(responseTokens)*completionRatio) | ||||
| 			} else { | ||||
| 				quota = textResponse.Usage.PromptTokens + textResponse.Usage.CompletionTokens*completionRatio | ||||
| 				quota = textResponse.Usage.PromptTokens + int(float64(textResponse.Usage.CompletionTokens)*completionRatio) | ||||
| 			} | ||||
| 			quota = int(float64(quota) * ratio) | ||||
| 			if ratio != 0 && quota <= 0 { | ||||
| 				quota = 1 | ||||
| 			} | ||||
| 			quotaDelta := quota - preConsumedQuota | ||||
| 			err := model.PostConsumeTokenQuota(tokenId, quotaDelta) | ||||
| 			if err != nil { | ||||
| 				common.SysError("Error consuming token remain quota: " + err.Error()) | ||||
| 			} | ||||
| 			userId := c.GetInt("id") | ||||
| 			model.RecordLog(userId, model.LogTypeConsume, fmt.Sprintf("使用模型 %s 消耗 %d 点额度(模型倍率 %.2f,分组倍率 %.2f)", textRequest.Model, quota, modelRatio, groupRatio)) | ||||
| 			model.UpdateUserUsedQuotaAndRequestCount(userId, quota) | ||||
| 			channelId := c.GetInt("channel_id") | ||||
| 			model.UpdateChannelUsedQuota(channelId, quota) | ||||
| 		} | ||||
| 	}() | ||||
|  | ||||
| @@ -374,3 +398,15 @@ func RelayNotImplemented(c *gin.Context) { | ||||
| 		"error": err, | ||||
| 	}) | ||||
| } | ||||
|  | ||||
| func RelayNotFound(c *gin.Context) { | ||||
| 	err := OpenAIError{ | ||||
| 		Message: fmt.Sprintf("API not found: %s:%s", c.Request.Method, c.Request.URL.Path), | ||||
| 		Type:    "one_api_error", | ||||
| 		Param:   "", | ||||
| 		Code:    "api_not_found", | ||||
| 	} | ||||
| 	c.JSON(http.StatusOK, gin.H{ | ||||
| 		"error": err, | ||||
| 	}) | ||||
| } | ||||
|   | ||||
| @@ -2,6 +2,7 @@ package controller | ||||
|  | ||||
| import ( | ||||
| 	"encoding/json" | ||||
| 	"fmt" | ||||
| 	"github.com/gin-contrib/sessions" | ||||
| 	"github.com/gin-gonic/gin" | ||||
| 	"net/http" | ||||
| @@ -351,6 +352,9 @@ func UpdateUser(c *gin.Context) { | ||||
| 		}) | ||||
| 		return | ||||
| 	} | ||||
| 	if originUser.Quota != updatedUser.Quota { | ||||
| 		model.RecordLog(originUser.Id, model.LogTypeManage, fmt.Sprintf("管理员将用户额度从 %d 点修改为 %d 点", originUser.Quota, updatedUser.Quota)) | ||||
| 	} | ||||
| 	c.JSON(http.StatusOK, gin.H{ | ||||
| 		"success": true, | ||||
| 		"message": "", | ||||
|   | ||||
| @@ -7,6 +7,7 @@ import ( | ||||
| 	"one-api/common" | ||||
| 	"one-api/model" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
| ) | ||||
|  | ||||
| type ModelRequest struct { | ||||
| @@ -15,6 +16,9 @@ type ModelRequest struct { | ||||
|  | ||||
| func Distribute() func(c *gin.Context) { | ||||
| 	return func(c *gin.Context) { | ||||
| 		userId := c.GetInt("id") | ||||
| 		userGroup, _ := model.GetUserGroup(userId) | ||||
| 		c.Set("group", userGroup) | ||||
| 		var channel *model.Channel | ||||
| 		channelId, ok := c.Get("channelId") | ||||
| 		if ok { | ||||
| @@ -64,8 +68,11 @@ func Distribute() func(c *gin.Context) { | ||||
| 				c.Abort() | ||||
| 				return | ||||
| 			} | ||||
| 			userId := c.GetInt("id") | ||||
| 			userGroup, _ := model.GetUserGroup(userId) | ||||
| 			if strings.HasPrefix(c.Request.URL.Path, "/v1/moderations") { | ||||
| 				if modelRequest.Model == "" { | ||||
| 					modelRequest.Model = "text-moderation-stable" | ||||
| 				} | ||||
| 			} | ||||
| 			channel, err = model.GetRandomSatisfiedChannel(userGroup, modelRequest.Model) | ||||
| 			if err != nil { | ||||
| 				c.JSON(200, gin.H{ | ||||
| @@ -82,12 +89,10 @@ func Distribute() func(c *gin.Context) { | ||||
| 		c.Set("channel_id", channel.Id) | ||||
| 		c.Set("channel_name", channel.Name) | ||||
| 		c.Request.Header.Set("Authorization", fmt.Sprintf("Bearer %s", channel.Key)) | ||||
| 		if channel.Type == common.ChannelTypeCustom || channel.Type == common.ChannelTypeAzure { | ||||
| 		c.Set("base_url", channel.BaseURL) | ||||
| 		if channel.Type == common.ChannelTypeAzure { | ||||
| 			c.Set("api_version", channel.Other) | ||||
| 		} | ||||
| 		} | ||||
| 		c.Next() | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -13,9 +13,6 @@ type Ability struct { | ||||
| } | ||||
|  | ||||
| func GetRandomSatisfiedChannel(group string, model string) (*Channel, error) { | ||||
| 	if group == "default" { | ||||
| 		return GetRandomChannel() | ||||
| 	} | ||||
| 	ability := Ability{} | ||||
| 	var err error = nil | ||||
| 	if common.UsingSQLite { | ||||
| @@ -33,16 +30,19 @@ func GetRandomSatisfiedChannel(group string, model string) (*Channel, error) { | ||||
|  | ||||
| func (channel *Channel) AddAbilities() error { | ||||
| 	models_ := strings.Split(channel.Models, ",") | ||||
| 	groups_ := strings.Split(channel.Group, ",") | ||||
| 	abilities := make([]Ability, 0, len(models_)) | ||||
| 	for _, model := range models_ { | ||||
| 		for _, group := range groups_ { | ||||
| 			ability := Ability{ | ||||
| 			Group:     channel.Group, | ||||
| 				Group:     group, | ||||
| 				Model:     model, | ||||
| 				ChannelId: channel.Id, | ||||
| 				Enabled:   channel.Status == common.ChannelStatusEnabled, | ||||
| 			} | ||||
| 			abilities = append(abilities, ability) | ||||
| 		} | ||||
| 	} | ||||
| 	return DB.Create(&abilities).Error | ||||
| } | ||||
|  | ||||
|   | ||||
| @@ -1,6 +1,7 @@ | ||||
| package model | ||||
|  | ||||
| import ( | ||||
| 	"gorm.io/gorm" | ||||
| 	"one-api/common" | ||||
| ) | ||||
|  | ||||
| @@ -20,6 +21,7 @@ type Channel struct { | ||||
| 	BalanceUpdatedTime int64   `json:"balance_updated_time" gorm:"bigint"` | ||||
| 	Models             string  `json:"models"` | ||||
| 	Group              string  `json:"group" gorm:"type:varchar(32);default:'default'"` | ||||
| 	UsedQuota          int64   `json:"used_quota" gorm:"bigint;default:0"` | ||||
| } | ||||
|  | ||||
| func GetAllChannels(startIdx int, num int, selectAll bool) ([]*Channel, error) { | ||||
| @@ -136,3 +138,10 @@ func UpdateChannelStatusById(id int, status int) { | ||||
| 		common.SysError("failed to update channel status: " + err.Error()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func UpdateChannelUsedQuota(id int, quota int) { | ||||
| 	err := DB.Model(&Channel{}).Where("id = ?", id).Update("used_quota", gorm.Expr("used_quota + ?", quota)).Error | ||||
| 	if err != nil { | ||||
| 		common.SysError("failed to update channel used quota: " + err.Error()) | ||||
| 	} | ||||
| } | ||||
|   | ||||
							
								
								
									
										70
									
								
								model/log.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										70
									
								
								model/log.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,70 @@ | ||||
| package model | ||||
|  | ||||
| import ( | ||||
| 	"gorm.io/gorm" | ||||
| 	"one-api/common" | ||||
| ) | ||||
|  | ||||
| type Log struct { | ||||
| 	Id        int    `json:"id"` | ||||
| 	UserId    int    `json:"user_id" gorm:"index"` | ||||
| 	CreatedAt int64  `json:"created_at" gorm:"bigint"` | ||||
| 	Type      int    `json:"type" gorm:"index"` | ||||
| 	Content   string `json:"content"` | ||||
| } | ||||
|  | ||||
| const ( | ||||
| 	LogTypeUnknown = iota | ||||
| 	LogTypeTopup | ||||
| 	LogTypeConsume | ||||
| 	LogTypeManage | ||||
| 	LogTypeSystem | ||||
| ) | ||||
|  | ||||
| func RecordLog(userId int, logType int, content string) { | ||||
| 	if logType == LogTypeConsume && !common.LogConsumeEnabled { | ||||
| 		return | ||||
| 	} | ||||
| 	log := &Log{ | ||||
| 		UserId:    userId, | ||||
| 		CreatedAt: common.GetTimestamp(), | ||||
| 		Type:      logType, | ||||
| 		Content:   content, | ||||
| 	} | ||||
| 	err := DB.Create(log).Error | ||||
| 	if err != nil { | ||||
| 		common.SysError("failed to record log: " + err.Error()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func GetAllLogs(logType int, startIdx int, num int) (logs []*Log, err error) { | ||||
| 	var tx *gorm.DB | ||||
| 	if logType == LogTypeUnknown { | ||||
| 		tx = DB | ||||
| 	} else { | ||||
| 		tx = DB.Where("type = ?", logType) | ||||
| 	} | ||||
| 	err = tx.Order("id desc").Limit(num).Offset(startIdx).Find(&logs).Error | ||||
| 	return logs, err | ||||
| } | ||||
|  | ||||
| func GetUserLogs(userId int, logType int, startIdx int, num int) (logs []*Log, err error) { | ||||
| 	var tx *gorm.DB | ||||
| 	if logType == LogTypeUnknown { | ||||
| 		tx = DB.Where("user_id = ?", userId) | ||||
| 	} else { | ||||
| 		tx = DB.Where("user_id = ? and type = ?", userId, logType) | ||||
| 	} | ||||
| 	err = tx.Order("id desc").Limit(num).Offset(startIdx).Omit("id").Find(&logs).Error | ||||
| 	return logs, err | ||||
| } | ||||
|  | ||||
| func SearchAllLogs(keyword string) (logs []*Log, err error) { | ||||
| 	err = DB.Where("type = ? or content LIKE ?", keyword, keyword+"%").Order("id desc").Limit(common.MaxRecentItems).Find(&logs).Error | ||||
| 	return logs, err | ||||
| } | ||||
|  | ||||
| func SearchUserLogs(userId int, keyword string) (logs []*Log, err error) { | ||||
| 	err = DB.Where("user_id = ? and type = ?", userId, keyword).Order("id desc").Limit(common.MaxRecentItems).Omit("id").Find(&logs).Error | ||||
| 	return logs, err | ||||
| } | ||||
| @@ -79,6 +79,10 @@ func InitDB() (err error) { | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		err = db.AutoMigrate(&Log{}) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		err = createRootAccountIfNeed() | ||||
| 		return err | ||||
| 	} else { | ||||
|   | ||||
| @@ -34,6 +34,7 @@ func InitOptionMap() { | ||||
| 	common.OptionMap["TurnstileCheckEnabled"] = strconv.FormatBool(common.TurnstileCheckEnabled) | ||||
| 	common.OptionMap["RegisterEnabled"] = strconv.FormatBool(common.RegisterEnabled) | ||||
| 	common.OptionMap["AutomaticDisableChannelEnabled"] = strconv.FormatBool(common.AutomaticDisableChannelEnabled) | ||||
| 	common.OptionMap["LogConsumeEnabled"] = strconv.FormatBool(common.LogConsumeEnabled) | ||||
| 	common.OptionMap["ChannelDisableThreshold"] = strconv.FormatFloat(common.ChannelDisableThreshold, 'f', -1, 64) | ||||
| 	common.OptionMap["SMTPServer"] = "" | ||||
| 	common.OptionMap["SMTPFrom"] = "" | ||||
| @@ -58,6 +59,7 @@ func InitOptionMap() { | ||||
| 	common.OptionMap["QuotaRemindThreshold"] = strconv.Itoa(common.QuotaRemindThreshold) | ||||
| 	common.OptionMap["PreConsumedQuota"] = strconv.Itoa(common.PreConsumedQuota) | ||||
| 	common.OptionMap["ModelRatio"] = common.ModelRatio2JSONString() | ||||
| 	common.OptionMap["GroupRatio"] = common.GroupRatio2JSONString() | ||||
| 	common.OptionMap["TopUpLink"] = common.TopUpLink | ||||
| 	common.OptionMapRWMutex.Unlock() | ||||
| 	loadOptionsFromDatabase() | ||||
| @@ -133,6 +135,8 @@ func updateOptionMap(key string, value string) (err error) { | ||||
| 			common.RegisterEnabled = boolValue | ||||
| 		case "AutomaticDisableChannelEnabled": | ||||
| 			common.AutomaticDisableChannelEnabled = boolValue | ||||
| 		case "LogConsumeEnabled": | ||||
| 			common.LogConsumeEnabled = boolValue | ||||
| 		} | ||||
| 	} | ||||
| 	switch key { | ||||
| @@ -177,6 +181,8 @@ func updateOptionMap(key string, value string) (err error) { | ||||
| 		common.PreConsumedQuota, _ = strconv.Atoi(value) | ||||
| 	case "ModelRatio": | ||||
| 		err = common.UpdateModelRatioByJSONString(value) | ||||
| 	case "GroupRatio": | ||||
| 		err = common.UpdateGroupRatioByJSONString(value) | ||||
| 	case "TopUpLink": | ||||
| 		common.TopUpLink = value | ||||
| 	case "ChannelDisableThreshold": | ||||
|   | ||||
| @@ -2,6 +2,7 @@ package model | ||||
|  | ||||
| import ( | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"one-api/common" | ||||
| ) | ||||
|  | ||||
| @@ -65,6 +66,7 @@ func Redeem(key string, userId int) (quota int, err error) { | ||||
| 		if err != nil { | ||||
| 			common.SysError("更新兑换码状态失败:" + err.Error()) | ||||
| 		} | ||||
| 		RecordLog(userId, LogTypeTopup, fmt.Sprintf("通过兑换码充值 %d 点额度", redemption.Quota)) | ||||
| 	}() | ||||
| 	return redemption.Quota, nil | ||||
| } | ||||
|   | ||||
| @@ -2,6 +2,7 @@ package model | ||||
|  | ||||
| import ( | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"gorm.io/gorm" | ||||
| 	"one-api/common" | ||||
| 	"strings" | ||||
| @@ -22,6 +23,8 @@ type User struct { | ||||
| 	VerificationCode string `json:"verification_code" gorm:"-:all"`                                    // this field is only for Email verification, don't save it to database! | ||||
| 	AccessToken      string `json:"access_token" gorm:"type:char(32);column:access_token;uniqueIndex"` // this token is for system management | ||||
| 	Quota            int    `json:"quota" gorm:"type:int;default:0"` | ||||
| 	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'"` | ||||
| } | ||||
|  | ||||
| @@ -73,8 +76,14 @@ func (user *User) Insert() error { | ||||
| 	} | ||||
| 	user.Quota = common.QuotaForNewUser | ||||
| 	user.AccessToken = common.GetUUID() | ||||
| 	err = DB.Create(user).Error | ||||
| 	return err | ||||
| 	result := DB.Create(user) | ||||
| 	if result.Error != nil { | ||||
| 		return result.Error | ||||
| 	} | ||||
| 	if common.QuotaForNewUser > 0 { | ||||
| 		RecordLog(user.Id, LogTypeSystem, fmt.Sprintf("新用户注册赠送 %d 点额度", common.QuotaForNewUser)) | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (user *User) Update(updatePassword bool) error { | ||||
| @@ -255,3 +264,15 @@ func GetRootUserEmail() (email string) { | ||||
| 	DB.Model(&User{}).Where("role = ?", common.RoleRootUser).Select("email").Find(&email) | ||||
| 	return email | ||||
| } | ||||
|  | ||||
| func UpdateUserUsedQuotaAndRequestCount(id int, quota int) { | ||||
| 	err := DB.Model(&User{}).Where("id = ?", id).Updates( | ||||
| 		map[string]interface{}{ | ||||
| 			"used_quota":    gorm.Expr("used_quota + ?", quota), | ||||
| 			"request_count": gorm.Expr("request_count + ?", 1), | ||||
| 		}, | ||||
| 	).Error | ||||
| 	if err != nil { | ||||
| 		common.SysError("Failed to update user used quota and request count: " + err.Error()) | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -93,5 +93,15 @@ func SetApiRouter(router *gin.Engine) { | ||||
| 			redemptionRoute.PUT("/", controller.UpdateRedemption) | ||||
| 			redemptionRoute.DELETE("/:id", controller.DeleteRedemption) | ||||
| 		} | ||||
| 		logRoute := apiRouter.Group("/log") | ||||
| 		logRoute.GET("/", middleware.AdminAuth(), controller.GetAllLogs) | ||||
| 		logRoute.GET("/search", middleware.AdminAuth(), controller.SearchAllLogs) | ||||
| 		logRoute.GET("/self", middleware.UserAuth(), controller.GetUserLogs) | ||||
| 		logRoute.GET("/self/search", middleware.UserAuth(), controller.SearchUserLogs) | ||||
| 		groupRoute := apiRouter.Group("/group") | ||||
| 		groupRoute.Use(middleware.AdminAuth()) | ||||
| 		{ | ||||
| 			groupRoute.GET("/", controller.GetGroups) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -37,6 +37,6 @@ func SetRelayRouter(router *gin.Engine) { | ||||
| 		relayV1Router.POST("/fine-tunes/:id/cancel", controller.RelayNotImplemented) | ||||
| 		relayV1Router.GET("/fine-tunes/:id/events", controller.RelayNotImplemented) | ||||
| 		relayV1Router.DELETE("/models/:model", controller.RelayNotImplemented) | ||||
| 		relayV1Router.POST("/moderations", controller.RelayNotImplemented) | ||||
| 		relayV1Router.POST("/moderations", controller.Relay) | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -7,7 +7,9 @@ import ( | ||||
| 	"github.com/gin-gonic/gin" | ||||
| 	"net/http" | ||||
| 	"one-api/common" | ||||
| 	"one-api/controller" | ||||
| 	"one-api/middleware" | ||||
| 	"strings" | ||||
| ) | ||||
|  | ||||
| func SetWebRouter(router *gin.Engine, buildFS embed.FS, indexPage []byte) { | ||||
| @@ -16,6 +18,10 @@ func SetWebRouter(router *gin.Engine, buildFS embed.FS, indexPage []byte) { | ||||
| 	router.Use(middleware.Cache()) | ||||
| 	router.Use(static.Serve("/", common.EmbedFolder(buildFS, "web/build"))) | ||||
| 	router.NoRoute(func(c *gin.Context) { | ||||
| 		if strings.HasPrefix(c.Request.RequestURI, "/v1") { | ||||
| 			controller.RelayNotFound(c) | ||||
| 			return | ||||
| 		} | ||||
| 		c.Header("Cache-Control", "no-cache") | ||||
| 		c.Data(http.StatusOK, "text/html; charset=utf-8", indexPage) | ||||
| 	}) | ||||
|   | ||||
| @@ -22,6 +22,7 @@ import EditChannel from './pages/Channel/EditChannel'; | ||||
| import Redemption from './pages/Redemption'; | ||||
| import EditRedemption from './pages/Redemption/EditRedemption'; | ||||
| import TopUp from './pages/TopUp'; | ||||
| import Log from './pages/Log'; | ||||
|  | ||||
| const Home = lazy(() => import('./pages/Home')); | ||||
| const About = lazy(() => import('./pages/About')); | ||||
| @@ -250,6 +251,14 @@ function App() { | ||||
|         </PrivateRoute> | ||||
|         } | ||||
|       /> | ||||
|       <Route | ||||
|         path='/log' | ||||
|         element={ | ||||
|           <PrivateRoute> | ||||
|             <Log /> | ||||
|           </PrivateRoute> | ||||
|         } | ||||
|       /> | ||||
|       <Route | ||||
|         path='/about' | ||||
|         element={ | ||||
|   | ||||
| @@ -4,7 +4,7 @@ import { Link } from 'react-router-dom'; | ||||
| import { API, showError, showInfo, showSuccess, timestamp2string } from '../helpers'; | ||||
|  | ||||
| import { CHANNEL_OPTIONS, ITEMS_PER_PAGE } from '../constants'; | ||||
| import { renderGroup } from '../helpers/render'; | ||||
| import { renderGroup, renderNumber } from '../helpers/render'; | ||||
|  | ||||
| function renderTimestamp(timestamp) { | ||||
|   return ( | ||||
| @@ -27,6 +27,20 @@ function renderType(type) { | ||||
|   return <Label basic color={type2label[type].color}>{type2label[type].text}</Label>; | ||||
| } | ||||
|  | ||||
| function renderBalance(type, balance) { | ||||
|   switch (type) { | ||||
|     case 1: // OpenAI | ||||
|     case 8: // 自定义 | ||||
|       return <span>${balance.toFixed(2)}</span>; | ||||
|     case 5: // OpenAI-SB | ||||
|       return <span>¥{(balance / 10000).toFixed(2)}</span>; | ||||
|     case 10: // AI Proxy | ||||
|       return <span>{renderNumber(balance)}</span>; | ||||
|     default: | ||||
|       return <span>不支持</span>; | ||||
|   } | ||||
| } | ||||
|  | ||||
| const ChannelsTable = () => { | ||||
|   const [channels, setChannels] = useState([]); | ||||
|   const [loading, setLoading] = useState(true); | ||||
| @@ -336,7 +350,7 @@ const ChannelsTable = () => { | ||||
|                     <Popup | ||||
|                       content={channel.balance_updated_time ? renderTimestamp(channel.balance_updated_time) : '未更新'} | ||||
|                       key={channel.id} | ||||
|                       trigger={<span>${channel.balance.toFixed(2)}</span>} | ||||
|                       trigger={renderBalance(channel.type, channel.balance)} | ||||
|                       basic | ||||
|                     /> | ||||
|                   </Table.Cell> | ||||
| @@ -415,7 +429,8 @@ const ChannelsTable = () => { | ||||
|               <Button size='small' loading={loading} onClick={testAllChannels}> | ||||
|                 测试所有已启用通道 | ||||
|               </Button> | ||||
|               <Button size='small' onClick={updateAllChannelsBalance} loading={loading || updatingBalance}>更新所有已启用通道余额</Button> | ||||
|               <Button size='small' onClick={updateAllChannelsBalance} | ||||
|                       loading={loading || updatingBalance}>更新所有已启用通道余额</Button> | ||||
|               <Pagination | ||||
|                 floated='right' | ||||
|                 activePage={activePage} | ||||
|   | ||||
| @@ -1,11 +1,31 @@ | ||||
| import React from 'react'; | ||||
| import React, { useEffect, useState } from 'react'; | ||||
|  | ||||
| import { Container, Segment } from 'semantic-ui-react'; | ||||
| import { getFooterHTML, getSystemName } from '../helpers'; | ||||
|  | ||||
| const Footer = () => { | ||||
|   const systemName = getSystemName(); | ||||
|   const footer = getFooterHTML(); | ||||
|   const [footer, setFooter] = useState(getFooterHTML()); | ||||
|   let remainCheckTimes = 5; | ||||
|  | ||||
|   const loadFooter = () => { | ||||
|     let footer_html = localStorage.getItem('footer_html'); | ||||
|     if (footer_html) { | ||||
|       setFooter(footer_html); | ||||
|     } | ||||
|   }; | ||||
|  | ||||
|   useEffect(() => { | ||||
|     const timer = setInterval(() => { | ||||
|       if (remainCheckTimes <= 0) { | ||||
|         clearInterval(timer); | ||||
|         return; | ||||
|       } | ||||
|       remainCheckTimes--; | ||||
|       loadFooter(); | ||||
|     }, 200); | ||||
|     return () => clearTimeout(timer); | ||||
|   }, []); | ||||
|  | ||||
|   return ( | ||||
|     <Segment vertical> | ||||
|   | ||||
| @@ -41,6 +41,11 @@ const headerButtons = [ | ||||
|     icon: 'user', | ||||
|     admin: true, | ||||
|   }, | ||||
|   { | ||||
|     name: '日志', | ||||
|     to: '/log', | ||||
|     icon: 'book', | ||||
|   }, | ||||
|   { | ||||
|     name: '设置', | ||||
|     to: '/setting', | ||||
|   | ||||
							
								
								
									
										256
									
								
								web/src/components/LogsTable.js
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										256
									
								
								web/src/components/LogsTable.js
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,256 @@ | ||||
| import React, { useEffect, useState } from 'react'; | ||||
| import { Button, Label, Pagination, Select, Table } from 'semantic-ui-react'; | ||||
| import { API, isAdmin, showError, timestamp2string } from '../helpers'; | ||||
|  | ||||
| import { ITEMS_PER_PAGE } from '../constants'; | ||||
|  | ||||
| function renderTimestamp(timestamp) { | ||||
|   return ( | ||||
|     <> | ||||
|       {timestamp2string(timestamp)} | ||||
|     </> | ||||
|   ); | ||||
| } | ||||
|  | ||||
| const MODE_OPTIONS = [ | ||||
|   { key: 'all', text: '全部用户', value: 'all' }, | ||||
|   { key: 'self', text: '当前用户', value: 'self' }, | ||||
| ]; | ||||
|  | ||||
| const LOG_OPTIONS = [ | ||||
|   { key: '0', text: '全部', value: 0 }, | ||||
|   { key: '1', text: '充值', value: 1 }, | ||||
|   { key: '2', text: '消费', value: 2 }, | ||||
|   { key: '3', text: '管理', value: 3 }, | ||||
|   { key: '4', text: '系统', value: 4 } | ||||
| ]; | ||||
|  | ||||
| function renderType(type) { | ||||
|   switch (type) { | ||||
|     case 1: | ||||
|       return <Label basic color='green'> 充值 </Label>; | ||||
|     case 2: | ||||
|       return <Label basic color='olive'> 消费 </Label>; | ||||
|     case 3: | ||||
|       return <Label basic color='orange'> 管理 </Label>; | ||||
|     case 4: | ||||
|       return <Label basic color='purple'> 系统 </Label>; | ||||
|     default: | ||||
|       return <Label basic color='black'> 未知 </Label>; | ||||
|   } | ||||
| } | ||||
|  | ||||
| const LogsTable = () => { | ||||
|   const [logs, setLogs] = useState([]); | ||||
|   const [loading, setLoading] = useState(true); | ||||
|   const [activePage, setActivePage] = useState(1); | ||||
|   const [searchKeyword, setSearchKeyword] = useState(''); | ||||
|   const [searching, setSearching] = useState(false); | ||||
|   const [logType, setLogType] = useState(0); | ||||
|   const [mode, setMode] = useState('self'); // all, self | ||||
|   const showModePanel = isAdmin(); | ||||
|  | ||||
|   const loadLogs = async (startIdx) => { | ||||
|     let url = `/api/log/self/?p=${startIdx}&type=${logType}`; | ||||
|     if (mode === 'all') { | ||||
|       url = `/api/log/?p=${startIdx}&type=${logType}`; | ||||
|     } | ||||
|     const res = await API.get(url); | ||||
|     const { success, message, data } = res.data; | ||||
|     if (success) { | ||||
|       if (startIdx === 0) { | ||||
|         setLogs(data); | ||||
|       } else { | ||||
|         let newLogs = logs; | ||||
|         newLogs.push(...data); | ||||
|         setLogs(newLogs); | ||||
|       } | ||||
|     } else { | ||||
|       showError(message); | ||||
|     } | ||||
|     setLoading(false); | ||||
|   }; | ||||
|  | ||||
|   const onPaginationChange = (e, { activePage }) => { | ||||
|     (async () => { | ||||
|       if (activePage === Math.ceil(logs.length / ITEMS_PER_PAGE) + 1) { | ||||
|         // In this case we have to load more data and then append them. | ||||
|         await loadLogs(activePage - 1); | ||||
|       } | ||||
|       setActivePage(activePage); | ||||
|     })(); | ||||
|   }; | ||||
|  | ||||
|   const refresh = async () => { | ||||
|     setLoading(true); | ||||
|     await loadLogs(0); | ||||
|   }; | ||||
|  | ||||
|   useEffect(() => { | ||||
|     loadLogs(0) | ||||
|       .then() | ||||
|       .catch((reason) => { | ||||
|         showError(reason); | ||||
|       }); | ||||
|   }, []); | ||||
|  | ||||
|   useEffect(() => { | ||||
|     refresh().then(); | ||||
|   }, [mode, logType]); | ||||
|  | ||||
|   const searchLogs = async () => { | ||||
|     if (searchKeyword === '') { | ||||
|       // if keyword is blank, load files instead. | ||||
|       await loadLogs(0); | ||||
|       setActivePage(1); | ||||
|       return; | ||||
|     } | ||||
|     setSearching(true); | ||||
|     const res = await API.get(`/api/log/self/search?keyword=${searchKeyword}`); | ||||
|     const { success, message, data } = res.data; | ||||
|     if (success) { | ||||
|       setLogs(data); | ||||
|       setActivePage(1); | ||||
|     } else { | ||||
|       showError(message); | ||||
|     } | ||||
|     setSearching(false); | ||||
|   }; | ||||
|  | ||||
|   const handleKeywordChange = async (e, { value }) => { | ||||
|     setSearchKeyword(value.trim()); | ||||
|   }; | ||||
|  | ||||
|   const sortLog = (key) => { | ||||
|     if (logs.length === 0) return; | ||||
|     setLoading(true); | ||||
|     let sortedLogs = [...logs]; | ||||
|     sortedLogs.sort((a, b) => { | ||||
|       return ('' + a[key]).localeCompare(b[key]); | ||||
|     }); | ||||
|     if (sortedLogs[0].id === logs[0].id) { | ||||
|       sortedLogs.reverse(); | ||||
|     } | ||||
|     setLogs(sortedLogs); | ||||
|     setLoading(false); | ||||
|   }; | ||||
|  | ||||
|   return ( | ||||
|     <> | ||||
|       <Table basic> | ||||
|         <Table.Header> | ||||
|           <Table.Row> | ||||
|             <Table.HeaderCell | ||||
|               style={{ cursor: 'pointer' }} | ||||
|               onClick={() => { | ||||
|                 sortLog('created_time'); | ||||
|               }} | ||||
|               width={3} | ||||
|             > | ||||
|               时间 | ||||
|             </Table.HeaderCell> | ||||
|             { | ||||
|               showModePanel && ( | ||||
|                 <Table.HeaderCell | ||||
|                   style={{ cursor: 'pointer' }} | ||||
|                   onClick={() => { | ||||
|                     sortLog('user_id'); | ||||
|                   }} | ||||
|                   width={1} | ||||
|                 > | ||||
|                   用户 | ||||
|                 </Table.HeaderCell> | ||||
|               ) | ||||
|             } | ||||
|             <Table.HeaderCell | ||||
|               style={{ cursor: 'pointer' }} | ||||
|               onClick={() => { | ||||
|                 sortLog('type'); | ||||
|               }} | ||||
|               width={2} | ||||
|             > | ||||
|               类型 | ||||
|             </Table.HeaderCell> | ||||
|             <Table.HeaderCell | ||||
|               style={{ cursor: 'pointer' }} | ||||
|               onClick={() => { | ||||
|                 sortLog('content'); | ||||
|               }} | ||||
|               width={showModePanel ? 10 : 11} | ||||
|             > | ||||
|               详情 | ||||
|             </Table.HeaderCell> | ||||
|           </Table.Row> | ||||
|         </Table.Header> | ||||
|  | ||||
|         <Table.Body> | ||||
|           {logs | ||||
|             .slice( | ||||
|               (activePage - 1) * ITEMS_PER_PAGE, | ||||
|               activePage * ITEMS_PER_PAGE | ||||
|             ) | ||||
|             .map((log, idx) => { | ||||
|               if (log.deleted) return <></>; | ||||
|               return ( | ||||
|                 <Table.Row key={log.created_at}> | ||||
|                   <Table.Cell>{renderTimestamp(log.created_at)}</Table.Cell> | ||||
|                   { | ||||
|                     showModePanel && ( | ||||
|                       <Table.Cell><Label>{log.user_id}</Label></Table.Cell> | ||||
|                     ) | ||||
|                   } | ||||
|                   <Table.Cell>{renderType(log.type)}</Table.Cell> | ||||
|                   <Table.Cell>{log.content}</Table.Cell> | ||||
|                 </Table.Row> | ||||
|               ); | ||||
|             })} | ||||
|         </Table.Body> | ||||
|  | ||||
|         <Table.Footer> | ||||
|           <Table.Row> | ||||
|             <Table.HeaderCell colSpan={showModePanel ? '5' : '4'}> | ||||
|               { | ||||
|                 showModePanel && ( | ||||
|                   <Select | ||||
|                     placeholder='选择模式' | ||||
|                     options={MODE_OPTIONS} | ||||
|                     style={{ marginRight: '8px' }} | ||||
|                     name='mode' | ||||
|                     value={mode} | ||||
|                     onChange={(e, { name, value }) => { | ||||
|                       setMode(value); | ||||
|                     }} | ||||
|                   /> | ||||
|                 ) | ||||
|               } | ||||
|               <Select | ||||
|                 placeholder='选择明细分类' | ||||
|                 options={LOG_OPTIONS} | ||||
|                 style={{ marginRight: '8px' }} | ||||
|                 name='logType' | ||||
|                 value={logType} | ||||
|                 onChange={(e, { name, value }) => { | ||||
|                   setLogType(value); | ||||
|                 }} | ||||
|               /> | ||||
|               <Button size='small' onClick={refresh} loading={loading}>刷新</Button> | ||||
|               <Pagination | ||||
|                 floated='right' | ||||
|                 activePage={activePage} | ||||
|                 onPageChange={onPaginationChange} | ||||
|                 size='small' | ||||
|                 siblingRange={1} | ||||
|                 totalPages={ | ||||
|                   Math.ceil(logs.length / ITEMS_PER_PAGE) + | ||||
|                   (logs.length % ITEMS_PER_PAGE === 0 ? 1 : 0) | ||||
|                 } | ||||
|               /> | ||||
|             </Table.HeaderCell> | ||||
|           </Table.Row> | ||||
|         </Table.Footer> | ||||
|       </Table> | ||||
|     </> | ||||
|   ); | ||||
| }; | ||||
|  | ||||
| export default LogsTable; | ||||
| @@ -1,5 +1,5 @@ | ||||
| import React, { useEffect, useState } from 'react'; | ||||
| import { Button, Divider, Form, Grid, Header, Modal } from 'semantic-ui-react'; | ||||
| import { Button, Divider, Form, Grid, Header, Message, Modal } from 'semantic-ui-react'; | ||||
| import { API, showError, showSuccess } from '../helpers'; | ||||
| import { marked } from 'marked'; | ||||
|  | ||||
| @@ -10,13 +10,13 @@ const OtherSetting = () => { | ||||
|     About: '', | ||||
|     SystemName: '', | ||||
|     Logo: '', | ||||
|     HomePageContent: '', | ||||
|     HomePageContent: '' | ||||
|   }); | ||||
|   let [loading, setLoading] = useState(false); | ||||
|   const [showUpdateModal, setShowUpdateModal] = useState(false); | ||||
|   const [updateData, setUpdateData] = useState({ | ||||
|     tag_name: '', | ||||
|     content: '', | ||||
|     content: '' | ||||
|   }); | ||||
|  | ||||
|   const getOptions = async () => { | ||||
| @@ -43,7 +43,7 @@ const OtherSetting = () => { | ||||
|     setLoading(true); | ||||
|     const res = await API.put('/api/option/', { | ||||
|       key, | ||||
|       value, | ||||
|       value | ||||
|     }); | ||||
|     const { success, message } = res.data; | ||||
|     if (success) { | ||||
| @@ -97,7 +97,7 @@ const OtherSetting = () => { | ||||
|     } else { | ||||
|       setUpdateData({ | ||||
|         tag_name: tag_name, | ||||
|         content: marked.parse(body), | ||||
|         content: marked.parse(body) | ||||
|       }); | ||||
|       setShowUpdateModal(true); | ||||
|     } | ||||
| @@ -165,6 +165,7 @@ const OtherSetting = () => { | ||||
|             /> | ||||
|           </Form.Group> | ||||
|           <Form.Button onClick={submitAbout}>保存关于</Form.Button> | ||||
|           <Message>移除 One API 的版权标识必须首先获得授权,后续版本将通过授权码强制执行。</Message> | ||||
|           <Form.Group widths='equal'> | ||||
|             <Form.Input | ||||
|               label='页脚' | ||||
|   | ||||
| @@ -30,9 +30,11 @@ const SystemSetting = () => { | ||||
|     QuotaRemindThreshold: 0, | ||||
|     PreConsumedQuota: 0, | ||||
|     ModelRatio: '', | ||||
|     GroupRatio: '', | ||||
|     TopUpLink: '', | ||||
|     AutomaticDisableChannelEnabled: '', | ||||
|     ChannelDisableThreshold: 0, | ||||
|     LogConsumeEnabled: '', | ||||
|   }); | ||||
|   const [originInputs, setOriginInputs] = useState({}); | ||||
|   let [loading, setLoading] = useState(false); | ||||
| @@ -67,6 +69,7 @@ const SystemSetting = () => { | ||||
|       case 'TurnstileCheckEnabled': | ||||
|       case 'RegisterEnabled': | ||||
|       case 'AutomaticDisableChannelEnabled': | ||||
|       case 'LogConsumeEnabled': | ||||
|         value = inputs[key] === 'true' ? 'false' : 'true'; | ||||
|         break; | ||||
|       default: | ||||
| @@ -101,6 +104,7 @@ const SystemSetting = () => { | ||||
|       name === 'QuotaRemindThreshold' || | ||||
|       name === 'PreConsumedQuota' || | ||||
|       name === 'ModelRatio' || | ||||
|       name === 'GroupRatio' || | ||||
|       name === 'TopUpLink' | ||||
|     ) { | ||||
|       setInputs((inputs) => ({ ...inputs, [name]: value })); | ||||
| @@ -131,6 +135,13 @@ const SystemSetting = () => { | ||||
|       } | ||||
|       await updateOption('ModelRatio', inputs.ModelRatio); | ||||
|     } | ||||
|     if (originInputs['GroupRatio'] !== inputs.GroupRatio) { | ||||
|       if (!verifyJSON(inputs.GroupRatio)) { | ||||
|         showError('分组倍率不是合法的 JSON 字符串'); | ||||
|         return; | ||||
|       } | ||||
|       await updateOption('GroupRatio', inputs.GroupRatio); | ||||
|     } | ||||
|     if (originInputs['TopUpLink'] !== inputs.TopUpLink) { | ||||
|       await updateOption('TopUpLink', inputs.TopUpLink); | ||||
|     } | ||||
| @@ -329,6 +340,23 @@ const SystemSetting = () => { | ||||
|               placeholder='为一个 JSON 文本,键为模型名称,值为倍率' | ||||
|             /> | ||||
|           </Form.Group> | ||||
|           <Form.Group widths='equal'> | ||||
|             <Form.TextArea | ||||
|               label='分组倍率' | ||||
|               name='GroupRatio' | ||||
|               onChange={handleInputChange} | ||||
|               style={{ minHeight: 250, fontFamily: 'JetBrains Mono, Consolas' }} | ||||
|               autoComplete='new-password' | ||||
|               value={inputs.GroupRatio} | ||||
|               placeholder='为一个 JSON 文本,键为分组名称,值为倍率' | ||||
|             /> | ||||
|           </Form.Group> | ||||
|           <Form.Checkbox | ||||
|             checked={inputs.LogConsumeEnabled === 'true'} | ||||
|             label='启用额度消费日志记录' | ||||
|             name='LogConsumeEnabled' | ||||
|             onChange={handleInputChange} | ||||
|           /> | ||||
|           <Form.Button onClick={submitOperationConfig}>保存运营设置</Form.Button> | ||||
|           <Divider /> | ||||
|           <Header as='h3'> | ||||
|   | ||||
| @@ -4,7 +4,7 @@ import { Link } from 'react-router-dom'; | ||||
| import { API, showError, showSuccess } from '../helpers'; | ||||
|  | ||||
| import { ITEMS_PER_PAGE } from '../constants'; | ||||
| import { renderGroup, renderText } from '../helpers/render'; | ||||
| import { renderGroup, renderNumber, renderText } from '../helpers/render'; | ||||
|  | ||||
| function renderRole(role) { | ||||
|   switch (role) { | ||||
| @@ -197,7 +197,7 @@ const UsersTable = () => { | ||||
|                 sortUser('quota'); | ||||
|               }} | ||||
|             > | ||||
|               剩余额度 | ||||
|               统计信息 | ||||
|             </Table.HeaderCell> | ||||
|             <Table.HeaderCell | ||||
|               style={{ cursor: 'pointer' }} | ||||
| @@ -241,7 +241,11 @@ const UsersTable = () => { | ||||
|                   </Table.Cell> | ||||
|                   <Table.Cell>{renderGroup(user.group)}</Table.Cell> | ||||
|                   <Table.Cell>{user.email ? renderText(user.email, 30) : '无'}</Table.Cell> | ||||
|                   <Table.Cell>{user.quota}</Table.Cell> | ||||
|                   <Table.Cell> | ||||
|                     <Popup content='剩余额度' trigger={<Label>{renderNumber(user.quota)}</Label>} /> | ||||
|                     <Popup content='已用额度' trigger={<Label>{renderNumber(user.used_quota)}</Label>} /> | ||||
|                     <Popup content='请求次数' trigger={<Label>{renderNumber(user.request_count)}</Label>} /> | ||||
|                   </Table.Cell> | ||||
|                   <Table.Cell>{renderRole(user.role)}</Table.Cell> | ||||
|                   <Table.Cell>{renderStatus(user.status)}</Table.Cell> | ||||
|                   <Table.Cell> | ||||
|   | ||||
| @@ -8,8 +8,31 @@ export function renderText(text, limit) { | ||||
| } | ||||
|  | ||||
| export function renderGroup(group) { | ||||
|   if (group === "") { | ||||
|     return <Label>default</Label> | ||||
|   if (group === '') { | ||||
|     return <Label>default</Label>; | ||||
|   } | ||||
|   let groups = group.split(','); | ||||
|   groups.sort(); | ||||
|   return <> | ||||
|     {groups.map((group) => { | ||||
|       if (group === 'vip' || group === 'pro') { | ||||
|         return <Label color='yellow'>{group}</Label>; | ||||
|       } else if (group === 'svip' || group === 'premium') { | ||||
|         return <Label color='red'>{group}</Label>; | ||||
|       } | ||||
|       return <Label>{group}</Label>; | ||||
|     })} | ||||
|   </>; | ||||
| } | ||||
|  | ||||
| export function renderNumber(num) { | ||||
|   if (num >= 1000000000) { | ||||
|     return (num / 1000000000).toFixed(1) + 'B'; | ||||
|   } else if (num >= 1000000) { | ||||
|     return (num / 1000000).toFixed(1) + 'M'; | ||||
|   } else if (num >= 10000) { | ||||
|     return (num / 1000).toFixed(1) + 'k'; | ||||
|   } else { | ||||
|     return num; | ||||
|   } | ||||
|   return <Label>{group}</Label> | ||||
| } | ||||
| @@ -15,12 +15,13 @@ const EditChannel = () => { | ||||
|     key: '', | ||||
|     base_url: '', | ||||
|     other: '', | ||||
|     group: 'default', | ||||
|     models: [], | ||||
|     groups: ['default'] | ||||
|   }; | ||||
|   const [batch, setBatch] = useState(false); | ||||
|   const [inputs, setInputs] = useState(originInputs); | ||||
|   const [modelOptions, setModelOptions] = useState([]); | ||||
|   const [groupOptions, setGroupOptions] = useState([]); | ||||
|   const [basicModels, setBasicModels] = useState([]); | ||||
|   const [fullModels, setFullModels] = useState([]); | ||||
|   const handleInputChange = (e, { name, value }) => { | ||||
| @@ -36,6 +37,11 @@ const EditChannel = () => { | ||||
|       } else { | ||||
|         data.models = data.models.split(",") | ||||
|       } | ||||
|       if (data.group === "") { | ||||
|         data.groups = [] | ||||
|       } else { | ||||
|         data.groups = data.group.split(",") | ||||
|       } | ||||
|       setInputs(data); | ||||
|     } else { | ||||
|       showError(message); | ||||
| @@ -58,11 +64,25 @@ const EditChannel = () => { | ||||
|     } | ||||
|   }; | ||||
|  | ||||
|   const fetchGroups = async () => { | ||||
|     try { | ||||
|       let res = await API.get(`/api/group/`); | ||||
|       setGroupOptions(res.data.data.map((group) => ({ | ||||
|         key: group, | ||||
|         text: group, | ||||
|         value: group, | ||||
|       }))); | ||||
|     } catch (error) { | ||||
|       showError(error.message); | ||||
|     } | ||||
|   }; | ||||
|  | ||||
|   useEffect(() => { | ||||
|     if (isEdit) { | ||||
|       loadChannel().then(); | ||||
|     } | ||||
|     fetchModels().then(); | ||||
|     fetchGroups().then(); | ||||
|   }, []); | ||||
|  | ||||
|   const submit = async () => { | ||||
| @@ -79,6 +99,7 @@ const EditChannel = () => { | ||||
|     } | ||||
|     let res; | ||||
|     localInputs.models = localInputs.models.join(",") | ||||
|     localInputs.group = localInputs.groups.join(",") | ||||
|     if (isEdit) { | ||||
|       res = await API.put(`/api/channel/`, { ...localInputs, id: parseInt(channelId) }); | ||||
|     } else { | ||||
| @@ -167,13 +188,19 @@ const EditChannel = () => { | ||||
|             /> | ||||
|           </Form.Field> | ||||
|           <Form.Field> | ||||
|             <Form.Input | ||||
|             <Form.Dropdown | ||||
|               label='分组' | ||||
|               name='group' | ||||
|               placeholder={'请输入分组'} | ||||
|               placeholder={'请选择分组'} | ||||
|               name='groups' | ||||
|               fluid | ||||
|               multiple | ||||
|               selection | ||||
|               allowAdditions | ||||
|               additionLabel={'请在系统设置页面编辑分组倍率以添加新的分组:'} | ||||
|               onChange={handleInputChange} | ||||
|               value={inputs.group} | ||||
|               value={inputs.groups} | ||||
|               autoComplete='new-password' | ||||
|               options={groupOptions} | ||||
|             /> | ||||
|           </Form.Field> | ||||
|           <Form.Field> | ||||
| @@ -198,6 +225,20 @@ const EditChannel = () => { | ||||
|               handleInputChange(null, { name: 'models', value: fullModels }); | ||||
|             }}>填入所有模型</Button> | ||||
|           </div> | ||||
|           { | ||||
|             inputs.type === 1 && ( | ||||
|               <Form.Field> | ||||
|                 <Form.Input | ||||
|                   label='代理' | ||||
|                   name='base_url' | ||||
|                   placeholder={'请输入 OpenAI API 代理地址,如果不需要请留空,格式为:https://api.openai.com'} | ||||
|                   onChange={handleInputChange} | ||||
|                   value={inputs.base_url} | ||||
|                   autoComplete='new-password' | ||||
|                 /> | ||||
|               </Form.Field> | ||||
|             ) | ||||
|           } | ||||
|           { | ||||
|             batch ? <Form.Field> | ||||
|               <Form.TextArea | ||||
|   | ||||
							
								
								
									
										14
									
								
								web/src/pages/Log/index.js
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								web/src/pages/Log/index.js
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,14 @@ | ||||
| import React from 'react'; | ||||
| import { Header, Segment } from 'semantic-ui-react'; | ||||
| import LogsTable from '../../components/LogsTable'; | ||||
|  | ||||
| const Token = () => ( | ||||
|   <> | ||||
|     <Segment> | ||||
|       <Header as='h3'>额度明细</Header> | ||||
|       <LogsTable /> | ||||
|     </Segment> | ||||
|   </> | ||||
| ); | ||||
|  | ||||
| export default Token; | ||||
| @@ -17,11 +17,24 @@ const EditUser = () => { | ||||
|     quota: 0, | ||||
|     group: 'default' | ||||
|   }); | ||||
|   const [groupOptions, setGroupOptions] = useState([]); | ||||
|   const { username, display_name, password, github_id, wechat_id, email, quota, group } = | ||||
|     inputs; | ||||
|   const handleInputChange = (e, { name, value }) => { | ||||
|     setInputs((inputs) => ({ ...inputs, [name]: value })); | ||||
|   }; | ||||
|   const fetchGroups = async () => { | ||||
|     try { | ||||
|       let res = await API.get(`/api/group/`); | ||||
|       setGroupOptions(res.data.data.map((group) => ({ | ||||
|         key: group, | ||||
|         text: group, | ||||
|         value: group, | ||||
|       }))); | ||||
|     } catch (error) { | ||||
|       showError(error.message); | ||||
|     } | ||||
|   }; | ||||
|  | ||||
|   const loadUser = async () => { | ||||
|     let res = undefined; | ||||
| @@ -41,6 +54,9 @@ const EditUser = () => { | ||||
|   }; | ||||
|   useEffect(() => { | ||||
|     loadUser().then(); | ||||
|     if (userId) { | ||||
|       fetchGroups().then(); | ||||
|     } | ||||
|   }, []); | ||||
|  | ||||
|   const submit = async () => { | ||||
| @@ -101,13 +117,19 @@ const EditUser = () => { | ||||
|           { | ||||
|             userId && <> | ||||
|               <Form.Field> | ||||
|                 <Form.Input | ||||
|                 <Form.Dropdown | ||||
|                   label='分组' | ||||
|                   placeholder={'请选择分组'} | ||||
|                   name='group' | ||||
|                   placeholder={'请输入用户分组'} | ||||
|                   fluid | ||||
|                   search | ||||
|                   selection | ||||
|                   allowAdditions | ||||
|                   additionLabel={'请在系统设置页面编辑分组倍率以添加新的分组:'} | ||||
|                   onChange={handleInputChange} | ||||
|                   value={group} | ||||
|                   value={inputs.group} | ||||
|                   autoComplete='new-password' | ||||
|                   options={groupOptions} | ||||
|                 /> | ||||
|               </Form.Field> | ||||
|               <Form.Field> | ||||
|   | ||||
		Reference in New Issue
	
	Block a user