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

genai description for sqlite_vec

Browse Source
This commit is contained in:
Josh Hawkins 2024-10-04 13:23:26 -05:00
parent cb5b982b61
commit b31216da74
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
frigate/genai/__init__.py
View File

@ -31,12 +31,12 @@ class GenAIClient:
self, self,
camera_config: CameraConfig, camera_config: CameraConfig,
thumbnails: list[bytes], thumbnails: list[bytes],
metadata: dict[str, any], label: str,
) -> 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(
metadata["label"], camera_config.genai.prompt label, camera_config.genai.prompt
).format(**metadata) )
return self._send(prompt, thumbnails) return self._send(prompt, thumbnails)
def _init_provider(self): def _init_provider(self):