| """生图可用模型的判定规则（与「对话可用模型」对称）。 |
| |
| ``/api/image/models`` 给出**全部能生图的模型**：账号池上游内置生图模型 |
| （``gpt-image-2`` 及其 codex 变体）+ 第三方接入的生图模型。 |
| |
| 候选判据复用 services/chat_model_service.py 的 ``supports_image_generation``： |
| 优先看第三方的能力字段（``txt2img`` / ``img2img``），没有该字段的上游模型退回名字判定。 |
| 多模态模型（能力里同时有 ``chat`` 与 ``txt2img``）会同时出现在对话与生图两个列表里。 |
| |
| **默认全部启用。** 第三方生图模型能不能用，本来就由第三方生图配置（分组 / 模型开关、 |
| 计费、路由）决定；这里的开关只是**展示层过滤**，只影响模型是否出现在生图下拉里， |
| 不改变计费、路由、额度与对外 API 的行为。所以升级后不会莫名其妙少模型， |
| 管理员可以按需勾掉不想在列表里看到的。 |
| """ |
|
|
| from __future__ import annotations |
|
|
| from typing import Any |
|
|
| from services.chat_model_service import ( |
| SOURCE_THIRD_PARTY, |
| SOURCE_UPSTREAM, |
| _clean, |
| image_capabilities, |
| model_source, |
| supports_image_generation, |
| ) |
|
|
| |
| IMAGE_SOURCE_BUILTIN = "builtin" |
| IMAGE_SOURCE_THIRD_PARTY = "third_party" |
|
|
|
|
| def image_model_source(item: dict[str, Any]) -> str: |
| """生图模型的来源：``builtin`` = 账号池上游内置，``third_party`` = 第三方接入。""" |
|
|
| if model_source(item) == SOURCE_THIRD_PARTY: |
| return IMAGE_SOURCE_THIRD_PARTY |
| return IMAGE_SOURCE_BUILTIN |
|
|
|
|
| def _clean_id_set(value: object) -> set[str]: |
| if not isinstance(value, list): |
| return set() |
| return {_clean(item) for item in value if _clean(item)} |
|
|
|
|
| def resolve_image_models( |
| items: list[Any], |
| overrides: dict[str, Any] | None = None, |
| ) -> list[dict[str, Any]]: |
| """筛出能生图的模型，并标注 ``image_enabled`` / ``image_default_enabled``。 |
| |
| 只保留 ``supports_image_generation`` 为真的条目（对话模型不会混进来）； |
| 分不清能不能生图的模型一律不放进候选，避免生图下拉里出现选不了的模型。 |
| """ |
|
|
| 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 or not supports_image_generation(item): |
| continue |
| if model_id in include: |
| enabled = True |
| elif model_id in exclude: |
| enabled = False |
| else: |
| enabled = True |
| resolved.append({ |
| **item, |
| "image_enabled": enabled, |
| "image_default_enabled": True, |
| "image_source": image_model_source(item), |
| "image_capabilities": sorted(image_capabilities(item)), |
| }) |
| |
| resolved.sort(key=lambda entry: ( |
| 0 if entry["image_enabled"] else 1, |
| 0 if entry["image_source"] == IMAGE_SOURCE_BUILTIN else 1, |
| str(entry.get("id") or ""), |
| )) |
| return resolved |
|
|
|
|
| __all__ = [ |
| "IMAGE_SOURCE_BUILTIN", |
| "IMAGE_SOURCE_THIRD_PARTY", |
| "image_model_source", |
| "resolve_image_models", |
| "SOURCE_THIRD_PARTY", |
| "SOURCE_UPSTREAM", |
| ] |
|
|
