This commit is contained in:
mrT23
2024-04-14 12:00:19 +03:00
parent a4680ded93
commit 8f0f08006f
5 changed files with 49 additions and 9 deletions

View File

@ -15,7 +15,7 @@ class BaseAiHandler(ABC):
pass
@abstractmethod
async def chat_completion(self, model: str, system: str, user: str, temperature: float = 0.2):
async def chat_completion(self, model: str, system: str, user: str, temperature: float = 0.2, img_path: str = None):
"""
This method should be implemented to return a chat completion from the AI model.
Args: