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/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/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") 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) 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