administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-03-25 09:38:22 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix formatting

Browse Source
This commit is contained in:
user 2025-12-12 16:53:56 -05:00
parent 7b511bdebb
commit 61282e0754
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
frigate/embeddings/embeddings.py
View File

@ -529,7 +529,9 @@ class Embeddings:
)
# Calculate embedding for new trigger
embedding = self._calculate_trigger_embedding(trigger, trigger_name, camera.name)
embedding = self._calculate_trigger_embedding(
trigger, trigger_name, camera.name
)
Trigger.create(
camera=camera.name,
@ -585,13 +587,13 @@ class Embeddings:
f"Failed to delete thumbnail for trigger with data {event_id} in {camera}: {e}"
)
def _calculate_trigger_embedding(self, trigger, trigger_name: str, camera_name: str) -> bytes:
def _calculate_trigger_embedding(
self, trigger, trigger_name: str, camera_name: str
) -> bytes:
"""Calculate embedding for a trigger based on its type and data."""
if trigger.type == "description":
logger.debug(f"Generating embedding for trigger description {trigger_name}")
embedding = self.embed_description(
None, trigger.data, upsert=False
)
embedding = self.embed_description(None, trigger.data, upsert=False)
return embedding.astype(np.float32).tobytes()
elif trigger.type == "thumbnail":
@ -611,9 +613,7 @@ class Embeddings:
try:
with open(
os.path.join(
TRIGGER_DIR, camera_name, f"{trigger.data}.webp"
),
os.path.join(TRIGGER_DIR, camera_name, f"{trigger.data}.webp"),
"rb",
) as f:
thumbnail = f.read()
@ -633,4 +633,4 @@ class Embeddings:
else:
logger.warning(f"Unknown trigger type: {trigger.type}")
return b""
return b""