fix(model_service): sync model capability and is_omni to associated api_keys

This commit is contained in:
Timebomb2018
2026-04-21 12:15:14 +08:00
parent f3ac7e084d
commit 8011610acc

View File

@@ -369,6 +369,15 @@ class ModelConfigService:
raise BusinessException("模型名称已存在", BizCode.DUPLICATE_NAME)
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.refresh(model)
return model