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

Convert peewee model to dict before formatting for genai description

Browse Source
This commit is contained in:
Josh Hawkins 2024-10-13 21:59:59 -05:00
parent d49c5c546b
commit 47e1f72aca
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
frigate/genai/__init__.py
View File

@ -4,6 +4,8 @@ import importlib
import os import os
from typing import Optional from typing import Optional
from playhouse.shortcuts import model_to_dict
from frigate.config import CameraConfig, GenAIConfig, GenAIProviderEnum from frigate.config import CameraConfig, GenAIConfig, GenAIProviderEnum
from frigate.models import Event from frigate.models import Event
@ -36,8 +38,9 @@ class GenAIClient:
) -> 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(
event.label, camera_config.genai.prompt event.label,
).format(**event) camera_config.genai.prompt,
).format(**model_to_dict(event))
return self._send(prompt, thumbnails) return self._send(prompt, thumbnails)
def _init_provider(self): def _init_provider(self):