This commit is contained in:
Timothy J. Baek 2024-05-24 03:06:57 -07:00
parent 468c6398cd
commit 0a48114bd2

View File

@ -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(),
}
)