zishuo/gpt4free
1
0
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2025-10-13 03:53:50 +08: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
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] provider: ProviderType = __map__[provider]
if issubclass(provider, ProviderModelMixin): if issubclass(provider, ProviderModelMixin):
return [{"model": model, "default": model == provider.default_model} for model in provider.get_models()] 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: else:
return []; return []
@staticmethod @staticmethod
def get_image_models() -> list[dict]: def get_image_models() -> list[dict]:
@@ -245,4 +240,4 @@ def get_error_message(exception: Exception) -> str:
provider = get_last_provider() provider = get_last_provider()
if provider is None: if provider is None:
return message return message
return f"{provider.__name__}: {message}" return f"{provider.__name__}: {message}"