From 6f3b828a8b38f07919a18c42f77b96f6bf2aecb1 Mon Sep 17 00:00:00 2001 From: h88782481 <54714341+h88782481@users.noreply.github.com> Date: Sun, 3 Nov 2024 19:12:30 +0800 Subject: [PATCH 1/3] =?UTF-8?q?feat:=E6=B7=BB=E5=8A=A0=E6=B8=85=E7=A9=BAse?= =?UTF-8?q?edmap=E7=9A=84=E6=8E=A5=E5=8F=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 2 +- gateway/backend.py | 6 ++++++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 294b676..68f5db1 100644 --- a/README.md +++ b/README.md @@ -140,7 +140,7 @@ curl --location 'http://127.0.0.1:5005/v1/chat/completions' \ | | SCHEDULED_REFRESH | `false` | `false` | 是否定时刷新 `AccessToken` ，开启后每次启动程序将会全部非强制刷新一次，每4天晚上3点全部强制刷新一次。 | | | RANDOM_TOKEN | `true` | `true` | 是否随机选取后台 `Token` ，开启后随机后台账号，关闭后为顺序轮询 | | 网关功能 | ENABLE_GATEWAY | `false` | `false` | 是否启用网关模式，开启后可以使用镜像站，但也将会不设防 | -| | AUTOSEED | `false` | `true` | 是否启用随机账号模式，默认启用，输入`seed`后随机匹配后台`Token`。关闭之后需要手动对接接口，来进行`Token`管控。 | +| | AUTO_SEED | `false` | `true` | 是否启用随机账号模式，默认启用，输入`seed`后随机匹配后台`Token`。关闭之后需要手动对接接口，来进行`Token`管控。 | ## 部署 diff --git a/gateway/backend.py b/gateway/backend.py index 5c005a9..5820500 100644 --- a/gateway/backend.py +++ b/gateway/backend.py @@ -130,6 +130,12 @@ if enable_gateway: data = await request.json() seed = data.get("seed") + if seed == "clear": + globals.seed_map.clear() + with open(globals.SEED_MAP_FILE, "w", encoding="utf-8") as f: + json.dump(globals.seed_map, f, indent=4) + return {"status": "success", "message": "All seeds deleted successfully"} + if not seed: raise HTTPException(status_code=400, detail="Missing required field: seed") From 6b0311faf6b0d6a9e095fb7de0c868d78d363ea9 Mon Sep 17 00:00:00 2001 From: h88782481 <54714341+h88782481@users.noreply.github.com> Date: Sun, 3 Nov 2024 19:59:01 +0800 Subject: [PATCH 2/3] fix bug --- chatgpt/authorization.py | 2 +- utils/configs.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/chatgpt/authorization.py b/chatgpt/authorization.py index 44451ff..8b7016e 100644 --- a/chatgpt/authorization.py +++ b/chatgpt/authorization.py @@ -14,7 +14,7 @@ from utils.Logger import logger def get_req_token(req_token, seed=None): - if configs.auto_seed: + if configs.auto_seed or not seed: available_token_list = list(set(globals.token_list) - set(globals.error_token_list)) length = len(available_token_list) if seed and length > 0: diff --git a/utils/configs.py b/utils/configs.py index d3da87e..57f18a5 100644 --- a/utils/configs.py +++ b/utils/configs.py @@ -86,4 +86,5 @@ logger.info("SCHEDULED_REFRESH: " + str(scheduled_refresh)) logger.info("RANDOM_TOKEN: " + str(random_token)) logger.info("------------------------- Gateway --------------------------") logger.info("ENABLE_GATEWAY: " + str(enable_gateway)) +logger.info("AUTO_SEED: " + str(auto_seed)) logger.info("-" * 60) From 70783bd19464ff5844e36cc674d034bce226987d Mon Sep 17 00:00:00 2001 From: lanqian528 <5499636+lanqian528@users.noreply.github.com> Date: Sun, 3 Nov 2024 20:03:58 +0800 Subject: [PATCH 3/3] v1.6.8-beta1 merge --- version.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/version.txt b/version.txt index a74a498..b21c430 100644 --- a/version.txt +++ b/version.txt @@ -1 +1 @@ -1.6.7-beta7 \ No newline at end of file +1.6.8-beta1 \ No newline at end of file