From 73e98a185df1d8bee65b2bda19a4295b7dd7ae2f Mon Sep 17 00:00:00 2001
From: snaily <snaily@linux.do>
Date: Sun, 13 Apr 2025 12:45:23 +0800
Subject: [PATCH] =?UTF-8?q?fix:=E4=BF=AE=E5=A4=8Dgemini=E6=A0=BC=E5=BC=8F?=
 =?UTF-8?q?=E4=B8=8D=E8=83=BD=E6=9F=A5=E8=AF=A2=E6=A8=A1=E5=9E=8B=E5=88=97?=
 =?UTF-8?q?=E8=A1=A8=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/router/gemini_routes.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/app/router/gemini_routes.py b/app/router/gemini_routes.py
index 2206a14..d7e4ac4 100644
--- a/app/router/gemini_routes.py
+++ b/app/router/gemini_routes.py
@@ -53,8 +53,8 @@ async def list_models(
     model_mapping = {x.get("name", "").split("/", maxsplit=1)[1]: x for x in models_json["models"]}
     
     # 添加搜索模型
-    if model_service.search_models:
-        for name in model_service.search_models:
+    if settings.SEARCH_MODELS:
+        for name in settings.SEARCH_MODELS:
             model = model_mapping.get(name)
             if not model:
                 continue
@@ -68,8 +68,8 @@ async def list_models(
             models_json["models"].append(item)
     
     # 添加图像生成模型
-    if model_service.image_models:
-        for name in model_service.image_models:
+    if settings.IMAGE_MODELS:
+        for name in settings.IMAGE_MODELS:
             model = model_mapping.get(name)
             if not model:
                 continue