From c42dc195241cf486934f77bd3736cdc8a25fdb9e Mon Sep 17 00:00:00 2001
From: Rui Alves <alvesrui951@gmail.com>
Date: Wed, 16 Oct 2024 13:53:57 +0100
Subject: [PATCH] Added responses for more review.py endpoints

---
 frigate/api/defs/generic_response.py | 6 ++++++
 frigate/api/review.py                | 7 ++++---
 2 files changed, 10 insertions(+), 3 deletions(-)
 create mode 100644 frigate/api/defs/generic_response.py

diff --git a/frigate/api/defs/generic_response.py b/frigate/api/defs/generic_response.py
new file mode 100644
index 000000000..dbf9434f9
--- /dev/null
+++ b/frigate/api/defs/generic_response.py
@@ -0,0 +1,6 @@
+from pydantic import BaseModel
+
+
+class GenericResponse(BaseModel):
+    success: bool
+    message: str
diff --git a/frigate/api/review.py b/frigate/api/review.py
index 40e5fd806..61eead5f7 100644
--- a/frigate/api/review.py
+++ b/frigate/api/review.py
@@ -13,6 +13,7 @@ from fastapi.responses import JSONResponse
 from peewee import Case, DoesNotExist, fn, operator
 from playhouse.shortcuts import model_to_dict
 
+from frigate.api.defs.generic_response import GenericResponse
 from frigate.api.defs.review_query_parameters import (
     ReviewActivityMotionQueryParams,
     ReviewQueryParams,
@@ -572,7 +573,7 @@ def audio_activity(params: ReviewActivityMotionQueryParams = Depends()):
     return JSONResponse(content=normalized)
 
 
-@router.get("/review/event/{event_id}")
+@router.get("/review/event/{event_id}", response_model=ReviewSegmentResponse)
 def get_review_from_event(event_id: str):
     try:
         return model_to_dict(
@@ -584,7 +585,7 @@ def get_review_from_event(event_id: str):
         return "Review item not found", 404
 
 
-@router.get("/review/{event_id}")
+@router.get("/review/{event_id}", response_model=ReviewSegmentResponse)
 def get_review(event_id: str):
     try:
         return model_to_dict(ReviewSegment.get(ReviewSegment.id == event_id))
@@ -592,7 +593,7 @@ def get_review(event_id: str):
         return "Review item not found", 404
 
 
-@router.delete("/review/{event_id}/viewed")
+@router.delete("/review/{event_id}/viewed", response_model=GenericResponse)
 def set_not_reviewed(event_id: str):
     try:
         review: ReviewSegment = ReviewSegment.get(ReviewSegment.id == event_id)