| """系统可用模型的判定规则（与「对话可用模型」「生图可用模型」对称）。 |
| |
| ``/api/system/models`` 给出**账号池当前能调用的全部模型**：上游账号池模型 |
| （含生图模型与 ``research`` / ``python`` 这类工具类伪模型）+ 第三方接入模型。 |
| |
| **默认全部启用。** 这里的开关只是**展示层过滤** —— 只决定这些模型是否出现在 |
| 管理页「系统可用模型」列表里，不改动上游账号池、第三方配置、计费、路由、 |
| 额度，也不影响对外 ``/v1/models``（后者必须保持全量，OpenAI 兼容）。 |
| |
| 之所以给系统列表单独一份开关，而不是复用对话 / 生图两边的配置：管理员想把 |
| 不关心的模型从列表里收起来时，不应该被迫去改「对话可用模型」或 |
| 「生图可用模型」的名单——三份名单的用途完全不同。 |
| """ |
|
|
| from __future__ import annotations |
|
|
| from typing import Any |
|
|
| from services.chat_model_service import ( |
| SOURCE_THIRD_PARTY, |
| SOURCE_UPSTREAM, |
| _clean, |
| _clean_id_set, |
| model_source, |
| ) |
|
|
|
|
| def resolve_system_models( |
| items: list[Any], |
| overrides: dict[str, Any] | None = None, |
| ) -> list[dict[str, Any]]: |
| """给每个模型标注 ``system_enabled`` / ``system_source``。 |
| |
| 不清洗、不过滤任何模型：系统列表就是"能调用到的全部模型"， |
| 只是允许管理员把不关心的收起来（``system_enabled`` = False）。 |
| """ |
|
|
| source = overrides if isinstance(overrides, dict) else {} |
| include = _clean_id_set(source.get("include")) |
| exclude = _clean_id_set(source.get("exclude")) |
|
|
| resolved: list[dict[str, Any]] = [] |
| for item in items or []: |
| if not isinstance(item, dict): |
| continue |
| model_id = _clean(item.get("id")) |
| if not model_id: |
| continue |
| if model_id in include: |
| enabled = True |
| elif model_id in exclude: |
| enabled = False |
| else: |
| enabled = True |
| resolved.append({ |
| **item, |
| "system_enabled": enabled, |
| "system_source": model_source(item), |
| }) |
| |
| resolved.sort(key=lambda entry: ( |
| 0 if entry["system_enabled"] else 1, |
| 0 if entry["system_source"] == SOURCE_UPSTREAM else 1, |
| str(entry.get("id") or ""), |
| )) |
| return resolved |
|
|
|
|
| __all__ = [ |
| "resolve_system_models", |
| "SOURCE_THIRD_PARTY", |
| "SOURCE_UPSTREAM", |
| ] |
|
|
