From d3e7a5748acd75d7d20f5387bad5adf0fdd5a8ac Mon Sep 17 00:00:00 2001 From: inter Date: Mon, 8 Sep 2025 16:36:15 +0800 Subject: [PATCH] Add File --- backend/apps/system/crud/aimodel_manage.py | 26 ++++++++++++++++++++++ 1 file changed, 26 insertions(+) create mode 100644 backend/apps/system/crud/aimodel_manage.py diff --git a/backend/apps/system/crud/aimodel_manage.py b/backend/apps/system/crud/aimodel_manage.py new file mode 100644 index 0000000..77d6cb5 --- /dev/null +++ b/backend/apps/system/crud/aimodel_manage.py @@ -0,0 +1,26 @@ + +from apps.system.models.system_model import AiModelDetail +from common.core.db import engine +from sqlmodel import Session, select +from common.utils.crypto import sqlbot_encrypt +from common.utils.utils import SQLBotLogUtil + +async def async_model_info(): + with Session(engine) as session: + model_list = session.exec(select(AiModelDetail)).all() + any_model_change = False + if model_list: + for model in model_list: + if model.api_domain.startswith("http"): + if model.api_key: + model.api_key = await sqlbot_encrypt(model.api_key) + if model.api_domain: + model.api_domain = await sqlbot_encrypt(model.api_domain) + session.add(model) + any_model_change = True + if any_model_change: + session.commit() + SQLBotLogUtil.info("✅ 异步加密已有模型的密钥和地址完成") + + + \ No newline at end of file