Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-15 19:37:47 +01:00
Code Issues Packages Projects Releases Wiki Activity

refac: show connection type for custom models

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2025-12-02 06:19:48 -05:00
parent 954aaa6bdc
commit 39f7575b64
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
backend/open_webui/utils/models.py
View File

@@ -191,6 +191,8 @@ async def get_all_models(request, refresh: bool = False, user: UserModel = None)
):
# Custom model based on a base model
owned_by = "openai"
connection_type = None
pipe = None
for m in models:
@@ -201,6 +203,8 @@ async def get_all_models(request, refresh: bool = False, user: UserModel = None)
owned_by = m.get("owned_by", "unknown")
if "pipe" in m:
pipe = m["pipe"]
connection_type = m.get("connection_type", None)
break
model = {
@@ -209,6 +213,7 @@ async def get_all_models(request, refresh: bool = False, user: UserModel = None)
"object": "model",
"created": custom_model.created_at,
"owned_by": owned_by,
"connection_type": connection_type,
"preset": True,
**({"pipe": pipe} if pipe is not None else {}),
}