Merge pull request #179 from h88782481/dev

feat:添加清空seedmap的接口

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`管控。    |
 
 ## 部署
 

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:

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

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)

version.txt

@@ -1 +1 @@
-1.6.7-beta7
+1.6.8-beta1