administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-03-29 11:30:17 +03:00
Code Issues Actions 1 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 # 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( Trigger.create(
camera=camera.name, camera=camera.name,
@ -585,13 +587,13 @@ class Embeddings:
f"Failed to delete thumbnail for trigger with data {event_id} in {camera}: {e}" 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.""" """Calculate embedding for a trigger based on its type and data."""
if trigger.type == "description": if trigger.type == "description":
logger.debug(f"Generating embedding for trigger description {trigger_name}") logger.debug(f"Generating embedding for trigger description {trigger_name}")
embedding = self.embed_description( embedding = self.embed_description(None, trigger.data, upsert=False)
None, trigger.data, upsert=False
)
return embedding.astype(np.float32).tobytes() return embedding.astype(np.float32).tobytes()
elif trigger.type == "thumbnail": elif trigger.type == "thumbnail":
@ -611,9 +613,7 @@ class Embeddings:
try: try:
with open( with open(
os.path.join( os.path.join(TRIGGER_DIR, camera_name, f"{trigger.data}.webp"),
TRIGGER_DIR, camera_name, f"{trigger.data}.webp"
),
"rb", "rb",
) as f: ) as f:
thumbnail = f.read() thumbnail = f.read()
@ -633,4 +633,4 @@ class Embeddings:
else: else:
logger.warning(f"Unknown trigger type: {trigger.type}") logger.warning(f"Unknown trigger type: {trigger.type}")
return b"" return b""