From ced82b09d089a6999a96a1f7939abceaa35fdc31 Mon Sep 17 00:00:00 2001 From: Inferencebot Date: Fri, 16 Oct 2026 10:26:04 -0600 Subject: [PATCH] portal: filter image-gen models from chat picker by name (gateway strips mode from /v1/models) --- app/main.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/app/main.py b/app/main.py index d5d27c5..70b77ef 100644 --- a/app/main.py +++ b/app/main.py @@ -2017,9 +2017,14 @@ async def list_models(request: Request): mid = (m.get("id") or "").lower() return any(h in mid for h in embedding_name_hints) + image_name_hints = ("qwen-image",) + def is_image_model(m) -> bool: mode = ((m.get("info") or {}).get("meta") or {}).get("mode") or "" - return mode in image_modes + if mode in image_modes: + return True + mid = (m.get("id") or "").lower() + return any(h in mid for h in image_name_hints) payload["data"] = [ m for m in payload.get("data", [])