From 948d5eb6a60a138dcfc539fd9e36f885200190de Mon Sep 17 00:00:00 2001
From: NuklearRabbit <145918611+NuklearRabbit@users.noreply.github.com>
Date: Mon, 10 Aug 2026 02:10:02 +0200
Subject: [PATCH] fix: normalize return risk timestamps

---
 backend/app/api/routers/bookings.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/backend/app/api/routers/bookings.py b/backend/app/api/routers/bookings.py
index 5ee060f..e1bc619 100644
--- a/backend/app/api/routers/bookings.py
+++ b/backend/app/api/routers/bookings.py
@@ -14,6 +14,7 @@ from app.schemas import (
     CurrentUser,
     NextBookingRisk,
     RegisterReturnRequest,
+    RegisterReturnResult,
     ReturnPreviewResult,
 )
 from app.services.returns import preview_vehicle_return, register_vehicle_return
@@ -133,7 +134,7 @@ def preview_return(
     )
 
 
-@router.post("/{public_ref}/return")
+@router.post("/{public_ref}/return", response_model=RegisterReturnResult)
 def register_return(
     public_ref: str,
     body: RegisterReturnRequest,
@@ -141,7 +142,7 @@ def register_return(
     idempotency_key: str = Header(..., alias="Idempotency-Key", min_length=8, max_length=128),
     db: Session = Depends(get_db),
     user: CurrentUser = Depends(get_current_user),
-) -> dict:
+) -> RegisterReturnResult:
     status_code, result = register_vehicle_return(db, public_ref, body, idempotency_key, user)
     response.status_code = status_code
-    return result
+    return RegisterReturnResult(**result)