From 3a4e032799cf88588f530f7b9b1f91a6249021c6 Mon Sep 17 00:00:00 2001 From: Paillat Date: Sun, 18 Feb 2024 22:49:45 +0100 Subject: [PATCH] Add SettingsEngine to ENGINES dictionary --- src/engines/__init__.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/engines/__init__.py b/src/engines/__init__.py index fa5f670..2d88daa 100644 --- a/src/engines/__init__.py +++ b/src/engines/__init__.py @@ -6,7 +6,7 @@ from . import ScriptEngine from . import LLMEngine from . import CaptioningEngine from . import AssetsEngine - +from . import SettingsEngine class EngineDict(TypedDict): classes: list[BaseEngine] @@ -14,6 +14,12 @@ class EngineDict(TypedDict): ENGINES: dict[str, EngineDict] = { + "SettingsEngine": { + "classes": [SettingsEngine.SettingsEngine], + "multiple": False, + "show_dropdown": False, + }, + "SimpleLLMEngine": { "classes": [LLMEngine.OpenaiLLMEngine, LLMEngine.AnthropicLLMEngine], "multiple": False,