新版本重构了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

57
model/ai_models/aliyun.py Normal file
View File

@@ -0,0 +1,57 @@
import toml
from model.logger import setup_logger
from http import HTTPStatus
from dashscope import Application
from model.sql_tools import init_db
from model.sql_tools import get_uuid
from model.sql_tools import add_uuid
logger = setup_logger()
def main(message, group_openid):
with open('./config.toml', 'r', encoding='utf-8') as f:
config = toml.load(f)
error_message = config.get("error_message")
# aliyun
aliyun_app_id = config.get("aliyun_app_id")
aliyun_api_key = config.get("aliyun_api_key")
init_db()
uuid = get_uuid(group_openid)
if uuid == "":
logger.info("未找到 UUID")
else:
logger.info(f"找到 UUID: {uuid}")
# 发送请求
response = Application.call(
# 若没有配置环境变量可用百炼API Key将下行替换为api_key="sk-xxx"。但不建议在生产环境中直接将API Key硬编码到代码中以减少API Key泄露风险。
api_key=aliyun_api_key,
app_id=aliyun_app_id,
prompt= message,
session_id = group_openid
)
# 检查响应状态码
if response.status_code != HTTPStatus.OK:
logger.info(f"请求失败!状态码: {response.status_code}")
logger.info(f"错误信息: {response.message}") # 打印错误信息
return error_message
else:
logger.info("请求成功!返回结果:")
print(f"message: {response.output.text}, session_id: {response.output.session_id}") # test
# 存储uuid
if uuid == "":
uuid = response.output.session_id
add_uuid(group_openid, uuid)
# 提取 answer 值
answer = response.output.text
return answer