mirror of
https://github.com/Paillat-dev/viralfactory.git
synced 2026-01-02 09:16:19 +00:00
Add MetadataEngine and ShortsMetadataEngine classes
This commit is contained in:
16
src/engines/MetadataEngine/ShortsMetadataEngine.py
Normal file
16
src/engines/MetadataEngine/ShortsMetadataEngine.py
Normal file
@@ -0,0 +1,16 @@
|
||||
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__)
|
||||
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)
|
||||
|
||||
def get_options(self):
|
||||
return []
|
||||
Reference in New Issue
Block a user