From c8b88b4f1f3724d8aaaa464275d77ad79e0ade17 Mon Sep 17 00:00:00 2001 From: DrSmoothl <1787882683@qq.com> Date: Sat, 15 Mar 2025 14:22:54 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E6=94=AF=E6=8C=81=E7=81=AB?= =?UTF-8?q?=E5=B1=B1=E5=BC=95=E6=93=8EAPI?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- webui.py | 67 ++++++++++++++++++++++++++++++++++++++++++++------------ 1 file changed, 53 insertions(+), 14 deletions(-) diff --git a/webui.py b/webui.py index 6cc2fa0c8..a65750111 100644 --- a/webui.py +++ b/webui.py @@ -10,11 +10,7 @@ import json is_share = False debug = True -try: - config_data = toml.load("config/bot_config.toml") -except Exception as e: - logger.error(f"读取配置文件失败，请检查配置文件是否存在，错误信息为：{e}") - sys.exit(1) +config_data = toml.load("config/bot_config.toml") CONFIG_VERSION = config_data["inner"]["version"] PARSED_CONFIG_VERSION = float(CONFIG_VERSION[2:]) @@ -76,8 +72,29 @@ def save_to_env_file(env_variables, filename=".env.prod"): f.write(f"{var[4:]}={value}\n") # 移除env_前缀 logger.info(f"配置已保存到 {filename}") + +#载入env文件并解析 env_config_file = ".env.prod" # 配置文件路径 env_config_data = parse_env_config(env_config_file) +if "env_VOLCENGINE_BASE_URL" in env_config_data: + logger.info("VOLCENGINE_BASE_URL 已存在，使用默认值") + env_config_data["env_VOLCENGINE_BASE_URL"] = "https://ark.cn-beijing.volces.com/api/v3" +else: + logger.info("VOLCENGINE_BASE_URL 不存在，已创建并使用默认值") + env_config_data["env_VOLCENGINE_BASE_URL"] = "https://ark.cn-beijing.volces.com/api/v3" + +if "env_VOLCENGINE_KEY" in env_config_data: + logger.info("VOLCENGINE_KEY 已存在，保持不变") +else: + logger.info("VOLCENGINE_KEY 不存在，已创建并使用默认值") + env_config_data["env_VOLCENGINE_KEY"] = "volc_key" +save_to_env_file(env_config_data, env_config_file) +MODEL_PROVIDER_LIST = [ + "VOLCENGINE", + "CHAT_ANY_WHERE", + "SILICONFLOW", + "DEEP_SEEK" +] #env读取保存结束 #============================================== @@ -166,7 +183,7 @@ def format_list_to_str(lst): #env保存函数 -def save_trigger(server_address, server_port, final_result_list,t_mongodb_host,t_mongodb_port,t_mongodb_database_name,t_chatanywhere_base_url,t_chatanywhere_key,t_siliconflow_base_url,t_siliconflow_key,t_deepseek_base_url,t_deepseek_key): +def save_trigger(server_address, server_port, final_result_list,t_mongodb_host,t_mongodb_port,t_mongodb_database_name,t_chatanywhere_base_url,t_chatanywhere_key,t_siliconflow_base_url,t_siliconflow_key,t_deepseek_base_url,t_deepseek_key,t_volcengine_base_url,t_volcengine_key): final_result_lists = format_list_to_str(final_result_list) env_config_data["env_HOST"] = server_address env_config_data["env_PORT"] = server_port @@ -180,6 +197,8 @@ def save_trigger(server_address, server_port, final_result_list,t_mongodb_host,t env_config_data["env_SILICONFLOW_KEY"] = t_siliconflow_key env_config_data["env_DEEP_SEEK_BASE_URL"] = t_deepseek_base_url env_config_data["env_DEEP_SEEK_KEY"] = t_deepseek_key + env_config_data["env_VOLCENGINE_BASE_URL"] = t_volcengine_base_url + env_config_data["env_VOLCENGINE_KEY"] = t_volcengine_key save_to_env_file(env_config_data) logger.success("配置已保存到 .env.prod 文件中") return "配置已保存" @@ -520,12 +539,24 @@ with gr.Blocks(title="MaimBot配置文件编辑") as app: value=env_config_data["env_DEEP_SEEK_KEY"], interactive=True ) + with gr.Row(): + volcengine_base_url = gr.Textbox( + label="VolcEngine的BaseURL", + value=env_config_data["env_VOLCENGINE_BASE_URL"], + interactive=True + ) + with gr.Row(): + volcengine_key = gr.Textbox( + label="VolcEngine的key", + value=env_config_data["env_VOLCENGINE_KEY"], + interactive=True + ) with gr.Row(): save_env_btn = gr.Button("保存环境配置",variant="primary") with gr.Row(): save_env_btn.click( save_trigger, - inputs=[server_address,server_port,final_result,mongodb_host,mongodb_port,mongodb_database_name,chatanywhere_base_url,chatanywhere_key,siliconflow_base_url,siliconflow_key,deepseek_base_url,deepseek_key], + inputs=[server_address,server_port,final_result,mongodb_host,mongodb_port,mongodb_database_name,chatanywhere_base_url,chatanywhere_key,siliconflow_base_url,siliconflow_key,deepseek_base_url,deepseek_key,volcengine_base_url,volcengine_key], outputs=[gr.Textbox( label="保存结果", interactive=False @@ -827,12 +858,20 @@ with gr.Blocks(title="MaimBot配置文件编辑") as app: gr.Markdown( """### 模型设置""" ) + with gr.Row(): + gr.Markdown( + """### 注意\n + 如果你是用的是火山引擎的API，建议查看[这篇文档](https://zxmucttizt8.feishu.cn/wiki/MQj7wp6dki6X8rkplApc2v6Enkd)中的修改火山API部分\n + 因为修改至火山API涉及到修改源码部分，由于自己修改源码造成的问题MaiMBot官方并不因此负责！\n + 感谢理解，感谢你使用MaiMBot + """ + ) with gr.Tabs(): with gr.TabItem("1-主要回复模型"): with gr.Row(): model1_name = gr.Textbox(value=config_data['model']['llm_reasoning']['name'], label="模型1的名称") with gr.Row(): - model1_provider = gr.Dropdown(choices=["SILICONFLOW","DEEP_SEEK", "CHAT_ANY_WHERE"], value=config_data['model']['llm_reasoning']['provider'], label="模型1（主要回复模型）提供商") + model1_provider = gr.Dropdown(choices=MODEL_PROVIDER_LIST, value=config_data['model']['llm_reasoning']['provider'], label="模型1（主要回复模型）提供商") with gr.Row(): model1_pri_in = gr.Number(value=config_data['model']['llm_reasoning']['pri_in'], label="模型1（主要回复模型）的输入价格（非必填，可以记录消耗）") with gr.Row(): @@ -841,12 +880,12 @@ with gr.Blocks(title="MaimBot配置文件编辑") as app: with gr.Row(): model2_name = gr.Textbox(value=config_data['model']['llm_normal']['name'], label="模型2的名称") with gr.Row(): - model2_provider = gr.Dropdown(choices=["SILICONFLOW","DEEP_SEEK", "CHAT_ANY_WHERE"], value=config_data['model']['llm_normal']['provider'], label="模型2提供商") + model2_provider = gr.Dropdown(choices=MODEL_PROVIDER_LIST, value=config_data['model']['llm_normal']['provider'], label="模型2提供商") with gr.TabItem("3-次要模型"): with gr.Row(): model3_name = gr.Textbox(value=config_data['model']['llm_reasoning_minor']['name'], label="模型3的名称") with gr.Row(): - model3_provider = gr.Dropdown(choices=["SILICONFLOW","DEEP_SEEK", "CHAT_ANY_WHERE"], value=config_data['model']['llm_reasoning_minor']['provider'], label="模型3提供商") + model3_provider = gr.Dropdown(choices=MODEL_PROVIDER_LIST, value=config_data['model']['llm_reasoning_minor']['provider'], label="模型3提供商") with gr.TabItem("4-情感&主题模型"): with gr.Row(): gr.Markdown( @@ -855,7 +894,7 @@ with gr.Blocks(title="MaimBot配置文件编辑") as app: with gr.Row(): emotion_model_name = gr.Textbox(value=config_data['model']['llm_emotion_judge']['name'], label="情感模型名称") with gr.Row(): - emotion_model_provider = gr.Dropdown(choices=["SILICONFLOW","DEEP_SEEK", "CHAT_ANY_WHERE"], value=config_data['model']['llm_emotion_judge']['provider'], label="情感模型提供商") + emotion_model_provider = gr.Dropdown(choices=MODEL_PROVIDER_LIST, value=config_data['model']['llm_emotion_judge']['provider'], label="情感模型提供商") with gr.Row(): gr.Markdown( """### 主题模型设置""" @@ -863,11 +902,11 @@ with gr.Blocks(title="MaimBot配置文件编辑") as app: with gr.Row(): topic_judge_model_name = gr.Textbox(value=config_data['model']['llm_topic_judge']['name'], label="主题判断模型名称") with gr.Row(): - topic_judge_model_provider = gr.Dropdown(choices=["SILICONFLOW","DEEP_SEEK", "CHAT_ANY_WHERE"], value=config_data['model']['llm_topic_judge']['provider'], label="主题判断模型提供商") + topic_judge_model_provider = gr.Dropdown(choices=MODEL_PROVIDER_LIST, value=config_data['model']['llm_topic_judge']['provider'], label="主题判断模型提供商") with gr.Row(): summary_by_topic_model_name = gr.Textbox(value=config_data['model']['llm_summary_by_topic']['name'], label="主题总结模型名称") with gr.Row(): - summary_by_topic_model_provider = gr.Dropdown(choices=["SILICONFLOW","DEEP_SEEK", "CHAT_ANY_WHERE"], value=config_data['model']['llm_summary_by_topic']['provider'], label="主题总结模型提供商") + summary_by_topic_model_provider = gr.Dropdown(choices=MODEL_PROVIDER_LIST, value=config_data['model']['llm_summary_by_topic']['provider'], label="主题总结模型提供商") with gr.TabItem("5-识图模型"): with gr.Row(): gr.Markdown( @@ -876,7 +915,7 @@ with gr.Blocks(title="MaimBot配置文件编辑") as app: with gr.Row(): vlm_model_name = gr.Textbox(value=config_data['model']['vlm']['name'], label="识图模型名称") with gr.Row(): - vlm_model_provider = gr.Dropdown(choices=["SILICONFLOW","DEEP_SEEK", "CHAT_ANY_WHERE"], value=config_data['model']['vlm']['provider'], label="识图模型提供商") + vlm_model_provider = gr.Dropdown(choices=MODEL_PROVIDER_LIST, value=config_data['model']['vlm']['provider'], label="识图模型提供商") with gr.Row(): save_model_btn = gr.Button("保存回复&模型设置",variant="primary", elem_id="save_model_btn") with gr.Row():