xtekky/gpt4free
1
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2024-11-09 22:28:52 +03:00
Code Issues Packages Projects Releases Wiki Activity

Check supports_stream in best_providers list

Browse Source
This commit is contained in:
Heiner Lohaus 2023-09-18 15:27:36 +02:00
parent 3b8dfff974
commit 78bee08527
2 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
g4f/__init__.py
View File

@ -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

3
g4f/models.py
View File

@ -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
]
)