administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-04-27 17:17:40 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Adjust logic to only send important items

Browse Source
This commit is contained in:
Nicolas Mowen 2025-08-11 13:35:53 -06:00
parent e2532ad6e5
commit f5f5d4cd1b
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
frigate/data_processing/post/review_descriptions.py
View File

@ -7,6 +7,7 @@ import os
import shutil import shutil
import threading import threading
from pathlib import Path from pathlib import Path
from typing import Any
import cv2 import cv2
@ -122,7 +123,7 @@ class ReviewDescriptionProcessor(PostProcessorApi):
if topic == EmbeddingsRequestEnum.summarize_review.value: if topic == EmbeddingsRequestEnum.summarize_review.value:
start_ts = request_data["start_ts"] start_ts = request_data["start_ts"]
end_ts = request_data["end_ts"] end_ts = request_data["end_ts"]
items = [ items: list[dict[str, Any]] = [
r["data"]["metadata"] r["data"]["metadata"]
for r in ( for r in (
ReviewSegment.select(ReviewSegment.data) ReviewSegment.select(ReviewSegment.data)
@ -141,7 +142,18 @@ class ReviewDescriptionProcessor(PostProcessorApi):
logger.debug("No review items with metadata found during time period") logger.debug("No review items with metadata found during time period")
return None return None
return self.genai_client.generate_review_summary(start_ts, end_ts, items) important_items = filter(
lambda item: item.get("potential_threat_level", 0) > 0
or item.get("other_concerns"),
items,
)
if not important_items:
return "No concerns were found during this time period."
return self.genai_client.generate_review_summary(
start_ts, end_ts, important_items
)
else: else:
return None return None