mirror of
https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git
synced 2026-11-09 17:57:06 +08:00
68 lines
2.3 KiB
Python
68 lines
2.3 KiB
Python
from sqlmodel import Session, select, or_
|
|
|
|
from apps.system.models.system_model import AiModelDetail, AiModelBrief, AiModelWorkspaceMapping
|
|
from common.core.db import engine
|
|
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:
|
|
current_model_change = False
|
|
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)
|
|
any_model_change = True
|
|
current_model_change = True
|
|
if model.supplier and model.supplier == 12:
|
|
model.supplier = 15
|
|
any_model_change = True
|
|
current_model_change = True
|
|
if current_model_change:
|
|
session.add(model)
|
|
if any_model_change:
|
|
session.commit()
|
|
SQLBotLogUtil.info("✅ 异步加密已有模型的密钥和地址完成")
|
|
|
|
|
|
def get_ai_model_list_by_workspace(session: Session, workspace_id: int, with_default: bool = True):
|
|
sub_stmt = (
|
|
select(AiModelWorkspaceMapping.ai_model_id)
|
|
.where(AiModelWorkspaceMapping.workspace_id == workspace_id)
|
|
.distinct()
|
|
)
|
|
|
|
# 查询：关联的模型 + default_model 为 True 的模型，默认模型排第一
|
|
base_condition = AiModelDetail.id.in_(sub_stmt)
|
|
if with_default:
|
|
where_condition = or_(base_condition, AiModelDetail.default_model == True)
|
|
else:
|
|
where_condition = base_condition
|
|
stmt = (
|
|
select(
|
|
AiModelDetail.id,
|
|
AiModelDetail.name,
|
|
AiModelDetail.default_model,
|
|
AiModelDetail.supplier,
|
|
)
|
|
.where(where_condition)
|
|
.order_by(AiModelDetail.default_model.desc())
|
|
)
|
|
rows = session.exec(stmt).all()
|
|
|
|
return [
|
|
AiModelBrief(
|
|
id=row[0],
|
|
name=row[1],
|
|
default_model=row[2],
|
|
supplier=row[3],
|
|
)
|
|
for row in rows
|
|
]
|