gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新支持火山引擎API

Browse Source
This commit is contained in:
DrSmoothl
2025-03-15 14:22:54 +08:00
parent dac1a6f769
commit c8b88b4f1f
1 changed files with 53 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

67
webui.py
View File

@@ -10,11 +10,7 @@ import json
is_share = False is_share = False
debug = True debug = True
try: config_data = toml.load("config/bot_config.toml")
config_data = toml.load("config/bot_config.toml")
except Exception as e:
logger.error(f"读取配置文件失败，请检查配置文件是否存在，错误信息为：{e}")
sys.exit(1)
CONFIG_VERSION = config_data["inner"]["version"] CONFIG_VERSION = config_data["inner"]["version"]
PARSED_CONFIG_VERSION = float(CONFIG_VERSION[2:]) 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_前缀 f.write(f"{var[4:]}={value}\n") # 移除env_前缀
logger.info(f"配置已保存到 {filename}") logger.info(f"配置已保存到 {filename}")
#载入env文件并解析
env_config_file = ".env.prod" # 配置文件路径 env_config_file = ".env.prod" # 配置文件路径
env_config_data = parse_env_config(env_config_file) 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读取保存结束 #env读取保存结束
#============================================== #==============================================
@@ -166,7 +183,7 @@ def format_list_to_str(lst):
#env保存函数 #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) final_result_lists = format_list_to_str(final_result_list)
env_config_data["env_HOST"] = server_address env_config_data["env_HOST"] = server_address
env_config_data["env_PORT"] = server_port 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_SILICONFLOW_KEY"] = t_siliconflow_key
env_config_data["env_DEEP_SEEK_BASE_URL"] = t_deepseek_base_url 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_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) save_to_env_file(env_config_data)
logger.success("配置已保存到 .env.prod 文件中") logger.success("配置已保存到 .env.prod 文件中")
return "配置已保存" return "配置已保存"
@@ -520,12 +539,24 @@ with gr.Blocks(title="MaimBot配置文件编辑") as app:
value=env_config_data["env_DEEP_SEEK_KEY"], value=env_config_data["env_DEEP_SEEK_KEY"],
interactive=True 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(): with gr.Row():
save_env_btn = gr.Button("保存环境配置",variant="primary") save_env_btn = gr.Button("保存环境配置",variant="primary")
with gr.Row(): with gr.Row():
save_env_btn.click( save_env_btn.click(
save_trigger, 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( outputs=[gr.Textbox(
label="保存结果", label="保存结果",
interactive=False interactive=False
@@ -827,12 +858,20 @@ with gr.Blocks(title="MaimBot配置文件编辑") as app:
gr.Markdown( 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.Tabs():
with gr.TabItem("1-主要回复模型"): with gr.TabItem("1-主要回复模型"):
with gr.Row(): with gr.Row():
model1_name = gr.Textbox(value=config_data['model']['llm_reasoning']['name'], label="模型1的名称") model1_name = gr.Textbox(value=config_data['model']['llm_reasoning']['name'], label="模型1的名称")
with gr.Row(): 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(): with gr.Row():
model1_pri_in = gr.Number(value=config_data['model']['llm_reasoning']['pri_in'], label="模型1（主要回复模型）的输入价格（非必填，可以记录消耗）") model1_pri_in = gr.Number(value=config_data['model']['llm_reasoning']['pri_in'], label="模型1（主要回复模型）的输入价格（非必填，可以记录消耗）")
with gr.Row(): with gr.Row():
@@ -841,12 +880,12 @@ with gr.Blocks(title="MaimBot配置文件编辑") as app:
with gr.Row(): with gr.Row():
model2_name = gr.Textbox(value=config_data['model']['llm_normal']['name'], label="模型2的名称") model2_name = gr.Textbox(value=config_data['model']['llm_normal']['name'], label="模型2的名称")
with gr.Row(): 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.TabItem("3-次要模型"):
with gr.Row(): with gr.Row():
model3_name = gr.Textbox(value=config_data['model']['llm_reasoning_minor']['name'], label="模型3的名称") model3_name = gr.Textbox(value=config_data['model']['llm_reasoning_minor']['name'], label="模型3的名称")
with gr.Row(): 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.TabItem("4-情感&主题模型"):
with gr.Row(): with gr.Row():
gr.Markdown( gr.Markdown(
@@ -855,7 +894,7 @@ with gr.Blocks(title="MaimBot配置文件编辑") as app:
with gr.Row(): with gr.Row():
emotion_model_name = gr.Textbox(value=config_data['model']['llm_emotion_judge']['name'], label="情感模型名称") emotion_model_name = gr.Textbox(value=config_data['model']['llm_emotion_judge']['name'], label="情感模型名称")
with gr.Row(): 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(): with gr.Row():
gr.Markdown( gr.Markdown(
"""### 主题模型设置""" """### 主题模型设置"""
@@ -863,11 +902,11 @@ with gr.Blocks(title="MaimBot配置文件编辑") as app:
with gr.Row(): with gr.Row():
topic_judge_model_name = gr.Textbox(value=config_data['model']['llm_topic_judge']['name'], label="主题判断模型名称") topic_judge_model_name = gr.Textbox(value=config_data['model']['llm_topic_judge']['name'], label="主题判断模型名称")
with gr.Row(): 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(): with gr.Row():
summary_by_topic_model_name = gr.Textbox(value=config_data['model']['llm_summary_by_topic']['name'], label="主题总结模型名称") summary_by_topic_model_name = gr.Textbox(value=config_data['model']['llm_summary_by_topic']['name'], label="主题总结模型名称")
with gr.Row(): 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.TabItem("5-识图模型"):
with gr.Row(): with gr.Row():
gr.Markdown( gr.Markdown(
@@ -876,7 +915,7 @@ with gr.Blocks(title="MaimBot配置文件编辑") as app:
with gr.Row(): with gr.Row():
vlm_model_name = gr.Textbox(value=config_data['model']['vlm']['name'], label="识图模型名称") vlm_model_name = gr.Textbox(value=config_data['model']['vlm']['name'], label="识图模型名称")
with gr.Row(): 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(): with gr.Row():
save_model_btn = gr.Button("保存回复&模型设置",variant="primary", elem_id="save_model_btn") save_model_btn = gr.Button("保存回复&模型设置",variant="primary", elem_id="save_model_btn")
with gr.Row(): with gr.Row():