完善OPENAI_API_BASE设置

config_example.json

@@ -10,6 +10,8 @@
     "minimax_api_key": "",
     "minimax_group_id": "",
     "language": "auto",
+    // 支持自定义OpenAI API Base
+    // "openai_api_base": "https://api.openai.com",
     // 如果使用代理，请取消注释下面的两行，并替换代理URL
     // "https_proxy": "http://127.0.0.1:1079",
     // "http_proxy": "http://127.0.0.1:1079",

modules/config.py

@@ -104,8 +104,8 @@ auth_list = config.get("users", []) # 实际上是使用者的列表
 authflag = len(auth_list) > 0  # 是否开启认证的状态值，改为判断auth_list长度
 
 # 处理自定义的api_host，优先读环境变量的配置，如果存在则自动装配
-api_host = os.environ.get("api_host", config.get("api_host", ""))
-if api_host:
+api_host = os.environ.get("OPENAI_API_BASE", config.get("openai_api_base", None))
+if api_host is not None:
     shared.state.set_api_host(api_host)
 
 default_chuanhu_assistant_model = config.get("default_chuanhu_assistant_model", "gpt-3.5-turbo")

modules/shared.py

@@ -57,5 +57,3 @@ class State:
 
 
 state = State()
-if (host:=os.environ.get("OPENAI_API_BASE", None)) is not None:
-    state.set_api_host(host)