Update ollama docs and add genai debug logging (#15012)

This commit is contained in:
Josh Hawkins
2024-11-15 15:24:17 -06:00
committed by GitHub
parent 206ed06905
commit ad85f8882b
2 changed files with 6 additions and 2 deletions

View File

@@ -1,6 +1,7 @@
"""Generative AI module for Frigate."""
import importlib
import logging
import os
from typing import Optional
@@ -9,6 +10,8 @@ from playhouse.shortcuts import model_to_dict
from frigate.config import CameraConfig, FrigateConfig, GenAIConfig, GenAIProviderEnum
from frigate.models import Event
logger = logging.getLogger(__name__)
PROVIDERS = {}
@@ -41,6 +44,7 @@ class GenAIClient:
event.label,
camera_config.genai.prompt,
).format(**model_to_dict(event))
logger.debug(f"Sending images to genai provider with prompt: {prompt}")
return self._send(prompt, thumbnails)
def _init_provider(self):