from abc import ABC

from ..exceptions.exceptions import InputErrorException
from ..schemas import BaseCompletion


class BaseLLM(ABC):

    def __init__(self, model_name: str, params: dict, **kwargs):
        self.__model_name = model_name
        self.__params = params

    @classmethod
    async def create(cls, config_data: dict):
        pass

    @property
    def model_name(self) -> str:
        return self.__model_name

    @model_name.setter
    def model_name(self, model_name):
        if model_name is None:
            raise InputErrorException("Invalid model_name {}".format(model_name))
        self.__model_name = model_name

    @property
    def params(self) -> dict:
        return self.__params

    def completion(self, prompt) -> BaseCompletion:
        pass

    async def async_completion(self, prompt) -> BaseCompletion:
        pass