Paillat-dev/viralfactory
1
0
Fork 0
mirror of https://github.com/Paillat-dev/viralfactory.git synced 2026-01-02 09:16:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Formatting

Browse Source
This commit is contained in:
Paillat
2024-02-15 17:53:27 +01:00
parent 53b627f4ed
commit e8cf4bad50
1 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/engines/LLMEngine/BaseLLMEngine.py
View File

@@ -3,8 +3,17 @@ from ..BaseEngine import BaseEngine
import openai
class BaseLLMEngine(BaseEngine):
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:
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