diff --git a/backend/open_webui/apps/webui/routers/models.py b/backend/open_webui/apps/webui/routers/models.py index a99c65d76..a5cb2395e 100644 --- a/backend/open_webui/apps/webui/routers/models.py +++ b/backend/open_webui/apps/webui/routers/models.py @@ -18,8 +18,18 @@ router = APIRouter() @router.get("/", response_model=list[ModelResponse]) -async def get_models(user=Depends(get_verified_user)): - return Models.get_all_models() +async def get_models(id: Optional[str] = None, user=Depends(get_verified_user)): + if id: + model = Models.get_model_by_id(id) + if model: + return [model] + else: + raise HTTPException( + status_code=status.HTTP_401_UNAUTHORIZED, + detail=ERROR_MESSAGES.NOT_FOUND, + ) + else: + return Models.get_all_models() ############################ @@ -50,24 +60,6 @@ async def add_new_model( ) -############################ -# GetModelById -############################ - - -@router.get("/", response_model=Optional[ModelModel]) -async def get_model_by_id(id: str, user=Depends(get_verified_user)): - model = Models.get_model_by_id(id) - - if model: - return model - else: - raise HTTPException( - status_code=status.HTTP_401_UNAUTHORIZED, - detail=ERROR_MESSAGES.NOT_FOUND, - ) - - ############################ # UpdateModelById ############################