This commit is contained in:
Timothy J. Baek 2024-09-13 00:41:20 -04:00
parent 5970dadead
commit 47e9c12fc2

View File

@ -18,8 +18,18 @@ router = APIRouter()
@router.get("/", response_model=list[ModelResponse]) @router.get("/", response_model=list[ModelResponse])
async def get_models(user=Depends(get_verified_user)): async def get_models(id: Optional[str] = None, user=Depends(get_verified_user)):
return Models.get_all_models() 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 # UpdateModelById
############################ ############################