增加默认命名模型设置（可选）

config_example.json

@@ -74,6 +74,7 @@
     //     "sk-xxxxxxxxxxxxxxxxxxxxxxxx2",
     //     "sk-xxxxxxxxxxxxxxxxxxxxxxxx3"
     // ],
+    // "rename_model": "GPT-4o-mini", //指定默认命名模型
     // 自定义OpenAI API Base
     // "openai_api_base": "https://api.openai.com",
     // 自定义使用代理（请替换代理URL）

modules/config.py

@@ -335,6 +335,18 @@ def update_doc_config(two_column_pdf):
 share = config.get("share", False)
 autobrowser = config.get("autobrowser", True)
 
+#设置默认命名model
+rename_model = config.get("rename_model", None)
+try:
+    if rename_model is not None:
+        if rename_model in presets.MODELS:
+            presets.RENAME_MODEL = presets.MODELS.index(rename_model)
+        else:
+            presets.RENAME_MODEL = presets.MODELS.index(next((k for k, v in presets.MODEL_METADATA.items() if v.get("model_name") == rename_model), None))
+        logging.info("默认命名模型设置为了：" + str(presets.MODELS[presets.RENAME_MODEL]))
+except ValueError:
+    logging.error("你填写的默认命名模型" + rename_model + "不存在！请从下面的列表中挑一个填写：" + str(presets.MODELS))
+
 # avatar
 bot_avatar = config.get("bot_avatar", "default")
 user_avatar = config.get("user_avatar", "default")

modules/models/OpenAIVision.py

@@ -275,7 +275,7 @@ def _single_query_at_once(self, history, temperature=1.0):
             "temperature": f"{temperature}",
         }
         payload = {
-            "model": self.model_name,
+            "model": RENAME_MODEL if RENAME_MODEL is not None else self.model_name,
             "messages": history,
         }
 

modules/presets.py

@@ -512,6 +512,8 @@
 
 DEFAULT_MODEL = 0
 
+RENAME_MODEL = 0
+
 os.makedirs("models", exist_ok=True)
 os.makedirs("lora", exist_ok=True)
 os.makedirs("history", exist_ok=True)