"...pruning/git@developer.sourcefind.cn:OpenDAS/nni.git" did not exist on "26f47727bffb2a02ff2afbfa6d24d2250f2a76b8"
Commit 0a48114b authored by Timothy J. Baek's avatar Timothy J. Baek
Browse files

refac

parent 468c6398
...@@ -312,13 +312,20 @@ async def get_models(user=Depends(get_verified_user)): ...@@ -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(),
} }
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment