GithubBackup/gemini-balance
1
0
Fork 0
mirror of https://github.com/snailyp/gemini-balance.git synced 2026-09-04 23:19:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix:修复gemini格式不能查询模型列表的问题

Browse Source
This commit is contained in:
snaily
2025-04-13 12:45:23 +08:00
parent 73a7c81f85
commit 73e98a185d
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/router/gemini_routes.py
+4 -4
View File
@@ -53,8 +53,8 @@ async def list_models(
model_mapping = {x.get("name", "").split("/", maxsplit=1)[1]: x for x in models_json["models"]} model_mapping = {x.get("name", "").split("/", maxsplit=1)[1]: x for x in models_json["models"]}
# 添加搜索模型 # 添加搜索模型
if model_service.search_models: if settings.SEARCH_MODELS:
for name in model_service.search_models: for name in settings.SEARCH_MODELS:
model = model_mapping.get(name) model = model_mapping.get(name)
if not model: if not model:
continue continue
@@ -68,8 +68,8 @@ async def list_models(
models_json["models"].append(item) models_json["models"].append(item)
# 添加图像生成模型 # 添加图像生成模型
if model_service.image_models: if settings.IMAGE_MODELS:
for name in model_service.image_models: for name in settings.IMAGE_MODELS:
model = model_mapping.get(name) model = model_mapping.get(name)
if not model: if not model:
continue continue