OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat/improved-non-chat-model-substrings

Browse Source
This commit is contained in:
Olof Larsson 2025-06-26 10:28:10 +02:00
parent b5f4c85bb1
commit 548bbaf501
No known key found for this signature in database
GPG Key ID: EE38B743948AC983
1 changed files with 25 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
backend/open_webui/routers/openai.py
View File

@ -49,6 +49,29 @@ log = logging.getLogger(__name__)
log.setLevel(SRC_LOG_LEVELS["OPENAI"])
NON_CHAT_MODEL_SUBSTRINGS = [
# Legacy
"babbage",
"davinci",
# Image Gen
"dall-e",
"flux",
"sdxl",
# Embeddings / Rerankers
"embedding",
"bge-",
"e5-",
"reranker",
# Speech
"tts",
"stt",
"whisper",
]
##########################################
#
# Utility functions
@ -426,14 +449,7 @@ async def get_all_models(request: Request, user: UserModel) -> dict[str, list]:
not in request.app.state.config.OPENAI_API_BASE_URLS[idx]
or not any(
name in model["id"]
for name in [
"babbage",
"dall-e",
"davinci",
"embedding",
"tts",
"whisper",
]
for name in NON_CHAT_MODEL_SUBSTRINGS
)
)
]
@ -519,14 +535,7 @@ async def get_models(
for model in response_data.get("data", [])
if not any(
name in model["id"]
for name in [
"babbage",
"dall-e",
"davinci",
"embedding",
"tts",
"whisper",
]
for name in NON_CHAT_MODEL_SUBSTRINGS
)
]