add model config : gpt_35_turbo_0613

@@ -1,226 +1,229 @@

1 from __future__ import annotations

2 2 from dataclasses import dataclass

3 from .typing import Union

4 from .Provider import BaseProvider

5 from .Provider import (

3 from .typing import Union

4 from .Provider import BaseProvider

5 from .Provider import (

6 6 ChatgptLogin,

7 7 CodeLinkAva,

8 ChatgptAi,

9 ChatBase,

10 Vercel,

11 DeepAi,

12 Aivvm,

13 Bard,

14 H2o

8 ChatgptAi,

9 ChatBase,

10 Vercel,

11 DeepAi,

12 Aivvm,

13 Bard,

14 H2o,

15 Yqcloud

15 16 )

16 17

18

17 19 @dataclass

18 20 class Model:

19 21 name: str

20 22 base_provider: str

21 23 best_provider: Union[type[BaseProvider], tuple[type[BaseProvider]]] = None

22 24

25

23 26 # Config for HuggingChat, OpenAssistant

24 27 # Works for Liaobots, H2o, OpenaiChat, Yqcloud, You

25 28 default = Model(

26 name = "",

27 base_provider = "huggingface")

29 name="",

30 base_provider="huggingface")

28 31

29 32 # GPT-3.5 / GPT-4

30 33 gpt_35_turbo = Model(

31 name = 'gpt-3.5-turbo',

32 base_provider = 'openai',

33 best_provider = [

34 name='gpt-3.5-turbo',

35 base_provider='openai',

36 best_provider=[

34 37 DeepAi, CodeLinkAva, ChatgptLogin, ChatgptAi, ChatBase, Aivvm

35 38 ]

36 39 )

37 40

38 41 gpt_4 = Model(

39 name = 'gpt-4',

40 base_provider = 'openai')

42 name='gpt-4',

43 base_provider='openai')

41 44

42 45 # Bard

43 46 palm = Model(

44 name = 'palm',

45 base_provider = 'google',

46 best_provider = Bard)

47 name='palm',

48 base_provider='google',

49 best_provider=Bard)

47 50

48 51 # H2o

49 52 falcon_7b = Model(

50 name = 'h2oai/h2ogpt-gm-oasst1-en-2048-falcon-7b-v3',

51 base_provider = 'huggingface',

52 best_provider = H2o)

53 name='h2oai/h2ogpt-gm-oasst1-en-2048-falcon-7b-v3',

54 base_provider='huggingface',

55 best_provider=H2o)

53 56

54 57 falcon_40b = Model(

55 name = 'h2oai/h2ogpt-gm-oasst1-en-2048-falcon-40b-v1',

56 base_provider = 'huggingface',

57 best_provider = H2o)

58 name='h2oai/h2ogpt-gm-oasst1-en-2048-falcon-40b-v1',

59 base_provider='huggingface',

60 best_provider=H2o)

58 61

59 62 llama_13b = Model(

60 name = 'h2oai/h2ogpt-gm-oasst1-en-2048-open-llama-13b',

61 base_provider = 'huggingface',

62 best_provider = H2o)

63 name='h2oai/h2ogpt-gm-oasst1-en-2048-open-llama-13b',

64 base_provider='huggingface',

65 best_provider=H2o)

63 66

64 67 # Vercel

65 68 claude_instant_v1 = Model(

66 name = 'anthropic:claude-instant-v1',

67 base_provider = 'anthropic',

68 best_provider = Vercel)

69 name='anthropic:claude-instant-v1',

70 base_provider='anthropic',

71 best_provider=Vercel)

69 72

70 73 claude_v1 = Model(

71 name = 'anthropic:claude-v1',

72 base_provider = 'anthropic',

73 best_provider = Vercel)

74 name='anthropic:claude-v1',

75 base_provider='anthropic',

76 best_provider=Vercel)

74 77

75 78 claude_v2 = Model(

76 name = 'anthropic:claude-v2',

77 base_provider = 'anthropic',

78 best_provider = Vercel)

79 name='anthropic:claude-v2',

80 base_provider='anthropic',

81 best_provider=Vercel)

79 82

80 83 command_light_nightly = Model(

81 name = 'cohere:command-light-nightly',

82 base_provider = 'cohere',

83 best_provider = Vercel)

84 name='cohere:command-light-nightly',

85 base_provider='cohere',

86 best_provider=Vercel)

84 87

85 88 command_nightly = Model(

86 name = 'cohere:command-nightly',

87 base_provider = 'cohere',

88 best_provider = Vercel)

89 name='cohere:command-nightly',

90 base_provider='cohere',

91 best_provider=Vercel)

89 92

