新版本重构了ai模型提供商逻辑,新增MCUNC,阿里云百炼ai提供商,支持语言可配置,优化日志输出格式

This commit is contained in:
2025-09-24 17:55:33 +08:00
parent 0ddede55b5
commit 4d9e1bf355
15 changed files with 563 additions and 207 deletions

35
model/ai_model.py Normal file
View File

@@ -0,0 +1,35 @@
import toml
from model.logger import setup_logger
logger = setup_logger()
def main(message,qid,group_openid):
with open('./config.toml', 'r', encoding='utf-8') as f:
config = toml.load(f)
ai_service = config.get("ai_service")
enable_qid = config.get("enable_qid")
qid_prefix = config.get("qid_prefix")
qid_suffix = config.get("qid_suffix")
error_message = config.get("error_message")
if enable_qid:
message = f"{qid_prefix}{qid}{qid_suffix}{message}"
else:
pass
if ai_service == "dify":
from model.ai_models import dify
return dify.main(message, group_openid)
elif ai_service == "mcunc":
from model.ai_models import mcunc
return mcunc.main(message, group_openid)
elif ai_service == "xyit":
from model.ai_models import xyit
return xyit.main(message, group_openid)
else:
logger.error("未配置ai_service")
return error_message