src/engines/__init__.py

from .BaseEngine import BaseEngine
from . import TTSEngine
from . import ScriptEngine
from . import LLMEngine

ENGINES = {
    "SimpleLLMEngine": [LLMEngine.OpenaiLLMEngine, LLMEngine.AnthropicLLMEngine],
    "PowerfulLLMEngine": [LLMEngine.OpenaiLLMEngine, LLMEngine.AnthropicLLMEngine],
    "TTSEngine": [TTSEngine.CoquiTTSEngine, TTSEngine.ElevenLabsTTSEngine],
    "ScriptEngine": [
        ScriptEngine.ShowerThoughtsScriptEngine,
        ScriptEngine.CustomScriptEngine,
    ],
}
:rocket: 2024-02-14 17:49:51 +01:00			`from .BaseEngine import BaseEngine`
:rocket: Maaany things 2024-02-15 14:11:16 +01:00			`from . import TTSEngine`
:rocket: 2024-02-14 17:49:51 +01:00			`from . import ScriptEngine`
feat(GenerationContext.py): add new file GenerationContext.py to handle the context of generation engines feat(OpenaiLLMEngine.py): add orjson library for JSON serialization and deserialization, and implement the generate method to make API call to OpenAI chat completions endpoint feat(__init__.py): import OpenaiLLMEngine in LLMEngine package feat(BaseScriptEngine.py): add time_script method to the BaseScriptEngine class feat(CustomScriptEngine.py): add new file CustomScriptEngine.py to handle custom script generation, implement generate method to return the provided script, and add get_options method to provide a textbox for the prompt input feat(__init__.py): import CustomScriptEngine in ScriptEngine package feat(__init__.py): import LLMEngine package and add OpenaiLLMEngine to the ENGINES dictionary refactor(gradio_ui.py): change equal_height attribute of Row to False to allow different heights for input blocks 2024-02-15 11:23:36 +01:00			`from . import LLMEngine`
:rocket: 2024-02-14 17:49:51 +01:00
			`ENGINES = {`
Add new engines and fix naming 2024-02-15 17:52:47 +01:00			`"SimpleLLMEngine": [LLMEngine.OpenaiLLMEngine, LLMEngine.AnthropicLLMEngine],`
			`"PowerfulLLMEngine": [LLMEngine.OpenaiLLMEngine, LLMEngine.AnthropicLLMEngine],`
:rocket: 2024-02-14 17:49:51 +01:00			`"TTSEngine": [TTSEngine.CoquiTTSEngine, TTSEngine.ElevenLabsTTSEngine],`
Add new engines and fix naming 2024-02-15 17:52:47 +01:00			`"ScriptEngine": [`
			`ScriptEngine.ShowerThoughtsScriptEngine,`
			`ScriptEngine.CustomScriptEngine,`
			`],`
:rocket: 2024-02-14 17:49:51 +01:00			`}`