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

Formatting

Browse Source
This commit is contained in:
Paillat
2024-02-20 16:23:15 +01:00
parent ce128e400a
commit cced96d792
8 changed files with 37 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/engines/MetadataEngine/ShortsMetadataEngine.py
View File

@@ -2,15 +2,20 @@ from . import BaseMetadataEngine
from ...utils.prompting import get_prompt
class ShortsMetadataEngine(BaseMetadataEngine):
def __init__(self, **kwargs) -> None:
...
def get_metadata(self):
sytsem_prompt, chat_prompt = get_prompt("ShortsMetadata", by_file_location=__file__)
sytsem_prompt, chat_prompt = get_prompt(
"ShortsMetadata", by_file_location=__file__
)
chat_prompt = chat_prompt.replace("{script}", self.ctx.script)
return self.ctx.simplellmengine.generate(chat_prompt=chat_prompt, system_prompt=sytsem_prompt, json_mode=True)
return self.ctx.simplellmengine.generate(
chat_prompt=chat_prompt, system_prompt=sytsem_prompt, json_mode=True
)
def get_options(self):
return []
return []