added vegetation

api/app/routers/link.py

@@ -1,5 +1,9 @@
-from fastapi import APIRouter
+from typing import Annotated
 
+from fastapi import APIRouter, Depends
+from sqlalchemy.orm import Session
+
+from app.deps import get_db
 from app.models.link import LinkBudgetRequest, LinkBudgetResponse
 from app.routers.errors import as_501
 from app.services import link as link_service
@@ -8,5 +12,8 @@ router = APIRouter()
 
 
 @router.post("/budget", response_model=LinkBudgetResponse)
-def budget(request: LinkBudgetRequest) -> LinkBudgetResponse:
-    return as_501(lambda: link_service.link_budget(request))
+def budget(
+    request: LinkBudgetRequest,
+    db: Annotated[Session, Depends(get_db)],
+) -> LinkBudgetResponse:
+    return as_501(lambda: link_service.link_budget(request, db))