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

优化gpts结构

Browse Source
This commit is contained in:
Clivia
2024-02-19 21:52:50 +08:00
committed by GitHub
parent 6f66431bb5
commit 002ff558b0
1 changed files with 5 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
main.py
View File

@@ -234,7 +234,6 @@ def updateGptsKey():
# 根据 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}"
updateGptsKey()
headers = { headers = {
"Authorization": f"Bearer {KEY_FOR_GPTS_INFO}" "Authorization": f"Bearer {KEY_FOR_GPTS_INFO}"
} }
@@ -251,6 +250,7 @@ def fetch_gizmo_info(base_url, proxy_api_prefix, model_id):
# 将配置添加到全局列表 # 将配置添加到全局列表
def add_config_to_global_list(base_url, proxy_api_prefix, gpts_data): def add_config_to_global_list(base_url, proxy_api_prefix, gpts_data):
global gpts_configurations global gpts_configurations
updateGptsKey()
# print(f"gpts_data: {gpts_data}") # print(f"gpts_data: {gpts_data}")
for model_name, model_info in gpts_data.items(): for model_name, model_info in gpts_data.items():
# print(f"model_name: {model_name}") # print(f"model_name: {model_name}")
@@ -2614,21 +2614,17 @@ 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.........") logging.info("开始更新KEY_FOR_GPTS_INFO和GPTS配置信息......")
updateGptsKey()
# 配置GPTS
logger.info(f"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)
accessible_model_list = get_accessible_model_list() accessible_model_list = get_accessible_model_list()
logger.info(f"当前可用 GPTS 列表: {accessible_model_list}")
# 检查列表中是否有重复的模型名称 # 检查列表中是否有重复的模型名称
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配置信息成功......")
logger.info(f"当前可用 GPTS 列表: {accessible_model_list}")
logger.info(f"==========================================") logger.info(f"==========================================")