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

Ensure descriptions saved in chroma are non-empty

Browse Source
This commit is contained in:
Josh Hawkins 2024-09-21 12:38:55 -05:00
parent 176af55e8c
commit e98b402cdf
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frigate/embeddings/embeddings.py
View File

@ -126,9 +126,9 @@ class Embeddings:
thumbnails["ids"].append(event.id)
thumbnails["images"].append(img)
thumbnails["metadatas"].append(metadata)
if event.data.get("description") is not None:
if description := event.data.get("description", "").strip():
descriptions["ids"].append(event.id)
descriptions["documents"].append(event.data["description"])
descriptions["documents"].append(description)
descriptions["metadatas"].append(metadata)
if len(thumbnails["ids"]) > 0:

2
frigate/embeddings/maintainer.py
View File

@ -177,7 +177,7 @@ class EmbeddingMaintainer(threading.Thread):
camera_config, thumbnails, metadata
)
if description is None:
if not description:
logger.debug("Failed to generate description for %s", event.id)
return