from typing import Union, List

from src.model.model import Model


class OpenAIModel(Model):

    def infer(
        self, prompts: Union[str, List[str]], possible_answers, conditions=None
    ) -> Union[str, List[str]]:
        pass

    def generate(self, prompts: List[str], conditions=None) -> Union[str, List[str]]:
        pass