GithubBackup/gemini-balance
1
0
Fork 0
mirror of https://github.com/snailyp/gemini-balance.git synced 2026-05-06 20:32:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: 统一从model_service读取模型列表

Browse Source
This commit is contained in:
Toddy
2025-03-18 18:05:00 +00:00
parent ba292dbedd
commit 21444ed6c7
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
app/api/gemini_routes.py
View File

@@ -41,8 +41,8 @@ async def list_models(_=Depends(security_service.verify_key),
model_mapping = {x.get("name", "").split("/", maxsplit=1)[1]: x for x in models_json["models"]}
# 添加搜索模型
if settings.SEARCH_MODELS:
for name in settings.SEARCH_MODELS:
if model_service.search_models:
for name in model_service.search_models:
model = model_mapping.get(name, None)
if not model:
continue
@@ -56,8 +56,8 @@ async def list_models(_=Depends(security_service.verify_key),
models_json["models"].append(item)
# 添加图像生成模型
if settings.IMAGE_MODELS:
for name in settings.IMAGE_MODELS:
if model_service.image_models:
for name in model_service.image_models:
model = model_mapping.get(name, None)
if not model:
continue