From 987058e54e334591ba532c9585948cca85d606c9 Mon Sep 17 00:00:00 2001 From: Samon Yu Date: Wed, 5 Feb 2025 18:19:43 +0800 Subject: [PATCH] =?UTF-8?q?refactor(config):=20=E4=BC=98=E5=8C=96=E9=85=8D?= =?UTF-8?q?=E7=BD=AE=E6=96=87=E4=BB=B6=E6=A0=BC=E5=BC=8F=E5=92=8C=E5=86=85?= =?UTF-8?q?=E5=AE=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 调整缩进和格式，提高可读性 - 更新注释，明确配置项说明 - 移除冗余配置项 - 统一字符串引号使用单引号 --- config.py | 125 +++++++++++++++++++++--------------------------------- 1 file changed, 49 insertions(+), 76 deletions(-) diff --git a/config.py b/config.py index 1d23da2e..99e9d2f8 100644 --- a/config.py +++ b/config.py @@ -1,17 +1,17 @@ """ -以下所有配置也都支持利用环境变量覆写，环境变量配置格式见docker-compose.yml。 -读取优先级：环境变量 > config_private.py > config.py ---- --- --- --- --- --- --- --- --- --- --- --- --- --- --- --- --- --- --- --- --- -All the following configurations also support using environment variables to override, -and the environment variable configuration format can be seen in docker-compose.yml. -Configuration reading priority: environment variable > config_private.py > config.py + 以下所有配置也都支持利用环境变量覆写，环境变量配置格式见docker-compose.yml。 + 读取优先级：环境变量 > config_private.py > config.py + --- --- --- --- --- --- --- --- --- --- --- --- --- --- --- --- --- --- --- --- --- + All the following configurations also support using environment variables to override, + and the environment variable configuration format can be seen in docker-compose.yml. + Configuration reading priority: environment variable > config_private.py > config.py """ # [step 1-1]>> ( 接入GPT等模型 ) API_KEY = "sk-123456789xxxxxxxxxxxxxxxxxxxxxxxxxxxxxx123456789"。极少数情况下，还需要填写组织（格式如org-123456789abcdefghijklmno的），请向下翻，找 API_ORG 设置项 -API_KEY = "在此处填写APIKEY" # 可同时填写多个API-KEY，用英文逗号分割，例如API_KEY = "sk-openaikey1,sk-openaikey2,fkxxxx-api2dkey3,azure-apikey4" +API_KEY = "在此处填写APIKEY" # 可同时填写多个API-KEY，用英文逗号分割，例如API_KEY = "sk-openaikey1,sk-openaikey2,fkxxxx-api2dkey3,azure-apikey4" # [step 1-2]>> ( 接入通义 qwen-max ) 接入通义千问在线大模型，api-key获取地址 https://dashscope.console.aliyun.com/ -DASHSCOPE_API_KEY = "" # 阿里灵积云API_KEY +DASHSCOPE_API_KEY = "" # 阿里灵积云API_KEY # [step 1-3]>> ( 接入 deepseek-reasoner, 即 deepseek-r1 ) 深度求索(DeepSeek) API KEY，默认请求地址为"https://api.deepseek.com/v1/chat/completions" DEEPSEEK_API_KEY = "" @@ -107,28 +107,22 @@ DEFAULT_WORKER_NUM = 3 # 色彩主题, 可选 ["Default", "Chuanhu-Small-and-Beautiful", "High-Contrast"] # 更多主题, 请查阅Gradio主题商店: https://huggingface.co/spaces/gradio/theme-gallery 可选 ["Gstaff/Xkcd", "NoCrypt/Miku", ...] THEME = "Default" -AVAIL_THEMES = [ - "Default", - "Chuanhu-Small-and-Beautiful", - "High-Contrast", - "Gstaff/Xkcd", - "NoCrypt/Miku", -] +AVAIL_THEMES = ["Default", "Chuanhu-Small-and-Beautiful", "High-Contrast", "Gstaff/Xkcd", "NoCrypt/Miku"] FONT = "Theme-Default-Font" AVAIL_FONTS = [ - "默认值(Theme-Default-Font)", - "宋体(SimSun)", - "黑体(SimHei)", - "楷体(KaiTi)", - "仿宋(FangSong)", + "默认值(Theme-Default-Font)", + "宋体(SimSun)", + "黑体(SimHei)", + "楷体(KaiTi)", + "仿宋(FangSong)", "华文细黑(STHeiti Light)", - "华文楷体(STKaiti)", - "华文仿宋(STFangsong)", - "华文宋体(STSong)", - "华文中宋(STZhongsong)", - "华文新魏(STXinwei)", - "华文隶书(STLiti)", + "华文楷体(STKaiti)", + "华文仿宋(STFangsong)", + "华文宋体(STSong)", + "华文中宋(STZhongsong)", + "华文新魏(STXinwei)", + "华文隶书(STLiti)", "思源宋体(Source Han Serif CN VF@https://chinese-fonts-cdn.deno.dev/packages/syst/dist/SourceHanSerifCN/result.css)", "月星楷(Moon Stars Kai HW@https://chinese-fonts-cdn.deno.dev/packages/moon-stars-kai/dist/MoonStarsKaiHW-Regular/result.css)", "珠圆体(MaokenZhuyuanTi@https://chinese-fonts-cdn.deno.dev/packages/mkzyt/dist/猫啃珠圆体/result.css)", @@ -136,7 +130,7 @@ AVAIL_FONTS = [ "Helvetica", "ui-sans-serif", "sans-serif", - "system-ui", + "system-ui" ] @@ -153,7 +147,7 @@ CODE_HIGHLIGHT = True # 窗口布局 -LAYOUT = "LEFT-RIGHT" # "LEFT-RIGHT"（左右布局） # "TOP-DOWN"（上下布局） +LAYOUT = "LEFT-RIGHT" # "LEFT-RIGHT"（左右布局） # "TOP-DOWN"（上下布局） # 暗色模式 / 亮色模式 @@ -177,7 +171,7 @@ MAX_RETRY = 2 # 插件分类默认选项 -DEFAULT_FN_GROUPS = ["对话", "编程", "学术", "智能体"] +DEFAULT_FN_GROUPS = ['对话', '编程', '学术', '智能体'] # 定义界面上“询问多个GPT模型”插件应该使用哪些模型，请从AVAIL_LLM_MODELS中选择，并在不同模型之间用`&`间隔，例如"gpt-3.5-turbo&chatglm3&azure-gpt-4" @@ -191,25 +185,21 @@ QWEN_LOCAL_MODEL_SELECTION = "Qwen/Qwen-1_8B-Chat-Int8" # 百度千帆（LLM_MODEL="qianfan"） -BAIDU_CLOUD_API_KEY = "" -BAIDU_CLOUD_SECRET_KEY = "" -BAIDU_CLOUD_QIANFAN_MODEL = "ERNIE-Bot" # 可选 "ERNIE-Bot-4"(文心大模型4.0), "ERNIE-Bot"(文心一言), "ERNIE-Bot-turbo", "BLOOMZ-7B", "Llama-2-70B-Chat", "Llama-2-13B-Chat", "Llama-2-7B-Chat", "ERNIE-Speed-128K", "ERNIE-Speed-8K", "ERNIE-Lite-8K" +BAIDU_CLOUD_API_KEY = '' +BAIDU_CLOUD_SECRET_KEY = '' +BAIDU_CLOUD_QIANFAN_MODEL = 'ERNIE-Bot' # 可选 "ERNIE-Bot-4"(文心大模型4.0), "ERNIE-Bot"(文心一言), "ERNIE-Bot-turbo", "BLOOMZ-7B", "Llama-2-70B-Chat", "Llama-2-13B-Chat", "Llama-2-7B-Chat", "ERNIE-Speed-128K", "ERNIE-Speed-8K", "ERNIE-Lite-8K" # 如果使用ChatGLM3或ChatGLM4本地模型，请把 LLM_MODEL="chatglm3" 或LLM_MODEL="chatglm4"，并在此处指定模型路径 -CHATGLM_LOCAL_MODEL_PATH = "THUDM/glm-4-9b-chat" # 例如"/home/hmp/ChatGLM3-6B/" +CHATGLM_LOCAL_MODEL_PATH = "THUDM/glm-4-9b-chat" # 例如"/home/hmp/ChatGLM3-6B/" # 如果使用ChatGLM2微调模型，请把 LLM_MODEL="chatglmft"，并在此处指定模型路径 -CHATGLM_PTUNING_CHECKPOINT = ( - "" # 例如"/home/hmp/ChatGLM2-6B/ptuning/output/6b-pt-128-1e-2/checkpoint-100" -) +CHATGLM_PTUNING_CHECKPOINT = "" # 例如"/home/hmp/ChatGLM2-6B/ptuning/output/6b-pt-128-1e-2/checkpoint-100" # 本地LLM模型如ChatGLM的执行方式 CPU/GPU -LOCAL_MODEL_DEVICE = "cpu" # 可选 "cuda" -LOCAL_MODEL_QUANT = ( - "FP16" # 默认 "FP16" "INT4" 启用量化INT4版本 "INT8" 启用量化INT8版本 -) +LOCAL_MODEL_DEVICE = "cpu" # 可选 "cuda" +LOCAL_MODEL_QUANT = "FP16" # 默认 "FP16" "INT4" 启用量化INT4版本 "INT8" 启用量化INT8版本 # 设置gradio的并行线程数（不需要修改） @@ -244,16 +234,14 @@ API_ORG = "" # 如果需要使用Slack Claude，使用教程详情见 request_llms/README.md -SLACK_CLAUDE_BOT_ID = "" -SLACK_CLAUDE_USER_TOKEN = "" +SLACK_CLAUDE_BOT_ID = '' +SLACK_CLAUDE_USER_TOKEN = '' # 如果需要使用AZURE（方法一：单个azure模型部署）详情请见额外文档 docs\use_azure.md AZURE_ENDPOINT = "https://你亲手写的api名称.openai.azure.com/" -AZURE_API_KEY = ( - "填入azure openai api的密钥" # 建议直接在API_KEY处填写，该选项即将被弃用 -) -AZURE_ENGINE = "填入你亲手写的部署名" # 读 docs\use_azure.md +AZURE_API_KEY = "填入azure openai api的密钥" # 建议直接在API_KEY处填写，该选项即将被弃用 +AZURE_ENGINE = "填入你亲手写的部署名" # 读 docs\use_azure.md # 如果需要使用AZURE（方法二：多个azure模型部署+动态切换）详情请见额外文档 docs\use_azure.md @@ -263,14 +251,14 @@ AZURE_CFG_ARRAY = {} # 阿里云实时语音识别 配置难度较高 # 参考 https://github.com/binary-husky/gpt_academic/blob/master/docs/use_audio.md ENABLE_AUDIO = False -ALIYUN_TOKEN = "" # 例如 f37f30e0f9934c34a992f6f64f7eba4f -ALIYUN_APPKEY = "" # 例如 RoPlZrM88DnAFkZK -ALIYUN_ACCESSKEY = "" # （无需填写） -ALIYUN_SECRET = "" # （无需填写） +ALIYUN_TOKEN="" # 例如 f37f30e0f9934c34a992f6f64f7eba4f +ALIYUN_APPKEY="" # 例如 RoPlZrM88DnAFkZK +ALIYUN_ACCESSKEY="" # （无需填写） +ALIYUN_SECRET="" # （无需填写） # GPT-SOVITS 文本转语音服务的运行地址（将语言模型的生成文本朗读出来） -TTS_TYPE = "EDGE_TTS" # EDGE_TTS / LOCAL_SOVITS_API / DISABLE +TTS_TYPE = "EDGE_TTS" # EDGE_TTS / LOCAL_SOVITS_API / DISABLE GPT_SOVITS_URL = "" EDGE_TTS_VOICE = "zh-CN-XiaoxiaoNeural" @@ -283,7 +271,7 @@ XFYUN_API_KEY = "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa" # 接入智谱大模型 ZHIPUAI_API_KEY = "" -ZHIPUAI_MODEL = "" # 此选项已废弃，不再需要填写 +ZHIPUAI_MODEL = "" # 此选项已废弃，不再需要填写 # Claude API KEY @@ -293,9 +281,6 @@ ANTHROPIC_API_KEY = "" # 月之暗面 API KEY MOONSHOT_API_KEY = "" -# 硅基智能 API KEY -SILICONFLOW_API_KEY = "" - # 零一万物(Yi Model) API KEY YIMODEL_API_KEY = "" @@ -321,7 +306,7 @@ CUSTOM_API_KEY_PATTERN = "" # Google Gemini API-Key -GEMINI_API_KEY = "" +GEMINI_API_KEY = '' # HUGGINGFACE的TOKEN，下载LLAMA时起作用 https://huggingface.co/docs/hub/security-tokens @@ -331,19 +316,14 @@ HUGGINGFACE_ACCESS_TOKEN = "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" # GROBID服务器地址（填写多个可以均衡负载），用于高质量地读取PDF文档 # 获取方法：复制以下空间https://huggingface.co/spaces/qingxu98/grobid，设为public，然后GROBID_URL = "https://(你的hf用户名如qingxu98)-(你的填写的空间名如grobid).hf.space" GROBID_URLS = [ - "https://qingxu98-grobid.hf.space", - "https://qingxu98-grobid2.hf.space", - "https://qingxu98-grobid3.hf.space", - "https://qingxu98-grobid4.hf.space", - "https://qingxu98-grobid5.hf.space", - "https://qingxu98-grobid6.hf.space", - "https://qingxu98-grobid7.hf.space", - "https://qingxu98-grobid8.hf.space", + "https://qingxu98-grobid.hf.space","https://qingxu98-grobid2.hf.space","https://qingxu98-grobid3.hf.space", + "https://qingxu98-grobid4.hf.space","https://qingxu98-grobid5.hf.space", "https://qingxu98-grobid6.hf.space", + "https://qingxu98-grobid7.hf.space", "https://qingxu98-grobid8.hf.space", ] # Searxng互联网检索服务（这是一个huggingface空间，请前往huggingface复制该空间，然后把自己新的空间地址填在这里） -SEARXNG_URLS = [f"https://kaletianlre-beardvs{i}dd.hf.space/" for i in range(1, 5)] +SEARXNG_URLS = [ f"https://kaletianlre-beardvs{i}dd.hf.space/" for i in range(1,5) ] # 是否允许通过自然语言描述修改本页的配置，该功能具有一定的危险性，默认关闭 @@ -367,16 +347,8 @@ ARXIV_CACHE_DIR = "gpt_log/arxiv_cache" # 除了连接OpenAI之外，还有哪些场合允许使用代理，请尽量不要修改 -WHEN_TO_USE_PROXY = [ - "Connect_OpenAI", - "Download_LLM", - "Download_Gradio_Theme", - "Connect_Grobid", - "Warmup_Modules", - "Nougat_Download", - "AutoGen", - "Connect_OpenAI_Embedding", -] +WHEN_TO_USE_PROXY = ["Connect_OpenAI", "Download_LLM", "Download_Gradio_Theme", "Connect_Grobid", + "Warmup_Modules", "Nougat_Download", "AutoGen", "Connect_OpenAI_Embedding"] # 启用插件热加载 @@ -388,7 +360,8 @@ NUM_CUSTOM_BASIC_BTN = 4 # 媒体智能体的服务地址（这是一个huggingface空间，请前往huggingface复制该空间，然后把自己新的空间地址填在这里） -DAAS_SERVER_URLS = [f"https://niuziniu-biligpt{i}.hf.space/stream" for i in range(1, 5)] +DAAS_SERVER_URLS = [ f"https://niuziniu-biligpt{i}.hf.space/stream" for i in range(1,5) ] + """