Files
viralfactory/src/engines/LLMEngine/BaseLLMEngine.py

10 lines
332 B
Python
Raw Normal View History

2024-02-14 17:49:51 +01:00
from abc import ABC, abstractmethod
from ..BaseEngine import BaseEngine
import openai
class BaseLLMEngine(BaseEngine):
@abstractmethod
2024-02-15 14:11:16 +01:00
def generate(self, system_prompt: str, chat_prompt: str, max_tokens: int, temperature: float, top_p: float, frequency_penalty: float, presence_penalty: float) -> str | dict:
2024-02-14 17:49:51 +01:00
pass