i/Aether
1
0
Fork 0
mirror of https://github.com/fawney19/Aether.git synced 2026-09-02 01:10:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: 关闭 auto_fetch_models 时保留锁定的模型

Browse Source
This commit is contained in:
fawney19
2026-01-14 21:31:05 +08:00
parent 8909e6741a
commit 154ec94c58
1 changed files with 15 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/api/admin/endpoints/keys.py
View File

@@ -260,9 +260,21 @@ class AdminUpdateEndpointKeyAdapter(AdminApiAdapter):
logger.error(f"触发模型获取失败: {e}") logger.error(f"触发模型获取失败: {e}")
# 不抛出异常，避免影响 Key 更新操作 # 不抛出异常，避免影响 Key 更新操作
elif auto_fetch_enabled_before and not auto_fetch_enabled_after: elif auto_fetch_enabled_before and not auto_fetch_enabled_after:
# 关闭了 auto_fetch_models，将 allowed_models 设置为 null（不限制） # 关闭了 auto_fetch_models，只保留锁定的模型，清除自动获取的模型
logger.info("[AUTO_FETCH] Key %s 关闭自动获取模型，清空 allowed_models", self.key_id) locked = key.locked_models or []
key.allowed_models = None if locked:
key.allowed_models = locked
logger.info(
"[AUTO_FETCH] Key %s 关闭自动获取模型，保留 %d 个锁定模型",
self.key_id,
len(locked),
)
else:
key.allowed_models = None
logger.info(
"[AUTO_FETCH] Key %s 关闭自动获取模型，无锁定模型，清空 allowed_models",
self.key_id,
)
db.commit() db.commit()
db.refresh(key) db.refresh(key)