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

🎨 Run linter

Browse Source
This commit is contained in:
Paillat
2024-04-21 22:00:06 +02:00
parent f3db42e341
commit 0da6b1a518
2 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/engines/LLMEngine/AnthropicLLMEngine.py
View File

@@ -61,7 +61,9 @@ class AnthropicLLMEngine(BaseLLMEngine):
returnable = fix_busted_json.repair_json(content) returnable = fix_busted_json.repair_json(content)
returnable = orjson.loads(returnable) returnable = orjson.loads(returnable)
return returnable return returnable
except Exception as e: # noqa wait for library to imlement pep https://peps.python.org/pep-0352/ (Required Superclass for Exceptions except (
Exception
) as e: # noqa wait for library to imlement pep https://peps.python.org/pep-0352/ (Required Superclass for Exceptions
tries += 1 tries += 1
else: else:
return content return content

7
src/engines/Pipelines/ScriptedVideoPipeline.py
View File

@@ -76,8 +76,11 @@ class ScriptedVideoPipeline(BasePipeline):
chat = chat.replace("{assets_instructions}", str(self.assets_instructions)) chat = chat.replace("{assets_instructions}", str(self.assets_instructions))
chat = chat.replace("{video_transcript}", str(ctx.timed_script)) chat = chat.replace("{video_transcript}", str(ctx.timed_script))
assets: list[dict[str, str | float]] = ctx.powerfulllmengine.generate( assets: list[dict[str, str | float]] = ctx.powerfulllmengine.generate(
system_prompt=system, chat_prompt=chat, temperature=1, max_tokens=4096, system_prompt=system,
json_mode=True chat_prompt=chat,
temperature=1,
max_tokens=4096,
json_mode=True,
)["assets"] )["assets"]
for asset in assets: for asset in assets:
if asset["type"] == "stock": if asset["type"] == "stock":