From 34a98389f5fcea9c3f60be2fc8b56cab7ba8d432 Mon Sep 17 00:00:00 2001 From: yinpeng <2291314224@qq.com> Date: Tue, 11 Feb 2025 02:55:39 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E5=9B=BE=E7=89=87?= =?UTF-8?q?=E7=94=9F=E6=88=90=E6=A8=A1=E5=9E=8B=E9=87=8D=E5=A4=8D=E6=B7=BB?= =?UTF-8?q?=E5=8A=A0=E7=9A=84=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/services/model_service.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/app/services/model_service.py b/app/services/model_service.py index fcad8e4..b0a2fbd 100644 --- a/app/services/model_service.py +++ b/app/services/model_service.py @@ -52,15 +52,14 @@ class ModelService: "parent": None, } openai_format["data"].append(openai_model) - - if settings.CREATE_IMAGE_MODEL: - image_model = openai_model.copy() - image_model["id"] = f"{settings.CREATE_IMAGE_MODEL}-chat" - openai_format["data"].append(image_model) if model_id in self.model_search: search_model = openai_model.copy() search_model["id"] = f"{model_id}-search" openai_format["data"].append(search_model) + if settings.CREATE_IMAGE_MODEL: + image_model = openai_model.copy() + image_model["id"] = f"{settings.CREATE_IMAGE_MODEL}-chat" + openai_format["data"].append(image_model) return openai_format