feat: 从环境变量中读取配置api_host & 用户名是否配置改为判断auth长度 (#504)

* 从环境变量中读取配置api_host && authflag改为判断auth_list的长度

* chore: 更改README中提示和config-example 并调整shared写法

fix #507

- shared写法来自川虎在config废分支的提交，所以加co-author（

---------

Co-authored-by: Keldos <i@keldos.me>
Co-authored-by: Tuchuanhuhuhu <gzblog@hdu.edu.cn>

ChuanhuChatbot.py

@@ -5,6 +5,7 @@ import sys
 
 import gradio as gr
 
+from modules import config
 from modules.config import *
 from modules.utils import *
 from modules.presets import *
@@ -180,11 +181,12 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
                         )
 
                     with gr.Accordion("网络设置", open=False):
+                        # 优先展示自定义的api_host
                         apihostTxt = gr.Textbox(
                             show_label=True,
                             placeholder=f"在这里输入API-Host...",
                             label="API-Host",
-                            value="api.openai.com",
+                            value=config.api_host or shared.API_HOST,
                             lines=1,
                         )
                         changeAPIURLBtn = gr.Button("🔄 切换API地址")

README.md

@@ -99,18 +99,18 @@
 
 	这样设置的密钥以及其他设置项可以在拉取项目更新之后保留。
 
-	在项目文件夹中复制一份 `config_example.json`，并将其重命名为 `config.json`，在其中填入 API-Key、用户名密码（可选）、API host（可选）、代理地址（可选）等设置。用户名密码支持多用户。
+	在项目文件夹中复制一份 `config_example.json`，并将其重命名为 `config.json`，在其中填入 API-Key、用户名密码（可选）、API host（可选）、代理地址（可选）等设置。用户名密码支持多用户。示例：
 
 	```
 	{
 		"openai_api_key": "sk-xxxxxxxxxxxxxxxxxxxxxxxxx",
 		"users": [
-			["username-0", "password-0"],
-			["username-1", "password-1"]
+			["用户1的用户名", "用户1的密码"],
+			["用户2的用户名", "用户2的密码"]
 		], 
 	}
 	```
-
+	* 如果不设置用户名与密码，可以直接将“users”字段整段删去，或留空为 `"users": [], `
 	* *原本在 `api_key.txt` 和 `auth.json` 中填写相关设置的方法仍然可用，但不再推荐。*
 
 	</details>
@@ -181,6 +181,7 @@ docker pull tuchuanhuhuhu/chuanhuchatgpt:latest
 ```shell
 docker run -d --name chatgpt \
 	-e my_api_key="替换成API" \
+	-e api_host="替换成自定义的api请求地址" \
 	-e USERNAME="替换成用户名" \
 	-e PASSWORD="替换成密码" \
 	-v ~/chatGPThistory:/app/history \
@@ -302,6 +303,7 @@ docker run --detach \
 ```
 docker run -d --name chatgpt \
 	-e my_api_key="你的API" \
+	-e api_host="替换成自定义的api请求地址" \
 	-e USERNAME="替换成用户名" \
 	-e PASSWORD="替换成密码" \
 	-v ~/chatGPThistory:/app/history \

config_example.json

@@ -2,10 +2,7 @@
     "openai_api_key": "sk-xxxxxxxxxxxxxxxxxxxxxxxxx",
     "https_proxy": "http://127.0.0.1:1079",
     "http_proxy": "http://127.0.0.1:1079",
-    "users": [
-        ["username-0", "password-0"],
-        ["username-1", "password-1"]
-    ], 
+    "users": [], 
     "advance_docs": {
         "pdf": {
             "two_column": false,

modules/chat_func.py

@@ -18,7 +18,7 @@ import aiohttp
 from modules.presets import *
 from modules.llama_func import *
 from modules.utils import *
-import modules.shared as shared
+from . import shared
 from modules.config import retrieve_proxy
 
 # logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] [%(filename)s:%(lineno)d] %(message)s")

modules/config.py

@@ -5,6 +5,9 @@ import logging
 import sys
 import json
 
+from . import shared
+
+
 __all__ = [
     "my_api_key",
     "authflag",
@@ -31,9 +34,16 @@ if os.environ.get("dockerrun") == "yes":
 
 ## 处理 api-key 以及 允许的用户列表
 my_api_key = config.get("openai_api_key", "") # 在这里输入你的 API 密钥
-authflag = "users" in config
-auth_list = config.get("users", []) # 实际上是使用者的列表
 my_api_key = os.environ.get("my_api_key", my_api_key)
+
+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:
+    shared.state.set_api_host(api_host)
+
 if dockerflag:
     if my_api_key == "empty":
         logging.error("Please give a api key!")

modules/openai_func.py

@@ -10,7 +10,7 @@ from modules.presets import (
     read_timeout_prompt
 )
 
-from modules import shared
+from . import shared
 from modules.config import retrieve_proxy
 import os, datetime
 

modules/utils.py

@@ -23,7 +23,7 @@ from pygments.lexers import get_lexer_by_name
 from pygments.formatters import HtmlFormatter
 
 from modules.presets import *
-import modules.shared as shared
+from . import shared
 from modules.config import retrieve_proxy
 
 if TYPE_CHECKING: