From 47e1f72aca2ff6c1751c8338254917d736316e5c Mon Sep 17 00:00:00 2001
From: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com>
Date: Sun, 13 Oct 2024 21:59:59 -0500
Subject: [PATCH] Convert peewee model to dict before formatting for genai
 description

---
 frigate/genai/__init__.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/frigate/genai/__init__.py b/frigate/genai/__init__.py
index dccb74c1d..e2d509383 100644
--- a/frigate/genai/__init__.py
+++ b/frigate/genai/__init__.py
@@ -4,6 +4,8 @@ import importlib
 import os
 from typing import Optional
 
+from playhouse.shortcuts import model_to_dict
+
 from frigate.config import CameraConfig, GenAIConfig, GenAIProviderEnum
 from frigate.models import Event
 
@@ -36,8 +38,9 @@ class GenAIClient:
     ) -> Optional[str]:
         """Generate a description for the frame."""
         prompt = camera_config.genai.object_prompts.get(
-            event.label, camera_config.genai.prompt
-        ).format(**event)
+            event.label,
+            camera_config.genai.prompt,
+        ).format(**model_to_dict(event))
         return self._send(prompt, thumbnails)
 
     def _init_provider(self):