From 0a48114bd290689eb101a3d898c9b50ce19ced4a Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Fri, 24 May 2024 03:06:57 -0700 Subject: [PATCH] refac --- backend/main.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/backend/main.py b/backend/main.py index 04ed5d2d7..0c8ce8e8d 100644 --- a/backend/main.py +++ b/backend/main.py @@ -312,13 +312,20 @@ async def get_models(user=Depends(get_verified_user)): model["name"] = custom_model.name model["info"] = custom_model.model_dump() else: + + owned_by = "openai" + for model in models: + if custom_model.base_model_id == model["id"]: + owned_by = model["owned_by"] + break + models.append( { "id": custom_model.id, "name": custom_model.name, "object": "model", "created": custom_model.created_at, - "owned_by": "user", + "owned_by": owned_by, "info": custom_model.model_dump(), } )