File size: 4,147 Bytes
d7738e9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
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 = 'openai'
        best_provider: Provider.Provider = Provider.Naga

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

    class gpt_35_turbo_poe:
        name: str = 'gpt-3.5-turbo-poe'
        base_provider: str = 'poe'
        best_provider: Provider.Provider = Provider.Naga

    class gpt_35_turbo_16k:
        name: str = 'gpt-3.5-turbo-16k'
        base_provider: str = 'reversed'
        best_provider: Provider.Provider = Provider.Naga

    class gpt_35_turbo_16k_poe:
        name: str = 'gpt-3.5-turbo-16k-poe'
        base_provider: str = 'poe'
        best_provider: Provider.Provider = Provider.Naga

    class gpt_4:
        name: str = 'gpt-4'
        base_provider: str = 'reversed'
        best_provider: Provider.Provider = Provider.Naga

    class gpt_4_0314:
        name: str = 'gpt-4-0314'
        base_provider: str = 'reversed'
        best_provider: Provider.Provider = Provider.Naga

    class gpt_4_poe:
        name: str = 'gpt-4-poe'
        base_provider: str = 'poe'
        best_provider: Provider.Provider = Provider.Naga

    class gpt_4_32k:
        name: str = 'gpt-4-32k'
        base_provider: str = 'reversed'
        best_provider: Provider.Provider = Provider.Naga

    class gpt_4_32k_poe:
        name: str = 'gpt-4-32k-poe'
        base_provider: str = 'poe'
        best_provider: Provider.Provider = Provider.Naga

    class claude_instant_100k:
        name: str = 'claude-instant-100k'
        base_provider: str = 'anthropic'
        best_provider: Provider.Provider = Provider.Naga

    class claude_instant:
        name: str = 'claude-instant'
        base_provider: str = 'anthropic'
        best_provider: Provider.Provider = Provider.Naga

    class claude_2:
        name: str = 'claude-2'
        base_provider: str = 'anthropic'
        best_provider: Provider.Provider = Provider.Naga

    class claude_2_100k:
        name: str = 'claude-2-100k'
        base_provider: str = 'anthropic'
        best_provider: Provider.Provider = Provider.Naga

    class llama_2_7b_chat:
        name: str = 'llama-2-7b-chat'
        base_provider: str = 'llama'
        best_provider: Provider.Provider = Provider.Naga

    class llama_2_13b_chat:
        name: str = 'llama-2-13b-chat'
        base_provider: str = 'llama'
        best_provider: Provider.Provider = Provider.Naga

    class llama_2_70b_chat:
        name: str = 'llama-2-70b-chat'
        base_provider: str = 'llama'
        best_provider: Provider.Provider = Provider.Naga
    
    class oasst_sft_6_llama_30b:
        name: str = 'oasst-sft-6-llama-30b'
        base_provider: str = 'huggingface-chat'
        best_provider: Provider.Provider = Provider.Naga

    class falcon_180b_chat:
        name: str = 'falcon-180b-chat'
        base_provider: str = 'huggingface-chat'
        best_provider: Provider.Provider = Provider.Naga



class ModelUtils:
    convert: dict = {
        'gpt-3.5-turbo': Model.gpt_35_turbo,
        'gpt-3.5-turbo-0301': Model.gpt_35_turbo_0301,
        'gpt-3.5-turbo-poe': Model.gpt_35_turbo_poe,
        'gpt-3.5-turbo-16k': Model.gpt_35_turbo_16k,
        'gpt-3.5-turbo-16k-poe': Model.gpt_35_turbo_16k_poe,
        'gpt-4': Model.gpt_4,
        'gpt-4-0314': Model.gpt_4_0314,
        'gpt-4-poe': Model.gpt_4_poe,
        'gpt-4-32k': Model.gpt_4_32k,
        'gpt-4-32k-poe': Model.gpt_4_32k_poe,

        'claude-instant-100k': Model.claude_instant_100k,
        'claude-instant': Model.claude_instant,
        'claude-2': Model.claude_2,
        'claude-2-100k': Model.claude_2_100k,

        'oasst-sft-6-llama-30b': Model.oasst_sft_6_llama_30b,
        'llama-2-7b-chat': Model.llama_2_7b_chat,
        'llama-2-13b-chat': Model.llama_2_13b_chat,
        'llama-2-70b-chat': Model.llama_2_70b_chat,
        'falcon-180b-chat': Model.falcon_180b_chat
    }