mirror of
https://github.com/Paillat-dev/viralfactory.git
synced 2026-01-02 17:24:54 +00:00
Formatting
This commit is contained in:
@@ -3,14 +3,16 @@ from typing import TypedDict
|
||||
|
||||
from .. import BaseEngine
|
||||
|
||||
|
||||
class MetadataEngineSettings(TypedDict):
|
||||
title: str
|
||||
description: str
|
||||
|
||||
|
||||
class BaseMetadataEngine(BaseEngine):
|
||||
def __init__(self, **kwargs) -> None:
|
||||
...
|
||||
|
||||
@abstractmethod
|
||||
def get_metadata(self, input: str) -> MetadataEngineSettings:
|
||||
...
|
||||
...
|
||||
|
||||
@@ -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 []
|
||||
|
||||
@@ -1,2 +1,2 @@
|
||||
from .BaseMetadataEngine import BaseMetadataEngine
|
||||
from .ShortsMetadataEngine import ShortsMetadataEngine
|
||||
from .ShortsMetadataEngine import ShortsMetadataEngine
|
||||
|
||||
Reference in New Issue
Block a user