From d17ca523a04a9228d54a523ffae340c391d1885e Mon Sep 17 00:00:00 2001 From: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com> Date: Thu, 10 Oct 2024 18:01:15 -0500 Subject: [PATCH] ensure event_ids is passed as list --- frigate/api/event.py | 6 +++--- frigate/events/cleanup.py | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/frigate/api/event.py b/frigate/api/event.py index ca531d098..3a8d003ad 100644 --- a/frigate/api/event.py +++ b/frigate/api/event.py @@ -939,7 +939,7 @@ def set_description( new_description, ) else: - context.db.delete_embeddings_description(event_id) + context.db.delete_embeddings_description(event_ids=[event_id]) response_message = ( f"Event {event_id} description is now blank" @@ -1025,8 +1025,8 @@ def delete_event(request: Request, event_id: str): # If semantic search is enabled, update the index if request.app.frigate_config.semantic_search.enabled: context: EmbeddingsContext = request.app.embeddings - context.db.delete_embeddings_thumbnail(id=[event_id]) - context.db.delete_embeddings_description(id=[event_id]) + context.db.delete_embeddings_thumbnail(event_ids=[event_id]) + context.db.delete_embeddings_description(event_ids=[event_id]) return JSONResponse( content=({"success": True, "message": "Event " + event_id + " deleted"}), status_code=200, diff --git a/frigate/events/cleanup.py b/frigate/events/cleanup.py index 8fabf2b21..9442eabe9 100644 --- a/frigate/events/cleanup.py +++ b/frigate/events/cleanup.py @@ -229,8 +229,8 @@ class EventCleanup(threading.Thread): Event.delete().where(Event.id << chunk).execute() if self.config.semantic_search.enabled: - self.db.delete_embeddings_description(chunk) - self.db.delete_embeddings_thumbnail(chunk) + self.db.delete_embeddings_description(event_ids=[chunk]) + self.db.delete_embeddings_thumbnail(event_ids=[chunk]) logger.debug(f"Deleted {len(events_to_delete)} embeddings") logger.info("Exiting event cleanup...")