Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-08-22 13:47:29 +02:00
Code Issues Projects Releases Wiki Activity

Fix substitution on genai prompts (#14298)

Browse Source
This commit is contained in:
Josh Hawkins 2024-10-12 07:19:24 -05:00 committed by GitHub
parent 1e1610671e
commit 48c60621b6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/embeddings/maintainer.py
View File

@ -259,7 +259,7 @@ class EmbeddingMaintainer(threading.Thread):
camera_config = self.config.cameras[event.camera] camera_config = self.config.cameras[event.camera]
description = self.genai_client.generate_description( description = self.genai_client.generate_description(
camera_config, thumbnails, event.label camera_config, thumbnails, event
) )
if not description: if not description:

7
frigate/genai/__init__.py
View File

@ -5,6 +5,7 @@ import os
from typing import Optional from typing import Optional
from frigate.config import CameraConfig, GenAIConfig, GenAIProviderEnum from frigate.config import CameraConfig, GenAIConfig, GenAIProviderEnum
from frigate.models import Event
PROVIDERS = {} PROVIDERS = {}
@ -31,12 +32,12 @@ class GenAIClient:
self, self,
camera_config: CameraConfig, camera_config: CameraConfig,
thumbnails: list[bytes], thumbnails: list[bytes],
label: str, event: Event,
) -> Optional[str]: ) -> Optional[str]:
"""Generate a description for the frame.""" """Generate a description for the frame."""
prompt = camera_config.genai.object_prompts.get( prompt = camera_config.genai.object_prompts.get(
label, camera_config.genai.prompt event.label, camera_config.genai.prompt
).format(label=label) ).format(**event)
return self._send(prompt, thumbnails) return self._send(prompt, thumbnails)
def _init_provider(self): def _init_provider(self):