From d08a2c29efe6babdd70ec1a92e4dc1b74d8da8d3 Mon Sep 17 00:00:00 2001 From: Paillat Date: Thu, 22 Feb 2024 15:14:36 +0100 Subject: [PATCH] Formatting --- .../AssetsEngine/AssetsEngineSelector.py | 5 +++-- .../ScriptEngine/ShowerThoughtsScriptEngine.py | 18 +++++++----------- 2 files changed, 10 insertions(+), 13 deletions(-) diff --git a/src/engines/AssetsEngine/AssetsEngineSelector.py b/src/engines/AssetsEngine/AssetsEngineSelector.py index a259a43..350accc 100644 --- a/src/engines/AssetsEngine/AssetsEngineSelector.py +++ b/src/engines/AssetsEngine/AssetsEngineSelector.py @@ -30,7 +30,8 @@ class AssetsEngineSelector: )["assets"] clips: list = [] for engine in self.ctx.assetsengine: - assets_opts = [asset for asset in assets if asset["engine"] == engine.name] - assets_opts = [asset["args"] for asset in assets_opts] + assets_opts = [ + asset["args"] for asset in assets if asset["engine"] == engine.name + ] clips.extend(engine.get_assets(assets_opts)) self.ctx.index_3.extend(clips) diff --git a/src/engines/ScriptEngine/ShowerThoughtsScriptEngine.py b/src/engines/ScriptEngine/ShowerThoughtsScriptEngine.py index 9290845..3fc4078 100644 --- a/src/engines/ScriptEngine/ShowerThoughtsScriptEngine.py +++ b/src/engines/ScriptEngine/ShowerThoughtsScriptEngine.py @@ -23,17 +23,13 @@ class ShowerThoughtsScriptEngine(BaseScriptEngine): ) sys_prompt = sys_prompt.format(n_sentences=self.n_sentences) chat_prompt = chat_prompt.format(n_sentences=self.n_sentences) - self.ctx.script = ( - self.ctx.powerfulllmengine.generate( - system_prompt=sys_prompt, - chat_prompt=chat_prompt, - max_tokens=20 * self.n_sentences, - temperature=1.3, - json_mode=False, - ) - .strip() - .copy() - ) + self.ctx.script = self.ctx.powerfulllmengine.generate( + system_prompt=sys_prompt, + chat_prompt=chat_prompt, + max_tokens=20 * self.n_sentences, + temperature=1.3, + json_mode=False, + ).strip() @classmethod def get_options(cls) -> list: