diff --git a/endpoints/names.post.py b/endpoints/names.post.py index d71839e..e06bee9 100644 --- a/endpoints/names.post.py +++ b/endpoints/names.post.py @@ -1,11 +1,12 @@ -from fastapi import APIRouter, status +from fastapi import APIRouter from typing import List +from schemas.fruit import FruitNamesResponse from helpers.fruit_helpers import get_all_fruit_names router = APIRouter() -@router.post("/names", status_code=status.HTTP_200_OK, response_model=List[str]) +@router.post("/names", response_model=List[FruitNamesResponse]) async def get_fruit_names(): """Return all fruit names""" - names = get_all_fruit_names() - return names \ No newline at end of file + fruit_names = get_all_fruit_names() + return fruit_names \ No newline at end of file