UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-01-29 13:58:09 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy J. Baek 2024-05-24 03:06:57 -07:00
parent 468c6398cd
commit 0a48114bd2
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
backend/main.py
View File

@ -312,13 +312,20 @@ async def get_models(user=Depends(get_verified_user)):
model["name"] = custom_model.name model["name"] = custom_model.name
model["info"] = custom_model.model_dump() model["info"] = custom_model.model_dump()
else: else:
owned_by = "openai"
for model in models:
if custom_model.base_model_id == model["id"]:
owned_by = model["owned_by"]
break
models.append( models.append(
{ {
"id": custom_model.id, "id": custom_model.id,
"name": custom_model.name, "name": custom_model.name,
"object": "model", "object": "model",
"created": custom_model.created_at, "created": custom_model.created_at,
"owned_by": "user", "owned_by": owned_by,
"info": custom_model.model_dump(), "info": custom_model.model_dump(),
} }
) )