Add File

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("✅ 异步加密已有模型的密钥和地址完成")           
+            
+            
+