Files
viralfactory/src/engines/LLMEngine/BaseLLMEngine.py
2024-02-15 14:11:16 +01:00

10 lines
332 B
Python

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 | dict:
pass