File size: 802 Bytes
0c4cce1
 
 
 
 
 
 
 
 
 
 
37fe589
f2b20fc
0c4cce1
914c1ca
 
 
6d239e6
f095ec6
04fce89
f095ec6
 
 
914c1ca
0c4cce1
 
 
914c1ca
c24f458
0c4cce1
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
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
    }