summaryrefslogtreecommitdiffstats
path: root/g4f/models.py
diff options
context:
space:
mode:
authorabc <98614666+xtekky@users.noreply.github.com>2023-10-01 04:32:51 +0200
committerabc <98614666+xtekky@users.noreply.github.com>2023-10-01 04:32:51 +0200
commit83443ee91288e41ef2c4c710f25cd7f5334e292b (patch)
tree0b24459eb0ea74101d73b2536052ae8fa891e3f7 /g4f/models.py
parent~ | pypi v-0.1.4.0 `pip install -U g4f` (diff)
parent~ | Merge pull request #954 from Commenter123321/main (diff)
downloadgpt4free-83443ee91288e41ef2c4c710f25cd7f5334e292b.tar
gpt4free-83443ee91288e41ef2c4c710f25cd7f5334e292b.tar.gz
gpt4free-83443ee91288e41ef2c4c710f25cd7f5334e292b.tar.bz2
gpt4free-83443ee91288e41ef2c4c710f25cd7f5334e292b.tar.lz
gpt4free-83443ee91288e41ef2c4c710f25cd7f5334e292b.tar.xz
gpt4free-83443ee91288e41ef2c4c710f25cd7f5334e292b.tar.zst
gpt4free-83443ee91288e41ef2c4c710f25cd7f5334e292b.zip
Diffstat (limited to '')
-rw-r--r--g4f/models.py48
1 files changed, 31 insertions, 17 deletions
diff --git a/g4f/models.py b/g4f/models.py
index 7c2d6822..cca9e850 100644
--- a/g4f/models.py
+++ b/g4f/models.py
@@ -3,25 +3,25 @@ from dataclasses import dataclass
from .typing import Union
from .Provider import BaseProvider, RetryProvider
from .Provider import (
+ AItianhuSpace,
ChatgptLogin,
- ChatgptAi,
+ PerplexityAi,
+ ChatgptDuo,
+ ChatgptAi,
ChatBase,
+ AItianhu,
+ Wewordle,
+ Yqcloud,
+ Myshell,
Vercel,
DeepAi,
+ Aichat,
Aivvm,
- Bard,
- H2o,
GptGo,
- Bing,
- PerplexityAi,
- Wewordle,
- Yqcloud,
- AItianhu,
- AItianhuSpace,
- Aichat,
- Myshell,
+ Bard,
Aibn,
- ChatgptDuo,
+ Bing,
+ H2o,
)
@dataclass(unsafe_hash=True)
@@ -166,7 +166,17 @@ gpt_35_turbo_0613 = Model(
gpt_4_0613 = Model(
name = 'gpt-4-0613',
base_provider = 'openai',
- best_provider = Vercel)
+ best_provider = Aivvm)
+
+gpt_4_32k = Model(
+ name = 'gpt-4-32k',
+ base_provider = 'openai',
+ best_provider = Aivvm)
+
+gpt_4_32k_0613 = Model(
+ name = 'gpt-4-32k-0613',
+ base_provider = 'openai',
+ best_provider = Aivvm)
text_ada_001 = Model(
name = 'text-ada-001',
@@ -206,13 +216,17 @@ llama7b_v2_chat = Model(
class ModelUtils:
convert: dict[str, Model] = {
- # gpt-3.5 / gpt-4
+ # gpt-3.5
'gpt-3.5-turbo' : gpt_35_turbo,
'gpt-3.5-turbo-16k' : gpt_35_turbo_16k,
- 'gpt-4' : gpt_4,
- 'gpt-4-0613' : gpt_4_0613,
'gpt-3.5-turbo-16k-0613' : gpt_35_turbo_16k_0613,
+ # gpt-4
+ 'gpt-4' : gpt_4,
+ 'gpt-4-0613' : gpt_4_0613,
+ 'gpt-4-32k' : gpt_4_32k,
+ 'gpt-4-32k-0613' : gpt_4_32k_0613,
+
# Bard
'palm2' : palm,
'palm' : palm,
@@ -247,4 +261,4 @@ class ModelUtils:
'oasst-sft-1-pythia-12b' : oasst_sft_1_pythia_12b,
'oasst-sft-4-pythia-12b-epoch-3.5' : oasst_sft_4_pythia_12b_epoch_35,
'command-light-nightly' : command_light_nightly,
- } \ No newline at end of file
+ }