zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Application model use

Browse Source
This commit is contained in:
zhangzhanwei 2025-07-03 20:21:35 +08:00 committed by zhanweizhang7
parent a3c5cf452c
commit dd621b59b9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/models_provider/tools.py
View File

@ -107,8 +107,8 @@ def is_valid_credential(provider, model_type, model_name, model_credential: Dict
def get_model_by_id(_id, workspace_id): def get_model_by_id(_id, workspace_id):
model = QuerySet(Model).filter(id=_id).first() model = QuerySet(Model).filter(id=_id).first()
get_authorized_model = DatabaseModelManage.get_model("get_authorized_model") get_authorized_model = DatabaseModelManage.get_model("get_authorized_model")
if get_authorized_model is not None: if model and model.workspace_id!=workspace_id and get_authorized_model is not None:
model = get_authorized_model(model, workspace_id) model = get_authorized_model(QuerySet(Model).filter(id=_id), workspace_id).first()
if model is None: if model is None:
raise Exception(_("Model does not exist")) raise Exception(_("Model does not exist"))
return model return model