90 93 gpt_neox_20b = Model(

91 name = 'huggingface:EleutherAI/gpt-neox-20b',

92 base_provider = 'huggingface',

93 best_provider = Vercel)

94 name='huggingface:EleutherAI/gpt-neox-20b',

95 base_provider='huggingface',

96 best_provider=Vercel)

94 97

95 98 oasst_sft_1_pythia_12b = Model(

96 name = 'huggingface:OpenAssistant/oasst-sft-1-pythia-12b',

97 base_provider = 'huggingface',

98 best_provider = Vercel)

99 name='huggingface:OpenAssistant/oasst-sft-1-pythia-12b',

100 base_provider='huggingface',

101 best_provider=Vercel)

99 102

100 103 oasst_sft_4_pythia_12b_epoch_35 = Model(

101 name = 'huggingface:OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5',

102 base_provider = 'huggingface',

103 best_provider = Vercel)

104 name='huggingface:OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5',

105 base_provider='huggingface',

106 best_provider=Vercel)

104 107

105 108 santacoder = Model(

106 name = 'huggingface:bigcode/santacoder',

107 base_provider = 'huggingface',

108 best_provider = Vercel)

109 name='huggingface:bigcode/santacoder',

110 base_provider='huggingface',

111 best_provider=Vercel)

109 112

110 113 bloom = Model(

111 name = 'huggingface:bigscience/bloom',

112 base_provider = 'huggingface',

113 best_provider = Vercel)

114 name='huggingface:bigscience/bloom',

115 base_provider='huggingface',

116 best_provider=Vercel)

114 117

115 118 flan_t5_xxl = Model(

116 name = 'huggingface:google/flan-t5-xxl',

117 base_provider = 'huggingface',

118 best_provider = Vercel)

119 name='huggingface:google/flan-t5-xxl',

120 base_provider='huggingface',

121 best_provider=Vercel)

119 122

120 123 code_davinci_002 = Model(

121 name = 'openai:code-davinci-002',

122 base_provider = 'openai',

123 best_provider = Vercel)

124 name='openai:code-davinci-002',

125 base_provider='openai',

126 best_provider=Vercel)

124 127

125 128 gpt_35_turbo_16k = Model(

126 name = 'openai:gpt-3.5-turbo-16k',

127 base_provider = 'openai',

128 best_provider = Vercel)

129 name='openai:gpt-3.5-turbo-16k',

130 base_provider='openai',

131 best_provider=Vercel)

129 132

130 133 gpt_35_turbo_0613 = Model(

131 name = 'openai:gpt-3.5-turbo-0613',

132 base_provider = 'openai',

133 best_provider = [

134 DeepAi, CodeLinkAva, ChatgptLogin, ChatgptAi, ChatBase, Aivvm

134 name='openai:gpt-3.5-turbo-0613',

135 base_provider='openai',

136 best_provider=[

137 Yqcloud, Aivvm,ChatgptLogin

135 138 ])

136 139

137 140 gpt_35_turbo_16k_0613 = Model(

138 name = 'openai:gpt-3.5-turbo-16k-0613',

139 base_provider = 'openai')

141 name='openai:gpt-3.5-turbo-16k-0613',

142 base_provider='openai')

140 143

141 144 gpt_4_0613 = Model(

142 name = 'openai:gpt-4-0613',

143 base_provider = 'openai',

144 best_provider = Vercel)

145 name='openai:gpt-4-0613',

146 base_provider='openai',

147 best_provider=Vercel)

145 148

146 149 text_ada_001 = Model(

147 name = 'openai:text-ada-001',

148 base_provider = 'openai',

149 best_provider = Vercel)

150 name='openai:text-ada-001',

151 base_provider='openai',

152 best_provider=Vercel)

150 153

151 154 text_babbage_001 = Model(

152 name = 'openai:text-babbage-001',

153 base_provider = 'openai',

154 best_provider = Vercel)

155 name='openai:text-babbage-001',

156 base_provider='openai',

157 best_provider=Vercel)

155 158

156 159 text_curie_001 = Model(

157 name = 'openai:text-curie-001',

158 base_provider = 'openai',

159 best_provider = Vercel)

160 name='openai:text-curie-001',

161 base_provider='openai',

162 best_provider=Vercel)

160 163

161 164 text_davinci_002 = Model(

162 name = 'openai:text-davinci-002',

163 base_provider = 'openai',

164 best_provider = Vercel)

165 name='openai:text-davinci-002',

166 base_provider='openai',

167 best_provider=Vercel)

165 168

166 169 text_davinci_003 = Model(

167 name = 'openai:text-davinci-003',

168 base_provider = 'openai',

169 best_provider = Vercel)

170 name='openai:text-davinci-003',

