ChatGPT/RefreshToV1Api
1
0
Fork 0
mirror of https://github.com/Yanyutin753/RefreshToV1Api.git synced 2025-10-14 15:11:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

更新KEY_FOR_GPTS_INFO

Browse Source
This commit is contained in:
Clivia
2024-02-20 01:14:46 +08:00
committed by GitHub
parent 002ff558b0
commit 1e3e233adc
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
main.py
View File

@@ -48,6 +48,7 @@ if PROXY_API_PREFIX != '':
PROXY_API_PREFIX = "/" + PROXY_API_PREFIX PROXY_API_PREFIX = "/" + PROXY_API_PREFIX
UPLOAD_BASE_URL = CONFIG.get('backend_container_url', '') UPLOAD_BASE_URL = CONFIG.get('backend_container_url', '')
KEY_FOR_GPTS_INFO = CONFIG.get('key_for_gpts_info', '') KEY_FOR_GPTS_INFO = CONFIG.get('key_for_gpts_info', '')
KEY_FOR_GPTS_INFO_ACCESS_TOKEN = CONFIG.get('key_for_gpts_info', '')
API_PREFIX = CONFIG.get('backend_container_api_prefix', '') API_PREFIX = CONFIG.get('backend_container_api_prefix', '')
GPT_4_S_New_Names = CONFIG.get('gpt_4_s_new_name', 'gpt-4-s').split(',') GPT_4_S_New_Names = CONFIG.get('gpt_4_s_new_name', 'gpt-4-s').split(',')
GPT_4_MOBILE_NEW_NAMES = CONFIG.get('gpt_4_mobile_new_name', 'gpt-4-mobile').split(',') GPT_4_MOBILE_NEW_NAMES = CONFIG.get('gpt_4_mobile_new_name', 'gpt-4-mobile').split(',')
@@ -221,21 +222,22 @@ def ninjaGetAccessToken(getAccessTokenUrl, refresh_token):
def updateGptsKey(): def updateGptsKey():
global KEY_FOR_GPTS_INFO global KEY_FOR_GPTS_INFO
global KEY_FOR_GPTS_INFO_ACCESS_TOKEN
if not KEY_FOR_GPTS_INFO == '' and not KEY_FOR_GPTS_INFO.startswith("eyJhb"): if not KEY_FOR_GPTS_INFO == '' and not KEY_FOR_GPTS_INFO.startswith("eyJhb"):
if REFRESH_TOACCESS_ENABLEOAI: if REFRESH_TOACCESS_ENABLEOAI:
access_token = oaiGetAccessToken(KEY_FOR_GPTS_INFO) access_token = oaiGetAccessToken(KEY_FOR_GPTS_INFO)
else: else:
access_token = ninjaGetAccessToken(REFRESH_TOACCESS_NINJA_REFRESHTOACCESS_URL, KEY_FOR_GPTS_INFO) access_token = ninjaGetAccessToken(REFRESH_TOACCESS_NINJA_REFRESHTOACCESS_URL, KEY_FOR_GPTS_INFO)
if access_token.startswith("eyJhb"): if access_token.startswith("eyJhb"):
KEY_FOR_GPTS_INFO = access_token KEY_FOR_GPTS_INFO_ACCESS_TOKEN = access_token
logging.info("KEY_FOR_GPTS_INFO被更新:" + KEY_FOR_GPTS_INFO) logging.info("KEY_FOR_GPTS_INFO_ACCESS_TOKEN被更新:" + KEY_FOR_GPTS_INFO_ACCESS_TOKEN)
# 根据 ID 发送请求并获取配置信息 # 根据 ID 发送请求并获取配置信息
def fetch_gizmo_info(base_url, proxy_api_prefix, model_id): def fetch_gizmo_info(base_url, proxy_api_prefix, model_id):
url = f"{base_url}{proxy_api_prefix}/backend-api/gizmos/{model_id}" url = f"{base_url}{proxy_api_prefix}/backend-api/gizmos/{model_id}"
headers = { headers = {
"Authorization": f"Bearer {KEY_FOR_GPTS_INFO}" "Authorization": f"Bearer {KEY_FOR_GPTS_INFO_ACCESS_TOKEN}"
} }
response = requests.get(url, headers=headers) response = requests.get(url, headers=headers)
# logger.debug(f"fetch_gizmo_info_response: {response.text}") # logger.debug(f"fetch_gizmo_info_response: {response.text}")
@@ -2614,7 +2616,7 @@ def updateRefresh_dict():
success_num += 1 success_num += 1
logging.info("更新成功: " + str(success_num) + ", 失败: " + str(error_num)) logging.info("更新成功: " + str(success_num) + ", 失败: " + str(error_num))
logger.info(f"==========================================") logger.info(f"==========================================")
logging.info("开始更新KEY_FOR_GPTS_INFO和GPTS配置信息......") logging.info("开始更新KEY_FOR_GPTS_INFO_ACCESS_TOKEN和GPTS配置信息......")
# 加载配置并添加到全局列表 # 加载配置并添加到全局列表
gpts_data = load_gpts_config("./data/gpts.json") gpts_data = load_gpts_config("./data/gpts.json")
add_config_to_global_list(BASE_URL, PROXY_API_PREFIX, gpts_data) add_config_to_global_list(BASE_URL, PROXY_API_PREFIX, gpts_data)
@@ -2623,7 +2625,7 @@ def updateRefresh_dict():
# 检查列表中是否有重复的模型名称 # 检查列表中是否有重复的模型名称
if len(accessible_model_list) != len(set(accessible_model_list)): if len(accessible_model_list) != len(set(accessible_model_list)):
raise Exception("检测到重复的模型名称，请检查环境变量或配置文件。") raise Exception("检测到重复的模型名称，请检查环境变量或配置文件。")
logging.info("更新KEY_FOR_GPTS_INFO和GPTS配置信息成功......") logging.info("更新KEY_FOR_GPTS_INFO_ACCESS_TOKEN和GPTS配置信息成功......")
logger.info(f"当前可用 GPTS 列表: {accessible_model_list}") logger.info(f"当前可用 GPTS 列表: {accessible_model_list}")
logger.info(f"==========================================") logger.info(f"==========================================")