diff options
author | Heiner Lohaus <heiner@lohaus.eu> | 2023-09-12 04:40:10 +0200 |
---|---|---|
committer | Heiner Lohaus <heiner@lohaus.eu> | 2023-09-12 04:40:10 +0200 |
commit | 618addd001077d9e4ed4d2a7aa03769a0ee0901b (patch) | |
tree | 0a1f1eb3639c34472884d8d3d7bd102b12c891ff /g4f/models.py | |
parent | Update Readme (diff) | |
download | gpt4free-618addd001077d9e4ed4d2a7aa03769a0ee0901b.tar gpt4free-618addd001077d9e4ed4d2a7aa03769a0ee0901b.tar.gz gpt4free-618addd001077d9e4ed4d2a7aa03769a0ee0901b.tar.bz2 gpt4free-618addd001077d9e4ed4d2a7aa03769a0ee0901b.tar.lz gpt4free-618addd001077d9e4ed4d2a7aa03769a0ee0901b.tar.xz gpt4free-618addd001077d9e4ed4d2a7aa03769a0ee0901b.tar.zst gpt4free-618addd001077d9e4ed4d2a7aa03769a0ee0901b.zip |
Diffstat (limited to '')
-rw-r--r-- | g4f/models.py | 21 |
1 files changed, 13 insertions, 8 deletions
diff --git a/g4f/models.py b/g4f/models.py index 0c5eb961..e095ce7e 100644 --- a/g4f/models.py +++ b/g4f/models.py @@ -1,31 +1,37 @@ from __future__ import annotations from dataclasses import dataclass -from .Provider import Bard, BaseProvider, GetGpt, H2o, Liaobots, Vercel, Equing +from .Provider import BaseProvider, Bard, H2o, Vercel +from .Provider import Aichat, Aivvm, ChatBase, ChatgptAi, ChatgptLogin, CodeLinkAva +from .Provider import DeepAi, Vercel, Vitalentum, Ylokh, You, Yqcloud +from .typing import Union @dataclass class Model: name: str base_provider: str - best_provider: type[BaseProvider] + best_provider: Union[type[BaseProvider], tuple[type[BaseProvider]]] = None # Config for HuggingChat, OpenAssistant # Works for Liaobots, H2o, OpenaiChat, Yqcloud, You default = Model( name="", - base_provider="huggingface", - best_provider=H2o, + base_provider="huggingface" ) # GPT-3.5 / GPT-4 gpt_35_turbo = Model( name = 'gpt-3.5-turbo', base_provider = 'openai', - best_provider = GetGpt) + best_provider = ( + Vercel, Aichat, Aivvm, ChatBase, ChatgptAi, ChatgptLogin, + CodeLinkAva, DeepAi, Vitalentum, Ylokh, You, Yqcloud + ) +) gpt_4 = Model( name = 'gpt-4', base_provider = 'openai', - best_provider = Liaobots) +) # Bard palm = Model( @@ -117,8 +123,7 @@ gpt_35_turbo_16k = Model( gpt_35_turbo_16k_0613 = Model( name = 'openai:gpt-3.5-turbo-16k-0613', - base_provider = 'openai', - best_provider = Equing) + base_provider = 'openai') gpt_4_0613 = Model( name = 'openai:gpt-4-0613', |