171 base_provider='openai',

172 best_provider=Vercel)

170 173

171 174 llama13b_v2_chat = Model(

172 name = 'replicate:a16z-infra/llama13b-v2-chat',

173 base_provider = 'replicate',

174 best_provider = Vercel)

175 name='replicate:a16z-infra/llama13b-v2-chat',

176 base_provider='replicate',

177 best_provider=Vercel)

175 178

176 179 llama7b_v2_chat = Model(

177 name = 'replicate:a16z-infra/llama7b-v2-chat',

178 base_provider = 'replicate',

179 best_provider = Vercel)

180 name='replicate:a16z-infra/llama7b-v2-chat',

181 base_provider='replicate',

182 best_provider=Vercel)

180 183

181 184

182 185 class ModelUtils:

183 186 convert: dict[str, Model] = {

184 187 # GPT-3.5 / GPT-4

185 'gpt-3.5-turbo' : gpt_35_turbo,

186 'gpt-4' : gpt_4,

187

188 'gpt-3.5-turbo': gpt_35_turbo,

189 'gpt-4': gpt_4,

190

188 191 # Bard

189 'palm2' : palm,

190 'palm' : palm,

191 'google' : palm,

192 'google-bard' : palm,

193 'google-palm' : palm,

194 'bard' : palm,

195

192 'palm2': palm,

193 'palm': palm,

194 'google': palm,

195 'google-bard': palm,

196 'google-palm': palm,

197 'bard': palm,

198

196 199 # H2o

197 'falcon-40b' : falcon_40b,

198 'falcon-7b' : falcon_7b,

199 'llama-13b' : llama_13b,

200

200 'falcon-40b': falcon_40b,

201 'falcon-7b': falcon_7b,

202 'llama-13b': llama_13b,

203

201 204 # Vercel

202 'claude-instant-v1' : claude_instant_v1,

203 'claude-v1' : claude_v1,

204 'claude-v2' : claude_v2,

205 'command-nightly' : command_nightly,

206 'gpt-neox-20b' : gpt_neox_20b,

207 'santacoder' : santacoder,

208 'bloom' : bloom,

209 'flan-t5-xxl' : flan_t5_xxl,

210 'code-davinci-002' : code_davinci_002,

211 'gpt-3.5-turbo-16k' : gpt_35_turbo_16k,

212 'gpt-4-0613' : gpt_4_0613,

213 'text-ada-001' : text_ada_001,

214 'text-babbage-001' : text_babbage_001,

215 'text-curie-001' : text_curie_001,

216 'text-davinci-002' : text_davinci_002,

217 'text-davinci-003' : text_davinci_003,

218 'llama13b-v2-chat' : llama13b_v2_chat,

219 'llama7b-v2-chat' : llama7b_v2_chat,

220

221 'oasst-sft-1-pythia-12b' : oasst_sft_1_pythia_12b,

222 'oasst-sft-4-pythia-12b-epoch-3.5' : oasst_sft_4_pythia_12b_epoch_35,

223 'command-light-nightly' : command_light_nightly,

224 'gpt-3.5-turbo-16k-0613' : gpt_35_turbo_16k_0613,

225 'gpt-3.5-turbo-0613' : gpt_35_turbo_0613,

226 }

205 'claude-instant-v1': claude_instant_v1,

206 'claude-v1': claude_v1,

207 'claude-v2': claude_v2,

208 'command-nightly': command_nightly,

209 'gpt-neox-20b': gpt_neox_20b,

210 'santacoder': santacoder,

211 'bloom': bloom,ChatgptLogin

212 'flan-t5-xxl': flan_t5_xxl,

213 'code-davinci-002': code_davinci_002,

214 'gpt-3.5-turbo-16k': gpt_35_turbo_16k,

215 'gpt-4-0613': gpt_4_0613,

216 'text-ada-001': text_ada_001,

217 'text-babbage-001': text_babbage_001,

218 'text-curie-001': text_curie_001,

219 'text-davinci-002': text_davinci_002,

220 'text-davinci-003': text_davinci_003,

221 'llama13b-v2-chat': llama13b_v2_chat,

222 'llama7b-v2-chat': llama7b_v2_chat,

223

224 'oasst-sft-1-pythia-12b': oasst_sft_1_pythia_12b,

225 'oasst-sft-4-pythia-12b-epoch-3.5': oasst_sft_4_pythia_12b_epoch_35,

226 'command-light-nightly': command_light_nightly,

227 'gpt-3.5-turbo-16k-0613': gpt_35_turbo_16k_0613,

228 'gpt-3.5-turbo-0613': gpt_35_turbo_0613,

229 }

gpt4free

add model config : gpt_35_turbo_0613

代码差异