From 8011610accbb3d169a3566f8f7ad76903f71cb77 Mon Sep 17 00:00:00 2001
From: Timebomb2018 <18868801967@163.com>
Date: Tue, 21 Apr 2026 12:15:14 +0800
Subject: [PATCH] fix(model_service): sync model capability and is_omni to
 associated api_keys

---
 api/app/services/model_service.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/api/app/services/model_service.py b/api/app/services/model_service.py
index 12733b2a..72e46f4a 100644
--- a/api/app/services/model_service.py
+++ b/api/app/services/model_service.py
@@ -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