diff --git a/endpoints/fruits.get.py b/endpoints/fruits.get.py index 098bda1..723a551 100644 --- a/endpoints/fruits.get.py +++ b/endpoints/fruits.get.py @@ -1,12 +1,13 @@ -from fastapi import APIRouter +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", status_code=200, response_model=List[FruitSchema]) -async def get_fruits(): - """Get all fruit names""" - fruits = get_all_fruits() +@router.get("/fruits", response_model=List[FruitSchema]) +async def get_fruits(db: Session = Depends(get_db)): + fruits = get_all_fruits(db) return fruits \ No newline at end of file