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

10 lines
325 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
def generate(self, system_prompt: str, chat_prompt: str, max_tokens: int, temperature: float, top_p: float, frequency_penalty: float, presence_penalty: float) -> str:
pass