summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--g4f/__init__.py9
-rw-r--r--g4f/models.py3
2 files changed, 7 insertions, 5 deletions
diff --git a/g4f/__init__.py b/g4f/__init__.py
index d8681297..a49e60e9 100644
--- a/g4f/__init__.py
+++ b/g4f/__init__.py
@@ -16,14 +16,17 @@ class ChatCompletion:
auth : Union[str, None] = None, **kwargs: Any) -> Union[CreateResult, str]:
if isinstance(model, str):
- try:
+ if model in models.ModelUtils.convert:
model = models.ModelUtils.convert[model]
- except KeyError:
+ else:
raise Exception(f'The model: {model} does not exist')
if not provider:
if isinstance(model.best_provider, list):
- provider = random.choice(model.best_provider)
+ if stream:
+ provider = random.choice([p for p in model.best_provider if p.supports_stream])
+ else:
+ provider = random.choice(model.best_provider)
else:
provider = model.best_provider
diff --git a/g4f/models.py b/g4f/models.py
index 0e2680b6..9b01fa3c 100644
--- a/g4f/models.py
+++ b/g4f/models.py
@@ -7,7 +7,6 @@ from .Provider import (
CodeLinkAva,
ChatgptAi,
ChatBase,
- Yqcloud,
Vercel,
DeepAi,
Aivvm,
@@ -32,7 +31,7 @@ gpt_35_turbo = Model(
name = 'gpt-3.5-turbo',
base_provider = 'openai',
best_provider = [
- Yqcloud, DeepAi, CodeLinkAva, ChatgptLogin, ChatgptAi, ChatBase, Aivvm
+ DeepAi, CodeLinkAva, ChatgptLogin, ChatgptAi, ChatBase, Aivvm
]
)