[feat] 支持自定义自带的三种模型的模型名

docker-compose.yml

@@ -11,6 +11,9 @@ services:
       - PROXY_API_PREFIX=<Pandora-Next的PROXY_API_PREFIX>
       - UPLOAD_BASE_URL=<50011端口在公网可以访问到的地址，如：http://1.2.3.4:50011，如果使用了反代需填入反代后的域名如：https://pandora-backend-api.com>
       - KEY_FOR_GPTS_INFO=<一个仅用于获取GPTs信息的fk> # 如果不需要额外使用gpts，可以不填
+      - GPT_4_S_New_Name=gpt-4-s # gpt-4-s模型的自定义模型名称
+      - GPT_4_MOBILE_NEW_NAME=gpt-4-mobile # gpt-4-mobile模型的自定义模型名称
+      - GPT_3_5_NEW_NAME=gpt-3.5-turbo # gpt-3.5-turbo模型的自定义模型名称
     volumes:
       - ./log:/app/log
       - ./images:/app/images

main.py

@@ -50,17 +50,7 @@ def fetch_gizmo_info(base_url, proxy_api_prefix, model_id):
     else:
         return None
 
-gpts_configurations = [
-    {
-        "name":"gpt-4-s"
-    },
-    {
-        "name":"gpt-4-mobile"
-    },
-    {
-        "name":"gpt-3.5-turbo"
-    }
-]
+# gpts_configurations = []
 
 # 将配置添加到全局列表
 def add_config_to_global_list(base_url, proxy_api_prefix, gpts_data):
@@ -113,10 +103,14 @@ 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.4'
-UPDATE_INFO = '合并了 Upload 容器和 Backend 容器'
+VERSION = '0.1.5'
+# VERSION = 'test'
+UPDATE_INFO = '支持自定义自带的三种模型的名称'
+# UPDATE_INFO = '【仅供临时测试使用】 '
 
 with app.app_context():
+    global gpts_configurations  # 移到作用域的最开始
+
     # 输出版本信息
     print(f"==========================================")
     print(f"Version: {VERSION}")
@@ -134,6 +128,27 @@ 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')
+
+    # 更新 gpts_configurations 列表
+    gpts_configurations = [
+        {
+            "name": GPT_4_S_New_Name,
+            "ori_name": "gpt-4-s"
+        },
+        {
+            "name": GPT_4_MOBILE_NEW_NAME,
+            "ori_name": "gpt-4-mobile"
+        },
+        {
+            "name": GPT_3_5_NEW_NAME,
+            "ori_name": "gpt-3.5-turbo"
+        }
+    ]
+
     print(f"GPTS 配置信息")
 
     # 加载配置并添加到全局列表
@@ -141,7 +156,13 @@ with app.app_context():
     add_config_to_global_list(BASE_URL, PROXY_API_PREFIX, gpts_data)
     # print("当前可用GPTS：" + get_accessible_model_list())
     # 输出当前可用 GPTS name
-    print(f"当前可用 GPTS 列表: {get_accessible_model_list()}")
+    # 获取当前可用的 GPTS 模型列表
+    accessible_model_list = get_accessible_model_list()
+    print(f"当前可用 GPTS 列表: {accessible_model_list}")
+
+    # 检查列表中是否有重复的模型名称
+    if len(accessible_model_list) != len(set(accessible_model_list)):
+        raise Exception("检测到重复的模型名称，请检查环境变量或配置文件。")
 
     print(f"==========================================")
 
@@ -185,7 +206,13 @@ def send_text_prompt_and_get_response(messages, api_key, stream, model):
     payload = {}
 
     print(f"model: {model}")
-    if model == 'gpt-4-s':
+
+    # 查找模型配置
+    model_config = find_model_config(model)
+    if model_config:
+        # 检查是否有 ori_name
+        ori_model_name = model_config.get('ori_name', model)
+        if ori_model_name == 'gpt-4-s':
             payload = {
                 # 构建 payload
                 "action": "next",
@@ -197,7 +224,7 @@ def send_text_prompt_and_get_response(messages, api_key, stream, model):
                 "history_and_training_disabled": False,
                 "conversation_mode":{"kind":"primary_assistant"},"force_paragen":False,"force_rate_limit":False
             }
-    elif model == 'gpt-4-mobile':
+        elif ori_model_name == 'gpt-4-mobile':
             payload = {
                 # 构建 payload
                 "action": "next",
@@ -209,7 +236,7 @@ def send_text_prompt_and_get_response(messages, api_key, stream, model):
                 "history_and_training_disabled": False,
                 "conversation_mode":{"kind":"primary_assistant"},"force_paragen":False,"force_rate_limit":False
             }
-    elif model =='gpt-3.5-turbo':
+        elif ori_model_name =='gpt-3.5-turbo':
             payload = {
                 # 构建 payload
                 "action": "next",