Merge pull request #952 from SuanmoSuanyangTechnology/fix/Timebomb_031
fix(model_service)
This commit is contained in:
@@ -369,6 +369,15 @@ class ModelConfigService:
|
|||||||
raise BusinessException("模型名称已存在", BizCode.DUPLICATE_NAME)
|
raise BusinessException("模型名称已存在", BizCode.DUPLICATE_NAME)
|
||||||
|
|
||||||
model = ModelConfigRepository.update(db, model_id, model_data, tenant_id=tenant_id)
|
model = ModelConfigRepository.update(db, model_id, model_data, tenant_id=tenant_id)
|
||||||
|
|
||||||
|
# 同步更新关联 api_keys 的 capability 和 is_omni
|
||||||
|
if model_data.capability is not None or model_data.is_omni is not None:
|
||||||
|
for api_key in model.api_keys:
|
||||||
|
if model_data.capability is not None:
|
||||||
|
api_key.capability = model_data.capability
|
||||||
|
if model_data.is_omni is not None:
|
||||||
|
api_key.is_omni = model_data.is_omni
|
||||||
|
|
||||||
db.commit()
|
db.commit()
|
||||||
db.refresh(model)
|
db.refresh(model)
|
||||||
return model
|
return model
|
||||||
|
|||||||
Reference in New Issue
Block a user