diff --git a/endpoints/fruits.get.py b/endpoints/fruits.get.py index 41c9c56..595b118 100644 --- a/endpoints/fruits.get.py +++ b/endpoints/fruits.get.py @@ -2,11 +2,13 @@ from fastapi import APIRouter, Depends from sqlalchemy.orm import Session from typing import List from core.database import get_db +from schemas.fruit import FruitSchema from helpers.fruit_helpers import get_all_fruits router = APIRouter() -@router.get("/fruits", response_model=List[str]) -async def get_fruit_names(db: Session = Depends(get_db)): +@router.get("/fruits", response_model=List[FruitSchema]) +async def get_fruits(db: Session = Depends(get_db)): + """Get all fruits""" fruits = get_all_fruits(db) - return [fruit.name for fruit in fruits] \ No newline at end of file + return fruits \ No newline at end of file