From dee544a26cc2d122ccc0fcd930ecfebd63090ab9 Mon Sep 17 00:00:00 2001 From: Paillat Date: Thu, 15 Feb 2024 17:52:47 +0100 Subject: [PATCH] Add new engines and fix naming --- src/engines/__init__.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/engines/__init__.py b/src/engines/__init__.py index f49f99a..a2f548e 100644 --- a/src/engines/__init__.py +++ b/src/engines/__init__.py @@ -4,7 +4,11 @@ from . import ScriptEngine from . import LLMEngine ENGINES = { - "LLMEngine": [LLMEngine.OpenaiLLMEngine], + "SimpleLLMEngine": [LLMEngine.OpenaiLLMEngine, LLMEngine.AnthropicLLMEngine], + "PowerfulLLMEngine": [LLMEngine.OpenaiLLMEngine, LLMEngine.AnthropicLLMEngine], "TTSEngine": [TTSEngine.CoquiTTSEngine, TTSEngine.ElevenLabsTTSEngine], - "ScriptEngine": [ScriptEngine.ShowerThoughtsScriptEngine, ScriptEngine.CustomScriptEngine], + "ScriptEngine": [ + ScriptEngine.ShowerThoughtsScriptEngine, + ScriptEngine.CustomScriptEngine, + ], }