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

[feat] 自带的三个模型支持多名对一个模型映射

Browse Source
This commit is contained in:
Wizerd
2023-12-17 21:05:44 +08:00
parent 77bef75d3f
commit aa9f789fb5
2 changed files with 24 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
main.py
View File

@@ -103,9 +103,9 @@ PROXY_API_PREFIX = os.getenv('PROXY_API_PREFIX', '')
UPLOAD_BASE_URL = os.getenv('UPLOAD_BASE_URL', '')
KEY_FOR_GPTS_INFO = os.getenv('KEY_FOR_GPTS_INFO', '')
VERSION = '0.1.5'
VERSION = '0.1.6'
# VERSION = 'test'
UPDATE_INFO = '支持自定义自带的三种模型的名称'
UPDATE_INFO = '对于自带的三个模型，支持多个名字映射到同一个模型'
# UPDATE_INFO = '【仅供临时测试使用】 '
with app.app_context():
@@ -128,26 +128,29 @@ with app.app_context():
print(f"==========================================")
# 从环境变量中读取模型名称
GPT_4_S_New_Name = os.getenv('GPT_4_S_New_Name', 'gpt-4-s')
GPT_4_MOBILE_NEW_NAME = os.getenv('GPT_4_MOBILE_NEW_NAME', 'gpt-4-mobile')
GPT_3_5_NEW_NAME = os.getenv('GPT_3_5_NEW_NAME', 'gpt-3.5-turbo')
# 从环境变量中读取模型名称，支持用逗号分隔的多个名称
GPT_4_S_New_Names = os.getenv('GPT_4_S_New_Name', 'gpt-4-s').split(',')
GPT_4_MOBILE_NEW_NAMES = os.getenv('GPT_4_MOBILE_NEW_NAME', 'gpt-4-mobile').split(',')
GPT_3_5_NEW_NAMES = os.getenv('GPT_3_5_NEW_NAME', 'gpt-3.5-turbo').split(',')
# 更新 gpts_configurations 列表
gpts_configurations = [
{
"name": GPT_4_S_New_Name,
# 更新 gpts_configurations 列表，支持多个映射
gpts_configurations = []
for name in GPT_4_S_New_Names:
gpts_configurations.append({
"name": name.strip(),
"ori_name": "gpt-4-s"
},
{
"name": GPT_4_MOBILE_NEW_NAME,
})
for name in GPT_4_MOBILE_NEW_NAMES:
gpts_configurations.append({
"name": name.strip(),
"ori_name": "gpt-4-mobile"
},
{
"name": GPT_3_5_NEW_NAME,
})
for name in GPT_3_5_NEW_NAMES:
gpts_configurations.append({
"name": name.strip(),
"ori_name": "gpt-3.5-turbo"
}
]
})
print(f"GPTS 配置信息")