Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2024-12-19 19:06:16 +01:00
Code Issues Projects Releases Wiki Activity

Fix plus api (#13937)

Browse Source 
* Fix plus api

* Fix regenerate api
This commit is contained in:
Nicolas Mowen 2024-09-24 11:22:11 -06:00 committed by GitHub
parent a9ff795948
commit 7854e1c2c1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frigate/api/defs/events_body.py
View File

@ -28,3 +28,7 @@ class EventsCreateBody(BaseModel):
class EventsEndBody(BaseModel):
end_time: Optional[int] = datetime.now().timestamp()
class SubmitPlusBody(BaseModel):
include_annotation: int = Field(default=1)

9
frigate/api/event.py
View File

@ -22,6 +22,7 @@ from frigate.api.defs.events_body import (
EventsDescriptionBody,
EventsEndBody,
EventsSubLabelBody,
SubmitPlusBody,
)
from frigate.api.defs.events_query_parameters import (
DEFAULT_TIME_RANGE,
@ -616,7 +617,7 @@ def set_retain(event_id: str):
@router.post("/events/{event_id}/plus")
def send_to_plus(request: Request, event_id: str):
def send_to_plus(request: Request, event_id: str, body: SubmitPlusBody = None):
if not request.app.frigate_config.plus_api.is_active():
message = "PLUS_API_KEY environment variable is not set"
logger.error(message)
@ -630,9 +631,7 @@ def send_to_plus(request: Request, event_id: str):
status_code=400,
)
include_annotation = (
request.json.get("include_annotation") if request.is_json else None
)
include_annotation = body.include_annotation if body is not None else None
try:
event = Event.get(Event.id == event_id)
@ -945,7 +944,7 @@ def set_description(
)
@router.put("/events/<id>/description/regenerate")
@router.put("/events/{event_id}/description/regenerate")
def regenerate_description(request: Request, event_id: str):
try:
event: Event = Event.get(Event.id == event_id)