from g4f import Provider


class Model:
    class model:
        name: str
        base_provider: str
        best_provider: str

    class gpt_35_turbo:
        name: str = 'gpt-3.5-turbo'
        base_provider: str = 'Achyuth'
        best_provider: Provider.Provider = Provider.Wewordle

    class gpt_35_turbo_0301:
        name: str = 'gpt-3.5-turbo-0301'
        base_provider: str = 'openai'
        best_provider: Provider.Provider = Provider.GetGpt

    class falcon_40b:
        name: str = 'falcon-40b'
        base_provider: str = 'H2o'
        best_provider: Provider.Provider = Provider.H2o
           
class ModelUtils:
    convert: dict = {
        'gpt-3.5-turbo': Model.gpt_35_turbo,
        'gpt-3.5-turbo-0301': Model.gpt_35_turbo_0301,
        'falcon-40b': Model.falcon_40b
    }