Paillat-dev/viralfactory
1
0
Fork 0
mirror of https://github.com/Paillat-dev/viralfactory.git synced 2026-01-02 17:24:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add BackgroundEngine to engines.

Browse Source
This commit is contained in:
Paillat
2024-02-20 14:55:58 +01:00
parent c56257a2d3
commit 7322abf1c3
2 changed files with 52 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/engines/__init__.py
View File

@@ -7,6 +7,8 @@ from . import LLMEngine
from . import CaptioningEngine
from . import AssetsEngine
from . import SettingsEngine
from . import BackgroundEngine
class EngineDict(TypedDict):
classes: list[BaseEngine]
@@ -19,7 +21,6 @@ ENGINES: dict[str, EngineDict] = {
"multiple": False,
"show_dropdown": False,
},
"SimpleLLMEngine": {
"classes": [LLMEngine.OpenaiLLMEngine, LLMEngine.AnthropicLLMEngine],
"multiple": False,
@@ -47,4 +48,8 @@ ENGINES: dict[str, EngineDict] = {
"classes": [AssetsEngine.DallEAssetsEngine, NoneEngine],
"multiple": True,
},
"BackgroundEngine": {
"classes": [BackgroundEngine.SimpleBackgroundEngine, NoneEngine],
"multiple": False,
},
}