xtekky/gpt4free
1
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2024-12-23 19:11:48 +03:00
Code Issues Packages Projects Releases Wiki Activity

refactor(g4f/gui/server/api.py): streamline model retrieval logic

Browse Source
This commit is contained in:
kqlio67 2024-10-19 12:56:43 +03:00
parent 3dcacd842d
commit fda90aa8f5
1 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
g4f/gui/server/api.py
View File

@ -43,13 +43,8 @@ class Api():
provider: ProviderType = __map__[provider]
if issubclass(provider, ProviderModelMixin):
return [{"model": model, "default": model == provider.default_model} for model in provider.get_models()]
elif provider.supports_gpt_35_turbo or provider.supports_gpt_4:
return [
*([{"model": "gpt-4", "default": not provider.supports_gpt_4}] if provider.supports_gpt_4 else []),
*([{"model": "gpt-3.5-turbo", "default": not provider.supports_gpt_4}] if provider.supports_gpt_35_turbo else [])
]
else:
return [];
return []
@staticmethod
def get_image_models() -> list[dict]:
@ -245,4 +240,4 @@ def get_error_message(exception: Exception) -> str:
provider = get_last_provider()
if provider is None:
return message
return f"{provider.__name__}: {message}"
return f"{provider.__name__}: {message}"