From 72aa68cedcb5138eeec0f96eecee2fce58f27365 Mon Sep 17 00:00:00 2001
From: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com>
Date: Mon, 14 Oct 2024 07:23:10 -0500
Subject: [PATCH] Fix genai labels (#14330)

* Publish model state and embeddings reindex in dispatcher onConnect

* remove unneeded from explore

* add embeddings reindex progress to statusbar

* don't allow right click or show similar button if semantic search is disabled

* fix status bar

* Convert peewee model to dict before formatting for genai description

* add embeddings reindex progress to statusbar

* fix status bar

* 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):